ffmpeg4 解码264 输出rgb

      int play_file(char* file_name, FrameFunc tcallback(char* a, int size, int num, int height, int width))
      {
      	av_log_set_level(AV_LOG_WARNING);
     	unsigned version = avcodec_version();
     	printf("FFmpeg version: %d\n", version);
      	AVFormatContext *avFormatCtx = NULL;
     	int i, videoindex;
      	AVCodecContext  *avCodecCtx=NULL;
      	AVCodec *avCodec;
      	avformat_network_init();
     	printf("1111\n");
      	FILE *fp_open = fopen(file_name, "rb+");
     	unsigned char *aviobuffer = (unsigned char *)av_malloc(32768);
      	AVIOContext *avio = avio_alloc_context(aviobuffer, 32768, 0, (void*)fp_open, file_buffer, NULL, NULL);
     	printf("222\n");
      	avFormatCtx = avformat_alloc_context();
      	avFormatCtx->pb = avio;
     	//if (avformat_open_input(&pFormatCtx, patha.c_str(), NULL, NULL) != 0) {
     	if (avformat_open_input(&avFormatCtx, NULL, NULL, NULL) != 0) {
     		printf("Couldn't open input stream.\n");
     		return -1;
      	}
      	avCodec = NULL;
     	while (avCodec == NULL) {
     		printf("start find stream info \n");
     		if (avformat_find_stream_info(avFormatCtx, NULL) < 0) {
     			printf("Couldn't find stream info\n");
     			goto restart_stream;
     			continue;
      		}
      		videoindex = -1;
     		for (i = 0; i < avFormatCtx->nb_streams; i++)
     			if (avFormatCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
      if (videoindex == -1) {
       videoindex = i;
       }
      //break;
      			}
     		if (videoindex == -1) {
     			printf("Didn't find a video stream.\n");
     			goto restart_stream;
      		}
      		avCodecCtx = avcodec_alloc_context3(NULL);
      		avcodec_parameters_to_context(avCodecCtx, avFormatCtx->streams[videoindex]->codecpar);
      		avCodec = avcodec_find_decoder(avCodecCtx->codec_id);
     		if (avCodec == NULL) {
     			printf("Codec not found \n");
     			goto restart_stream;
     			//return -1;
      		}
     		if (avcodec_open2(avCodecCtx, avCodec, NULL) < 0) {
     			printf("Could not open codec.\n");
     			goto restart_stream;
     			continue;
     			//return -1;
      		}
     		goto ok;
      	restart_stream:
     		printf("restart 1 ");
      		avformat_free_context(avFormatCtx);
     		printf("restart 2 ");
     		//avformat_close_input(&pFormatCtx);
      		avFormatCtx = NULL;
      		avFormatCtx = avformat_alloc_context();
     		printf("restart 3 ");
      		av_freep(aviobuffer);
     		//printf("restart 4");
      		aviobuffer = (unsigned char *)av_malloc(32768);
     		printf("restart 4 ");
      		AVIOContext *avio2 = avio_alloc_context(aviobuffer, 32768, 0, &fp_open, file_buffer, NULL, NULL);
      		avFormatCtx->pb = avio2;
     		int open_ret = avformat_open_input(&avFormatCtx, NULL, NULL, NULL);
     		if (open_ret != 0) {
     			printf("2Couldn't open input stream %d\n", open_ret);
     			return -1;
      		}
      		avFormatCtx->probesize = 1000 * 1024;
      		avFormatCtx->max_analyze_duration = 10 * AV_TIME_BASE;
     		printf("restart 5\n");
      		avCodec = NULL;
     		continue;
      	ok:
     		break;
      	}
      	AVFrame *pFrameYUV;
      	AVFrame *pFrameRGB;
      	pFrameYUV = av_frame_alloc();
      	pFrameRGB = av_frame_alloc();
     	uint8_t *out_buffer;
     	struct SwsContext *img_convert_ctx;
      	img_convert_ctx = sws_getContext(avCodecCtx->width, avCodecCtx->height, avCodecCtx->pix_fmt, avCodecCtx->width, avCodecCtx->height, AV_PIX_FMT_RGB24, SWS_BICUBIC, NULL, NULL, NULL);
      	out_buffer = new uint8_t[avpicture_get_size(AV_PIX_FMT_RGB24, avCodecCtx->width, avCodecCtx->height)];
     	//avpicture_fill((AVPicture *)pFrameRGB, out_buffer, AV_PIX_FMT_RGB24, avCodecCtx->width, avCodecCtx->height);
      	av_image_fill_arrays(pFrameRGB->data, pFrameRGB->linesize, out_buffer, AV_PIX_FMT_RGB24, avCodecCtx->width, avCodecCtx->height, 1);
     	int ret, got_picture;
      	AVPacket packet;// = (AVPacket *)av_malloc(sizeof(AVPacket));
     	int need_decode = 1;
     	int is_key_frame = 0;
     	while (av_read_frame(avFormatCtx, &packet) >= 0) {
     		if (packet.stream_index == videoindex) {
      			DWORD start_time = GetTickCount();
     			if (avcodec_send_packet(avCodecCtx, &packet) != 0) {
      printf("avcodec_send_packet错误\n");
      break;
      			}
     		/* if (packet.flags &AV_PKT_FLAG_KEY) {
       is_key_frame = 1;
       need_decode = 1;
       printf("key frame %d\n", need_decode);
       }
       else {
       is_key_frame = 0;
       }
       if (!need_decode) {
       av_packet_unref(&packet);
       printf("not key frame \n");
       continue;
       }*/
     			while (avcodec_receive_frame(avCodecCtx, pFrameYUV) == 0) {
      int ret = avcodec_send_frame(avCodecCtx, pFrameYUV);
       pFrameYUV->pts++;
      printf("decode ok %d %u\n", avCodecCtx->flags, GetTickCount()-start_time);
      /*fwrite(pFrameYUV->data[0], (pCodecCtx->width)*(pCodecCtx->height) * 3, 1, output);*/
       sws_scale(img_convert_ctx, pFrameYUV->data, pFrameYUV->linesize, 0, avCodecCtx->height, pFrameRGB->data, pFrameRGB->linesize);
       tcallback((char*)pFrameRGB->data[0], avCodecCtx->height * avCodecCtx->width * 3, 1, avCodecCtx->height, avCodecCtx->width);
       avcodec_flush_buffers(avCodecCtx);
      //av_frame_free(&pFrameYUV);
      			}
      		}
      		av_packet_unref(&packet);
      	}
      	avcodec_close(avCodecCtx);
      	avformat_close_input(&avFormatCtx);
     	return 0;
      }
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/87456509