use crate::buffer::{ChannelConfig, ChannelConfigOptions};
use crate::context::{AsBaseAudioContext, AudioContextRegistration, AudioParamId};
use crate::param::{AudioParam, AudioParamOptions};
use crate::process::{AudioParamValues, AudioProcessor};
use crate::SampleRate;
use super::AudioNode;
pub struct GainOptions {
pub gain: f32,
pub channel_config: ChannelConfigOptions,
}
impl Default for GainOptions {
fn default() -> Self {
Self {
gain: 1.,
channel_config: ChannelConfigOptions::default(),
}
}
}
pub struct GainNode {
registration: AudioContextRegistration,
channel_config: ChannelConfig,
gain: AudioParam,
}
impl AudioNode for GainNode {
fn registration(&self) -> &AudioContextRegistration {
&self.registration
}
fn channel_config_raw(&self) -> &ChannelConfig {
&self.channel_config
}
fn number_of_inputs(&self) -> u32 {
1
}
fn number_of_outputs(&self) -> u32 {
1
}
}
impl GainNode {
pub fn new<C: AsBaseAudioContext>(context: &C, options: GainOptions) -> Self {
context.base().register(move |registration| {
let param_opts = AudioParamOptions {
min_value: f32::MIN,
max_value: f32::MAX,
default_value: 1.,
automation_rate: crate::param::AutomationRate::A,
};
let (param, proc) = context
.base()
.create_audio_param(param_opts, registration.id());
param.set_value_at_time(options.gain, 0.);
let render = GainRenderer { gain: proc };
let node = GainNode {
registration,
channel_config: options.channel_config.into(),
gain: param,
};
(node, Box::new(render))
})
}
pub fn gain(&self) -> &AudioParam {
&self.gain
}
}
struct GainRenderer {
gain: AudioParamId,
}
impl AudioProcessor for GainRenderer {
fn process(
&mut self,
inputs: &[crate::alloc::AudioBuffer],
outputs: &mut [crate::alloc::AudioBuffer],
params: AudioParamValues,
_timestamp: f64,
_sample_rate: SampleRate,
) {
let input = &inputs[0];
let output = &mut outputs[0];
let gain_values = params.get(&self.gain);
*output = input.clone();
output.modify_channels(|channel| {
channel
.iter_mut()
.zip(gain_values.iter())
.for_each(|(value, g)| *value *= g)
});
}
fn tail_time(&self) -> bool {
false
}
}