waycap-rs 0.1.2

High-level Wayland screen capture library with hardware-accelerated encoding
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

use crate::{
    types::{
        config::{AudioEncoder, QualityPreset, VideoEncoder},
        error::{Result, WaycapError},
    },
    Capture,
};

pub struct CaptureBuilder {
    video_encoder: Option<VideoEncoder>,
    audio_encoder: Option<AudioEncoder>,
    quality_preset: Option<QualityPreset>,
    include_cursor: bool,
    include_audio: bool,
    target_fps: u64,
}

impl Default for CaptureBuilder {
    fn default() -> Self {
        Self::new()
    }
}

impl CaptureBuilder {
    pub fn new() -> Self {
        Self {
            video_encoder: None,
            audio_encoder: None,
            quality_preset: None,
            include_cursor: false,
            include_audio: false,
            target_fps: 60,
        }
    }

    pub fn with_video_encoder(mut self, encoder: VideoEncoder) -> Self {
        self.video_encoder = Some(encoder);
        self
    }

    pub fn with_audio_encoder(mut self, encoder: AudioEncoder) -> Self {
        self.audio_encoder = Some(encoder);
        self
    }

    pub fn with_cursor_shown(mut self) -> Self {
        self.include_cursor = true;
        self
    }

    pub fn with_audio(mut self) -> Self {
        self.include_audio = true;
        self
    }

    pub fn with_quality_preset(mut self, quality: QualityPreset) -> Self {
        self.quality_preset = Some(quality);
        self
    }

    pub fn with_target_fps(mut self, fps: u64) -> Self {
        self.target_fps = fps;
        self
    }

    pub fn build(self) -> Result<Capture> {
        let video_encoder = match self.video_encoder {
            Some(enc) => enc,
            None => {
                return Err(WaycapError::Init(
                    "Video encoder was not specified".to_string(),
                ))
            }
        };

        let quality = match self.quality_preset {
            Some(qual) => qual,
            None => QualityPreset::Medium,
        };

        let mut audio_encoder = AudioEncoder::Opus;
        if self.include_audio {
            audio_encoder = match self.audio_encoder {
                Some(enc) => enc,
                None => {
                    return Err(WaycapError::Init(
                        "Include audio specified but no audio encoder chosen.".to_string(),
                    ))
                }
            }
        }

        Capture::new(
            video_encoder,
            audio_encoder,
            quality,
            self.include_cursor,
            self.include_audio,
            self.target_fps,
        )
    }
}