We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 98d20e7 commit 08068efCopy full SHA for 08068ef
pybindings/src/lib.rs
@@ -1,10 +1,15 @@
1
use numpy::PyReadwriteArray1;
2
-use phastft::fft_dif;
+use phastft::{fft as fft_rs, planner::Planner};
3
use pyo3::prelude::*;
4
5
#[pyfunction]
6
fn fft(mut reals: PyReadwriteArray1<f64>, mut imags: PyReadwriteArray1<f64>) {
7
- fft_dif(reals.as_slice_mut().unwrap(), imags.as_slice_mut().unwrap());
+ let mut planner = Planner::new(reals.len());
8
+ fft_rs(
9
+ reals.as_slice_mut().unwrap(),
10
+ imags.as_slice_mut().unwrap(),
11
+ &mut planner,
12
+ );
13
}
14
15
/// A Python module implemented in Rust.
0 commit comments