add video seek bar

Author: arch

funscript_editor/ui/funscript_editor_view.py

@@ -38,6 +38,10 @@ def setupUi(self, MainWindow):
         self.videoLayout.addWidget(self.videoPane)
         spacerItem = QtWidgets.QSpacerItem(40, 0, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Minimum)
         self.videoLayout.addItem(spacerItem)
+        self.seekBar = QtWidgets.QSlider(self.verticalLayoutWidget)
+        self.seekBar.setOrientation(QtCore.Qt.Horizontal)
+        self.seekBar.setObjectName("seekBar")
+        self.videoLayout.addWidget(self.seekBar)
         self.settingsLayout = QtWidgets.QGroupBox(self.splitterHorizontal)
         self.settingsLayout.setTitle("")
         self.settingsLayout.setObjectName("settingsLayout")

funscript_editor/ui/funscript_editor_view.ui

@@ -46,6 +46,13 @@
            </property>
           </spacer>
          </item>
+         <item>
+          <widget class="QSlider" name="seekBar">
+           <property name="orientation">
+            <enum>Qt::Horizontal</enum>
+           </property>
+          </widget>
+         </item>
         </layout>
        </widget>
        <widget class="QGroupBox" name="settingsLayout">

funscript_editor/ui/funscript_editor_window.py

@@ -106,6 +106,8 @@ def __setup_modul_bindings(self):
         self.video_player.set_frame_changed_callback(self.__frame_changed_handler)
 
     def __frame_changed_handler(self, frame_num):
+        if not self.ui.seekBar.isSliderDown():
+            self.ui.seekBar.setValue(frame_num)
         self.funscript_visualizer.set_frame(frame_num)
         if self.funscript is None: return
         self.ui.currentStrokeLabel.setText('{} ms'.format(\
@@ -126,6 +128,7 @@ def __setup_ui_binding(self):
         helpMenu.addAction("App Documentation", lambda : webbrowser.open(os.path.join(APP_DOCUMENTATION_DIR, 'index.html')))
         helpMenu.addAction("Code Documentation", lambda : webbrowser.open(os.path.join(CODE_DOCUMENTATION_DIR, 'index.html')))
         helpMenu.addAction(str('Version '+VERSION))
+        self.ui.seekBar.sliderReleased.connect(lambda: self.video_player.seek_frame(self.ui.seekBar.value()))
 
         self.__generateFunscript.connect(self.__generate_funscript)
 
@@ -269,6 +272,8 @@ def __open_video(self):
 
         self.video_player.set_funscript(self.funscript)
         self.funscript_visualizer.set_funscript(self.funscript)
+        self.ui.seekBar.setMaximum(max((0, self.video_player.get_length-1)))
+        self.ui.seekBar.setValue(0)
 
     def __new_funscript(self):
         self.funscript = Funscript(fps=self.video_player.get_fps)

funscript_editor/ui/video_player.py

@@ -51,11 +51,12 @@ def __init__(self,
         self.indicate_bussy = False
         self.bussy_indicator_pos = 0
         self.funscript = None
-        self.fps = None
+        self.fps = 60
         self.stroke_indicator_position = UI_CONFIG['player']['stroke_indicator_pos']
         self.stroke_indicator_size = UI_CONFIG['player']['stroke_indicator_size']
         self.quit = False
         self.duration = 0
+        self.length = 0
         self.key_callback = key_callback
         self.player.pause = True
         self.invert_stroke_indicator = False
@@ -260,6 +261,16 @@ def get_current_timestamp_in_millis(self) -> int:
         return int(ts * 1000)
 
 
+    @property
+    def get_length(self) -> int:
+        """ Get video length in frames
+
+        Returns:
+            int: number of frames in video
+        """
+        return self.length
+
+
     @property
     def get_current_frame(self) -> int:
         """ Get current frame number
@@ -374,6 +385,7 @@ def open(self, video_file :str) -> None:
         self.height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
         self.width = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
         self.duration = cap.get(cv2.CAP_PROP_FRAME_COUNT) / self.fps
+        self.length = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
         self.player.loadfile(video_file)
         self.video_file = video_file
 
@@ -403,6 +415,7 @@ def seek_frame(self, frame_number :int) -> None:
         Args:
             frame_number (int): frame number
         """
+        if self.length < 1: return
         self.player.seek(self.frame_to_millisec(frame_number) * 1.0 / 1000, "absolute", "exact")