guess video type

Author: arch

funscript_editor/algorithms/funscriptgenerator.py

@@ -929,9 +929,9 @@ def run(self) -> None:
             if self.params.metric not in ['x', 'y']:
                 self.params.track_men = True # we need 2 tracking points
 
-            if self.video_info.fps < 31.0 and self.params.skip_frames > 0:
+            if self.video_info.fps < 31.0 and self.params.skip_frames > 1:
                 self.logger.warning("The Video has less than 30 frames per seconds and you have set skip_frames to %d "\
-                        + "this can lead to inaccuracies when predicting the changepoint positions! (consider to set skip_frames to 0)" \
+                        + "this can lead to inaccuracies when predicting the changepoint positions! (consider to set skip_frames to 0 or 1)" \
                         , self.params.skip_frames)
 
             if self.params.raw_output:

funscript_editor/ui/funscript_generator_window.py

@@ -52,9 +52,13 @@ def __init__(self,
 
         cap = cv2.VideoCapture(video_file)
         fps = cap.get(cv2.CAP_PROP_FPS)
+        height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
+        width = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
+        video_aspect_ratio = float(width) / max((1, float(height)))
         cap.release()
 
         self.video_file = video_file
+        self.is_sbs_vr_video = True if 1.9 < video_aspect_ratio < 2.1 else False
         self.funscript = Funscript(fps)
         self.output_file = output_file
 
@@ -64,7 +68,7 @@ def __init__(self,
         self.__logger.info("Set End Time to Frame Number %d", self.end_frame)
 
         self.settings = {}
-        self.settings_dialog = SettingsDialog(self.settings)
+        self.settings_dialog = SettingsDialog(self.settings, include_vr = self.is_sbs_vr_video)
         self.settings_dialog.applySettings.connect(self.run)
         self.settings_dialog.show()
 

funscript_editor/ui/settings_dialog.py

@@ -13,8 +13,9 @@ class SettingsDialog(QtWidgets.QDialog):
         settings (dict): dict where to store the settings
     """
 
-    def __init__(self, settings):
+    def __init__(self, settings: dict, include_vr: bool = True):
         super(SettingsDialog, self).__init__()
+        self.include_vr = include_vr
         self.ui = settings_view.Ui_Form()
         self.form = QtWidgets.QDialog()
         self.ui.setupUi(self.form)
@@ -42,7 +43,7 @@ def __setup_ui_bindings(self):
         self.ui.trackingMethodComboBox.currentTextChanged.connect(lambda value: self.__set_setting('trackingMethod', value))
 
     def __setup_combo_boxes(self):
-        self.ui.videoTypeComboBox.addItems([PROJECTION[key]['name'] for key in PROJECTION.keys()])
+        self.ui.videoTypeComboBox.addItems([PROJECTION[key]['name'] for key in PROJECTION.keys() if 'vr' not in key or self.include_vr])
         self.ui.trackingMetricComboBox.addItems(['y', 'x', 'euclideanDistance', 'pitch'])
         self.ui.trackingMethodComboBox.addItems(['Woman', 'Woman + Men'])
 

funscript_editor/ui/video_player.py

@@ -413,7 +413,7 @@ def get_screenshot(self) -> np.ndarray:
             np.ndarray: opencv image
         """
         try: return cv2.cvtColor(np.array(self.player.screenshot_raw()), cv2.COLOR_RGB2BGR)
-        except: return None
+        except: return np.ndarray(0)
 
 
     def set_funscript(self, funscript :Funscript) -> None: