使用ffmpeg进行视频解码以及图像转换

1
2
3
4
5
6
7
8
9
10

AVInputFormat   *inputFmt;
AVFormatContext *fmtCtx = NULL;

inputFmt = av_find_input_format("avi"); /* 打开“AVI”的输入格式 */
if (inputFmt == NULL) {
/* Error processing */
}
if (av_open_input_file(&fmtCtx,"/test/test.avi", inputFmt, 0, NULL) != 0) {
/* Error processing */
}

1
2
3
4
5
6
7
8
9
1011
12

int i, found_stream_index;
AVCodecContext  *videoDecodeCtx = NULL;
for (i=0;i<fmtCtx->nb_streams;i++){
if (fmtCtx->streams[i]->codec->codec_type == AVMEDIA_VIDEO){
videoDecodeCtx = fmtCtx->streams[i]->codec;
found_stream_index = i;
break;
}
}
if (decodeCtx == NULL) {
/* 找不到视频流，错误处理 */
}

1
2
3
4
5
6
7
8
9
1011
12

AVCodec   *videoDecoder;
videoDecoder = avcodec_find_encoder(videoDecodeCtx->codec_id);
if (videoDecoder == NULL){
/* 找不到解码器，错误处理 */
}
if (videoDecoder->capabilities & CODEC_CAP_TRUNCATED){
videoDecodeCtx->flags |= CODEC_FLAG_TRUNCATED;
}

if (avcodec_open(videoDecodeCtx,videoDecoder) < 0){
/* 打不开解码器，错误处理 */
}

1
2
3
4
5
6
7
8
9
1011
12
13
14
15
16
17
18
19
20
21
22
23

AVPacket pkt;
AVFrame *frame = avcodec_alloc_frame();
int got_picture = 0, rc = 0;

while (1){
av_init_packet(&pkt);
rc = av_read_packet(fmtCtx,&pkt); /* 获取一个包的数据 */
if (rc != 0) break;
if (pkt.stream_index != found_stream_index)
goto free_pkt_continue; /* 不是所关心的视频流的数据，舍去 */
if ( avcodec_decode_video2 (videoDecodeCtx, frame, &got_picture, &pkt) < 0) {
/* 核心函数：解码。错误处理。 */
}

if (got_picture) {
{  /* 处理获得的图像(存储在AVFrame里) */ }
av_free(frame);
frame = avcodec_alloc_frame();
}

free_pkt_continue:
av_free_packet(&pkt);
}

1
2
3
4
5
6
7
8
9
1011
12
13
14
15
16
17
18
19

SwsContext *swsCtx;
int dst_width   = 320,  /* 目标宽度 */
dst_height  = 240, /* 目标高度 */
dst_pix_fmt = PIX_FMT_RGB24; /* 目标图像格式 */
AVFrame *convertedFrame = avcodec_alloc_frame();
swsCtx = sws_getContext (
videoDecodeCtx->width, videoDecodeCtx->height, videoDecodeCtx->pix_fmt,
dst_width, dst_height, dst_pix_fmt, SWS_FAST_BILINEAR,
NULL, NULL, NULL );
if (swsCtx == NULL) {
/* 转换上下文初始化失败，错误处理 */
}

/* 分配转换需要的内存 */
avpicture_fill ( (AVPicture *)convertedFrame,dst_pix_fmt, dst_width, dst_height);
if (sws_scale(swsCtx,frame->data,frame->linesize,0,
videoDecodeCtx->height,convertedFrame->data,convertedFrame->linesize) <= 0) {
/* 转换失败，错误处理 */
}