voice_activity_detector 0.2.1

Voice Activity Detector using the Silero VAD model.
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

use std::task::Poll;

use futures::Stream;
use pin_project::pin_project;

use crate::predict::PredictState;
use crate::Sample;

/// Predicts speech in a stream of audio samples.
#[pin_project]
pub struct PredictStream<'a, T, St>
where
    St: Stream,
{
    #[pin]
    pub(super) stream: St,
    pub(super) state: PredictState<'a, T>,
}

impl<T, St> Stream for PredictStream<'_, T, St>
where
    T: Sample,
    St: Stream<Item = T>,
{
    type Item = (Vec<T>, f32);

    fn poll_next(
        self: std::pin::Pin<&mut Self>,
        cx: &mut std::task::Context<'_>,
    ) -> Poll<Option<Self::Item>> {
        let mut this = self.project();
        loop {
            let sample = match this.stream.as_mut().poll_next(cx) {
                Poll::Pending => return Poll::Pending,
                Poll::Ready(None) => return Poll::Ready(None),
                Poll::Ready(Some(next)) => next,
            };
            match this.state.try_next(sample) {
                None => continue,
                Some(value) => return Poll::Ready(Some(value)),
            }
        }
    }
}