diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /third_party/rust/audio-mixer/benches/benchmark.rs | |
parent | Initial commit. (diff) | |
download | firefox-esr-upstream.tar.xz firefox-esr-upstream.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esrupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/rust/audio-mixer/benches/benchmark.rs')
-rw-r--r-- | third_party/rust/audio-mixer/benches/benchmark.rs | 95 |
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); |