#![allow(bare_trait_objects)]
#![allow(non_snake_case)]
#![feature(test)]
extern crate rustfft;
extern crate test;
use rustfft::algorithm::butterflies::*;
use rustfft::algorithm::*;
use rustfft::num_complex::Complex;
use rustfft::num_traits::Zero;
use rustfft::FftPlannerScalar;
use rustfft::{Direction, Fft, FftDirection, FftNum, Length};
use std::sync::Arc;
use test::Bencher;
struct Noop {
len: usize,
direction: FftDirection,
}
impl<T: FftNum> Fft<T> for Noop {
fn process_with_scratch(&self, _buffer: &mut [Complex<T>], _scratch: &mut [Complex<T>]) {}
fn process_outofplace_with_scratch(
&self,
_input: &mut [Complex<T>],
_output: &mut [Complex<T>],
_scratch: &mut [Complex<T>],
) {
}
fn get_inplace_scratch_len(&self) -> usize {
self.len
}
fn get_outofplace_scratch_len(&self) -> usize {
0
}
}
impl Length for Noop {
fn len(&self) -> usize {
self.len
}
}
impl Direction for Noop {
fn fft_direction(&self) -> FftDirection {
self.direction
}
}
fn bench_planned_f32(b: &mut Bencher, len: usize) {
let mut planner = rustfft::FftPlannerScalar::new();
let fft: Arc<dyn Fft<f32>> = planner.plan_fft_forward(len);
assert_eq!(fft.len(), len);
let mut buffer = vec![Complex::zero(); len];
let mut scratch = vec![Complex::zero(); fft.get_inplace_scratch_len()];
b.iter(|| {
fft.process_with_scratch(&mut buffer, &mut scratch);
});
}
#[bench]
fn planned32_p2_00000064(b: &mut Bencher) {
bench_planned_f32(b, 64);
}
#[bench]
fn planned32_p2_00000128(b: &mut Bencher) {
bench_planned_f32(b, 128);
}
#[bench]
fn planned32_p2_00000256(b: &mut Bencher) {
bench_planned_f32(b, 256);
}
#[bench]
fn planned32_p2_00000512(b: &mut Bencher) {
bench_planned_f32(b, 512);
}
#[bench]
fn planned32_p2_00001024(b: &mut Bencher) {
bench_planned_f32(b, 1024);
}
#[bench]
fn planned32_p2_00002048(b: &mut Bencher) {
bench_planned_f32(b, 2048);
}
#[bench]
fn planned32_p2_00004096(b: &mut Bencher) {
bench_planned_f32(b, 4096);
}
#[bench]
fn planned32_p2_00016384(b: &mut Bencher) {
bench_planned_f32(b, 16384);
}
#[bench]
fn planned32_p2_00065536(b: &mut Bencher) {
bench_planned_f32(b, 65536);
}
#[bench]
fn planned32_p2_01048576(b: &mut Bencher) {
bench_planned_f32(b, 1048576);
}
fn bench_planned_f64(b: &mut Bencher, len: usize) {
let mut planner = rustfft::FftPlannerScalar::new();
let fft: Arc<dyn Fft<f64>> = planner.plan_fft_forward(len);
let mut buffer = vec![Complex::zero(); len];
let mut scratch = vec![Complex::zero(); fft.get_inplace_scratch_len()];
b.iter(|| {
fft.process_with_scratch(&mut buffer, &mut scratch);
});
}
#[bench]
fn planned64_p2_00000064(b: &mut Bencher) {
bench_planned_f64(b, 64);
}
#[bench]
fn planned64_p2_00000128(b: &mut Bencher) {
bench_planned_f64(b, 128);
}
#[bench]
fn planned64_p2_00000256(b: &mut Bencher) {
bench_planned_f64(b, 256);
}
#[bench]
fn planned64_p2_00000512(b: &mut Bencher) {
bench_planned_f64(b, 512);
}
#[bench]
fn planned64_p2_00001024(b: &mut Bencher) {
bench_planned_f64(b, 1024);
}
#[bench]
fn planned64_p2_00002048(b: &mut Bencher) {
bench_planned_f64(b, 2048);
}
#[bench]
fn planned64_p2_00004096(b: &mut Bencher) {
bench_planned_f64(b, 4096);
}
#[bench]
fn planned64_p2_00016384(b: &mut Bencher) {
bench_planned_f64(b, 16384);
}
#[bench]
fn planned64_p2_00065536(b: &mut Bencher) {
bench_planned_f64(b, 65536);
}
#[bench]
fn planned64_p2_01048576(b: &mut Bencher) {
bench_planned_f64(b, 1048576);
}
fn bench_good_thomas(b: &mut Bencher, width: usize, height: usize) {
let mut planner = rustfft::FftPlannerScalar::new();
let width_fft = planner.plan_fft_forward(width);
let height_fft = planner.plan_fft_forward(height);
let fft: Arc<Fft<f32>> = Arc::new(GoodThomasAlgorithm::new(width_fft, height_fft));
let mut buffer = vec![Complex::zero(); width * height];
let mut scratch = vec![Complex::zero(); fft.get_inplace_scratch_len()];
b.iter(|| {
fft.process_with_scratch(&mut buffer, &mut scratch);
});
}
#[bench]
fn good_thomas_0002_3(b: &mut Bencher) {
bench_good_thomas(b, 2, 3);
}
#[bench]
fn good_thomas_0003_4(b: &mut Bencher) {
bench_good_thomas(b, 3, 4);
}
#[bench]
fn good_thomas_0004_5(b: &mut Bencher) {
bench_good_thomas(b, 4, 5);
}
#[bench]
fn good_thomas_0007_32(b: &mut Bencher) {
bench_good_thomas(b, 7, 32);
}
#[bench]
fn good_thomas_0032_27(b: &mut Bencher) {
bench_good_thomas(b, 32, 27);
}
fn bench_good_thomas_setup(b: &mut Bencher, width: usize, height: usize) {
let mut planner = rustfft::FftPlannerScalar::new();
let width_fft = planner.plan_fft_forward(width);
let height_fft = planner.plan_fft_forward(height);
b.iter(|| {
let fft: Arc<Fft<f32>> = Arc::new(GoodThomasAlgorithm::new(
Arc::clone(&width_fft),
Arc::clone(&height_fft),
));
test::black_box(fft);
});
}
#[bench]
fn good_thomas_setup_0002_3(b: &mut Bencher) {
bench_good_thomas_setup(b, 2, 3);
}
#[bench]
fn good_thomas_setup_0003_4(b: &mut Bencher) {
bench_good_thomas_setup(b, 3, 4);
}
#[bench]
fn good_thomas_setup_0004_5(b: &mut Bencher) {
bench_good_thomas_setup(b, 4, 5);
}
#[bench]
fn good_thomas_setup_0007_32(b: &mut Bencher) {
bench_good_thomas_setup(b, 7, 32);
}
#[bench]
fn good_thomas_setup_0032_27(b: &mut Bencher) {
bench_good_thomas_setup(b, 32, 27);
}
#[bench]
fn good_thomas_setup_0256_243(b: &mut Bencher) {
bench_good_thomas_setup(b, 256, 243);
}
#[bench]
fn good_thomas_setup_2048_3(b: &mut Bencher) {
bench_good_thomas_setup(b, 2048, 3);
}
#[bench]
fn good_thomas_setup_2048_2187(b: &mut Bencher) {
bench_good_thomas_setup(b, 2048, 2187);
}
fn bench_mixed_radix(b: &mut Bencher, width: usize, height: usize) {
let mut planner = rustfft::FftPlannerScalar::new();
let width_fft = planner.plan_fft_forward(width);
let height_fft = planner.plan_fft_forward(height);
let fft: Arc<Fft<_>> = Arc::new(MixedRadix::new(width_fft, height_fft));
let mut buffer = vec![
Complex {
re: 0_f32,
im: 0_f32
};
fft.len()
];
let mut scratch = vec![
Complex {
re: 0_f32,
im: 0_f32
};
fft.get_inplace_scratch_len()
];
b.iter(|| {
fft.process_with_scratch(&mut buffer, &mut scratch);
});
}
#[bench]
fn mixed_radix_0002_3(b: &mut Bencher) {
bench_mixed_radix(b, 2, 3);
}
#[bench]
fn mixed_radix_0003_4(b: &mut Bencher) {
bench_mixed_radix(b, 3, 4);
}
#[bench]
fn mixed_radix_0004_5(b: &mut Bencher) {
bench_mixed_radix(b, 4, 5);
}
#[bench]
fn mixed_radix_0007_32(b: &mut Bencher) {
bench_mixed_radix(b, 7, 32);
}
#[bench]
fn mixed_radix_0032_27(b: &mut Bencher) {
bench_mixed_radix(b, 32, 27);
}
fn plan_butterfly_fft(len: usize) -> Arc<Fft<f32>> {
match len {
2 => Arc::new(Butterfly2::new(FftDirection::Forward)),
3 => Arc::new(Butterfly3::new(FftDirection::Forward)),
4 => Arc::new(Butterfly4::new(FftDirection::Forward)),
5 => Arc::new(Butterfly5::new(FftDirection::Forward)),
6 => Arc::new(Butterfly6::new(FftDirection::Forward)),
7 => Arc::new(Butterfly7::new(FftDirection::Forward)),
8 => Arc::new(Butterfly8::new(FftDirection::Forward)),
16 => Arc::new(Butterfly16::new(FftDirection::Forward)),
32 => Arc::new(Butterfly32::new(FftDirection::Forward)),
_ => panic!("Invalid butterfly size: {}", len),
}
}
fn bench_mixed_radix_small(b: &mut Bencher, width: usize, height: usize) {
let width_fft = plan_butterfly_fft(width);
let height_fft = plan_butterfly_fft(height);
let fft: Arc<Fft<_>> = Arc::new(MixedRadixSmall::new(width_fft, height_fft));
let mut signal = vec![
Complex {
re: 0_f32,
im: 0_f32
};
width * height
];
let mut spectrum = signal.clone();
b.iter(|| {
fft.process_with_scratch(&mut signal, &mut spectrum);
});
}
#[bench]
fn mixed_radix_small_0002_3(b: &mut Bencher) {
bench_mixed_radix_small(b, 2, 3);
}
#[bench]
fn mixed_radix_small_0003_4(b: &mut Bencher) {
bench_mixed_radix_small(b, 3, 4);
}
#[bench]
fn mixed_radix_small_0004_5(b: &mut Bencher) {
bench_mixed_radix_small(b, 4, 5);
}
#[bench]
fn mixed_radix_small_0007_32(b: &mut Bencher) {
bench_mixed_radix_small(b, 7, 32);
}
fn bench_good_thomas_small(b: &mut Bencher, width: usize, height: usize) {
let width_fft = plan_butterfly_fft(width);
let height_fft = plan_butterfly_fft(height);
let fft: Arc<Fft<_>> = Arc::new(GoodThomasAlgorithmSmall::new(width_fft, height_fft));
let mut signal = vec![
Complex {
re: 0_f32,
im: 0_f32
};
width * height
];
let mut spectrum = signal.clone();
b.iter(|| {
fft.process_with_scratch(&mut signal, &mut spectrum);
});
}
#[bench]
fn good_thomas_small_0002_3(b: &mut Bencher) {
bench_good_thomas_small(b, 2, 3);
}
#[bench]
fn good_thomas_small_0003_4(b: &mut Bencher) {
bench_good_thomas_small(b, 3, 4);
}
#[bench]
fn good_thomas_small_0004_5(b: &mut Bencher) {
bench_good_thomas_small(b, 4, 5);
}
#[bench]
fn good_thomas_small_0007_32(b: &mut Bencher) {
bench_good_thomas_small(b, 7, 32);
}
#[allow(dead_code)]
fn bench_raders_scalar(b: &mut Bencher, len: usize) {
let mut planner = rustfft::FftPlannerScalar::new();
let inner_fft = planner.plan_fft_forward(len - 1);
let fft: Arc<Fft<_>> = Arc::new(RadersAlgorithm::new(inner_fft));
let mut buffer = vec![
Complex {
re: 0_f32,
im: 0_f32
};
len
];
let mut scratch = vec![
Complex {
re: 0_f32,
im: 0_f32
};
fft.get_inplace_scratch_len()
];
b.iter(|| {
fft.process_with_scratch(&mut buffer, &mut scratch);
});
}
#[allow(dead_code)]
fn bench_bluesteins_scalar_prime(b: &mut Bencher, len: usize) {
let mut planner = rustfft::FftPlannerScalar::new();
let inner_fft = planner.plan_fft_forward((len * 2 - 1).checked_next_power_of_two().unwrap());
let fft: Arc<Fft<f32>> = Arc::new(BluesteinsAlgorithm::new(len, inner_fft));
let mut buffer = vec![Zero::zero(); len];
let mut scratch = vec![Zero::zero(); fft.get_inplace_scratch_len()];
b.iter(|| {
fft.process_with_scratch(&mut buffer, &mut scratch);
});
}
fn bench_radix4(b: &mut Bencher, len: usize) {
assert!(len % 4 == 0);
let fft = Radix4::new(len, FftDirection::Forward);
let mut signal = vec![
Complex {
re: 0_f32,
im: 0_f32
};
len
];
let mut spectrum = signal.clone();
b.iter(|| {
fft.process_outofplace_with_scratch(&mut signal, &mut spectrum, &mut []);
});
}
#[bench]
fn radix4_______64(b: &mut Bencher) {
bench_radix4(b, 64);
}
#[bench]
fn radix4______256(b: &mut Bencher) {
bench_radix4(b, 256);
}
#[bench]
fn radix4_____1024(b: &mut Bencher) {
bench_radix4(b, 1024);
}
#[bench]
fn radix4____65536(b: &mut Bencher) {
bench_radix4(b, 65536);
}
fn bench_64_radix4(b: &mut Bencher, len: usize) {
assert!(len % 4 == 0);
let fft = Radix4::new(len, FftDirection::Forward);
let mut signal = vec![
Complex {
re: 0_f64,
im: 0_f64
};
len
];
let mut spectrum = signal.clone();
b.iter(|| {
fft.process_outofplace_with_scratch(&mut signal, &mut spectrum, &mut []);
});
}
#[bench]
fn radix4_64____64(b: &mut Bencher) {
bench_64_radix4(b, 64);
}
#[bench]
fn radix4_64___256(b: &mut Bencher) {
bench_64_radix4(b, 256);
}
#[bench]
fn radix4_64__1024(b: &mut Bencher) {
bench_64_radix4(b, 1024);
}
#[bench]
fn radix4_64_65536(b: &mut Bencher) {
bench_64_radix4(b, 65536);
}
fn get_mixed_radix_power2(len: usize) -> Arc<dyn Fft<f32>> {
match len {
8 => Arc::new(Butterfly8::new(FftDirection::Forward)),
16 => Arc::new(Butterfly16::new(FftDirection::Forward)),
32 => Arc::new(Butterfly32::new(FftDirection::Forward)),
_ => {
let zeroes = len.trailing_zeros();
assert!(zeroes % 2 == 0);
let half_zeroes = zeroes / 2;
let inner = get_mixed_radix_power2(1 << half_zeroes);
Arc::new(MixedRadix::new(Arc::clone(&inner), inner))
}
}
}
fn bench_mixed_radix_power2(b: &mut Bencher, len: usize) {
let fft = get_mixed_radix_power2(len);
let mut buffer = vec![Zero::zero(); len];
let mut scratch = vec![Zero::zero(); fft.get_inplace_scratch_len()];
b.iter(|| {
fft.process_with_scratch(&mut buffer, &mut scratch);
});
}
#[bench]
fn mixed_radix_power2__00000256(b: &mut Bencher) {
bench_mixed_radix_power2(b, 256);
}
#[bench]
fn mixed_radix_power2__00001024(b: &mut Bencher) {
bench_mixed_radix_power2(b, 1024);
}
#[bench]
fn mixed_radix_power2__00004096(b: &mut Bencher) {
bench_mixed_radix_power2(b, 4096);
}
#[bench]
fn mixed_radix_power2__00065536(b: &mut Bencher) {
bench_mixed_radix_power2(b, 65536);
}
fn get_mixed_radix_inline_power2(len: usize) -> Arc<dyn Fft<f32>> {
match len {
8 => Arc::new(Butterfly8::new(FftDirection::Forward)),
16 => Arc::new(Butterfly16::new(FftDirection::Forward)),
32 => Arc::new(Butterfly32::new(FftDirection::Forward)),
_ => {
let zeroes = len.trailing_zeros();
assert!(zeroes % 2 == 0);
let half_zeroes = zeroes / 2;
let inner = get_mixed_radix_inline_power2(1 << half_zeroes);
Arc::new(MixedRadix::new(Arc::clone(&inner), inner))
}
}
}
fn bench_mixed_radix_inline_power2(b: &mut Bencher, len: usize) {
let fft = get_mixed_radix_inline_power2(len);
let mut buffer = vec![Zero::zero(); len];
let mut scratch = vec![Zero::zero(); fft.get_inplace_scratch_len()];
b.iter(|| {
fft.process_with_scratch(&mut buffer, &mut scratch);
});
}
#[bench]
fn mixed_radix_power2_inline__00000256(b: &mut Bencher) {
bench_mixed_radix_inline_power2(b, 256);
}
#[bench]
fn mixed_radix_power2_inline__00001024(b: &mut Bencher) {
bench_mixed_radix_inline_power2(b, 1024);
}
#[bench]
fn mixed_radix_power2_inline__00004096(b: &mut Bencher) {
bench_mixed_radix_inline_power2(b, 4096);
}
#[bench]
fn mixed_radix_power2_inline__00065536(b: &mut Bencher) {
bench_mixed_radix_inline_power2(b, 65536);
}
fn bench_butterfly32(b: &mut Bencher, len: usize) {
let mut planner = rustfft::FftPlannerScalar::new();
let fft: Arc<dyn Fft<f32>> = planner.plan_fft_forward(len);
let mut buffer = vec![Complex::zero(); len * 10];
let mut scratch = vec![Complex::zero(); fft.get_inplace_scratch_len()];
b.iter(|| {
fft.process_with_scratch(&mut buffer, &mut scratch);
});
}
#[bench]
fn butterfly32_02(b: &mut Bencher) {
bench_butterfly32(b, 2);
}
#[bench]
fn butterfly32_03(b: &mut Bencher) {
bench_butterfly32(b, 3);
}
#[bench]
fn butterfly32_04(b: &mut Bencher) {
bench_butterfly32(b, 4);
}
#[bench]
fn butterfly32_05(b: &mut Bencher) {
bench_butterfly32(b, 5);
}
#[bench]
fn butterfly32_06(b: &mut Bencher) {
bench_butterfly32(b, 6);
}
#[bench]
fn butterfly32_07(b: &mut Bencher) {
bench_butterfly32(b, 7);
}
#[bench]
fn butterfly32_08(b: &mut Bencher) {
bench_butterfly32(b, 8);
}
#[bench]
fn butterfly32_09(b: &mut Bencher) {
bench_butterfly32(b, 9);
}
#[bench]
fn butterfly32_11(b: &mut Bencher) {
bench_butterfly32(b, 11);
}
#[bench]
fn butterfly32_12(b: &mut Bencher) {
bench_butterfly32(b, 12);
}
#[bench]
fn butterfly32_16(b: &mut Bencher) {
bench_butterfly32(b, 16);
}
fn bench_butterfly64(b: &mut Bencher, len: usize) {
let mut planner = rustfft::FftPlannerScalar::new();
let fft: Arc<dyn Fft<f64>> = planner.plan_fft_forward(len);
let mut buffer = vec![Complex::zero(); len * 10];
let mut scratch = vec![Complex::zero(); fft.get_inplace_scratch_len()];
b.iter(|| {
fft.process_with_scratch(&mut buffer, &mut scratch);
});
}
#[bench]
fn butterfly64_02(b: &mut Bencher) {
bench_butterfly64(b, 2);
}
#[bench]
fn butterfly64_03(b: &mut Bencher) {
bench_butterfly64(b, 3);
}
#[bench]
fn butterfly64_04(b: &mut Bencher) {
bench_butterfly64(b, 4);
}
#[bench]
fn butterfly64_05(b: &mut Bencher) {
bench_butterfly64(b, 5);
}
#[bench]
fn butterfly64_06(b: &mut Bencher) {
bench_butterfly64(b, 6);
}
#[bench]
fn butterfly64_07(b: &mut Bencher) {
bench_butterfly64(b, 7);
}
#[bench]
fn butterfly64_08(b: &mut Bencher) {
bench_butterfly64(b, 8);
}
#[bench]
fn butterfly64_09(b: &mut Bencher) {
bench_butterfly64(b, 9);
}
#[bench]
fn butterfly64_11(b: &mut Bencher) {
bench_butterfly64(b, 11);
}
#[bench]
fn butterfly64_12(b: &mut Bencher) {
bench_butterfly64(b, 12);
}
#[bench]
fn butterfly64_16(b: &mut Bencher) {
bench_butterfly64(b, 16);
}
fn bench_bluesteins_setup(b: &mut Bencher, len: usize) {
let inner_len = (len * 2 - 1).next_power_of_two();
let inner_fft = FftPlannerScalar::<f32>::new().plan_fft_forward(inner_len);
b.iter(|| {
test::black_box(BluesteinsAlgorithm::new(len, Arc::clone(&inner_fft)));
});
}
#[bench]
fn setup_bluesteins_0017(b: &mut Bencher) {
bench_bluesteins_setup(b, 17);
}
#[bench]
fn setup_bluesteins_0055(b: &mut Bencher) {
bench_bluesteins_setup(b, 55);
}
#[bench]
fn setup_bluesteins_0117(b: &mut Bencher) {
bench_bluesteins_setup(b, 117);
}
#[bench]
fn setup_bluesteins_0555(b: &mut Bencher) {
bench_bluesteins_setup(b, 555);
}
#[bench]
fn setup_bluesteins_1117(b: &mut Bencher) {
bench_bluesteins_setup(b, 1117);
}
#[bench]
fn setup_bluesteins_5555(b: &mut Bencher) {
bench_bluesteins_setup(b, 5555);
}