Added detelecine option

[transcoding] / encode.py

diff --git a/encode.py b/encode.py
index 29daa9577881f32caeda37f5f9f4b25306a12b44..31b765fe72a1652b6529a61af3199065b340c341 100755 (executable)
--- a/encode.py
+++ b/encode.py
@@ -1,16 +1,30 @@
 #!/usr/bin/env python
 
-import commands
 import optparse
+import re
 import subprocess
 import sys
 import os
-import tempfile
 import shutil
+import tempfile
 
 class FatalException(Exception):
        pass
 
+def mkarg(arg):
+       if re.match("^[a-zA-Z0-9\-\\.,/@_:=]*$", arg):
+               return arg
+
+       if "'" not in arg:
+               return "'%s'" % arg
+       out = "\""
+       for c in arg:
+               if c in "\\$\"`":
+                       out += "\\"
+               out += c
+       out += "\""
+       return out
+
 class Command(object):
        def __init__(self, profile, opts):
                self.profile = profile
@@ -33,7 +47,7 @@ class Command(object):
 
        def do_exec(self, args):
                if self.opts.dump:
-                       print "".join(map(commands.mkarg, args))[1:]
+                       print " ".join(map(mkarg, args))
                else:
                        if subprocess.Popen(args).wait() != 0:
                                raise FatalException("Failure executing command: %s" % args)
@@ -65,6 +79,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",
@@ -81,6 +109,11 @@ class Mencoder(Command):
                                cmd.append(var)
                if self.opts.deinterlace:
                        cmd += ["-vf-add", "pp=lb"]
+               if self.opts.detelecine:
+                       self.opts.ofps = "24000/1001"
+                       cmd += ["-vf-add", "pullup,softskip"]
+               try_opt("-fps", self.opts.ifps)
+               try_opt("-ofps", self.opts.ofps)
                try_opt("-ss", self.opts.startpos)
                try_opt("-endpos", self.opts.endpos)
                try_opt("-dvd-device", self.opts.dvd)
@@ -161,6 +194,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",
+               acodec="mp3lame",
+               aopts="abr:br=%(abitrate)d",
+       ),
+
        "xvid" :
        Profile(
                commands=[Mencoder],
@@ -248,11 +290,14 @@ def parse_args():
        parser = optparse.OptionParser(usage="%prog [options] input [output]")
        parser.add_option("--dvd", action="store", dest="dvd")
        parser.add_option("--deinterlace", action="store_true", dest="deinterlace")
+       parser.add_option("--detelecine", action="store_true", dest="detelecine")
        parser.add_option("--vfilters", action="store", dest="vfilters")
        parser.add_option("--afilters", action="store", dest="afilters")
        parser.add_option("--vbitrate", action="store", dest="vbitrate", type="int")
        parser.add_option("--abitrate", action="store", dest="abitrate", type="int")
        parser.add_option("--chapter", action="store", dest="chapter")
+       parser.add_option("--ifps", action="store", dest="ifps")
+       parser.add_option("--ofps", action="store", dest="ofps")
        parser.add_option("--startpos", action="store", dest="startpos")
        parser.add_option("--endpos", action="store", dest="endpos")
        parser.add_option("--audioid", action="store", dest="audioid")
@@ -314,7 +359,7 @@ def main():
                                command.run()
 
                except FatalException, e:
-                       print >>sys.stderr, "Error:", e.message
+                       print >>sys.stderr, "Error:", str(e)
                        sys.exit(1)
 
        finally: