56 lines
1.6 KiB
Rust
56 lines
1.6 KiB
Rust
|
#[cfg(target_os = "macos")]
|
||
|
|
use std::hint::black_box;
|
||
|
|
|
||
|
|
#[cfg(target_os = "macos")]
|
||
|
|
use affine_media_capture::macos::audio_buffer::{mix_audio_samples, mix_audio_samples_scalar};
|
||
|
|
#[cfg(target_os = "macos")]
|
||
|
|
use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
|
||
|
|
|
||
|
|
#[cfg(target_os = "macos")]
|
||
|
|
fn generate_test_samples() -> [f32; 1024] {
|
||
|
|
let mut samples = [0.0; 1024];
|
||
|
|
|
||
|
|
// Generate a simple sine wave with some variation
|
||
|
|
for (i, sample) in samples.iter_mut().enumerate() {
|
||
|
|
let t = i as f32 / 1024.0;
|
||
|
|
// Create a complex waveform with multiple frequencies
|
||
|
|
let value = 0.008 * (2.0 * std::f32::consts::PI * t * 5.0).sin()
|
||
|
|
+ 0.004 * (2.0 * std::f32::consts::PI * t * 10.0).cos()
|
||
|
|
+ 0.002 * (2.0 * std::f32::consts::PI * t * 20.0).sin();
|
||
|
|
*sample = value;
|
||
|
|
}
|
||
|
|
|
||
|
|
samples
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(target_os = "macos")]
|
||
|
|
fn bench_audio_mix(c: &mut Criterion) {
|
||
|
|
let mut group = c.benchmark_group("audio mix");
|
||
|
|
|
||
|
|
let input = generate_test_samples();
|
||
|
|
let output = generate_test_samples();
|
||
|
|
|
||
|
|
group.bench_function(BenchmarkId::from_parameter("simd"), |b| {
|
||
|
|
b.iter(|| {
|
||
|
|
let mixed = mix_audio_samples(&input, &output);
|
||
|
|
black_box(mixed);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
group.bench_function(BenchmarkId::from_parameter("scalar"), |b| {
|
||
|
|
b.iter(|| {
|
||
|
|
let mut mixed = vec![0.0; 1024];
|
||
|
|
mix_audio_samples_scalar(&input, &output, &mut mixed, 0, input.len());
|
||
|
|
black_box(mixed);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(target_os = "macos")]
|
||
|
|
criterion_group!(benches, bench_audio_mix);
|
||
|
|
#[cfg(target_os = "macos")]
|
||
|
|
criterion_main!(benches);
|
||
|
|
|
||
|
|
#[cfg(not(target_os = "macos"))]
|
||
|
|
fn main() {}
|