Fix default value. Refactor.

IvoVellekoop · Nov 15, 2024 · 4c4ed38 · 4c4ed38
1 parent fff52c6
commit 4c4ed38
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/openwfs/calibration/fringe_analysis_slm_calibrator.py b/openwfs/calibration/fringe_analysis_slm_calibrator.py
@@ -65,7 +65,7 @@ def __init__(
             self.reference_slices = reference_slices
 
         if gray_values is None:
-            self.gray_values = np.arange(0, 255)
+            self.gray_values = np.arange(0, 256)
         else:
             self.gray_values = gray_values
 
@@ -98,10 +98,10 @@ def analyze(self, frames):
 
     @staticmethod
     def get_dominant_frequency(fft_data, dc_skip):
-        fft_data[..., :dc_skip, :dc_skip] = 0               # Remove DC peak
-        fft_data[..., :dc_skip, -dc_skip:] = 0              # Remove DC peak
-        fft_data[..., -dc_skip:, -dc_skip:] = 0             # Remove DC peak
-        fft_data[..., -dc_skip:, :dc_skip] = 0              # Remove DC peak
+        fft_data[..., :dc_skip, :dc_skip] = 0  # Remove DC peak
+        fft_data[..., :dc_skip, -dc_skip:] = 0  # Remove DC peak
+        fft_data[..., -dc_skip:, -dc_skip:] = 0  # Remove DC peak
+        fft_data[..., -dc_skip:, :dc_skip] = 0  # Remove DC peak
 
         # Find the index of the maximum value along the last axis
         s = fft_data.shape