前面介绍了采集屏幕图像并转换成YUV420p。传送门
YUV420p数据是原始的图像数据,100张1920x1080的图像 总大小就达到了300M。太可怕了! 可见,这样直接写入文件是行不通的。因此我们需要把它编码成h264之后在写入文件。至于什么是h264,前面有介绍,这里就不介绍了。
废话不多说,直接进入正文吧。
编码h264还是一样使用ffmpeg,方法如下:
1.打开编码器
1
2
3
4
|
AVCodecContext* pCodecCtx; AVCodec* pCodec;
uint8_t* picture_buf;
AVFrame* picture;
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
bool
H264Encoder::openEncoder()
{
int
size;
int
in_w = mWidth;
int
in_h = mHeight;
//宽高
//查找h264编码器
pCodec = avcodec_find_encoder(AV_CODEC_ID_H264);
if
(!pCodec)
{
fprintf
(stderr, "h264 codec not found
");
exit
(1);
}
pCodecCtx = avcodec_alloc_context3(pCodec);
pCodecCtx->codec_id = AV_CODEC_ID_H264;
pCodecCtx->codec_type = AVMEDIA_TYPE_VIDEO;
pCodecCtx->pix_fmt = PIX_FMT_YUV420P;
pCodecCtx->width = in_w;
pCodecCtx->height = in_h;
pCodecCtx->time_base.num = 1;
pCodecCtx->time_base.den = 15;
//帧率(既一秒钟多少张图片)
pCodecCtx->bit_rate = mBitRate;
//比特率(调节这个大小可以改变编码后视频的质量)
pCodecCtx->gop_size=12;
// some formats want stream headers to be separate
if
(pCodecCtx->flags & AVFMT_GLOBALHEADER)
pCodecCtx->flags |= CODEC_FLAG_GLOBAL_HEADER;
// Set Option
AVDictionary *param = 0;
//H.264
//av_dict_set(¶m, "preset", "slow", 0);
av_dict_set(¶m,
"preset"
,
"superfast"
, 0);
av_dict_set(¶m,
"tune"
,
"zerolatency"
, 0);
//实现实时编码
pCodec = avcodec_find_encoder(pCodecCtx->codec_id);
if
(!pCodec){
printf
("Can not find video encoder! 没有找到合适的编码器!
");
return
false
;
}
if
(avcodec_open2(pCodecCtx, pCodec,¶m) < 0){
printf
("Failed to open video encoder! 编码器打开失败!
");
return
false
;
}
picture = avcodec_alloc_frame();
size = avpicture_get_size(pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height);
//计算需要用到的数据大小
picture_buf = (uint8_t *)av_malloc(size);
//分配空间
avpicture_fill((AVPicture *)picture, picture_buf, pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height);
return
true
;
}
|
2.编码
编码之前的数据必须是Yuv420p格式,我们前面已经得到了这样的数据。编码使用avcodec_encode_video2来实现:
1
2
3
4
5
6
7
8
9
10
11
12
13
|
picture->data[0] = node.buffer;
// 亮度Y
picture->data[1] = node.buffer + y_size;
// U
picture->data[2] = node.buffer + y_size*5/4;
// V
int
got_picture=0;
//编码
int
ret = avcodec_encode_video2(pCodecCtx, &pkt,picture, &got_picture);
if
(got_picture==1)
{
// bool isKeyFrame = pkt.flags & AV_PKT_FLAG_KEY; //判断是否关键帧
int
w =
fwrite
(pkt.data,1,pkt.size,h264Fp);
//写入文件中 (h264的裸数据 直接写入文件 也可以播放 因为这里包含H264关键帧)
}
|
最近时间比较少,代码就不做太多的解释了,直接上完整的工程吧。
完整工程下载地址:http://download.csdn.net/detail/qq214517703/9654163
最后生成的out.h264可以直接用普通播放器打开播放。
保存后的文件比起之前可真是小了几百倍啊!
注:由于h264没有时间戳,只有帧率,这里设定的是15,然而我们采集桌面的时候,一秒却采集不到15张,所以用播放器播放的时候 速度上会快很多 是正常现象。
学习音视频技术欢迎访问 http://blog.yundiantech.com
音视频技术交流讨论欢迎加 QQ群 121376426