scientific-cal 0.2.2

scientific cal
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

use num_complex::Complex;
use num_traits::{Float, FromPrimitive, One, Zero};

use crate::spectrum::{SpectrumError, factory::DefaultFFTFactory};

pub trait FFTApply<T> where
    T: Float + FromPrimitive + Zero + Clone + One + 'static,
{
    fn apply(&self, data: &Vec<T>) -> Result<Vec<Complex<T>>, SpectrumError>;
}

pub trait IFFTApply<T> where
    T: Float + FromPrimitive + Zero + Clone + One + 'static,
{
    fn apply(&self, data: &mut Vec<Complex<T>>) -> Result<Vec<T>, SpectrumError>;
}

pub trait FFTFactory<T>
where
    T: Float + FromPrimitive + Clone + Zero + One + 'static,
{
    fn create_fft(&self) -> Box<dyn FFTApply<T>>;
    fn create_ifft(&self) -> Box<dyn IFFTApply<T>>;
}

pub fn spectrum_factory<T>() -> Box<dyn FFTFactory<T>>
where
    T: Float + FromPrimitive + Clone + Zero + One + 'static,
{
    Box::new(DefaultFFTFactory)
}

pub fn fft<T: Float + FromPrimitive + Clone + Zero + One + 'static>(
    data: &Vec<T>,
) -> Result<Vec<Complex<T>>, SpectrumError> {
    let fft = spectrum_factory::<T>().create_fft();
    fft.apply(data)
}

pub fn ifft<T: Float + FromPrimitive + Clone + Zero + One + 'static>(
    data: &mut Vec<Complex<T>>,
) -> Result<Vec<T>, SpectrumError> {
    let ifft = spectrum_factory::<T>().create_ifft();
    ifft.apply(data)
}