Changeset 444 for trunk/UTIL/PYTHON/videosink.py

Timestamp:

Dec 1, 2011, 3:58:28 PM (13 years ago)

Author:

acolaitis

Message:

PYTHON. Improved quality of movies for really small weight. Also added an option to force even better quality (option is --quality, weight is also reasonnable but rendering is slower).

File:

• trunk/UTIL/PYTHON/videosink.py (modified) (3 diffs)

trunk/UTIL/PYTHON/videosink.py

@@ -1 +1 @@
 import numpy as np
 import subprocess
+from os import system
 
 class VideoSink(object) :
@@ -10 +11 @@
                     '-demuxer', 'rawvideo',
                     '-rawvideo', 'w=%i:h=%i'%size[::-1]+":fps=%i:format=%s"%(rate,byteorder),
-                    '-o', filename+'.avi',
-                    '-ovc', 'lavc',
+                    '-o', filename+'_raw.avi',
+                    '-nosound',
+                    '-ovc', 'x264',
+#                    '-ovc', 'lavc',
                     )
             self.p = subprocess.Popen(cmdstring, stdin=subprocess.PIPE, shell=False)
@@ -18 +21 @@
             #assert image.shape[0:2] == self.size
             self.p.stdin.write(image.tostring())
+
+    def first_pass(self,filename="output",quality=False) :
+
+
+             bitrate="7200"
+             if quality:bitrate="50000"
+             cmdstring  = ('mencoder',
+                     filename+'_raw.avi',
+                     '-of', 'rawvideo',
+                     '-nosound',
+                     '-ovc', 'x264',
+                     '-x264encopts', 'subq=1:frameref=1:bitrate='+bitrate+':bframes=1:pass=1',
+                     '-vf', 'scale=1280:720',
+                     '-o', filename+'_first.264'
+                    )
+             subprocess.call(cmdstring,shell=False)
+
+    def second_pass(self,filename="output",quality=False) :
+             bitrate="7200"
+             if quality:bitrate="50000"
+             cmdstring  = ('mencoder',
+                     filename+'_first.264',
+                     '-of', 'rawvideo',
+                     '-nosound',
+                     '-ovc', 'x264',
+                     '-x264encopts', 'subq=6:frameref=5:bitrate='+bitrate+':me=umh:partitions=all:bframes=1:me_range=16:cabac:weightb:deblock:pass=2',
+                     '-vf', 'scale=1280:720',
+                     '-o', filename+'.avi'
+                    )
+             system('rm -f '+filename+'_raw.avi')
+             subprocess.call(cmdstring,shell=False)
+             system('rm -f '+filename+'_first.264')
+
     def close(self) :
             self.p.stdin.close()
+            self.p.wait()
+