视频解码流程
视频解码的具体步骤如下:
查找指定的解码器 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数据。整个流程包括了查找解码器、初始化解析器、分配和打开解码器上下文、读取和解析数据包、发送数据包到解码器、接收解码后的帧并写入文件,以及最后的资源释放。