libwebrtc 0.3.35

Livekit safe bindings to libwebrtc
Documentation 
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

// Copyright 2025 LiveKit, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use cxx::UniquePtr;
use webrtc_sys::apm::ffi as sys_apm;

use crate::{RtcError, RtcErrorType};

pub struct AudioProcessingModule {
    sys_handle: UniquePtr<sys_apm::AudioProcessingModule>,
}

impl AudioProcessingModule {
    pub fn new(
        echo_canceller_enabled: bool,
        gain_controller_enabled: bool,
        high_pass_filter_enabled: bool,
        noise_suppression_enabled: bool,
    ) -> Self {
        Self {
            sys_handle: unsafe {
                sys_apm::create_apm(
                    echo_canceller_enabled,
                    gain_controller_enabled,
                    high_pass_filter_enabled,
                    noise_suppression_enabled,
                )
            },
        }
    }

    pub fn process_stream(
        &mut self,
        data: &mut [i16],
        sample_rate: i32,
        num_channels: i32,
    ) -> Result<(), RtcError> {
        let samples_per_10ms = (sample_rate as usize / 100) * num_channels as usize;
        assert!(
            data.len() % samples_per_10ms == 0 && data.len() >= samples_per_10ms,
            "slice must have a multiple of 10ms worth of samples"
        );

        for chunk in data.chunks_mut(samples_per_10ms) {
            if unsafe {
                self.sys_handle.pin_mut().process_stream(
                    chunk.as_mut_ptr(),
                    chunk.len(),
                    chunk.as_mut_ptr(),
                    chunk.len(),
                    sample_rate,
                    num_channels,
                )
            } != 0
            {
                return Err(RtcError {
                    error_type: RtcErrorType::Internal,
                    message: "Failed to process stream".to_string(),
                });
            }
        }

        Ok(())
    }

    pub fn process_reverse_stream(
        &mut self,
        data: &mut [i16],
        sample_rate: i32,
        num_channels: i32,
    ) -> Result<(), RtcError> {
        let samples_per_10ms = (sample_rate as usize / 100) * num_channels as usize;
        assert!(
            data.len() % samples_per_10ms == 0 && data.len() >= samples_per_10ms,
            "slice must have a multiple of 10ms worth of samples"
        );

        for chunk in data.chunks_mut(samples_per_10ms) {
            if unsafe {
                self.sys_handle.pin_mut().process_reverse_stream(
                    chunk.as_mut_ptr(),
                    chunk.len(),
                    chunk.as_mut_ptr(),
                    chunk.len(),
                    sample_rate,
                    num_channels,
                )
            } != 0
            {
                return Err(RtcError {
                    error_type: RtcErrorType::Internal,
                    message: "Failed to process reverse stream".to_string(),
                });
            }
        }

        Ok(())
    }

    pub fn set_stream_delay_ms(&mut self, delay_ms: i32) -> Result<(), RtcError> {
        if self.sys_handle.pin_mut().set_stream_delay_ms(delay_ms) == 0 {
            Ok(())
        } else {
            Err(RtcError {
                error_type: RtcErrorType::Internal,
                message: "Failed to set stream delay".to_string(),
            })
        }
    }
}