从零开始学习音视频编程技术(十六) 采集屏幕编码H264
时间:2016年10月14日 人气:...


前面介绍了采集屏幕图像并转换成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(&param, "preset", "slow", 0);
    av_dict_set(&param, "preset", "superfast", 0);
    av_dict_set(&param, "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,&param) < 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