audio_processor_testing_helpers/generators.rs
1// Augmented Audio: Audio libraries and applications
2// Copyright (c) 2022 Pedro Tacla Yamada
3//
4// The MIT License (MIT)
5//
6// Permission is hereby granted, free of charge, to any person obtaining a copy
7// of this software and associated documentation files (the "Software"), to deal
8// in the Software without restriction, including without limitation the rights
9// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10// copies of the Software, and to permit persons to whom the Software is
11// furnished to do so, subject to the following conditions:
12//
13// The above copyright notice and this permission notice shall be included in
14// all copies or substantial portions of the Software.
15//
16// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22// THE SOFTWARE.
23pub use augmented_oscillator::generators::*;
24pub use augmented_oscillator::Oscillator;
25use std::time::Duration;
26
27/// Create a sine wave buffer with this duration
28pub fn sine_buffer(sample_rate: f32, frequency: f32, length: Duration) -> Vec<f32> {
29 oscillator_buffer(sample_rate, frequency, length, sine_generator)
30}
31
32/// Create a sine wave buffer with this duration
33pub fn oscillator_buffer(
34 sample_rate: f32,
35 frequency: f32,
36 length: Duration,
37 generator_fn: fn(f32) -> f32,
38) -> Vec<f32> {
39 let mut source = Oscillator::new(generator_fn);
40 source.set_sample_rate(sample_rate);
41 source.set_frequency(frequency);
42 let mut output = Vec::new();
43 let length_samples = (length.as_secs_f32() * sample_rate).ceil();
44 output.resize(length_samples as usize, 0.0);
45 for sample in &mut output {
46 *sample = source.next_sample();
47 }
48 output
49}