fix: adjust zero-crossing detection: shift by 1, include all data poi…

…nts (#19)
patrk · Apr 5, 2024 · 4f9d2b4 · 4f9d2b4
1 parent aac8222
commit 4f9d2b4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyPTE/core/pyPTE.py b/pyPTE/core/pyPTE.py
@@ -20,9 +20,9 @@ def get_delay(phase: npt.NDArray) -> int:
     """
     phase = phase
     m, n = phase.shape
-    c1 = n * (m - 2)
-    r_phase = np.roll(phase, 2, axis=0)
-    phase_product = np.multiply(phase, r_phase)[1:-1]
+    c1 = n * m
+    r_phase = np.roll(phase, 1, axis=0)
+    phase_product = np.multiply(phase, r_phase)
     c2 = (phase_product < 0).sum()
     delay = int(np.round(c1 / c2))