1 files changed, 95 insertions, 0 deletions
diff --git a/third_party/rust/audio-mixer/benches/benchmark.rs b/third_party/rust/audio-mixer/benches/benchmark.rs
new file mode 100644
index 0000000000..11c47e55c7
--- /dev/null
+++ b/third_party/rust/audio-mixer/benches/benchmark.rs
@@ -0,0 +1,95 @@
+use audio_mixer::{Channel, Mixer};
+use criterion::{black_box, criterion_group, criterion_main, Criterion};
+
+use std::any::{Any, TypeId};
+
+fn criterion_benchmark(c: &mut Criterion) {
+    let frames = 512;
+    c.bench_function("downmix_f32", |b| {
+        b.iter(|| downmix::<f32>(black_box(frames)))
+    });
+    c.bench_function("downmix_i16", |b| {
+        b.iter(|| downmix::<i16>(black_box(frames)))
+    });
+    c.bench_function("upmix_f32", |b| b.iter(|| upmix::<f32>(black_box(frames))));
+    c.bench_function("upmix_i16", |b| b.iter(|| upmix::<i16>(black_box(frames))));
+}
+
+fn downmix<T>(frames: usize)
+where
+    T: Clone + Default + From<u8> + ?Sized + Any,
+{
+    // Downmix from 5.1 to stereo.
+    let input_channels = [
+        Channel::FrontLeft,
+        Channel::FrontRight,
+        Channel::FrontCenter,
+        Channel::LowFrequency,
+        Channel::BackLeft,
+        Channel::BackRight,
+    ];
+    let output_channels = [Channel::FrontLeft, Channel::FrontRight];
+    mix::<T>(&input_channels, &output_channels, frames);
+}
+
+fn upmix<T>(frames: usize)
+where
+    T: Clone + Default + From<u8> + ?Sized + Any,
+{
+    // upmix from mono to stereo.
+    let input_channels = [Channel::FrontCenter];
+    let output_channels = [Channel::FrontLeft, Channel::FrontRight];
+    mix::<T>(&input_channels, &output_channels, frames);
+}
+
+fn mix<T>(input_channels: &[Channel], output_channels: &[Channel], frames: usize)
+where
+    T: Clone + Default + From<u8> + ?Sized + Any,
+{
+    if TypeId::of::<T>() == TypeId::of::<f32>() {
+        let (input_buffer, mut output_buffer) = create_buffers::<f32>(
+            input_channels.len() * frames,
+            output_channels.len() * frames,
+        );
+        let mut in_buf = input_buffer.chunks(input_channels.len());
+        let mut out_buf = output_buffer.chunks_mut(output_channels.len());
+        let mixer = Mixer::<f32>::new(input_channels, output_channels);
+        for _ in 0..frames {
+            mixer.mix(in_buf.next().unwrap(), out_buf.next().unwrap());
+        }
+    } else if TypeId::of::<T>() == TypeId::of::<i16>() {
+        let (input_buffer, mut output_buffer) = create_buffers::<i16>(
+            input_channels.len() * frames,
+            output_channels.len() * frames,
+        );
+        let mut in_buf = input_buffer.chunks(input_channels.len());
+        let mut out_buf = output_buffer.chunks_mut(output_channels.len());
+        let mixer = Mixer::<i16>::new(input_channels, output_channels);
+        for _ in 0..frames {
+            mixer.mix(in_buf.next().unwrap(), out_buf.next().unwrap());
+        }
+    } else {
+        panic!("Unsupport type");
+    }
+}
+
+fn create_buffers<T: Clone + Default + From<u8>>(
+    input_size: usize,
+    output_size: usize,
+) -> (Vec<T>, Vec<T>) {
+    let mut input_buffer = default_buffer::<T>(input_size);
+    for (i, data) in input_buffer.iter_mut().enumerate() {
+        *data = T::from((i + 1) as u8);
+    }
+    let output_buffer = default_buffer::<T>(output_size);
+    (input_buffer, output_buffer)
+}
+
+fn default_buffer<T: Clone + Default>(size: usize) -> Vec<T> {
+    let mut v = Vec::with_capacity(size);
+    v.resize(size, T::default());
+    v
+}
+
+criterion_group!(benches, criterion_benchmark);
+criterion_main!(benches);