ffmpeg视屏编码(1)-api-example.c解读

时间:2023-01-08 04:21:10

参考ffmpeg源代码库自带的api-example.c例子中的video_encode_example函数。

// 初始化codec库
avcodec_init();

// 注册编码器
avcodec_register_all();

static void video_encode_example(const char *filename)
{
    AVCodec *codec;   // 编码器
    AVCodecContext *c= NULL;  // 编解码环境
    int i, out_size, size, x, y, outbuf_size;
    FILE *f; // 输出文件
    AVFrame *picture; // 当前帧
    uint8_t *outbuf, *picture_buf;

    codec = avcodec_find_encoder(CODEC_ID_MPEG1VIDEO);   //  初始化编码器
    if (!codec) {
        fprintf(stderr, "codec not found/n");
        exit(1);
    }

    c= avcodec_alloc_context();  // 初始化编解码环境
    picture= avcodec_alloc_frame();  // 初始化帧

    // 初始化采样率
    c->bit_rate = 400000;
    // 分辨率
    c->width = 352;
    c->height = 288;
    // 帧数
    c->time_base= (AVRational){1,25};
    c->gop_size = 10; /* emit one intra frame every ten frames */
    c->max_b_frames=1;
    c->pix_fmt = PIX_FMT_YUV420P;

    // 打开编码器
    if (avcodec_open(c, codec) < 0) {
        fprintf(stderr, "could not open codec/n");
        exit(1);
    }

    // 打开输出文件
    f = fopen(filename, "wb");
    if (!f) {
        fprintf(stderr, "could not open %s/n", filename);
        exit(1);
    }

    // 分配输出缓存
    outbuf_size = 100000;
    outbuf = malloc(outbuf_size);
    //  分配图像缓存
    size = c->width * c->height;
    picture_buf = malloc((size * 3) / 2); /* size for YUV 420 */ 
    // 2*4个像素点用YUV420采样结果如下
    // Y U | Y | Y U | Y
    // Y V | Y | Y V | Y

    picture->data[0] = picture_buf;  // 亮度
    picture->data[1] = picture->data[0] + size;  // 色度
    picture->data[2] = picture->data[1] + size / 4; // 色度
    picture->linesize[0] = c->width;
    picture->linesize[1] = c->width / 2;
    picture->linesize[2] = c->width / 2;

    // 对1秒的视频进行编码
    for(i=0;i<25;i++) {
        fflush(stdout);
        // 生成1幅图像,存放到picture中
        // 图像的像素
        for(y=0;y<c->height;y++) {
            for(x=0;x<c->width;x++) {
                picture->data[0][y * picture->linesize[0] + x] = x + y + i * 3;
            }
        }

        // 图像的色度
        for(y=0;y<c->height/2;y++) {
            for(x=0;x<c->width/2;x++) {
                picture->data[1][y * picture->linesize[1] + x] = 128 + y + i * 2;
                picture->data[2][y * picture->linesize[2] + x] = 64 + x + i * 5;
            }
        }

        // 对一幅图像进行编码
        out_size = avcodec_encode_video(c, outbuf, outbuf_size, picture);        
        fwrite(outbuf, 1, out_size, f); // 结果存放到文件中
    }

    // 将延时帧写入文件,直到所有帧都写入文件
    for(; out_size; i++) {
        fflush(stdout); 
        out_size = avcodec_encode_video(c, outbuf, outbuf_size, NULL);
        fwrite(outbuf, 1, out_size, f);
    }

    // 写入序列结束代码到文件
    outbuf[0] = 0x00;
    outbuf[1] = 0x00;
    outbuf[2] = 0x01;
    outbuf[3] = 0xb7;
    fwrite(outbuf, 1, 4, f);
    fclose(f);
    free(picture_buf);
    free(outbuf);

    // 关闭解码器
    avcodec_close(c);
    av_free(c);
    av_free(picture);
}