スズトキヅキ's Vignette

1
#include <thread>
2
#include <atomic>
3
#include <queue>
4

5
extern "C" {
6
#include <libavformat/avformat.h>
7
#include <libavcodec/avcodec.h>
8
}
9

10
std::atomic<bool> running{true};
11
AVFormatContext* fmt_ctx = nullptr;
12
int vidx = -1, aidx = -1;
13

14
std::queue<AVPacket*> video_queue;
15
std::queue<AVPacket*> audio_queue;
16

17
void demux_thread(const char* url) {
18
  //DEMUX
19
}
20

21
void video_decode_thread() {
22
    if (vidx < 0) return;
23
    const AVCodec* codec = avcodec_find_decoder(fmt_ctx->streams[vidx]->codecpar->codec_id);
24
    AVCodecContext* cctx = avcodec_alloc_context3(codec);
25
    avcodec_parameters_to_context(cctx, fmt_ctx->streams[vidx]->codecpar);
26
    avcodec_open2(cctx, codec, nullptr);
27

28
    while (running) {
29
        if (video_queue.empty()) continue;
30
        AVPacket* pkt = video_queue.front();
31
        video_queue.pop();
32

33
        AVFrame* frame = av_frame_alloc();
34
        avcodec_send_packet(cctx, pkt);
35
        avcodec_receive_frame(cctx, frame);
36
        av_frame_free(&frame);
37
        av_packet_free(&pkt);
38
    }
39
    avcodec_free_context(&cctx);
40
}
41

42
void audio_decode_thread() {
43
    if (aidx < 0) return;
44
    const AVCodec* codec = avcodec_find_decoder(fmt_ctx->streams[aidx]->codecpar->codec_id);
45
    AVCodecContext* cctx = avcodec_alloc_context3(codec);
46
    avcodec_parameters_to_context(cctx, fmt_ctx->streams[aidx]->codecpar);
47
    avcodec_open2(cctx, codec, nullptr);
48

49
    while (running) {
50
        if (audio_queue.empty()) continue;
51
        AVPacket* pkt = audio_queue.front();
52
        audio_queue.pop();
53

54
        AVFrame* frame = av_frame_alloc();
55
        avcodec_send_packet(cctx, pkt);
56
        avcodec_receive_frame(cctx, frame);
57
        av_frame_free(&frame);
58
        av_packet_free(&pkt);
59
    }
60
    avcodec_free_context(&cctx);
61
}
62

63
int main() {
64
    av_register_all();
65
    avformat_network_init();
66

67
    std::thread demux(demux_thread, "/test.mp4");
68
    std::thread vdec(video_decode_thread);
69
    std::thread adec(audio_decode_thread);
70

71
    getchar();
72
    running = false;
73

74
    demux.join();
75
    vdec.join();
76
    adec.join();
77
    avformat_close_input(&fmt_ctx);
78
    avformat_network_deinit();
79
    return 0;
80
}

1
// 1. 初始化重采样上下文（设置输入/输出参数）
2
SwrContext* swr_ctx = swr_alloc_set_opts(
3
    nullptr,
4
    av_get_default_channel_layout(2),  // 输出：双声道
5
    AV_SAMPLE_FMT_S16,                // 输出：16位整型
6
    44100,                            // 输出：44100Hz采样率
7
    av_get_default_channel_layout(codec_ctx->channels), // 输入：解码帧声道
8
    codec_ctx->sample_fmt,            // 输入：解码帧采样格式
9
    codec_ctx->sample_rate,           // 输入：解码帧采样率
10
    0, nullptr
11
);
12
swr_init(swr_ctx);
13

14
// 2. 执行重采样（将解码后的PCM帧转换为目标格式）
15
uint8_t* out_buf = nullptr;
16
int out_samples = swr_convert(
17
    swr_ctx,
18
    &out_buf,  // 输出缓冲区
19
    frame->nb_samples,  // 输出样本数
20
    (const uint8_t**)frame->data,  // 输入PCM数据（解码帧）
21
    frame->nb_samples   // 输入样本数
22
);
23

24
// 3. 释放资源
25
swr_free(&swr_ctx);