add todo

Author: arch

funscript_editor/algorithms/funscriptgenerator.py

@@ -275,7 +275,9 @@ def scale_score(self, status: str, metric : str = 'y') -> None:
                 max_distance_frame = FFmpegStream.get_frame(self.params.video_path, max_distance_frame_num)
                 max_distance_frame = FFmpegStream.get_projection(max_distance_frame, self.projection_config)
                 center_line = self.ui.line_selector(max_distance_frame, "draw line on center of dick")
+                max_distance_roll_score = self.score['roll'][max_distance_frame_num - self.params.start_frame]
                 print('center line', center_line)
+                # TODO calc angel of line and apply max_distance_roll_score to  get real center offset and appl offset to all items in  roll score
 
             min_tracking_points = self.get_tracking_points_by_frame_number(min_frame - self.params.start_frame)
             max_tracking_points = self.get_tracking_points_by_frame_number(max_frame - self.params.start_frame)

funscript_editor/ui/opencvui.py

@@ -100,11 +100,12 @@ def was_key_pressed(self, key: str) -> bool:
 
 
 class DrawSingleLineWidget(object):
-    def __init__(self, background_img, window_name, preview_scaling):
+    def __init__(self, background_img, window_name, preview_scaling, color=(36,255,12)):
         self.original_image = background_img
         self.clone = self.original_image.copy()
         self.window_name = window_name
         self.preview_scaling = preview_scaling
+        self.color = color
 
         cv2.namedWindow(window_name)
         cv2.setMouseCallback(window_name, self.extract_coordinates)
@@ -122,7 +123,7 @@ def extract_coordinates(self, event, x, y, flags, parameters):
         elif event == cv2.EVENT_LBUTTONUP:
             self.end_coordinate = (x,y)
             self.clone = self.original_image.copy()
-            cv2.line(self.clone, self.start_coordinate, self.end_coordinate, (36,255,12), 2)
+            cv2.line(self.clone, self.start_coordinate, self.end_coordinate, self.color, 2)
 
     def show_image(self):
         return self.clone