Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
2**k2)
# take the modulus and go back in Fourier
U2 = modulus_complex(ifft1d_c2c(U2_hat))
if average:
U2_hat = fft1d_c2c(U2)
# Convolve with phi_J
k2_J = max(J - k2 - k1 - oversampling, 0)
S2_J_hat = subsample_fourier(U2_hat * phi[k1 + k2],
2**k2_J)
S2_J = unpad(real(ifft1d_c2c(S2_J_hat)),
ind_start[k1 + k2 + k2_J],
ind_end[k1 + k2 + k2_J])
else:
# just take the real value and unpad
S2_J = unpad(
real(U2), ind_start[k1 + k2], ind_end[k1 + k2])
if vectorize:
S[:, cc[2], :] = S2_J.squeeze(dim=1)
cc[2] += 1
else:
S[n1, n2] = S2_J
return S
k1 = max(j1 - oversampling, 0)
assert psi1[n1]['xi'] < 0.5 / (2**k1)
U1_hat = subsample_fourier(U0_hat * psi1[n1][0], 2**k1)
# Take the modulus
U1 = modulus_complex(ifft1d_c2c(U1_hat))
if average or max_order > 1:
U1_hat = fft1d_c2c(U1)
if average:
# Convolve with phi_J
k1_J = max(J - k1 - oversampling, 0)
S1_J_hat = subsample_fourier(U1_hat * phi[k1], 2**k1_J)
S1_J = unpad(real(ifft1d_c2c(S1_J_hat)),
ind_start[k1_J + k1], ind_end[k1_J + k1])
else:
# just take the real value and unpad
S1_J = unpad(real(U1), ind_start[k1], ind_end[k1])
if vectorize:
S[:, cc[1], :] = S1_J.squeeze(dim=1)
cc[1] += 1
else:
S[(n1,)] = S1_J
if max_order == 2:
# 2nd order
for n2 in range(len(psi2)):
j2 = psi2[n2]['j']
if j2 > j1:
assert psi2[n2]['xi'] < psi1[n1]['xi']
# convolution + downsampling
k2 = max(j2 - k1 - oversampling, 0)
U2_hat = subsample_fourier(U1_hat * psi2[n2][k1],
2**k2)
# take the modulus and go back in Fourier