我可以回答这个问题。以下是一份ffmpeg异步解码的代码示例：
```
#include <iostream>
#include <thread>
#include <mutex>
#include <condition_variable>
#include <queue>
#include <chrono>
extern "C" {
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libavutil/imgutils.h>
#include <libswscale/swscale.h>
}
using namespace std;
// 用于存储解码后的帧
struct Frame {
AVFrame* frame;
int64_t pts;
};
// 解码线程
void decode_thread(AVFormatContext* fmt_ctx, AVCodecContext* codec_ctx, queue<Frame>& frames, mutex& frames_mutex, condition_variable& frames_cv) {
AVPacket pkt;
av_init_packet(&pkt);
pkt.data = nullptr;
pkt.size = 0;
while (av_read_frame(fmt_ctx, &pkt) >= 0) {
if (pkt.stream_index == codec_ctx->stream_index) {
int ret = avcodec_send_packet(codec_ctx, &pkt);
if (ret < 0) {
cerr << "Error sending packet to decoder: " << av_err2str(ret) << endl;
break;
}
while (ret >= 0) {
AVFrame* frame = av_frame_alloc();
ret = avcodec_receive_frame(codec_ctx, frame);
if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
av_frame_free(&frame);
break;
} else if (ret < 0) {
cerr << "Error receiving frame from decoder: " << av_err2str(ret) << endl;
av_frame_free(&frame);
break;
}
Frame f = {frame, av_frame_get_best_effort_timestamp(frame)};
unique_lock<mutex> lock(frames_mutex);
frames.push(f);
lock.unlock();
frames_cv.notify_one();
}
}
av_packet_unref(&pkt);
}
av_packet_unref(&pkt);
}
// 显示线程
void display_thread(queue<Frame>& frames, mutex& frames_mutex, condition_variable& frames_cv) {
AVFrame* frame = nullptr;
AVPixelFormat pix_fmt = AV_PIX_FMT_RGBA;
int width = 0, height = 0;
SwsContext* sws_ctx = nullptr;
uint8_t* buffer = nullptr;
int buffer_size = 0;
while (true) {
unique_lock<mutex> lock(frames_mutex);
frames_cv.wait(lock, [&frames]{ return !frames.empty(); });
Frame f = frames.front();
frames.pop();
lock.unlock();
if (f.frame) {
if (!frame) {
width = f.frame->width;
height = f.frame->height;
pix_fmt = AV_PIX_FMT_RGBA;
sws_ctx = sws_getContext(width, height, f.frame->format, width, height, pix_fmt, SWS_BILINEAR, nullptr, nullptr, nullptr);
buffer_size = av_image_get_buffer_size(pix_fmt, width, height, 1);
buffer = (uint8_t*)av_malloc(buffer_size);
}
sws_scale(sws_ctx, f.frame->data, f.frame->linesize, 0, height, &buffer, &width);
av_frame_free(&f.frame);
frame = av_frame_alloc();
av_image_fill_arrays(frame->data, frame->linesize, buffer, pix_fmt, width, height, 1);
frame->width = width;
frame->height = height;
frame->format = pix_fmt;
frame->pts = f.pts;
// 显示帧
cout << "Displaying frame with PTS " << frame->pts << endl;
av_frame_free(&frame);
}
}
av_free(buffer);
sws_freeContext(sws_ctx);
}
int main(int argc, char* argv[]) {
if (argc < 2) {
cerr << "Usage: " << argv[0] << " <input_file>" << endl;
return 1;
}
av_register_all();
avcodec_register_all();
AVFormatContext* fmt_ctx = nullptr;
int ret = avformat_open_input(&fmt_ctx, argv[1], nullptr, nullptr);
if (ret < 0) {
cerr << "Error opening input file: " << av_err2str(ret) << endl;
return 1;
}
ret = avformat_find_stream_info(fmt_ctx, nullptr);
if (ret < 0) {
cerr << "Error finding stream information: " << av_err2str(ret) << endl;
avformat_close_input(&fmt_ctx);
return 1;
}
AVCodec* codec = nullptr;
int stream_index = av_find_best_stream(fmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, &codec, 0);
if (stream_index < 0) {
cerr << "Error finding video stream: " << av_err2str(stream_index) << endl;
avformat_close_input(&fmt_ctx);
return 1;
}
AVCodecContext* codec_ctx = avcodec_alloc_context3(codec);
if (!codec_ctx) {
cerr << "Error allocating codec context" << endl;
avformat_close_input(&fmt_ctx);
return 1;
}
ret = avcodec_parameters_to_context(codec_ctx, fmt_ctx->streams[stream_index]->codecpar);
if (ret < 0) {
cerr << "Error setting codec parameters: " << av_err2str(ret) << endl;
avcodec_free_context(&codec_ctx);
avformat_close_input(&fmt_ctx);
return 1;
}
ret = avcodec_open2(codec_ctx, codec, nullptr);
if (ret < 0) {
cerr << "Error opening codec: " << av_err2str(ret) << endl;
avcodec_free_context(&codec_ctx);
avformat_close_input(&fmt_ctx);
return 1;
}
queue<Frame> frames;
mutex frames_mutex;
condition_variable frames_cv;
thread decode_th(decode_thread, fmt_ctx, codec_ctx, ref(frames), ref(frames_mutex), ref(frames_cv));
thread display_th(display_thread, ref(frames), ref(frames_mutex), ref(frames_cv));
decode_th.join();
display_th.join();
avcodec_free_context(&codec_ctx);
avformat_close_input(&fmt_ctx);
return 0;
}
```
566
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
