Commit 78243d0d authored by Matthijs's avatar Matthijs
Browse files

P_M_masked tested, should properly apply the mask in Fourier space now

parent 8b1077ab
......@@ -69,6 +69,7 @@ class P_M_masked(P_M):
-------
array_like - p_M: the projection IN THE PHYSICAL (time) DOMAIN
"""
constrained = super(P_M_masked, self).work(u)
fourier_space_iterate = np.fft.fft2(u, axes=None)
constrained = magproj(fourier_space_iterate, self.M)
update = np.where(self.mask, u, constrained)
return update
return np.fft.ifft2(update, axes=None)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment