web_audio_api/node/

convolver.rs

use std::any::Any;

use fft_convolver::FFTConvolver;

use crate::buffer::AudioBuffer;
use crate::context::{AudioContextRegistration, BaseAudioContext};
use crate::render::{
    AudioParamValues, AudioProcessor, AudioRenderQuantum, AudioWorkletGlobalScope,
};
use crate::RENDER_QUANTUM_SIZE;

use super::{AudioNode, AudioNodeOptions, ChannelConfig, ChannelCountMode, ChannelInterpretation};

/// Scale buffer by an equal-power normalization
// see - <https://webaudio.github.io/web-audio-api/#dom-convolvernode-normalize>
fn normalize_buffer(buffer: &AudioBuffer) -> f32 {
    let gain_calibration = 0.00125;
    let gain_calibration_sample_rate = 44100.;
    let min_power = 0.000125;

    // Normalize by RMS power.
    let number_of_channels = buffer.number_of_channels();
    let length = buffer.length();
    let sample_rate = buffer.sample_rate();

    let mut power: f32 = buffer
        .channels()
        .iter()
        .map(|c| c.as_slice().iter().map(|&s| s * s).sum::<f32>())
        .sum();

    power = (power / (number_of_channels * length) as f32).sqrt();

    // Protect against accidental overload.
    if !power.is_finite() || power.is_nan() || power < min_power {
        power = min_power;
    }

    let mut scale = 1. / power;

    // Calibrate to make perceived volume same as unprocessed.
    scale *= gain_calibration;

    // Scale depends on sample-rate.
    scale *= gain_calibration_sample_rate / sample_rate;

    // True-stereo compensation.
    if number_of_channels == 4 {
        scale *= 0.5;
    }

    scale
}

/// `ConvolverNode` options
//dictionary ConvolverOptions : AudioNodeOptions {
//  AudioBuffer? buffer;
//  boolean disableNormalization = false;
//};
#[derive(Clone, Debug)]
pub struct ConvolverOptions {
    /// The desired buffer for the ConvolverNode
    pub buffer: Option<AudioBuffer>,
    /// The opposite of the desired initial value for the normalize attribute
    pub disable_normalization: bool,
    /// AudioNode options
    pub audio_node_options: AudioNodeOptions,
}

impl Default for ConvolverOptions {
    fn default() -> Self {
        Self {
            buffer: None,
            disable_normalization: false,
            audio_node_options: AudioNodeOptions {
                channel_count: 2,
                channel_count_mode: ChannelCountMode::ClampedMax,
                channel_interpretation: ChannelInterpretation::Speakers,
            },
        }
    }
}

/// Assert that the channel count is valid for the ConvolverNode
/// see <https://webaudio.github.io/web-audio-api/#audionode-channelcount-constraints>
///
/// # Panics
///
/// This function panics if given count is greater than 2
///
#[track_caller]
#[inline(always)]
fn assert_valid_channel_count(count: usize) {
    assert!(
        count <= 2,
        "NotSupportedError - ConvolverNode channel count cannot be greater than two"
    );
}

/// Assert that the channel count mode is valid for the ConvolverNode
/// see <https://webaudio.github.io/web-audio-api/#audionode-channelcountmode-constraints>
///
/// # Panics
///
/// This function panics if given count mode is [`ChannelCountMode::Max`]
///
#[track_caller]
#[inline(always)]
fn assert_valid_channel_count_mode(mode: ChannelCountMode) {
    assert_ne!(
        mode,
        ChannelCountMode::Max,
        "NotSupportedError - ConvolverNode channel count mode cannot be set to max"
    );
}

/// Processing node which applies a linear convolution effect given an impulse response.
///
/// - MDN documentation: <https://developer.mozilla.org/en-US/docs/Web/API/ConvolverNode>
/// - specification: <https://webaudio.github.io/web-audio-api/#ConvolverNode>
/// - see also: [`BaseAudioContext::create_convolver`]
///
/// The current implementation only handles mono-to-mono convolutions. The provided impulse
/// response buffer and the input signal will be downmixed appropriately.
///
/// # Usage
///
/// ```no_run
/// use std::fs::File;
///
/// use web_audio_api::context::{AudioContext, BaseAudioContext};
/// use web_audio_api::node::{AudioNode, AudioScheduledSourceNode, ConvolverNode, ConvolverOptions};
///
/// let context = AudioContext::default();
/// let file = File::open("samples/vocals-dry.wav").unwrap();
/// let audio_buffer = context.decode_audio_data_sync(file).unwrap();
///
/// let impulse_file = File::open("samples/small-room-response.wav").unwrap();
/// let impulse_buffer = context.decode_audio_data_sync(impulse_file).unwrap();
///
/// let mut src = context.create_buffer_source();
/// src.set_buffer(audio_buffer);
///
/// let mut convolve = ConvolverNode::new(&context, ConvolverOptions::default());
/// convolve.set_buffer(impulse_buffer);
///
/// src.connect(&convolve);
/// convolve.connect(&context.destination());
/// src.start();
/// std::thread::sleep(std::time::Duration::from_millis(4_000));
/// ```
///
/// # Examples
///
/// - `cargo run --release --example convolution`
///
#[derive(Debug)]
pub struct ConvolverNode {
    /// Represents the node instance and its associated audio context
    registration: AudioContextRegistration,
    /// Info about audio node channel configuration
    channel_config: ChannelConfig,
    /// Perform equal power normalization on response buffer
    normalize: bool,
    /// The response buffer, nullable
    buffer: Option<AudioBuffer>,
}

impl AudioNode for ConvolverNode {
    fn registration(&self) -> &AudioContextRegistration {
        &self.registration
    }

    fn channel_config(&self) -> &ChannelConfig {
        &self.channel_config
    }

    fn number_of_inputs(&self) -> usize {
        1
    }

    fn number_of_outputs(&self) -> usize {
        1
    }

    // see <https://webaudio.github.io/web-audio-api/#audionode-channelcount-constraints>
    fn set_channel_count(&self, count: usize) {
        assert_valid_channel_count(count);
        self.channel_config.set_count(count, self.registration());
    }

    // see <https://webaudio.github.io/web-audio-api/#audionode-channelcountmode-constraints>
    fn set_channel_count_mode(&self, mode: ChannelCountMode) {
        assert_valid_channel_count_mode(mode);
        self.channel_config
            .set_count_mode(mode, self.registration());
    }
}

impl ConvolverNode {
    /// returns a `ConvolverNode` instance
    ///
    /// # Arguments
    ///
    /// * `context` - audio context in which the audio node will live.
    /// * `options` - convolver options
    ///
    /// # Panics
    ///
    /// Panics when an AudioBuffer is provided via the `ConvolverOptions` with a sample rate
    /// different from the audio context sample rate.
    pub fn new<C: BaseAudioContext>(context: &C, options: ConvolverOptions) -> Self {
        let ConvolverOptions {
            buffer,
            disable_normalization,
            audio_node_options,
        } = options;

        assert_valid_channel_count(audio_node_options.channel_count);
        assert_valid_channel_count_mode(audio_node_options.channel_count_mode);

        let mut node = context.base().register(move |registration| {
            let renderer = ConvolverRenderer {
                convolvers: None,
                impulse_length: 0,
                impulse_number_of_channels: 0,
                tail_count: 0,
            };

            let node = Self {
                registration,
                channel_config: audio_node_options.into(),
                normalize: !disable_normalization,
                buffer: None,
            };

            (node, Box::new(renderer))
        });

        // renderer has been sent to render thread, we can send it messages
        if let Some(buffer) = buffer {
            node.set_buffer(buffer);
        }

        node
    }

    /// Get the current impulse response buffer
    pub fn buffer(&self) -> Option<&AudioBuffer> {
        self.buffer.as_ref()
    }

    /// Set or update the impulse response buffer
    ///
    /// # Panics
    ///
    /// Panics when the sample rate of the provided AudioBuffer differs from the audio context
    /// sample rate.
    pub fn set_buffer(&mut self, buffer: AudioBuffer) {
        // If the buffer number of channels is not 1, 2, 4, or if the sample-rate of the buffer is
        // not the same as the sample-rate of its associated BaseAudioContext, a NotSupportedError
        // MUST be thrown.

        let sample_rate = buffer.sample_rate();
        assert_eq!(
            sample_rate,
            self.context().sample_rate(),
            "NotSupportedError - sample rate of the convolution buffer must match the audio context"
        );

        let number_of_channels = buffer.number_of_channels();
        assert!(
            [1, 2, 4].contains(&number_of_channels),
            "NotSupportedError - the convolution buffer must consist of 1, 2 or 4 channels"
        );

        // normalize before padding because the length of the buffer affects the scale
        let scale = if self.normalize {
            normalize_buffer(&buffer)
        } else {
            1.
        };

        let mut convolvers = Vec::<FFTConvolver<f32>>::new();
        // @note - value defined by "rule of thumb", to be explored further
        let partition_size = RENDER_QUANTUM_SIZE * 8;

        // Handle multichannel IR
        // cf. https://webaudio.github.io/web-audio-api/#Convolution-channel-configurations
        // Note that in case of mono IR we create 2 convolvers to properly handle stereo input
        for index in 0..number_of_channels.max(2) {
            // make sure we don't try to access an inexisting channel, cf. note above
            let channel = index.min(number_of_channels - 1);

            let mut scaled_channel = vec![0.; buffer.length()];
            scaled_channel
                .iter_mut()
                .zip(buffer.get_channel_data(channel))
                .for_each(|(o, i)| *o = *i * scale);

            let mut convolver = FFTConvolver::<f32>::default();
            convolver
                .init(partition_size, &scaled_channel)
                .expect("Unable to initialize convolution engine");

            convolvers.push(convolver);
        }

        let msg = ConvolverInfosMessage {
            convolvers: Some(convolvers),
            impulse_length: buffer.length(),
            impulse_number_of_channels: number_of_channels,
        };

        self.registration.post_message(msg);
        self.buffer = Some(buffer);
    }

    /// Denotes if the response buffer will be scaled with an equal-power normalization
    pub fn normalize(&self) -> bool {
        self.normalize
    }

    /// Update the `normalize` setting. This will only have an effect when `set_buffer` is called.
    pub fn set_normalize(&mut self, value: bool) {
        self.normalize = value;
    }
}

struct ConvolverInfosMessage {
    convolvers: Option<Vec<FFTConvolver<f32>>>,
    impulse_length: usize,
    impulse_number_of_channels: usize,
}

struct ConvolverRenderer {
    convolvers: Option<Vec<FFTConvolver<f32>>>,
    impulse_length: usize,
    impulse_number_of_channels: usize,
    tail_count: usize,
}

impl AudioProcessor for ConvolverRenderer {
    fn process(
        &mut self,
        inputs: &[AudioRenderQuantum],
        outputs: &mut [AudioRenderQuantum],
        _params: AudioParamValues<'_>,
        _scope: &AudioWorkletGlobalScope,
    ) -> bool {
        // single input/output node
        let input = &inputs[0];
        let output = &mut outputs[0];

        // handle tail time and active processing
        // return early if input is silent and we reached the end of the convolution
        if input.is_silent() {
            if self.tail_count >= self.impulse_length {
                output.make_silent();
                return false;
            }

            self.tail_count += RENDER_QUANTUM_SIZE;
        } else {
            self.tail_count = 0;
        }

        let convolvers = match &mut self.convolvers {
            None => {
                // no convolution buffer set, passthrough
                *output = input.clone();
                return !input.is_silent();
            }
            Some(convolvers) => convolvers,
        };

        // https://webaudio.github.io/web-audio-api/#Convolution-channel-configurations
        match (input.number_of_channels(), self.impulse_number_of_channels) {
            (1, 1) => {
                output.set_number_of_channels(1);

                let i = &input.channel_data(0)[..];
                let o = &mut output.channel_data_mut(0)[..];
                let _ = convolvers[0].process(i, o);
            }
            (1, 2) => {
                output.set_number_of_channels(2);

                let i = &input.channel_data(0)[..];

                let o_left = &mut output.channel_data_mut(0)[..];
                let _ = convolvers[0].process(i, o_left);

                let o_right = &mut output.channel_data_mut(1)[..];
                let _ = convolvers[1].process(i, o_right);
            }
            (2, 1) => {
                output.set_number_of_channels(2);

                let i_left = &input.channel_data(0)[..];
                let o_left = &mut output.channel_data_mut(0)[..];
                let _ = convolvers[0].process(i_left, o_left);

                let i_right = &input.channel_data(1)[..];
                let o_right = &mut output.channel_data_mut(1)[..];
                let _ = convolvers[1].process(i_right, o_right);
            }
            (2, 2) => {
                output.set_number_of_channels(2);

                let i_left = &input.channel_data(0)[..];
                let o_left = &mut output.channel_data_mut(0)[..];
                let _ = convolvers[0].process(i_left, o_left);

                let i_right = &input.channel_data(1)[..];
                let o_right = &mut output.channel_data_mut(1)[..];
                let _ = convolvers[1].process(i_right, o_right);
            }
            (2, 4) => {
                output.set_number_of_channels(4);

                let i_left = &input.channel_data(0)[..];

                let o_0 = &mut output.channel_data_mut(0)[..];
                let _ = convolvers[0].process(i_left, o_0);
                let o_1 = &mut output.channel_data_mut(1)[..];
                let _ = convolvers[1].process(i_left, o_1);

                let i_right = &input.channel_data(1)[..];

                let o_2 = &mut output.channel_data_mut(2)[..];
                let _ = convolvers[2].process(i_right, o_2);
                let o_3 = &mut output.channel_data_mut(3)[..];
                let _ = convolvers[3].process(i_right, o_3);

                // mix output back to stereo
                let o_2 = output.channel_data(2).clone();
                let o_3 = output.channel_data(3).clone();

                output
                    .channel_data_mut(0)
                    .iter_mut()
                    .zip(o_2.iter())
                    .for_each(|(l, sl)| *l += *sl);

                output
                    .channel_data_mut(1)
                    .iter_mut()
                    .zip(o_3.iter())
                    .for_each(|(r, sr)| *r += *sr);

                output.set_number_of_channels(2);
            }
            (1, 4) => {
                output.set_number_of_channels(4);

                let i = &input.channel_data(0)[..];

                let o_0 = &mut output.channel_data_mut(0)[..];
                let _ = convolvers[0].process(i, o_0);
                let o_1 = &mut output.channel_data_mut(1)[..];
                let _ = convolvers[1].process(i, o_1);
                let o_2 = &mut output.channel_data_mut(2)[..];
                let _ = convolvers[2].process(i, o_2);
                let o_3 = &mut output.channel_data_mut(3)[..];
                let _ = convolvers[3].process(i, o_3);

                // mix output back to stereo
                let o_2 = output.channel_data(2).clone();
                let o_3 = output.channel_data(3).clone();

                output
                    .channel_data_mut(0)
                    .iter_mut()
                    .zip(o_2.iter())
                    .for_each(|(l, sl)| *l += *sl);

                output
                    .channel_data_mut(1)
                    .iter_mut()
                    .zip(o_3.iter())
                    .for_each(|(r, sr)| *r += *sr);

                output.set_number_of_channels(2);
            }
            _ => unreachable!(),
        }

        true
    }

    fn onmessage(&mut self, msg: &mut dyn Any) {
        if let Some(msg) = msg.downcast_mut::<ConvolverInfosMessage>() {
            let ConvolverInfosMessage {
                convolvers,
                impulse_length,
                impulse_number_of_channels,
            } = msg;
            // Avoid deallocation in the render thread by swapping the convolver.
            std::mem::swap(&mut self.convolvers, convolvers);
            self.impulse_length = *impulse_length;
            self.impulse_number_of_channels = *impulse_number_of_channels;

            return;
        }

        log::warn!("ConvolverRenderer: Dropping incoming message {msg:?}");
    }
}

#[cfg(test)]
mod tests {
    use float_eq::assert_float_eq;

    use crate::context::{BaseAudioContext, OfflineAudioContext};
    use crate::node::{AudioBufferSourceNode, AudioBufferSourceOptions, AudioScheduledSourceNode};

    use super::*;

    #[test]
    #[should_panic]
    fn test_buffer_sample_rate_matches() {
        let context = OfflineAudioContext::new(1, 128, 44100.);

        let ir = vec![1.];
        let ir = AudioBuffer::from(vec![ir; 1], 48000.); // sample_rate differs
        let options = ConvolverOptions {
            buffer: Some(ir),
            ..ConvolverOptions::default()
        };

        let _ = ConvolverNode::new(&context, options);
    }

    #[test]
    #[should_panic]
    fn test_buffer_must_have_1_2_4_channels() {
        let context = OfflineAudioContext::new(1, 128, 48000.);

        let ir = vec![1.];
        let ir = AudioBuffer::from(vec![ir; 3], 48000.); // three channels
        let options = ConvolverOptions {
            buffer: Some(ir),
            ..ConvolverOptions::default()
        };

        let _ = ConvolverNode::new(&context, options);
    }

    #[test]
    fn test_constructor_options_buffer() {
        let sample_rate = 44100.;
        let mut context = OfflineAudioContext::new(1, 10, sample_rate);

        let ir = vec![1.];
        let calibration = 0.00125;
        let channel_data = vec![0., 1., 0., -1., 0.];
        let expected = [0., calibration, 0., -calibration, 0., 0., 0., 0., 0., 0.];

        // identity ir
        let ir = AudioBuffer::from(vec![ir; 1], sample_rate);
        let options = ConvolverOptions {
            buffer: Some(ir),
            ..ConvolverOptions::default()
        };
        let conv = ConvolverNode::new(&context, options);
        conv.connect(&context.destination());

        let buffer = AudioBuffer::from(vec![channel_data; 1], sample_rate);
        let mut src = context.create_buffer_source();
        src.connect(&conv);
        src.set_buffer(buffer);
        src.start();

        let output = context.start_rendering_sync();

        assert_float_eq!(output.get_channel_data(0), &expected[..], abs_all <= 1E-6);
    }

    fn test_convolve(signal: &[f32], impulse_resp: Option<Vec<f32>>, length: usize) -> AudioBuffer {
        let sample_rate = 44100.;
        let mut context = OfflineAudioContext::new(1, length, sample_rate);

        let input = AudioBuffer::from(vec![signal.to_vec()], sample_rate);
        let mut src = AudioBufferSourceNode::new(&context, AudioBufferSourceOptions::default());
        src.set_buffer(input);
        src.start();

        let mut conv = ConvolverNode::new(&context, ConvolverOptions::default());
        if let Some(ir) = impulse_resp {
            conv.set_buffer(AudioBuffer::from(vec![ir.to_vec()], sample_rate));
        }

        src.connect(&conv);
        conv.connect(&context.destination());

        context.start_rendering_sync()
    }

    #[test]
    fn test_passthrough() {
        let output = test_convolve(&[0., 1., 0., -1., 0.], None, 10);
        let expected = [0., 1., 0., -1., 0., 0., 0., 0., 0., 0.];
        assert_float_eq!(output.get_channel_data(0), &expected[..], abs_all <= 1E-6);
    }

    #[test]
    fn test_empty() {
        let ir = vec![];
        let output = test_convolve(&[0., 1., 0., -1., 0.], Some(ir), 10);
        let expected = [0.; 10];
        assert_float_eq!(output.get_channel_data(0), &expected[..], abs_all <= 1E-6);
    }

    #[test]
    fn test_zeroed() {
        let ir = vec![0., 0., 0., 0., 0., 0.];
        let output = test_convolve(&[0., 1., 0., -1., 0.], Some(ir), 10);
        let expected = [0.; 10];
        assert_float_eq!(output.get_channel_data(0), &expected[..], abs_all <= 1E-6);
    }

    #[test]
    fn test_identity() {
        let ir = vec![1.];
        let calibration = 0.00125;
        let output = test_convolve(&[0., 1., 0., -1., 0.], Some(ir), 10);
        let expected = [0., calibration, 0., -calibration, 0., 0., 0., 0., 0., 0.];
        assert_float_eq!(output.get_channel_data(0), &expected[..], abs_all <= 1E-6);
    }

    #[test]
    fn test_two_id() {
        let ir = vec![1., 1.];
        let calibration = 0.00125;
        let output = test_convolve(&[0., 1., 0., -1., 0.], Some(ir), 10);
        let expected = [
            0.,
            calibration,
            calibration,
            -calibration,
            -calibration,
            0.,
            0.,
            0.,
            0.,
            0.,
        ];
        assert_float_eq!(output.get_channel_data(0), &expected[..], abs_all <= 1E-6);
    }

    #[test]
    fn test_should_have_tail_time() {
        // impulse response of length 256
        const IR_LEN: usize = 256;
        let ir = vec![1.; IR_LEN];

        // unity input signal
        let input = &[1.];

        // render into a buffer of size 512
        let output = test_convolve(input, Some(ir), 512);

        // we expect non-zero output in the range 0 to IR_LEN
        let output = output.channel_data(0).as_slice();
        assert!(!output[..IR_LEN].iter().any(|v| *v <= 1E-6));
        assert_float_eq!(&output[IR_LEN..], &[0.; 512 - IR_LEN][..], abs_all <= 1E-6);
    }

    #[test]
    fn test_channel_config_1_chan_in_1_chan_ir() {
        let number_of_channels = 1;
        let length = 128;
        let sample_rate = 44100.;
        let mut context = OfflineAudioContext::new(number_of_channels, length, sample_rate);

        let input = AudioBuffer::from(vec![vec![1.]], sample_rate);
        let ir = AudioBuffer::from(vec![vec![0., 1.]], sample_rate);

        let mut src = AudioBufferSourceNode::new(
            &context,
            AudioBufferSourceOptions {
                buffer: Some(input),
                ..AudioBufferSourceOptions::default()
            },
        );

        let conv = ConvolverNode::new(
            &context,
            ConvolverOptions {
                buffer: Some(ir),
                disable_normalization: true,
                ..ConvolverOptions::default()
            },
        );

        src.connect(&conv);
        conv.connect(&context.destination());
        src.start();

        let result = context.start_rendering_sync();

        let mut expected = [0.; 128];
        expected[1] = 1.;

        assert_float_eq!(
            result.get_channel_data(0)[..],
            expected[..],
            abs_all <= 1e-7
        );
    }

    #[test]
    fn test_channel_config_1_chan_in_2_chan_ir() {
        let number_of_channels = 2;
        let length = 128;
        let sample_rate = 44100.;
        let mut context = OfflineAudioContext::new(number_of_channels, length, sample_rate);

        let input = AudioBuffer::from(vec![vec![1.]], sample_rate);
        let ir = AudioBuffer::from(vec![vec![0., 1., 0.], vec![0., 0., 1.]], sample_rate);

        let mut src = AudioBufferSourceNode::new(
            &context,
            AudioBufferSourceOptions {
                buffer: Some(input),
                ..AudioBufferSourceOptions::default()
            },
        );

        let conv = ConvolverNode::new(
            &context,
            ConvolverOptions {
                buffer: Some(ir),
                disable_normalization: true,
                ..ConvolverOptions::default()
            },
        );

        src.connect(&conv);
        conv.connect(&context.destination());
        src.start();

        let result = context.start_rendering_sync();

        let mut expected_left = [0.; 128];
        expected_left[1] = 1.;

        let mut expected_right = [0.; 128];
        expected_right[2] = 1.;

        assert_eq!(result.number_of_channels(), 2);
        assert_float_eq!(
            result.get_channel_data(0)[..],
            expected_left[..],
            abs_all <= 1e-7
        );
        assert_float_eq!(
            result.get_channel_data(1)[..],
            expected_right[..],
            abs_all <= 1e-7
        );
    }

    #[test]
    fn test_channel_config_2_chan_in_1_chan_ir() {
        let number_of_channels = 2;
        let length = 128;
        let sample_rate = 44100.;
        let mut context = OfflineAudioContext::new(number_of_channels, length, sample_rate);

        let input = AudioBuffer::from(vec![vec![1., 0.], vec![0., 1.]], sample_rate);
        let ir = AudioBuffer::from(vec![vec![0., 1.]], sample_rate);

        let mut src = AudioBufferSourceNode::new(
            &context,
            AudioBufferSourceOptions {
                buffer: Some(input),
                ..AudioBufferSourceOptions::default()
            },
        );

        let conv = ConvolverNode::new(
            &context,
            ConvolverOptions {
                buffer: Some(ir),
                disable_normalization: true,
                ..ConvolverOptions::default()
            },
        );

        src.connect(&conv);
        conv.connect(&context.destination());
        src.start();

        let result = context.start_rendering_sync();

        let mut expected_left = [0.; 128];
        expected_left[1] = 1.;

        let mut expected_right = [0.; 128];
        expected_right[2] = 1.;

        assert_eq!(result.number_of_channels(), 2);
        assert_float_eq!(
            result.get_channel_data(0)[..],
            expected_left[..],
            abs_all <= 1e-7
        );
        assert_float_eq!(
            result.get_channel_data(1)[..],
            expected_right[..],
            abs_all <= 1e-7
        );
    }

    #[test]
    fn test_channel_config_2_chan_in_2_chan_ir() {
        let number_of_channels = 2;
        let length = 128;
        let sample_rate = 44100.;
        let mut context = OfflineAudioContext::new(number_of_channels, length, sample_rate);

        let input = AudioBuffer::from(vec![vec![1., 0.], vec![0., 1.]], sample_rate);
        let ir = AudioBuffer::from(vec![vec![0., 1., 0.], vec![0., 0., 1.]], sample_rate);

        let mut src = AudioBufferSourceNode::new(
            &context,
            AudioBufferSourceOptions {
                buffer: Some(input),
                ..AudioBufferSourceOptions::default()
            },
        );

        let conv = ConvolverNode::new(
            &context,
            ConvolverOptions {
                buffer: Some(ir),
                disable_normalization: true,
                ..ConvolverOptions::default()
            },
        );

        src.connect(&conv);
        conv.connect(&context.destination());
        src.start();

        let result = context.start_rendering_sync();

        let mut expected_left = [0.; 128];
        expected_left[1] = 1.;

        let mut expected_right = [0.; 128];
        expected_right[3] = 1.;

        assert_eq!(result.number_of_channels(), 2);
        assert_float_eq!(
            result.get_channel_data(0)[..],
            expected_left[..],
            abs_all <= 1e-7
        );
        assert_float_eq!(
            result.get_channel_data(1)[..],
            expected_right[..],
            abs_all <= 1e-7
        );
    }

    #[test]
    fn test_channel_config_2_chan_in_4_chan_ir() {
        let number_of_channels = 2;
        let length = 128;
        let sample_rate = 44100.;
        let mut context = OfflineAudioContext::new(number_of_channels, length, sample_rate);

        let input = AudioBuffer::from(vec![vec![1., 0.], vec![0., 1.]], sample_rate);
        let ir = AudioBuffer::from(
            vec![
                vec![0., 1., 0., 0., 0.], // in 0 -> out 0
                vec![0., 0., 1., 0., 0.], // in 0 -> out 1
                vec![0., 0., 0., 1., 0.], // in 1 -> out 0
                vec![0., 0., 0., 0., 1.], // in 1 -> out 1
            ],
            sample_rate,
        );

        let mut src = AudioBufferSourceNode::new(
            &context,
            AudioBufferSourceOptions {
                buffer: Some(input),
                ..AudioBufferSourceOptions::default()
            },
        );

        let conv = ConvolverNode::new(
            &context,
            ConvolverOptions {
                buffer: Some(ir),
                disable_normalization: true,
                ..ConvolverOptions::default()
            },
        );

        src.connect(&conv);
        conv.connect(&context.destination());
        src.start();

        let result = context.start_rendering_sync();

        let mut expected_left = [0.; 128];
        expected_left[1] = 1.;
        expected_left[4] = 1.;

        let mut expected_right = [0.; 128];
        expected_right[2] = 1.;
        expected_right[5] = 1.;

        assert_eq!(result.number_of_channels(), 2);
        assert_float_eq!(
            result.get_channel_data(0)[..],
            expected_left[..],
            abs_all <= 1e-7
        );
        assert_float_eq!(
            result.get_channel_data(1)[..],
            expected_right[..],
            abs_all <= 1e-7
        );
    }

    #[test]
    fn test_channel_config_1_chan_in_4_chan_ir() {
        let number_of_channels = 2;
        let length = 128;
        let sample_rate = 44100.;
        let mut context = OfflineAudioContext::new(number_of_channels, length, sample_rate);

        let input = AudioBuffer::from(vec![vec![1., 0.]], sample_rate);
        let ir = AudioBuffer::from(
            vec![
                vec![0., 1., 0., 0., 0.], // in 0 -> out 0
                vec![0., 0., 1., 0., 0.], // in 0 -> out 1
                vec![0., 0., 0., 1., 0.], // in 0 -> out 0
                vec![0., 0., 0., 0., 1.], // in 0 -> out 1
            ],
            sample_rate,
        );

        let mut src = AudioBufferSourceNode::new(
            &context,
            AudioBufferSourceOptions {
                buffer: Some(input),
                ..AudioBufferSourceOptions::default()
            },
        );

        let conv = ConvolverNode::new(
            &context,
            ConvolverOptions {
                buffer: Some(ir),
                disable_normalization: true,
                ..ConvolverOptions::default()
            },
        );

        src.connect(&conv);
        conv.connect(&context.destination());
        src.start();

        let result = context.start_rendering_sync();

        let mut expected_left = [0.; 128];
        expected_left[1] = 1.;
        expected_left[3] = 1.;

        let mut expected_right = [0.; 128];
        expected_right[2] = 1.;
        expected_right[4] = 1.;

        assert_eq!(result.number_of_channels(), 2);
        assert_float_eq!(
            result.get_channel_data(0)[..],
            expected_left[..],
            abs_all <= 1e-7
        );
        assert_float_eq!(
            result.get_channel_data(1)[..],
            expected_right[..],
            abs_all <= 1e-7
        );
    }
}