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
75
76
mod kse;
mod she;
pub use self::kse::KSE;
pub use self::she::SHE;
use fftw::array::*;
use fftw::plan::*;
use fftw::types::*;
use ndarray::*;
pub struct Pair {
pub r: AlignedVec<f64>,
pub c: AlignedVec<c64>,
r2c: R2CPlan64,
c2r: C2RPlan64,
}
impl Pair {
pub fn new(n: usize) -> Self {
let nf = n / 2 + 1;
let mut r = AlignedVec::new(n);
let mut c = AlignedVec::new(nf);
let r2c = R2CPlan::new(&[n], &mut r, &mut c, Flag::MEASURE).unwrap();
let c2r = C2RPlan::new(&[n], &mut c, &mut r, Flag::MEASURE).unwrap();
Pair { r, c, r2c, c2r }
}
pub fn r2c(&mut self) {
self.r2c.r2c(&mut self.r, &mut self.c).unwrap();
let n = 1.0 / self.r.len() as f64;
for v in self.c.iter_mut() {
*v *= n;
}
}
pub fn c2r(&mut self) {
self.c2r.c2r(&mut self.c, &mut self.r).unwrap();
}
pub fn to_r<'a>(&'a mut self, c: &[c64]) -> &'a [f64] {
self.c.copy_from_slice(c);
self.c2r();
&self.r
}
pub fn to_c<'a>(&'a mut self, r: &[f64]) -> &'a [c64] {
self.r.copy_from_slice(r);
self.r2c();
&self.c
}
pub fn real_view(&self) -> ArrayView1<f64> {
ArrayView::from_shape(self.r.len(), &self.r).unwrap()
}
pub fn coeff_view(&self) -> ArrayView1<c64> {
ArrayView::from_shape(self.c.len(), &self.c).unwrap()
}
pub fn real_view_mut(&mut self) -> ArrayViewMut1<f64> {
ArrayViewMut::from_shape(self.r.len(), &mut self.r).unwrap()
}
pub fn coeff_view_mut(&mut self) -> ArrayViewMut1<c64> {
ArrayViewMut::from_shape(self.c.len(), &mut self.c).unwrap()
}
}
impl Clone for Pair {
fn clone(&self) -> Self {
Pair::new(self.r.len())
}
}