Example #1
0
# -*- coding:utf-8 -*-

import matplotlib.pyplot as plt
import numpy as np
import time
import sys,os
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/..')
import FFT

start_time = time.time()
M_MIN = 1
M_MAX = 100
total_time = np.zeros([int(M_MAX-M_MIN+1),2])
for i in range(M_MIN, M_MAX+1) :
    temp_N = int(np.power(2,i))
    x = np.arange(temp_N)
    X = FFT.DIT(x)
    temp_time = time.time() - start_time
    total_time[i-M_MIN] = [i, temp_time]
    print ("N={0},elapsed_time:{1}".format(temp_N, temp_time) + "[sec]")
elapsed_time = time.time() - start_time
print ("total_time:{0}".format(elapsed_time) + "[sec]")

# もしすべての計算が終了していたら
if (len(total_time) == M_MAX) :
    np.save('total_time_FFT.npy',total_time)

total_time = np.array(total_time).T
plt.plot(total_time[0],total_time[1])
plt.savefig('FFT.png')
Example #2
0
#!/usr/bin/env python
# -*- coding:utf-8 -*-

import sys,os
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/..')
import FFT


f_a = [6, 1, 3, -4]
f_b = [2, -2, -3, 7]

F_a = FFT.DIT(f_a)
F_b = FFT.DIT(f_b)

print("(a)")
print("f(n)="+str(f_a))
print("F(n)="+str(F_a))

print("(b)")
print("f(n)="+str(f_b))
print("F(n)="+str(F_b))
Example #3
0
#!/usr/bin/env python
# -*- coding:utf-8 -*-

import sys,os
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/..')
import FFT


f = [6, 1, 3, -4]

F = FFT.DIT(f)

print("f(n)="+str(f))
print("F(n)="+str(F))