]> code.delx.au - transcoding/blobdiff - encode.py
Added high quality x264 encoding profile
[transcoding] / encode.py
index 57acdfc4c641ae5d8794c01be305fafc5378a5c5..28767bf546de98f81e6465be011024c4ffe86dc6 100755 (executable)
--- a/encode.py
+++ b/encode.py
@@ -26,6 +26,10 @@ class Command(object):
                        return
                if subprocess.Popen(["which", cmd], stdout=open("/dev/null", "w")).wait() != 0:
                        raise FatalException("Command '%s' is required" % cmd)
+       
+       def check_no_file(self, path):
+               if os.path.exists(path):
+                       raise FatalException("Output file '%s' exists." % path)
 
        def do_exec(self, args):
                if self.opts.dump:
@@ -45,6 +49,7 @@ class MP4Box(Command):
        def check(self):
                self.check_command("mencoder")
                self.check_command("MP4Box")
+               self.check_no_file(self.opts.output + ".mp4")
 
        def run(self):
                p = self.profile
@@ -60,6 +65,20 @@ class MP4Box(Command):
                self.do_exec(["rm", "-f", video, audio, input])
 
 
+
+class MKVMerge(Command):
+       def check(self):
+               self.check_command("mkvmerge")
+               self.check_no_file(self.opts.output + ".mkv")
+
+       def run(self):
+               input = self.opts.output + ".avi" # From Mencoder command
+               output = self.opts.output + ".mkv"
+               self.do_exec(["mkvmerge", "-o", output, input])
+               self.do_exec(["rm", "-f", input])
+
+
+
 class Mencoder(Command):
        codec2opts = {
                "lavc": "-lavcopts",
@@ -75,7 +94,7 @@ class Mencoder(Command):
                                cmd.append(opt)
                                cmd.append(var)
                if self.opts.deinterlace:
-                       cmd += ["-vf-add", "pp=ci"]
+                       cmd += ["-vf-add", "pp=lb"]
                try_opt("-ss", self.opts.startpos)
                try_opt("-endpos", self.opts.endpos)
                try_opt("-dvd-device", self.opts.dvd)
@@ -122,6 +141,7 @@ class Mencoder(Command):
 
        def check(self):
                self.check_command("mencoder")
+               self.check_no_file(self.opts.output + ".avi")
 
        def run(self):
                self.do_exec(self.pass1())
@@ -155,6 +175,15 @@ profiles = {
                aopts="br=%(abitrate)d:mpeg=4:object=2",
        ),
 
+       "x264" :
+       Profile(
+               commands=[Mencoder, MKVMerge],
+               vcodec="x264",
+               vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:subq=6:frameref=6:me=umh:partitions=all:bframes=4:b_adapt:qcomp=0.7:keyint=250:threads=2",
+               acodec="mp3lame",
+               aopts="abr:br=%(abitrate)d",
+       ),
+
        "xvid" :
        Profile(
                commands=[Mencoder],
@@ -179,10 +208,11 @@ profiles = {
        Profile(
                commands=[Mencoder, MP4Box],
                vcodec="x264",
-               vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:vbv_maxrate=1500:vbv_bufsize=2000:nocabac:me=umh:partitions=all:trellis=1:subq=7:bframes=0:direct_pred=auto:level_idc=30:global_header:turbo",
+               vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:vbv_maxrate=1500:vbv_bufsize=2000:nocabac:me=umh:partitions=all:trellis=1:subq=7:bframes=0:direct_pred=auto:level_idc=30:turbo",
                acodec="faac",
-               aopts="br=%(abitrate)d:mpeg=4:object=2:raw",
-               extra2=["-channels", "2", "-srate", "48000"]
+               aopts="br=%(abitrate)d:mpeg=4:object=2",
+               extra=["-vf-add", "scale=480:-10"],
+               extra2=["-channels", "2", "-srate", "48000"],
        ),
 
        "nokiax264" :
@@ -193,11 +223,39 @@ profiles = {
                        "abitrate": 96,
                },
                vcodec="x264",
-               vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:nocabac:me=umh:partitions=all:trellis=1:subq=7:bframes=0:direct_pred=auto:global_header",
+               vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:nocabac:me=umh:partitions=all:trellis=1:subq=7:bframes=0:direct_pred=auto",
                acodec="faac",
-               aopts="br=%(abitrate)d:mpeg=4:object=2:raw",
+               aopts="br=%(abitrate)d:mpeg=4:object=2",
                extra=["-vf-add", "scale=320:-10"],
        ),
+
+       "n97xvid" :
+       Profile(
+               commands=[Mencoder, MP4Box],
+               default_opts={
+                       "vbitrate": 1000,
+                       "abitrate": 96,
+               },
+               vcodec="xvid",
+               vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:vhq=4:autoaspect:max_bframes=0",
+               acodec="faac",
+               aopts="br=%(abitrate)d:mpeg=4:object=2",
+               extra=["-vf-add", "scale=640:-10"],
+       ),
+
+       "n97x264" :
+       Profile(
+               commands=[Mencoder, MP4Box],
+               default_opts={
+                       "vbitrate": 1000,
+                       "abitrate": 96,
+               },
+               vcodec="x264",
+               vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:vbv_maxrate=2000:vbv_bufsize=2000:nocabac:me=umh:partitions=all:trellis=1:subq=7:bframes=0:direct_pred=auto:level_idc=20",
+               acodec="faac",
+               aopts="br=%(abitrate)d:mpeg=4:object=2",
+               extra=["-vf-add", "scale=640:-10"],
+       ),
 }