You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.6 KiB
60 lines
1.6 KiB
extern crate assert;
|
|
extern crate complex;
|
|
extern crate fft;
|
|
|
|
use complex::c64;
|
|
|
|
mod fixtures;
|
|
|
|
#[test]
|
|
fn complex_forward() {
|
|
let mut data = fixtures::TIME_DATA.to_vec();
|
|
fft::complex::forward(as_c64_mut(&mut data));
|
|
assert::close(&data, &fixtures::FREQUENCY_DATA_FOR_COMPLEX[..], 1e-14);
|
|
}
|
|
|
|
#[test]
|
|
fn complex_inverse() {
|
|
let mut data = fixtures::FREQUENCY_DATA_FOR_COMPLEX.to_vec();
|
|
fft::complex::inverse(as_c64_mut(&mut data));
|
|
assert::close(&data, &fixtures::TIME_DATA[..], 1e-14);
|
|
}
|
|
|
|
#[test]
|
|
fn real_forward() {
|
|
let mut data = fixtures::TIME_DATA.to_vec();
|
|
{
|
|
let mut data = to_c64(&data);
|
|
fft::complex::forward(&mut data);
|
|
assert::close(as_f64(&data), &fixtures::FREQUENCY_DATA_FOR_REAL_UNPACKED[..], 1e-13);
|
|
}
|
|
{
|
|
fft::real::forward(&mut data);
|
|
assert::close(&data, &fixtures::FREQUENCY_DATA_FOR_REAL[..], 1e-13);
|
|
let data = fft::real::unpack(&data);
|
|
assert::close(as_f64(&data), &fixtures::FREQUENCY_DATA_FOR_REAL_UNPACKED[..], 1e-13);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn real_inverse() {
|
|
let mut data = fixtures::FREQUENCY_DATA_FOR_REAL.to_vec();
|
|
fft::real::inverse(&mut data);
|
|
assert::close(&data, &fixtures::TIME_DATA[..], 1e-14);
|
|
}
|
|
|
|
fn as_f64<'l>(slice: &'l [c64]) -> &'l [f64] {
|
|
unsafe {
|
|
std::slice::from_raw_parts(slice.as_ptr() as *const _, 2 * slice.len())
|
|
}
|
|
}
|
|
|
|
fn as_c64_mut<'l>(slice: &'l mut [f64]) -> &'l mut [c64] {
|
|
unsafe {
|
|
std::slice::from_raw_parts_mut(slice.as_mut_ptr() as *mut _, slice.len() / 2)
|
|
}
|
|
}
|
|
|
|
fn to_c64(slice: &[f64]) -> Vec<c64> {
|
|
slice.iter().map(|&re| c64(re, 0.0)).collect()
|
|
}
|