filter_97fft.py
1.45 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
title = "|FFT|"
tip = "applies sqrt(real^2 + imag^2) from theb data"
onein = True
import numpy as np
from guidata.dataset.datatypes import DataSet
from guidata.dataset.dataitems import (IntItem, FloatArrayItem, StringItem,
ChoiceItem, FloatItem, DictItem,
BoolItem)
from guiqwt.config import _
class NOD3_App:
def __init__(self, parent):
self.parent = parent
self.parent.activateWindow()
def compute_app(self, **args):
class FuncParam(DataSet):
Map = ChoiceItem("Map", (("Amplitude", "Amplitude"), ("Phase", "Phase")),
default="Amplitude")
name = title.replace(" ", "")
if args == {}:
param = FuncParam(_(title), "Choose Amplitude or Phase:")
else:
param = self.parent.ScriptParameter(name, args)
# if no parameter needed set param to None. activate next line
#param = None
self.parent.compute_11(name, lambda m, p: self.function(m, p), param, onein)
def function(self, m, p):
mask = np.isnan(m.data)
data = self.parent.nan_check(m.data, np.nanmin(m.data))
fftdata = np.fft.fft2(data)
fftdata = np.fft.fftshift(fftdata)
if p.Map == "Amplitude":
m.data = np.abs(fftdata)
else:
m.data = np.angle(fftdata)
#m.data = np.where(mask, np.nan, data)
return m, p