视频解码基础讲解

视频解码流程

视频解码的具体步骤如下：

查找指定的解码器 avcodec_find_decoder
根据指定的解码器ID初始化相应裸流的解析器 av_parser_init
分配解码器上下文 avcodec_alloc_context3
打开解码器和关联解码器上下文 avcodec_open2
读取原始裸流 fread
解析出一个完整的数据包 av_parser_parse2
发送给解码器 avcodec_send_packet
接收解码后的帧 avcodec_receive_frame
循环读取帧并将YUV数据写入文件 fwrite
文件读取完毕后冲刷解码器，以读取出缓存在解码器的数据帧 flush decoder
结束

关键函数和示例代码

查找指定的解码器

AVCodec* codec = avcodec_find_decoder(AV_CODEC_ID_H264);
if (!codec) {
    std::cerr << "Codec not found" << std::endl;
    return -1;
}

初始化解析器

AVCodecParserContext* parser = av_parser_init(codec->id);
if (!parser) {
    std::cerr << "Parser not found" << std::endl;
    return -1;
}

分配解码器上下文

AVCodecContext* codecContext = avcodec_alloc_context3(codec);
if (!codecContext) {
    std::cerr << "Could not allocate video codec context" << std::endl;
    return -1;
}

打开解码器

if (avcodec_open2(codecContext, codec, NULL) < 0) {
    std::cerr << "Could not open codec" << std::endl;
    return -1;
}

读取原始裸流

FILE* inputFile = fopen(inputFileName, "rb");
if (!inputFile) {
    std::cerr << "Could not open input file" << std::endl;
    return -1;
}

解析数据包并发送到解码器

uint8_t inbuf[INBUF_SIZE + AV_INPUT_BUFFER_PADDING_SIZE];
AVPacket* packet = av_packet_alloc();
AVFrame* frame = av_frame_alloc();

while (!feof(inputFile)) {
    size_t dataSize = fread(inbuf, 1, INBUF_SIZE, inputFile);
    if (dataSize <= 0)
        break;

    uint8_t* data = inbuf;
    while (dataSize > 0) {
        int ret = av_parser_parse2(parser, codecContext, &packet->data, &packet->size, data, dataSize, AV_NOPTS_VALUE, AV_NOPTS_VALUE, 0);
        if (ret < 0) {
            std::cerr << "Error while parsing" << std::endl;
            return -1;
        }
        data += ret;
        dataSize -= ret;

        if (packet->size)
            decode(codecContext, packet, frame, outputFile);
    }
}

解码函数

void decode(AVCodecContext* codecContext, AVPacket* packet, AVFrame* frame, FILE* outputFile) {
    int ret = avcodec_send_packet(codecContext, packet);
    if (ret < 0) {
        std::cerr << "Error sending a packet for decoding" << std::endl;
        return;
    }

    while (ret >= 0) {
        ret = avcodec_receive_frame(codecContext, frame);
        if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
            return;
        else if (ret < 0) {
            std::cerr << "Error during decoding" << std::endl;
            return;
        }

        // 将YUV数据写入文件
        for (int i = 0; i < frame->height; i++) {
            fwrite(frame->data[0] + i * frame->linesize[0], 1, frame->width, outputFile);
        }
        for (int i = 0; i < frame->height / 2; i++) {
            fwrite(frame->data[1] + i * frame->linesize[1], 1, frame->width / 2, outputFile);
        }
        for (int i = 0; i < frame->height / 2; i++) {
            fwrite(frame->data[2] + i * frame->linesize[2], 1, frame->width / 2, outputFile);
        }
    }
}

冲刷解码器

avcodec_send_packet(codecContext, NULL);
while (avcodec_receive_frame(codecContext, frame) == 0) {
    for (int i = 0; i < frame->height; i++) {
        fwrite(frame->data[0] + i * frame->linesize[0], 1, frame->width, outputFile);
    }
    for (int i = 0; i < frame->height / 2; i++) {
        fwrite(frame->data[1] + i * frame->linesize[1], 1, frame->width / 2, outputFile);
    }
    for (int i = 0; i < frame->height / 2; i++) {
        fwrite(frame->data[2] + i * frame->linesize[2], 1, frame->width / 2, outputFile);
    }
}

关闭文件和释放资源

fclose(inputFile);
fclose(outputFile);
av_parser_close(parser);
avcodec_free_context(&codecContext);
av_frame_free(&frame);
av_packet_free(&packet);

总结

通过以上步骤和代码示例，你可以了解如何使用FFmpeg进行视频解码，从而将H.264等格式的视频文件解码为YUV数据。整个流程包括了查找解码器、初始化解析器、分配和打开解码器上下文、读取和解析数据包、发送数据包到解码器、接收解码后的帧并写入文件，以及最后的资源释放。