Merge pull request #152 from husarion/ros1-fix-coding-style-lights

Add floating precision
husarion · Jul 24, 2023 · 79f5b89 · 79f5b89
2 parents 09a2e86 + 97e8f8c
commit 79f5b89
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/panther_lights/src/animation/battery_animation.py b/panther_lights/src/animation/battery_animation.py
@@ -67,7 +67,7 @@ def _color_rising(self, battery_percent: float, progress: float) -> np.array:
         frame = np.zeros((self._num_led, 3), dtype=np.uint8)
         led_to_disp = battery_percent * self._num_led
 
-        middle_led = (self._num_led - 1) / 2
+        middle_led = (self._num_led - 1) / 2.0
         ind_1 = int(np.ceil(middle_led - progress * led_to_disp / 2.0))
         ind_2 = int(np.floor(middle_led + progress * led_to_disp / 2.0))
 
@@ -79,9 +79,9 @@ def _color_rising(self, battery_percent: float, progress: float) -> np.array:
     def _fill_frame(self, frame: np.array, battery_percent: float, progress: float) -> np.array:
         led_to_disp = battery_percent * self._num_led
 
-        middle_led = (self._num_led - 1) / 2
-        ind_1 = int(np.ceil(middle_led - (1 - progress) * led_to_disp / 2.0))
-        ind_2 = int(np.floor(middle_led + (1 - progress) * led_to_disp / 2.0))
+        middle_led = (self._num_led - 1) / 2.0
+        ind_1 = int(np.ceil(middle_led - (1.0 - progress) * led_to_disp / 2.0))
+        ind_2 = int(np.floor(middle_led + (1.0 - progress) * led_to_disp / 2.0))
 
         rgb = self._calculate_color(battery_percent)
         frame[ind_1] = rgb