+import optparse
+import re
+import subprocess
+import sys
+import os
+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
+
+def midentify(source, field):
+ process = subprocess.Popen(
+ ["mplayer", "-frames", "0", "-identify", source],
+ stdout=subprocess.PIPE,
+ stderr=subprocess.PIPE,
+ )
+ for line in process.stdout:
+ try:
+ key, value = line.split("=")
+ except ValueError:
+ continue
+ if key == field:
+ return value.strip()
+
+
+
+class Command(object):
+ codec2exts = {
+ "xvid": "m4v",
+ "x264": "h264",
+ "faac": "aac",
+ "mp3lame": "mp3",
+ "copyac3": "ac3",
+ }
+
+ def __init__(self, profile, opts):
+ self.profile = profile
+ self.opts = opts
+ self.audio_tmp = "audio." + self.codec2exts[profile.acodec]
+ self.video_tmp = "video." + self.codec2exts[profile.vcodec]
+
+ def print_install_message(self):
+ print >>sys.stderr, "Problem with command: %s", self.name
+ if self.package:
+ print >>sys.stderr, "Try running:\n# aptitude install %s", self.package
+
+ def check_command(self, cmd):
+ if self.opts.dump:
+ 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:
+ print " ".join(map(mkarg, args))
+ else:
+ if subprocess.Popen(args).wait() != 0:
+ raise FatalException("Failure executing command: %s" % args)
+
+
+class MP4Box(Command):
+ def check(self):
+ self.check_command("MP4Box")
+ self.check_no_file(self.opts.output + ".mp4")
+
+ def run(self):
+ if self.opts.dump:
+ fps = "???"
+ else:
+ fps = midentify(self.video_tmp, "ID_VIDEO_FPS")
+
+ output = self.opts.output + ".mp4"
+ self.do_exec([
+ "MP4Box",
+ "-fps", fps,
+ "-add", self.video_tmp,
+ "-add", self.audio_tmp,
+ output
+ ])
+
+
+
+class MKVMerge(Command):
+ def check(self):
+ self.check_command("mkvmerge")
+ self.check_no_file(self.opts.output + ".mkv")
+
+ def run(self):
+ if self.opts.dump:
+ fps = "???"
+ else:
+ fps = midentify(self.video_tmp, "ID_VIDEO_FPS")
+
+ self.do_exec([
+ "mkvmerge",
+ "-o", self.opts.output + ".mkv",
+ "--default-duration", "0:%sfps"%fps,
+ self.video_tmp,
+ self.audio_tmp,
+ ])
+
+
+
+class MencoderMux(Command):
+ def check(self):
+ self.check_command("mencoder")
+ self.check_no_file(self.opts.output + ".avi")
+
+ def run(self):
+ self.do_exec([
+ "mencoder",
+ "-o", self.opts.output + ".avi",
+ "-oac", "copy", "-ovc", "copy",
+ "-noskip", "-mc", "0",
+ "-audiofile", self.audio_tmp,
+ self.video_tmp,
+ ])
+
+
+
+class Mencoder(Command):
+ codec2opts = {
+ "xvid": "-xvidencopts",
+ "x264": "-x264encopts",
+ "faac": "-faacopts",
+ "mp3lame": "-lameopts",
+ }
+
+ def insert_options(self, cmd):
+ def try_opt(opt, var):
+ if var is not None:
+ cmd.append(opt)
+ 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)
+ try_opt("-chapter", self.opts.chapter)
+ try_opt("-aid", self.opts.audioid)
+ try_opt("-sid", self.opts.subtitleid)
+ try_opt("-vf-add", self.opts.vfilters)
+ try_opt("-af-add", self.opts.afilters)
+ cmd += ["-vf-add", "harddup"]
+
+ def subst_values(self, cmd, vpass):
+ subst = {
+ "vbitrate": self.opts.vbitrate,
+ "abitrate": self.opts.abitrate,
+ "vpass": vpass,
+ }
+
+ return [x % subst for x in cmd]
+
+ def passn(self, n):
+ p = self.profile
+
+ acodec = p.acodec
+ if self.opts.copyac3:
+ acodec = "copy"
+ p.acodec = "copyac3"
+ p.aopts = None
+
+ cmd = []
+ cmd += ["mencoder", self.opts.input]
+ self.insert_options(cmd)
+ cmd += ["-ovc", p.vcodec, self.codec2opts[p.vcodec], p.vopts]
+ cmd += ["-oac", acodec]
+ if p.aopts:
+ cmd += [self.codec2opts[p.acodec], p.aopts]
+ cmd += self.profile.extra1 + self.profile.extra
+ cmd = self.subst_values(cmd, vpass=n)
+
+ return cmd
+
+
+ def pass1(self):
+ cmd = self.passn(1)
+ cmd += ["-o", self.audio_tmp, "-of", "rawaudio"]
+ return cmd
+
+ def pass2(self):
+ cmd = self.passn(2)
+ cmd += ["-o", self.video_tmp, "-of", "rawvideo"]
+ return cmd
+
+ def check(self):
+ self.check_command("mencoder")
+ self.check_no_file(self.audio_tmp)
+ self.check_no_file(self.video_tmp)
+
+ def run(self):
+ self.do_exec(self.pass1())
+ self.do_exec(self.pass2())
+
+
+
+class Profile(object):
+ def __init__(self, commands, **kwargs):
+ self.default_opts = {
+ "vbitrate": 1000,
+ "abitrate": 192,
+ }
+ self.extra = []
+ self.extra1 = []
+ self.extra2 = []
+ self.commands = commands
+ self.__dict__.update(kwargs)
+
+ def __contains__(self, keyname):
+ return hasattr(self, keyname)
+
+
+profiles = {
+ "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, MencoderMux],
+ vcodec="xvid",
+ vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:vhq=4:autoaspect",
+ acodec="mp3lame",
+ aopts="abr:br=%(abitrate)d",
+ ),
+
+ "apple-quicktime" :
+ Profile(
+ commands=[Mencoder, MP4Box],
+ vcodec="x264",
+ vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:me=umh:partitions=all:trellis=1:subq=7:bframes=1:direct_pred=auto",
+ acodec="faac",
+ aopts="br=%(abitrate)d:mpeg=4:object=2",
+ ),
+
+ "ipod-xvid" :
+ Profile(
+ commands=[Mencoder, MP4Box],
+ 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=480:-10"],
+ ),
+
+ "ipod-x264" :
+ 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:turbo",
+ acodec="faac",
+ aopts="br=%(abitrate)d:mpeg=4:object=2",
+ extra=["-vf-add", "scale=480:-10"],
+ extra2=["-channels", "2", "-srate", "48000"],
+ ),
+
+ "nokia-n97" :
+ Profile(
+ commands=[Mencoder, MP4Box],
+ default_opts={
+ "vbitrate": 256,
+ "abitrate": 64,
+ },
+ 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"],
+ ),