rodio/source/
repeat.rs

1use std::time::Duration;
2
3use crate::source::buffered::Buffered;
4
5use super::SeekError;
6use crate::common::{ChannelCount, SampleRate};
7use crate::Source;
8
9/// Internal function that builds a `Repeat` object.
10pub fn repeat<I>(input: I) -> Repeat<I>
11where
12    I: Source,
13{
14    let input = input.buffered();
15    Repeat {
16        inner: input.clone(),
17        next: input,
18    }
19}
20
21/// A source that repeats the given source.
22pub struct Repeat<I>
23where
24    I: Source,
25{
26    inner: Buffered<I>,
27    next: Buffered<I>,
28}
29
30impl<I> Iterator for Repeat<I>
31where
32    I: Source,
33{
34    type Item = <I as Iterator>::Item;
35
36    #[inline]
37    fn next(&mut self) -> Option<<I as Iterator>::Item> {
38        if let Some(value) = self.inner.next() {
39            return Some(value);
40        }
41
42        self.inner = self.next.clone();
43        self.inner.next()
44    }
45
46    #[inline]
47    fn size_hint(&self) -> (usize, Option<usize>) {
48        // infinite
49        (0, None)
50    }
51}
52
53impl<I> Source for Repeat<I>
54where
55    I: Iterator + Source,
56{
57    #[inline]
58    fn current_span_len(&self) -> Option<usize> {
59        if self.inner.is_exhausted() {
60            self.next.current_span_len()
61        } else {
62            self.inner.current_span_len()
63        }
64    }
65
66    #[inline]
67    fn channels(&self) -> ChannelCount {
68        if self.inner.is_exhausted() {
69            self.next.channels()
70        } else {
71            self.inner.channels()
72        }
73    }
74
75    #[inline]
76    fn sample_rate(&self) -> SampleRate {
77        if self.inner.is_exhausted() {
78            self.next.sample_rate()
79        } else {
80            self.inner.sample_rate()
81        }
82    }
83
84    #[inline]
85    fn total_duration(&self) -> Option<Duration> {
86        None
87    }
88
89    #[inline]
90    fn try_seek(&mut self, pos: Duration) -> Result<(), SeekError> {
91        self.inner.try_seek(pos)
92    }
93}
94
95impl<I> Clone for Repeat<I>
96where
97    I: Source,
98{
99    #[inline]
100    fn clone(&self) -> Repeat<I> {
101        Repeat {
102            inner: self.inner.clone(),
103            next: self.next.clone(),
104        }
105    }
106}
rodio/source/repeat.rs

rodio/source/
repeat.rs
