From bdac4b9ee260cc3dfa0827d0199fa38cb5b3ee2c Mon Sep 17 00:00:00 2001 From: Marshall Lochbaum Date: Sat, 28 May 2022 20:38:34 -0400 Subject: Use FFTW's 1d Fourier transform --- fftw.bqn | 8 ++++---- 1 file 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 } -- cgit v1.2.3