Added function for enabling / disabling the video preview

2 files changed, 12 insertions, 5 deletions

diff --git a/freeseer-gui-qt.py b/freeseer-gui-qt.py
index 077ac76..7c674ac 100755
--- a/freeseer-gui-qt.py
+++ b/freeseer-gui-qt.py
@@ -56,7 +56,8 @@ class GSTPlayerWidget(QtGui.QWidget):
        
     def __init__(self, parent):
         QtGui.QWidget.__init__(self, parent)
-        self.freeseer = FreeSeeR(self.winId())
+        self.freeseer = FreeSeeR()
+        self.freeseer.enable_preview(self.winId())
 
     def change_videosrc(self):
         self.freeseer.change_videosrc(str(CONFIG['VODRIVER']), str(CONFIG['VODEVICE']))
diff --git a/freeseer.py b/freeseer.py
index d32aec9..d8fdd58 100755
--- a/freeseer.py
+++ b/freeseer.py
@@ -26,8 +26,8 @@ pygst.require("0.10")
 import gst
 
 class FreeSeeR:
-    def __init__(self, window_id):
-        self.window_id = window_id
+    def __init__(self):
+        self.window_id = None
         
         self.viddrv = 'v4lsrc'
         self.viddev = '/dev/video0'
@@ -58,14 +58,13 @@ class FreeSeeR:
         self.filesink.set_property("location", 'default.mkv')
 
         # GST Add Components
-        self.player.add(self.vidsrc, self.cspace, self.vidtee, self.vidqueue1, self.vidqueue2, self.vidcodec, self.vidsink)
+        self.player.add(self.vidsrc, self.cspace, self.vidtee, self.vidqueue1, self.vidcodec)
         self.player.add(self.sndsrc, self.sndqueue, self.audioconvert, self.sndcodec)
         self.player.add(self.mux, self.filesink)
 
         # GST Link Components
         gst.element_link_many(self.vidsrc, self.cspace, self.vidtee)
         gst.element_link_many(self.vidtee, self.vidqueue1, self.vidcodec, self.mux)
-        gst.element_link_many(self.vidtee, self.vidqueue2, self.vidsink)
         gst.element_link_many(self.sndsrc, self.sndqueue, self.audioconvert, self.sndcodec, self.mux)
         gst.element_link_many(self.mux, self.filesink)
 
@@ -127,3 +126,10 @@ class FreeSeeR:
         self.player.add(self.sndcodec)
         gst.element_link_many(self.audioconvert, self.sndcodec, self.mux)
 
+    def enable_preview(self, window_id):
+        self.window_id = window_id
+        self.player.add(self.vidqueue2, self.vidsink)
+        gst.element_link_many(self.vidtee, self.vidqueue2, self.vidsink)
+
+    def disable_preview(self):
+        self.player.remove(self.vidqueue2, self.vidsink)
\ No newline at end of file