1
// 初始化和配置一个 H.264 视频编码器,并将其与 RTSP 输出流关联
2
int PushOpencv::open_codec(int width, int height, int den) {
3
    int ret = 0;
4
    avformat_network_init();
5
    // const AVCodec *codec = avcodec_find_encoder(AV_CODEC_ID_H264);  // CPU初始编码器
6

7
    const AVCodec *codec = avcodec_find_encoder_by_name("h264_nvenc"); // GPU硬件加速编码器
8

9
    if (!codec)
10
    {
11
        throw std::logic_error("Can`t find h264 encoder!"); // 找不到264编码器
12
    }
13
    // b 创建编码器上下文
14
    outputVc = avcodec_alloc_context3(codec);
15
    if (!outputVc)
16
    {
17
        throw std::logic_error("avcodec_alloc_context3 failed!"); // 创建编码器失败
18
    }
19
    // c 配置编码器参数
20
    outputVc->flags |= AV_CODEC_FLAG_GLOBAL_HEADER; // 全局参数
21
    outputVc->codec_id = codec->id;
22
    outputVc->codec_type = AVMEDIA_TYPE_VIDEO;
23
    outputVc->thread_count = 4;
24

25
    outputVc->bit_rate = 1024 * 1024 * 8; // 压缩后每秒视频的bit位大小为50kb
26
    outputVc->width = width;
27
    outputVc->height = height;
28
    outputVc->time_base = {1, den};
29
    outputVc->framerate = {den, 1};
30

31
    outputVc->gop_size = 5;
32
    outputVc->max_b_frames = 0;
33
    outputVc->qmax = 30;
34
    outputVc->qmin = 10;
35
    outputVc->pix_fmt = AV_PIX_FMT_YUV420P;
36

37
    // // CPU进行编码配置
38
    // av_opt_set(outputVc->priv_data, "preset", "ultrafast", 0); //preset 决定了编码器的速度与质量权衡,superfast 是一种快速模式。
39
    // av_opt_set(outputVc->priv_data, "tune", "zerolatency", 0); //tune 的 zerolatency 配置适合实时流媒体应用,可减少编码延迟。
40
    // av_opt_set(outputVc->priv_data, "profile", "main", 0);  // Main profile 兼容性更好
41
    // av_opt_set(outputVc->priv_data, "nal-hrd", "none", 0);  // 关闭 HRD,减少延迟
42
    // // profile 是 H.264 的一种简化模式,适合低延迟、低复杂度的编码场景::防止转换成webrtc播放卡顿
43
    // av_opt_set(outputVc->priv_data, "profile", "baseline", 0);    // 设置 baseline profile
44

45
    //GPU硬解码详细配置参数
46
    av_opt_set(outputVc->priv_data, "preset", "p3", 0);   // 低延迟高质量
47
    av_opt_set(outputVc->priv_data, "tune", "ll", 0);
48
    av_opt_set(outputVc->priv_data, "profile", "main", 0);
49
    av_opt_set(outputVc->priv_data, "rc", "vbr", 0); //cbr (Constant Bitrate): 恒定比特率模式。编码器将尽量保持视频流的比特率不变,适用于带宽稳定的场景。vbr (Variable Bitrate): 可变比特率模式。编码器根据视频内容的复杂度调整比特率,适用于不太受带宽限制的场景,能够在保持相同视频质量的同时节省带宽。
50
    // av_opt_set(outputVc->priv_data, "bitrate", "4000000", 0);   // 4Mbps
51
    av_opt_set(outputVc->priv_data, "maxrate", "4000000", 0);
52
    av_opt_set(outputVc->priv_data, "bufsize", "8000000", 0);
53
    av_opt_set(outputVc->priv_data, "gop", "30", 0);
54
    av_opt_set(outputVc->priv_data, "bframes", "0", 0);
55
    av_opt_set(outputVc->priv_data, "spatial-aq", "1", 0);
56
    av_opt_set(outputVc->priv_data, "temporal-aq", "1", 0);
57
    av_opt_set(outputVc->priv_data, "zerolatency", "1", 0);
58

59
    // d 打开编码器上下文
60
    ret = avcodec_open2(outputVc, codec, 0);
61
    std::cout << "avcodec_open2 success!" << std::endl;
62

63
    ret = avformat_alloc_output_context2(&output, nullptr, "rtsp", url.c_str());
64

65
    vs = avformat_new_stream(output, outputVc->codec);
66
    vs->codecpar->codec_tag = 0;
67
    // 从编码器复制参数
68
    avcodec_parameters_from_context(vs->codecpar, outputVc);
69

70
    av_dump_format(output, 0, url.c_str(), 1);
71

72
    ret = avio_open(&output->pb, url.c_str(), AVIO_FLAG_WRITE);
73
    return ret;
74
}