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

use rustfft::num_complex::Complex;
use rustfft::num_traits::Num;

/// Performs conversions between real and complex slices.

pub trait RealToComplex<T> {
    /// Produces a vec of complex, given a slice of reals.
    fn to_complex_vec(&self) -> Vec<Complex<T>>;
}
pub trait ComplexToReal<T> {
    /// Produces a vec of reals, given a slice of complex.
    fn to_real_vec(&self) -> Vec<T>;
}

#[inline(always)]
pub fn real_to_complex<T: Copy + Num>(this: &[T]) -> Vec<Complex<T>> {
    this.iter().fold(Vec::with_capacity(this.len()), |mut acc, sample| {
        acc.push(Complex::from(*sample));

        acc
    })
}

#[inline(always)]
pub fn complex_to_real<T: Copy + Num>(this: &[Complex<T>]) -> Vec<T> {
    this.iter().fold(Vec::with_capacity(this.len()), |mut acc, sample| {
        acc.push(sample.re);

        acc
    })
}

impl<T: Copy + Num> RealToComplex<T> for [T] {
    #[inline(always)]
    fn to_complex_vec(&self) -> Vec<Complex<T>> {
        real_to_complex(self)
    }
}
impl<T: Copy + Num> ComplexToReal<T> for [Complex<T>] {
    #[inline(always)]
    fn to_real_vec(&self) -> Vec<T> {
        complex_to_real(self)
    }
}