fcwt 0.1.3

Fast Continuous Wavelet Transforms
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
64
65
66
67
68
69
70
71
72
73
74

use rustfft::num_complex::Complex;
use core::ops::{Index, IndexMut};
use rayon::prelude::*;


pub struct CwtResult<T> {
    scales: usize,
    samples: usize,
    data: Vec<Vec<Complex<T>>>,
}

impl CwtResult<f32> {
    #[inline]
    pub fn new(scales: usize, samples: usize) -> Self {
        let data = Vec::with_capacity(scales*samples);
        Self {
            scales,
            samples,
            data
        }
    }

    /// Get the number of scales
    #[inline]
    pub fn num_scales(&self) -> usize {
        self.scales
    }

    /// Get the number of samples per scale
    #[inline]
    pub fn num_samples(&self) -> usize {
        self.samples
    }

    #[inline]
    pub fn rows(&self) -> &Vec<Vec<Complex<f32>>> {
        &self.data
    }

    #[inline]
    pub fn push_row(&mut self, value: Vec<Complex<f32>>) {
        self.data.push(value);
    }

    #[inline]
    pub fn clear(&mut self) {
        self.data.clear();
    }

    pub fn normalize(&mut self) {
        self.data.par_iter_mut().for_each(|row| {
            let size = row.len() as f32;

            row.par_iter_mut().for_each(|field| {
                *field = field.unscale(size);
            });
        });
    }

}

impl Index<usize> for CwtResult<f32> {
    type Output = [Complex<f32>];
    
    fn index(&self, y: usize) -> &Self::Output {
        &self.data[y]
    }
}

impl IndexMut<usize> for CwtResult<f32> {    
    fn index_mut(&mut self, y: usize) -> &mut Self::Output {
        &mut self.data[y]
    }
}