aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMarshall Lochbaum <mwlochbaum@gmail.com>2022-05-28 20:38:34 -0400
committerMarshall Lochbaum <mwlochbaum@gmail.com>2022-05-28 20:38:34 -0400
commitbdac4b9ee260cc3dfa0827d0199fa38cb5b3ee2c (patch)
treef81e6520cfe34fc0a26220480e551a0c60a8c838
parent18af8ef85a6b26dafeeab42c9f8b503af9586380 (diff)
Use FFTW's 1d Fourier transform
-rw-r--r--fftw.bqn8
1 files changed, 4 insertions, 4 deletions
diff --git a/fftw.bqn b/fftw.bqn
index 40c900c..55f0cbe 100644
--- a/fftw.bqn
+++ b/fftw.bqn
@@ -3,7 +3,7 @@
plan ← "*:i32"
Fn ← "/usr/lib/libfftw3.so.3"⊸•FFI
-createPlan ← Fn plan‿"fftw_plan_dft"‿"i32"‿"*i32"‿"*f64"‿"&f64"‿"i32"‿"i32"
+createPlan ← Fn plan‿"fftw_plan_dft_1d"‿"i32"‿"*f64"‿"&f64"‿"i32"‿"i32"
destroyPlan ← Fn ""‿"fftw_destroy_plan"‿plan
executePlan ← Fn ""‿"fftw_execute"‿plan
@@ -14,10 +14,10 @@ executePlan ← Fn ""‿"fftw_execute"‿plan
Enc ← ⍉⌽
𝕩 ↩ (2↑≍)⍟(2>=) 𝕩
dir ← -⟜¬ 𝕨≡¯1
- sh ← 1↓≢𝕩
+ sh ← 1⊑≢𝕩
in ← ⥊ Enc 𝕩
- plan‿out ← CreatePlan ⟨≠sh,sh,in,0¨in,dir,2⋆6⟩
+ plan‿out ← CreatePlan ⟨sh,in,0¨in,dir,2⋆6⟩
ExecutePlan ⟨plan⟩
DestroyPlan ⟨plan⟩
- ÷⟜(×´sh˙)⍟(1=dir) Enc⁼ (sh∾2) ⥊ out
+ ÷⟜sh⍟(1=dir) Enc⁼ sh‿2 ⥊ out
}