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), true);
    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, true);
    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()
}