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
use std::ops;
pub trait Generator: Sync + Send {
fn generate(&self, range: ops::Range<usize>, sample_rate: u32) -> Vec<i16>;
}
pub struct Silence(());
impl Silence {
pub fn new() -> Self {
Silence(())
}
}
impl Generator for Silence {
fn generate(&self, range: ops::Range<usize>, _: u32) -> Vec<i16> {
range.map(|_| i16::default()).collect::<Vec<_>>()
}
}
pub struct Tone {
frequency: f32,
amplitude: f32,
}
impl Tone {
pub fn new() -> Self {
Self {
frequency: 1000f32,
amplitude: 0.3f32,
}
}
}
impl Generator for Tone {
fn generate(&self, range: ops::Range<usize>, sample_rate: u32) -> Vec<i16> {
use std::f32::consts::PI;
let sample_rate = sample_rate as f32;
range
.into_iter()
.enumerate()
.map(|(i, _)| {
let mag = (i as f32) * self.frequency * 2f32 * PI / sample_rate;
(mag.sin() * self.amplitude * (std::i16::MAX as f32)) as i16
})
.collect()
}
}