Updated scripts to handle complex numbers in Restriction

DIG-Kaust · Nov 26, 2024 · 1317998 · 1317998
1 parent 15c3fe7
commit 1317998
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/scripts/2d/Interpolation_slopes.py b/scripts/2d/Interpolation_slopes.py
@@ -154,6 +154,9 @@ def main(parser):
 
     ######### Multi-channel interpolation #########
 
+    # Turn to complex since we are now working with a complex model
+    Rop.dtype = np.complex128 
+
     # Let's start by setting the slope regularization term
     D1op0 = FirstDerivative(dims=(nxorig, nt), axis=0, sampling=dx, order=5, edge=True, dtype="complex128")
     D1op1 = FirstDerivative(dims=(nxorig, nt), axis=1, sampling=dt, order=5, edge=True, dtype="complex128")
@@ -201,7 +204,7 @@ def main(parser):
 
     if not true_solution:
         fig, axs = plt.subplots(2, 2, figsize=(12, 12), gridspec_kw={'height_ratios': [2, 1]})
-        axs[0, 0].imshow(data.T, cmap='gray', aspect='auto', vmin=-vlim, vmax=vlim,
+        axs[0, 0].imshow(data.real.T, cmap='gray', aspect='auto', vmin=-vlim, vmax=vlim,
                          extent=(x[0] / 1000, x[-1] / 1000, t[-1], t[0]))
         axs[0, 0].set_title('Original')
         axs[0, 0].set_xlabel('Offset (m)')
@@ -232,7 +235,7 @@ def main(parser):
         axs[0, 0].set_title('Original')
         axs[0, 0].set_xlabel('Offset (m)')
         axs[0, 0].set_ylabel('TWT (s)')
-        axs[0, 1].imshow(data.T, cmap='gray', aspect='auto', vmin=-vlim, vmax=vlim,
+        axs[0, 1].imshow(data.real.T, cmap='gray', aspect='auto', vmin=-vlim, vmax=vlim,
                          extent=(x[0]/1000, x[-1]/1000, t[-1], t[0]))
         axs[0, 1].set_title('Subsampled')
         axs[0, 1].set_xlabel('Offset (m)')

diff --git a/scripts/3d/Interpolation_slopes.py b/scripts/3d/Interpolation_slopes.py
@@ -171,6 +171,9 @@ def main(parser):
 
     ######### Multi-channel interpolation #########
 
+    # Turn to complex since we are now working with a complex model
+    Rop.dtype = np.complex128 
+
     # Let's start by setting the slope regularization term
     D1op0 = FirstDerivative(dims=(nyorig, nx, nt), axis=0, sampling=dy, order=5, edge=True, dtype=np_floatcconv)
     D1op1 = FirstDerivative(dims=(nyorig, nx, nt), axis=2, sampling=dt, order=5, edge=True, dtype=np_floatcconv)