Adjust pybindings to work with new API

QuState · Feb 4, 2024 · 08068ef · 08068ef
1 parent 98d20e7
commit 08068ef
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pybindings/src/lib.rs b/pybindings/src/lib.rs
@@ -1,10 +1,15 @@
 use numpy::PyReadwriteArray1;
-use phastft::fft_dif;
+use phastft::{fft as fft_rs, planner::Planner};
 use pyo3::prelude::*;
 
 #[pyfunction]
 fn fft(mut reals: PyReadwriteArray1<f64>, mut imags: PyReadwriteArray1<f64>) {
-    fft_dif(reals.as_slice_mut().unwrap(), imags.as_slice_mut().unwrap());
+    let mut planner = Planner::new(reals.len());
+    fft_rs(
+        reals.as_slice_mut().unwrap(),
+        imags.as_slice_mut().unwrap(),
+        &mut planner,
+    );
 }
 
 /// A Python module implemented in Rust.