Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
fs = []
fhs = []
for i in range(36):
t = i * 2 * np.pi / 36.
R = np.array([[np.cos(-t), -np.sin(-t)],
[np.sin(-t), np.cos(-t)]])
RC = np.einsum('ij,abj->abi', R, C - np.array([p0, q0])[None, None, :]) + np.array([p0, q0])[None, None, :]
#Rfr = map_wrap(f1.real, RC.transpose(2, 0, 1))
#Rfi = map_wrap(f1.imag, RC.transpose(2, 0, 1))
#Rf = Rfr + Rfi * 1j
Rf = np.exp(1j * 2 * np.pi * (5 * RC[..., 0]) / nx)
Rfh = shift_fft(Rf)
Rfh = T2FFT.analyze(Rf)
fs.append(Rf)
fhs.append(Rfh)
return fs, fhs