gao/analyseScript
1
0
Fork 0
Code Issues Pull Requests repo.project_board Releases Wiki Activity

update

Browse Source
This commit is contained in:
Jianshun Gao 2023-05-19 17:23:20 +02:00
parent 759f772976
commit 8fddd8bb21
1 changed files with 37 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
Analyser/FFTAnalyser.py
View File

@ -7,49 +7,44 @@ from datetime import datetime
import xrft import xrft
import finufft import finufft
def fft(dataArray, **kwargs):
return xrft.fft(dataArray, **kwargs)
class FFTAnalyser(): def ifft(dataArray, **kwargs):
return xrft.ifft(dataArray, **kwargs)
def __init__(self) -> None:
pass
def fft(self, dataArray, **kwargs):
return xrft.fft(dataArray, **kwargs)
def ifft(self, dataArray, **kwargs):
return xrft.ifft(dataArray, **kwargs)
def fft_nutou(self, dataArray, modeNum):
data = dataArray.to_numpy()
data = data.astype('complex128')
time = dataArray[dataArray.dims[0]].to_numpy()
if isinstance(time[0], type(np.datetime64(500,'s'))):
time = time.astype(float)
time = time - time[0]
freqUpLim = 1 / np.min(np.abs(time - np.roll(time, 1))) * 1e9
else:
time = time.astype(float)
time = time - time[0]
freqUpLim = 1 / np.min(np.abs(time - np.roll(time, 1)))
print(freqUpLim)
time = time / time.max() * 2 * np.pi
# calculate the transform def fft_nutou(dataArray, modeNum, **kwargs):
res = xr.DataArray(
data=finufft.nufft1d1(time, data, modeNum),
dims=['freq'],
coords={
"freq":np.linspace(-freqUpLim/2, freqUpLim/2, modeNum)
}
)
return res
def ifft_nutou(self, dataArray, modeNum): data = dataArray.to_numpy()
pass data = data.astype('complex128')
time = dataArray[dataArray.dims[0]].to_numpy()
if isinstance(time[0], type(np.datetime64(500,'s'))):
time = time.astype(float)
time = time - time[0]
freqUpLim = 1 / np.min(np.abs(time - np.roll(time, 1))) * 1e9
else:
time = time.astype(float)
time = time - time[0]
freqUpLim = 1 / np.min(np.abs(time - np.roll(time, 1)))
print(freqUpLim)
time = time / time.max() * 2 * np.pi
# calculate the transform
res = xr.DataArray(
data=finufft.nufft1d1(time, data, modeNum, **kwargs),
dims=['freq'],
coords={
"freq":np.linspace(-freqUpLim/2, freqUpLim/2, modeNum)
}
)
return res
def ifft_nutou(dataArray, modeNum):
pass