visu for roll

Author: arch

funscript_editor/algorithms/funscriptgenerator.py

@@ -274,10 +274,10 @@ def scale_score(self, status: str, metric : str = 'y') -> None:
             max_tracking_points = self.get_tracking_points_by_frame_number(max_frame - self.params.start_frame)
 
             for points in min_tracking_points:
-                imgMin = OpenCV_GUI.draw_point_to_image(imgMin, points)
+                imgMin = OpenCV_GUI.draw_point_to_image(imgMin, points, connect_points=(metric == 'roll'))
 
             for points in max_tracking_points:
-                imgMax = OpenCV_GUI.draw_point_to_image(imgMax, points)
+                imgMax = OpenCV_GUI.draw_point_to_image(imgMax, points, connect_points=(metric == 'roll'))
 
             # print('min_tracking_points', min_tracking_points, 'max_tracking_points', max_tracking_points)
 

funscript_editor/ui/opencvui.py

@@ -260,13 +260,19 @@ def draw_box_to_image(image: np.ndarray, bbox, color: tuple = (255, 0, 255)) ->
 
 
     @staticmethod
-    def draw_point_to_image(image: np.ndarray, point, color: tuple = (255, 0, 255)) -> np.ndarray:
+    def draw_point_to_image(
+            image: np.ndarray,
+            point,
+            color: tuple = (255, 0, 255),
+            connect_points: bool = False
+            ) -> np.ndarray:
         """ Draw an point to given image
 
         Args:
             image (np.ndarray): image
             point (tuple or list): points (x,y) or list of points
             color (tuple): RGB color values for the box
+            connect_points (bool): connect points with an line
 
         Returns:
             np.ndarray: opencv image
@@ -283,6 +289,15 @@ def draw_point_to_image(image: np.ndarray, point, color: tuple = (255, 0, 255))
                     2
                 )
 
+        if len(point) > 1 and connect_points:
+            for i in range(len(point)-1):
+                cv2.line(
+                        image,
+                        point[i],
+                        point[i+1],
+                        color,
+                        2)
+
         return image