From b27167ac87282646c43e3e453d8e55b5ee4cf640 Mon Sep 17 00:00:00 2001
From: prisae <mail@werthmuller.org>
Date: Tue, 6 Dec 2016 17:07:07 -0600
Subject: [PATCH] replace i/rfft with _fftpack.drfft, ~1/4 speed-up

---
 pyfftlog.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/pyfftlog.py b/pyfftlog.py
index 10c5ed5..c764154 100644
--- a/pyfftlog.py
+++ b/pyfftlog.py
@@ -180,7 +180,9 @@
 """
 import numpy as np
 from scipy.special import loggamma
-from scipy.fftpack import rfft, irfft
+from scipy.fftpack._fftpack import drfft
+# from scipy.fftpack import rfft, irfft
+# from scipy.fftpack.basic import _raw_fft
 
 
 def fhti(n, mu, dlnr, q=0, kr=1, kropt=0):
@@ -594,7 +596,9 @@ def fhtq(a, xsave, tdir=1):
     n = a.size
 
     # normal FFT
-    a = rfft(a)
+    # a = rfft(a)
+    # _raw_fft(a, n, -1, 1, 1, _fftpack.drfft)
+    a = drfft(a, n, 1, 0)
 
     m = np.arange(1, n/2, dtype=int)  # index variable
     if q == 0:  # unbiased (q = 0) transform
@@ -660,7 +664,9 @@ def fhtq(a, xsave, tdir=1):
                 a[-1] /= ar
 
     # normal FFT back
-    a = irfft(a)
+    # a = irfft(a)
+    # _raw_fft(a, n, -1, -1, 1, _fftpack.drfft)
+    a = drfft(a, n, -1, 1)
 
     # reverse the array and at the same time undo the FFTs' multiplication by n
     # => Just reverse the array, the rest is already done in drfft.