HackRF-Treasure-Chest/Software/Universal Radio Hacker/tests/performance/fft.py
2022-09-22 13:46:47 -07:00

43 lines
1.2 KiB
Python

import numpy as np
import time
def init_array(size: int):
result = np.empty((size, size), dtype=np.complex64)
result.real = np.random.rand(size)
result.imag = np.random.rand(size)
return result
def numpy_fft(array, window_size:int):
np.fft.fft(array, window_size)
def scipy_fft(array, window_size: int):
import scipy.fftpack
scipy.fftpack.fft(array, window_size)
def pyfftw_fft(array):
import pyfftw
pyfftw.interfaces.cache.enable()
fft_a = pyfftw.interfaces.numpy_fft.fft(array, threads=2, overwrite_input=True)
if __name__ == '__main__':
print("Size\tIterations\tNumpy\tScipy\tPyFFTW")
iterations = 70
arr = init_array(1024)
numpy_time = time.time()
for _ in range(iterations):
numpy_fft(arr, 1024)
numpy_time = time.time()-numpy_time
scipy_time = time.time()
for _ in range(iterations):
scipy_fft(arr, 1024)
scipy_time = time.time()-scipy_time
pyfftw_time = time.time()
for _ in range(iterations):
pyfftw_fft(arr)
pyfftw_time = time.time()-pyfftw_time
print("{0}\t{1}\t\t\t{2:.4f}\t{3:.4f}\t{4:.4f}".format(1024, iterations, numpy_time, scipy_time, pyfftw_time))