signal_processing 0.3.0

A signal processing library.
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

use core::{iter::Sum, ops::{DivAssign, MulAssign}};

use num::Float;
use option_trait::Maybe;

pub fn coifaux<T, S>(order: usize, scale: S) -> Option<Vec<T>>
where
    T: Float + Sum + MulAssign + DivAssign,
    S: Maybe<T>
{
    if let Some(mut h) = crate::gen::wavelet::coifwavf::<T>(order)
    {
        let one = T::one();
    
        let scale = scale.into_option()
            .unwrap_or(one);

        h.reverse();
        for h in h.iter_mut()
            .skip(1)
            .step_by(2)
        {
            *h = -*h
        }
        
        let s = h.iter()
            .copied()
            .sum::<T>();
        if !s.is_zero()
        {
            for c in h.iter_mut()
            {
                *c /= s;
                *c *= scale
            }
        }

        return Some(h)
    }
    None
}

#[cfg(test)]
mod test
{
    use linspace::Linspace;

    use crate::plot;

    #[test]
    fn test()
    {
        let phi = crate::gen::wavelet::coifaux(17, ())
            .unwrap();

        plot::plot_curves("ϕ[n]", "plots/phi_n_coifaux.png", [
                &(0.0..phi.len() as f64).linspace(phi.len())
                    .into_iter()
                    .zip(phi)
                    .collect::<Vec<_>>()
            ]).unwrap()
    }
}