Added interval to apply OpenCV effects

Author: BrennoCaldato

src/classes/effect_init.py

@@ -95,6 +95,7 @@
             "y": 0.05,
             "width": 0.25,
             "height": 0.25,
+            "first-frame": 1,
             "type": "rect"
         },
 

src/windows/process_effect.py

@@ -59,6 +59,12 @@ def __init__(self, clip_id, effect_name, effect_params):
         self.effect_name = effect_name
         self.context = {}
 
+        timeline_instance = get_app().window.timeline_sync.timeline
+        for clip_instance in timeline_instance.Clips():
+            if clip_instance.Id() == self.clip_id:
+                self.clip_instance = clip_instance
+                break
+
         # Load UI from designer & init
         ui_util.load_ui(self, self.ui_path)
         ui_util.init_ui(self)
@@ -233,10 +239,11 @@ def rect_select_clicked(self, widget, param):
         reader_path = c.data.get('reader', {}).get('path','')
         f = File.get(path=reader_path)
         if f:
-            win = SelectRegion(f, c)
+            win = SelectRegion(f, self.clip_instance)
             # Run the dialog event loop - blocking interaction on this window during that time
             result = win.exec_()
             if result == QDialog.Accepted:
+                # self.first_frame = win.current_frame
                 # Region selected (get coordinates if any)
                 topLeft = win.videoPreview.regionTopLeftHandle
                 bottomRight = win.videoPreview.regionBottomRightHandle
@@ -266,6 +273,7 @@ def rect_select_clicked(self, widget, param):
                                                            "scaled_x": topLeft.x() / viewPortSize.width(), "scaled_y": topLeft.y() / viewPortSize.height(),
                                                            "scaled_width": (bottomRight.x() - topLeft.x()) / viewPortSize.width(),
                                                            "scaled_height": (bottomRight.y() - topLeft.y()) / viewPortSize.height(),
+                                                           "first-frame": win.current_frame
                                                            })
                     log.info(self.context)
 
@@ -286,11 +294,6 @@ def accept(self):
 
         # DO WORK HERE, and periodically set progressBar value
         # Access C++ timeline and find the Clip instance which this effect should be applied to
-        timeline_instance = get_app().window.timeline_sync.timeline
-        for clip_instance in timeline_instance.Clips():
-            if clip_instance.Id() == self.clip_id:
-                self.clip_instance = clip_instance
-                break
 
         # Create effect Id and protobuf data path
         ID = get_app().project.generate_id()
@@ -365,6 +368,10 @@ def generateJson(self, protobufPath):
             tracker_dict = self.context["region"]
             bbox = (tracker_dict["x"],tracker_dict["y"],tracker_dict["width"],tracker_dict["height"])
             jsonString += ',"bbox": {"x": %d, "y": %d, "w": %d, "h": %d}' % (bbox)
+
+            # Get processing start frame
+            print(self.context["region"])
+            jsonString +=',"first_frame": %d' % (self.context["region"]["first-frame"])
 
         # Finish JSON string
         jsonString+='}'

src/windows/region.py

@@ -78,22 +78,30 @@ def __init__(self, file=None, clip=None):
         self.start_image = None
         self.end_frame = 1
         self.end_image = None
+        self.current_frame = 1
+
+        self.clip = clip
+        self.clip.Open()
+        self.clip_position = self.clip.Position()
+        self.clip.Position(0)
 
         # Keep track of file object
-        self.file = file
-        self.file_path = file.absolute_path()
-        self.video_length = int(file.data['video_length'])
-        self.fps_num = int(file.data['fps']['num'])
-        self.fps_den = int(file.data['fps']['den'])
-        self.fps = float(self.fps_num) / float(self.fps_den)
-        self.width = int(file.data['width'])
-        self.height = int(file.data['height'])
-        self.sample_rate = int(file.data['sample_rate'])
-        self.channels = int(file.data['channels'])
-        self.channel_layout = int(file.data['channel_layout'])
+        # self.file = file
+        # self.file_path = file.absolute_path()
+
+        c_info = clip.Reader().info
+        self.fps = c_info.fps.ToInt() #float(self.fps_num) / float(self.fps_den)
+        self.fps_num = self.fps #int(file.data['fps']['num'])
+        self.fps_den = 1 #int(file.data['fps']['den'])
+        self.width = c_info.width #int(file.data['width'])
+        self.height = c_info.height #int(file.data['height'])
+        self.sample_rate = c_info.sample_rate #int(file.data['sample_rate'])
+        self.channels = c_info.channels #int(file.data['channels'])
+        self.channel_layout = c_info.channel_layout #int(file.data['channel_layout'])
+        self.video_length = int(self.clip.Duration() * self.fps) + 1 #int(file.data['video_length'])
 
         # Open video file with Reader
-        log.info(self.file_path)
+        log.info(self.clip.Reader())
 
         # Add Video Widget
         self.videoPreview = VideoWidget()
@@ -110,7 +118,7 @@ def __init__(self, file=None, clip=None):
 
         try:
             # Add clip for current preview file
-            self.clip = openshot.Clip(self.file_path)
+            # self.clip = openshot.Clip(self.file_path)
 
             # Show waveform for audio files
             if not self.clip.Reader().info.has_video and self.clip.Reader().info.has_audio:
@@ -123,6 +131,7 @@ def __init__(self, file=None, clip=None):
             self.r.info.video_length = self.video_length
 
             self.r.AddClip(self.clip)
+
         except:
             log.error('Failed to load media file into region select player: %s' % self.file_path)
             return
@@ -147,8 +156,8 @@ def __init__(self, file=None, clip=None):
 
         # Determine if a start or end attribute is in this file
         start_frame = 1
-        if 'start' in self.file.data.keys():
-            start_frame = (float(self.file.data['start']) * self.fps) + 1
+        # if 'start' in self.file.data.keys():
+        #     start_frame = (float(self.file.data['start']) * self.fps) + 1
 
         # Display start frame (and then the previous frame)
         QTimer.singleShot(500, functools.partial(self.sliderVideo.setValue, start_frame + 1))
@@ -166,7 +175,7 @@ def __init__(self, file=None, clip=None):
         self.sliderVideo.valueChanged.connect(self.sliderVideo_valueChanged)
         self.initialized = True
 
-        get_app().window.SelectRegionSignal.emit(clip.id)
+        get_app().window.SelectRegionSignal.emit(clip.Id())
 
     def actionPlay_Triggered(self):
         # Trigger play button (This action is invoked from the preview thread, so it must exist here)
@@ -175,6 +184,7 @@ def actionPlay_Triggered(self):
     def movePlayhead(self, frame_number):
         """Update the playhead position"""
 
+        self.current_frame = frame_number
         # Move slider to correct frame position
         self.sliderIgnoreSignal = True
         self.sliderVideo.setValue(frame_number)
@@ -222,10 +232,16 @@ def sliderVideo_valueChanged(self, new_frame):
 
     def accept(self):
         """ Ok button clicked """
+
+        self.clip.Position(self.clip_position)
+
         self.shutdownPlayer()
         super(SelectRegion, self).accept()
 
     def shutdownPlayer(self):
+
+        self.clip.Position(self.clip_position)
+
         log.info('shutdownPlayer')
 
         # Stop playback
@@ -237,11 +253,15 @@ def shutdownPlayer(self):
         self.preview_parent.background.wait(5000)
 
         # Close readers
+        self.r.RemoveClip(self.clip)
         self.r.Close()
-        self.clip.Close()
+        # self.clip.Close()
         self.r.ClearAllCache()
 
     def reject(self):
+
+        self.clip.Position(self.clip_position)
+
         # Cancel dialog
         self.shutdownPlayer()
         super(SelectRegion, self).reject()