参考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);
}