import math
import cmath
import numpy as np
import imageio
import MyModule as mm
import time
x=[1,2,3,4]
x1=mm.complex_idft(mm.complex_dft(x))
x2=mm.complex_idft(mm.fft(x))
x3=mm.ifft(mm.complex_dft(x))
x4=mm.ifft(mm.fft(x))
y=[5,6,7,8]
output1=mm.conv(x,y)
output2=mm.ifft(mm.fft(x)*mm.fft(y))
img=imageio.imread('lena.png')#[256-128:256+128,256-128:256+128]
t1=time.time()
imageio.imwrite('lenareconstructed.png',mm.ifft2d(mm.fft2d(img)))
t2=time.time()
print('img ifft takes %10.2f sec' %( t2-t1))
