Homec4science
Diffusion SP4E Homework (master) Browse

SP4EHW_IM/HW3/PythonTest/test_fft.py
5a38a7ecaf4dmaster

Last Change
Raw File

test_fft.py
View Options

"""
Igor and Mahmoud Homework No. 3.
"""
# Import libraries
import numpy as np
import csv
import sys
class Frequencies:
"""
This class should compute the frequenceis in both 1D and 2D.
"""
@staticmethod
def computeFreq1D(N):
"""
This function is going to compute for 1D
"""
freq = np.fft.fftfreq(N) * N;
return freq
@staticmethod
def computeFreq2D(N):
"""
This function is going to compute for 2D
"""
#Detailed info shall be found in this link:
#https://gitlab.chara.gsu.edu/lebouquj/mircx_pipeline/blob/0bb527a4282a0c09d8c4820ff7f69a7f2aa5a2c8/vis.py
UnityMatrix = np.ones(N);
freq = np.einsum('i,j->ij', UnityMatrix, Frequencies.computeFreq1D(N))
return freq
@staticmethod
def writeFreq(N, Matrix, Name = ""):
"""
This function should convert the input to lists and write them in a file
that saved as Name(Optional)_N(Auto Size)_FFT.csv.
The file will be saved in the current directory of the current
test_fft.py file
"""
fileName=Name
fileName += str(N);
fileName+= "FFT.csv"
with open(fileName, 'w') as csvfile:
temp = csv.writer(csvfile)
temp.writerows(map(lambda x: [x], Matrix))
#map(lambda x: [x], fin_ids)
if __name__ == "__main__":
# The main file callout
# calling the arguments
if len(sys.argv) == 1:
N = int(5); # Test Default value
D = 1;
print("********** Warning **********")
print("The program used default value of 5 as size of the output mat. and one Dim.")
filename="";
pass # passing nothing Just Executing the .py file (Passing the file itself)
elif len(sys.argv) == 2: # The size of the matrices
N = int(sys.argv[1])
D = 1;
filename="";
elif len(sys.argv) == 3: # Input the optional Filename
N = int(sys.argv[1])
D = int(sys.argv[2]);
filename = "";
elif len(sys.argv) == 4: # Input the optional Filename
N = int(sys.argv[1])
D = int(sys.argv[2]);
filename = str(sys.argv[3])
else: # rais an Error for any different inputs
raise ValueError('Invalid Inputs. N (Integer size of the output mat.), filename (optional)')
if D == 1:
OneD = Frequencies.computeFreq1D(N); # 1D problem
Frequencies.writeFreq(N, OneD, filename)
elif D == 2:
TwoD = Frequencies.computeFreq2D(N); # 2D problem
Frequencies.writeFreq(N, TwoD, filename)
else:
raise ValueError("Enter a valid Dimention - One or Two are allowed only")
c4science · Help