fix interaction between padding and wrap safety

mehta-lab · Sep 10, 2024 · 7c7f2a5 · 7c7f2a5
1 parent 22a059b
commit 7c7f2a5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/waveorder/models/isotropic_fluorescent_thick_3d.py b/waveorder/models/isotropic_fluorescent_thick_3d.py
@@ -57,9 +57,9 @@ def calculate_transfer_function(
         index_of_refraction_media,
         numerical_aperture_detection,
     )
-
+    zyx_out_shape = (zyx_shape[0] + 2 * z_padding,) + zyx_shape[1:]
     return sampling.nd_fourier_central_cuboid(
-        optical_transfer_function, zyx_shape
+        optical_transfer_function, zyx_out_shape
     )
 
 

diff --git a/waveorder/models/phase_thick_3d.py b/waveorder/models/phase_thick_3d.py
@@ -73,12 +73,13 @@ def calculate_transfer_function(
         )
     )
 
+    zyx_out_shape = (zyx_shape[0] + 2 * z_padding,) + zyx_shape[1:]
     return (
         sampling.nd_fourier_central_cuboid(
-            real_potential_transfer_function, zyx_shape
+            real_potential_transfer_function, zyx_out_shape
         ),
         sampling.nd_fourier_central_cuboid(
-            imag_potential_transfer_function, zyx_shape
+            imag_potential_transfer_function, zyx_out_shape
         ),
     )