wayclip-daemon 0.2.0

Recording and management of Wayclip daemon
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140

use gstreamer::Element;
use wayclip_core::{
    models::error::WayclipError,
    settings::{
        UserSettings,
        recording::{CodecType, VideoCodec},
    },
};

use crate::common::gst::element::{
    GStreamerElement, GStreamerElementProperty, GStreamerElementPropertyValue,
};

pub struct VideoEncoderFactory;

impl VideoEncoderFactory {
    pub fn build_encoder(user_settings: &UserSettings) -> Result<Element, WayclipError> {
        let codec = &user_settings.recording.video.codec;
        let bitrate = user_settings.recording.video.bitrate_kbps.0;
        let gop = user_settings.recording.video.fps.0.max(1) as i32;

        match codec.get_backend() {
            CodecType::NVIDIA => Self::build_nvidia_pipeline(codec, bitrate, gop),
            CodecType::VAAPI => Self::build_vaapi_pipeline(codec, bitrate, gop),
            CodecType::Software => Self::build_software_pipeline(codec, bitrate, gop),
        }
    }

    fn build_nvidia_pipeline(
        codec: &VideoCodec,
        bitrate: u32,
        gop: i32,
    ) -> Result<Element, WayclipError> {
        GStreamerElement::new(
            codec.get_encoder(),
            vec![
                GStreamerElementProperty::new(
                    "bitrate",
                    GStreamerElementPropertyValue::typed(bitrate),
                ),
                GStreamerElementProperty::new(
                    "gop-size",
                    GStreamerElementPropertyValue::typed(gop),
                ),
                GStreamerElementProperty::new(
                    "rc-mode",
                    GStreamerElementPropertyValue::from_str("cbr"),
                ),
            ],
        )
        .build_element()
    }

    fn build_vaapi_pipeline(
        codec: &VideoCodec,
        bitrate: u32,
        gop: i32,
    ) -> Result<Element, WayclipError> {
        GStreamerElement::new(
            codec.get_encoder(),
            vec![
                GStreamerElementProperty::new(
                    "bitrate",
                    GStreamerElementPropertyValue::typed(bitrate),
                ),
                GStreamerElementProperty::new(
                    "key-int-max",
                    GStreamerElementPropertyValue::typed(gop),
                ),
            ],
        )
        .build_element()
    }

    fn build_software_pipeline(
        codec: &VideoCodec,
        bitrate: u32,
        gop: i32,
    ) -> Result<Element, WayclipError> {
        let threads = std::thread::available_parallelism()
            .map(|n| n.get() as u32)
            .unwrap_or(4);

        let properties = match codec {
            VideoCodec::H264(_) => vec![
                GStreamerElementProperty::new(
                    "bitrate",
                    GStreamerElementPropertyValue::typed(bitrate),
                ),
                GStreamerElementProperty::new(
                    "key-int-max",
                    GStreamerElementPropertyValue::typed(gop),
                ),
                GStreamerElementProperty::new(
                    "speed-preset",
                    GStreamerElementPropertyValue::from_str("ultrafast"),
                ),
                GStreamerElementProperty::new(
                    "tune",
                    GStreamerElementPropertyValue::from_str("zerolatency"),
                ),
                GStreamerElementProperty::new(
                    "threads",
                    GStreamerElementPropertyValue::typed(threads),
                ),
                GStreamerElementProperty::new(
                    "sliced-threads",
                    GStreamerElementPropertyValue::typed(true),
                ),
            ],
            VideoCodec::H265(_) => vec![
                GStreamerElementProperty::new(
                    "bitrate",
                    GStreamerElementPropertyValue::typed(bitrate),
                ),
                GStreamerElementProperty::new(
                    "key-int-max",
                    GStreamerElementPropertyValue::typed(gop),
                ),
                GStreamerElementProperty::new(
                    "speed-preset",
                    GStreamerElementPropertyValue::from_str("ultrafast"),
                ),
            ],
            // AV1 uses target-bitrate instead
            VideoCodec::AV1(_) => vec![
                GStreamerElementProperty::new(
                    "target-bitrate",
                    GStreamerElementPropertyValue::typed(bitrate),
                ),
                GStreamerElementProperty::new(
                    "keyframe-max-dist",
                    GStreamerElementPropertyValue::typed(gop),
                ),
            ],
        };

        GStreamerElement::new(codec.get_encoder(), properties).build_element()
    }
}