前面介绍了采集屏幕图像并转换成YUV420p。传送门
YUV420p数据是原始的图像数据,100张1920x1080的图像 总大小就达到了300M。太可怕了! 可见,这样直接写入文件是行不通的。因此我们需要把它编码成h264之后在写入文件。至于什么是h264,前面有介绍,这里就不介绍了。
废话不多说,直接进入正文吧。
编码h264还是一样使用ffmpeg,方法如下:
1.打开编码器
AVCodecContext* pCodecCtx; AVCodec* pCodec; uint8_t* picture_buf; AVFrame* picture;
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来实现:
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