avi转yuv

开发环境

操作系统：win10
IDE：vs2019
第三方库：ffmpeg5.1

代码

#include <stdio.h>
#include <stdlib.h>

extern "C" {
#include "libavcodec/avcodec.h"
#include "libavfilter/avfilter.h"
#include "libavformat/avformat.h"
#include "libavutil/avutil.h"
#include "libavutil/ffversion.h"
#include "libswresample/swresample.h"
#include "libswscale/swscale.h"
#include "libpostproc/postprocess.h"
}

int decode_test() {
    char filePath[] = "../resources/vtest.avi";//文件地址
    int  videoStreamIndex = -1;//视频流所在流序列中的索引
    int ret = 0;//默认返回值

    //需要的变量名并初始化
    AVFormatContext* fmtCtx = NULL;
    AVPacket* pkt = NULL;
    AVCodecContext* codecCtx = NULL;
    AVCodecParameters* avCodecPara = NULL;
    AVCodec* codec = NULL;
    AVFrame* yuvFrame = av_frame_alloc();

    FILE* fp;
    fopen_s(&fp, "result.yuv", "w+b");
    if (fp == NULL) {
        printf("Cannot open file.\n");
        return -1;
    }

    do {
        //=========================== 创建AVFormatContext结构体 ===============================//
        //分配一个AVFormatContext，FFMPEG所有的操作都要通过这个AVFormatContext来进行
        fmtCtx = avformat_alloc_context();
        //==================================== 打开文件 ======================================//
        if ((ret = avformat_open_input(&fmtCtx, filePath, NULL, NULL)) != 0) {
            printf("cannot open video file\n");
            break;
        }

        //=================================== 获取视频流信息 ===================================//
        if ((ret = avformat_find_stream_info(fmtCtx, NULL)) < 0) {
            printf("cannot retrive video info\n");
            break;
        }

        //循环查找视频中包含的流信息，直到找到视频类型的流
        //便将其记录下来 保存到videoStreamIndex变量中
        for (unsigned int i = 0; i < fmtCtx->nb_streams; i++) {
            if (fmtCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
                videoStreamIndex = i;
                break;//找到视频流就退出
            }
        }

        //如果videoStream为-1 说明没有找到视频流
        if (videoStreamIndex == -1) {
            printf("cannot find video stream\n");
            break;
        }

        int time = fmtCtx->duration;
        int maxtime = time / AV_TIME_BASE / 60;
        int mintime = time / AV_TIME_BASE % 60;
        printf("时长 %d分 %d秒 ", maxtime, mintime);
        //打印输入和输出信息：长度 比特率 流格式等
        av_dump_format(fmtCtx, 0, filePath, 0);

        int VideoStream = -1, AudioStream = -1;
        VideoStream = av_find_best_stream(fmtCtx, AVMEDIA_TYPE_VIDEO, -1, -1, NULL, NULL);

        //=================================  查找解码器 ===================================//
        avCodecPara = fmtCtx->streams[videoStreamIndex]->codecpar;
        codec = (AVCodec*)avcodec_find_decoder(avCodecPara->codec_id);
        if (codec == NULL) {
            printf("cannot find decoder\n");
            break;
        }
        //根据解码器参数来创建解码器内容
        codecCtx = avcodec_alloc_context3(codec);
        avcodec_parameters_to_context(codecCtx, avCodecPara);
        if (codecCtx == NULL) {
            printf("Cannot alloc context.");
            break;
        }

        //================================  打开解码器 ===================================//
        if ((ret = avcodec_open2(codecCtx, codec, NULL)) < 0) { // 具体采用什么解码器ffmpeg经过封装 我们无须知道
            printf("cannot open decoder\n");
            break;
        }

        int w = codecCtx->width;//视频宽度
        int h = codecCtx->height;//视频高度

        //=========================== 分配AVPacket结构体 ===============================//
        int       i = 0;//用于帧计数
        pkt = av_packet_alloc();                      //分配一个packet
        av_new_packet(pkt, codecCtx->width * codecCtx->height); //调整packet的数据

        //===========================  读取视频信息 ===============================//
        while (av_read_frame(fmtCtx, pkt) >= 0) { //读取的是一帧视频  数据存入一个AVPacket的结构中
            if (pkt->stream_index == videoStreamIndex) {
                if (avcodec_send_packet(codecCtx, pkt) == 0) {
                    while (avcodec_receive_frame(codecCtx, yuvFrame) == 0) {
                        fwrite(yuvFrame->data[0], 1, w * h, fp);//y
                        fwrite(yuvFrame->data[1], 1, w * h / 4, fp);//u
                        fwrite(yuvFrame->data[2], 1, w * h / 4, fp);//v
                    }
                }
                i++;//只计算视频帧
            }
            av_packet_unref(pkt);//重置pkt的内容
        }
        printf("There are %d frames int total.\n", i);
    } while (0);
    //===========================释放所有指针===============================//
    av_packet_free(&pkt);
    avcodec_close(codecCtx);
    avcodec_parameters_free(&avCodecPara);

    av_free(codec);
    av_frame_free(&yuvFrame);
    //avformat_close_input(&fmtCtx);
    //avformat_free_context(fmtCtx);
    return ret;
}