#!/usr/bin/env python
-import commands, optparse, subprocess, sys, os
+from functools import partial
+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", source,
+ "-ao", "null", "-vo", "null",
+ "-frames", "0", "-identify",
+ ],
+ 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()
+
+def append_cmd(cmd, opt, var):
+ if var is not None:
+ cmd.append(opt)
+ cmd.append(str(var))
+
+def duplicate_opts(opts):
+ return optparse.Values(opts.__dict__)
+
+def insert_mplayer_options(cmd, o):
+ if o.mplayer_done:
+ return
+
+ do_opt = partial(append_cmd, cmd)
+ do_opt("-mc", o.mc)
+ do_opt("-ss", o.startpos)
+ do_opt("-endpos", o.endpos)
+ do_opt("-dvd-device", o.dvd)
+ do_opt("-chapter", o.chapter)
+ do_opt("-aid", o.audioid)
+ do_opt("-sid", o.subtitleid)
+ do_opt("-vf", o.vfilters)
+ do_opt("-af", o.afilters)
+
+ if o.deinterlace:
+ cmd += ["-vf-pre", "yadif"]
+ if o.detelecine:
+ cmd += ["-vf-pre", "pullup,softskip", "-ofps", "24000/1001"]
+ if o.noskip:
+ cmd += ["-noskip"]
+ if o.skipkb:
+ cmd += ["-sb", str(o.skipkb * 1024)]
-class MencoderCommand(object):
- codec2opts = {
- "lavc": "-lavcopts",
- "xvid": "-xvidencopts",
- "x264": "-x264encopts",
- "faac": "-faaccopts",
- "mp3lame": "-lameopts",
- }
+class Command(object):
def __init__(self, profile, opts):
self.profile = profile
self.opts = opts
+ self.__process = None
+ self.init()
+
+ def init(self):
+ pass
+
+ 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, wait=True):
+ if self.opts.dump:
+ print " ".join(map(mkarg, args))
+ else:
+ self.__process = subprocess.Popen(args)
+ self.__args = args
+ if wait:
+ self.wait()
+
+ def wait(self):
+ if self.__process == None:
+ return
+ if self.__process.wait() != 0:
+ raise FatalException("Failure executing command: %s" % self.__args)
+ self.__process = None
+
+
+class MP4Box(Command):
+ def init(self):
+ self.check_command("MP4Box")
+ self.check_no_file(self.opts.output + ".mp4")
+
+ def run(self):
+ o = self.opts
+ p = self.profile
+
+ if o.dump:
+ fps = "???"
+ else:
+ fps = midentify(p.video_tmp, "ID_VIDEO_FPS")
+
+ self.do_exec([
+ "MP4Box",
+ "-fps", fps,
+ "-add", p.video_tmp,
+ "-add", p.audio_tmp,
+ o.output + ".mp4"
+ ])
+
+
+
+class MKVMerge(Command):
+ def init(self):
+ self.check_command("mkvmerge")
+ self.check_no_file(self.opts.output + ".mkv")
- def insertOptions(self, cmd):
- def tryOpt(opt, var):
- if var is not None:
- cmd.append(opt)
- cmd.append(var)
- tryOpt("-ss", self.opts.startpos)
- tryOpt("-endpos", self.opts.endpos)
- tryOpt("-dvd-device", self.opts.dvd)
- tryOpt("-chapter", self.opts.chapter)
- tryOpt("-aid", self.opts.audioid)
- tryOpt("-sid", self.opts.subtitleid)
- tryOpt("-vf", self.opts.vfilters)
- tryOpt("-af", self.opts.afilters)
-
- def substValues(self, cmd):
- subst = {
- "vbitrate": self.opts.vbitrate,
- "abitrate": self.opts.abitrate,
- "input": self.opts.input,
- "output": self.opts.output,
- }
-
- return [x % subst for x in cmd]
+ def run(self):
+ o = self.opts
+ p = self.profile
+
+ if o.dump:
+ fps = "???"
+ else:
+ fps = midentify(p.video_tmp, "ID_VIDEO_FPS")
+
+ self.do_exec([
+ "mkvmerge",
+ "-o", o.output + ".mkv",
+ "--default-duration", "0:%sfps"%fps,
+ p.video_tmp,
+ p.audio_tmp,
+ ])
+
+
+
+class MencoderLossless(Command):
+ def init(self):
+ self.check_command("mencoder")
+ self.check_no_file("lossless.avi")
+
+ ofut = self.opts
+ self.opts = duplicate_opts(ofut)
+ ofut.input = "lossless.avi"
+ ofut.mplayer_done = True
+
+ def run(self):
+ fifo = False
+ if fifo:
+ os.mkfifo("lossless.avi")
+ o = self.opts
+ cmd = []
+ cmd += ["mencoder", self.opts.input, "-o", "lossless.avi"]
+ cmd += ["-noconfig", "all"]
+ cmd += ["-oac", "copy", "-ovc", "lavc", "-lavcopts", "vcodec=ffv1:autoaspect"]
+ insert_mplayer_options(cmd, self.opts)
+ cmd += ["-vf-add", "harddup"]
+ self.do_exec(cmd, wait=not fifo)
+
+
+
+class MPlayer(Command):
+ def init(self):
+ self.check_command("mplayer")
+ self.check_no_file("video.y4m")
+ self.check_no_file("audio.wav")
+
+ def run(self):
+ os.mkfifo("video.y4m")
+ os.mkfifo("audio.wav")
+ cmd = []
+ cmd += ["mplayer", self.opts.input]
+ cmd += ["-benchmark", "-noconsolecontrols", "-noconfig", "all"]
+ cmd += ["-vo", "yuv4mpeg:file=video.y4m"]
+ cmd += ["-ao", "pcm:waveheader:file=audio.wav"]
+ insert_mplayer_options(cmd, self.opts)
+ cmd += self.profile.mplayeropts
+ self.do_exec(cmd, wait=False)
+
+
+class MencoderCopyAC3(Command):
+ def init(self):
+ self.check_command("mplayer")
+ self.check_no_file("audio.ac3")
+ self.profile.audio_tmp = "audio.ac3"
- def pass1(self):
+ def run(self):
+ cmd = []
+ cmd += ["mencoder", self.opts.input]
+ cmd += ["-noconfig", "all"]
+ cmd += ["-ovc", "copy", "-oac", "copy"]
+ cmd += ["-of", "rawaudio", "-o", "audio.ac3"]
+ insert_mplayer_options(cmd, self.opts)
+ self.do_exec(cmd)
+
+
+class X264(Command):
+ def init(self):
+ self.check_command("x264")
+ self.profile.video_tmp = "video.h264"
+
+ def run(self):
+ p = self.profile
+ cmd = []
+ cmd += ["x264", "--no-progress"]
+ cmd += p.x264opts
+ cmd += ["-o", p.video_tmp]
+ cmd += ["video.y4m"]
+ self.do_exec(cmd, wait=False)
+
+
+class Lame(Command):
+ def init(self):
+ self.check_command("lame")
+ self.profile.audio_tmp = "audio.mp3"
+
+ def run(self):
+ p = self.profile
+ cmd = []
+ cmd += ["lame", "--quiet"]
+ cmd += p.lameopts
+ cmd += ["audio.wav"]
+ cmd += [p.audio_tmp]
+ self.do_exec(cmd, wait=False)
+
+
+class Faac(Command):
+ def init(self):
+ self.check_command("faac")
+ self.profile.audio_tmp = "audio.aac"
+
+ def run(self):
p = self.profile
cmd = []
- cmd += ["mencoder", "%(input)s", "-o", "/dev/null"]
- self.insertOptions(cmd)
- cmd += ["-ovc", p.vcodec, self.codec2opts[p.vcodec], "pass=1:"+p.vopts]
- cmd += ["-oac", "copy"]
- cmd = self.substValues(cmd)
- return cmd
-
- def pass2(self):
+ cmd += ["faac"]
+ cmd += ["-o", p.audio_tmp]
+ cmd += p.faacopts
+ cmd += ["audio.wav"]
+ self.do_exec(cmd, wait=False)
+
+
+class Mencoder(Command):
+ codec2opts = {
+ "xvid": "-xvidencopts",
+ "x264": "-x264encopts",
+ "faac": "-faacopts",
+ "mp3lame": "-lameopts",
+ }
+
+ def init(self):
+ o = self.opts
+ p = self.profile
+
+ self.check_command("mencoder")
+ self.check_no_file(o.output + ".avi")
+
+ p.video_tmp = o.output + ".avi"
+ p.audio_tmp = o.output + ".avi"
+
+ def run(self):
+ o = self.opts
p = self.profile
+
cmd = []
- cmd += ["mencoder", "%(input)s", "-o", "%(output)s"]
- self.insertOptions(cmd)
- cmd += ["-ovc", p.vcodec, self.codec2opts[p.vcodec], "pass=2:"+p.vopts]
- cmd += ["-oac", p.acodec, self.codec2opts[p.acodec], p.aopts]
- if self.opts.episode_name:
- cmd += ["-info", "name='%s'" % self.opts.episode_name]
- cmd += self.profile.extra
- cmd = self.substValues(cmd)
- return cmd
+ cmd += ["mencoder", o.input]
+ cmd += ["-noconfig", "all"]
+ insert_mplayer_options(cmd, o)
+ cmd += ["-vf-add", "harddup"]
+ cmd += ["-ovc", p.vcodec, self.codec2opts[p.vcodec], p.vopts]
+ cmd += ["-oac", p.acodec]
+ if p.aopts:
+ cmd += [self.codec2opts[p.acodec], p.aopts]
+ cmd += self.profile.mplayeropts
+ cmd += ["-o", self.opts.output + ".avi"]
+
+ self.do_exec(cmd)
+
+
+class MencoderDemux(Command):
+ codec2exts = {
+ "xvid": "m4v",
+ "x264": "h264",
+ "faac": "aac",
+ "mp3lame": "mp3",
+ "copyac3": "ac3",
+ }
+
+ def init(self):
+ o = self.opts
+ p = self.profile
+
+ self.check_command("mencoder")
+ p.audio_tmp = "audio." + self.codec2exts[p.acodec]
+ p.video_tmp = "video." + self.codec2exts[p.vcodec]
+ self.check_no_file(p.audio_tmp)
+ self.check_no_file(p.video_tmp)
+
+ def run(self):
+ o = self.opts
+ p = self.profile
+
+ cmd = ["mencoder", "-ovc", "copy", "-oac", "copy", o.output + ".avi"]
+ cmd += ["-noconfig", "all", "-noskip", "-mc", "0"]
+ self.do_exec(cmd + ["-of", "rawaudio", "-o", p.audio_tmp])
+ self.do_exec(cmd + ["-of", "rawvideo", "-o", p.video_tmp])
+ self.do_exec(["rm", "-f", o.output + ".avi"])
+
+
class Profile(object):
- def __init__(self, CommandClass, **kwargs):
- self.CommandClass = CommandClass
+ def __init__(self, commands, **kwargs):
+ self.commands = commands
self.__dict__.update(kwargs)
+ def __contains__(self, keyname):
+ return hasattr(self, keyname)
+
+class Wait(object):
+ def __init__(self, commands):
+ self.commands = commands[:]
+
+ def run(self):
+ for command in self.commands:
+ command.wait()
+
+
+
profiles = {
- "qt7" :
+ "x264/lame" :
Profile(
- CommandClass=MencoderCommand,
- vcodec="x264",
- vopts="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",
+ commands=[MPlayer, X264, Lame, Wait, MKVMerge],
+ mplayeropts=[],
+ x264opts=["--preset", "veryslow", "--crf", "20"],
+ lameopts=["--preset", "medium"],
+ ),
+
+ "x264/copyac3" :
+ Profile(
+ commands=[MPlayer, X264, Wait, MencoderCopyAC3, MKVMerge],
+ mplayeropts=["-nosound"],
+ x264opts=["--preset", "veryslow", "--crf", "20"],
+ ),
+
+ "x264/lame/fast" :
+ Profile(
+ commands=[MPlayer, X264, Lame, Wait, MKVMerge],
+ mplayeropts=[],
+ x264opts=["--preset", "fast", "--crf", "23"],
+ lameopts=["--preset", "medium"],
),
- "xvid" :
+ "xvid/lame" :
Profile(
- CommandClass=MencoderCommand,
+ commands=[Mencoder],
+ mplayeropts=["-ffourcc", "DX50"],
vcodec="xvid",
- vopts="bitrate=%(vbitrate)d:vhq=4:autoaspect",
+ vopts="fixed_quant=2:vhq=4:autoaspect",
acodec="mp3lame",
- aopts="abr:br=%(abitrate)d",
- extra=["-ffourcc", "DX50"],
+ aopts="cbr:br=128",
+ ),
+
+ "apple-quicktime" :
+ Profile(
+ commands=[MPlayer, X264, Faac, Wait, MP4Box],
+ mplayeropts=[],
+ x264opts=["--crf", "20", "--bframes", "1"],
+ faacopts=["-q", "100", "--mpeg-vers", "4"],
),
- "ipodxvid" :
+
+ "nokia-n97" :
Profile(
- CommandClass=MencoderCommand,
+ commands=[Mencoder, MencoderDemux, MP4Box],
+ mplayeropts=["-vf-add", "scale=640:-10"],
vcodec="xvid",
- vopts="bitrate=%(vbitrate)d:vhq=4:autoaspect:max_bframes=0",
+ vopts="bitrate=384:vhq=4:autoaspect:max_bframes=0",
acodec="faac",
- aopts="br=%(abitrate)d:mpeg=4:object=2",
+ aopts="br=64:mpeg=4:object=2",
),
}
+mappings = {
+ "x264": "x264/lame",
+ "xvid": "xvid/lame",
+}
+for x, y in mappings.iteritems():
+ profiles[x] = profiles[y]
+
if sys.argv[0].find(profile_name) >= 0:
break
else:
- profile_name = "xvid"
+ profile_name = "xvid/lame"
- parser = optparse.OptionParser(usage="%prog [options] input output")
+ parser = optparse.OptionParser(usage="%prog [options] input [output]")
parser.add_option("--dvd", action="store", dest="dvd")
+ parser.add_option("--fixmux", action="store_true", dest="fixmux")
+ parser.add_option("--deinterlace", action="store_true", dest="deinterlace")
+ parser.add_option("--detelecine", action="store_true", dest="detelecine")
+ parser.add_option("--mc", action="store", dest="mc", type="float")
+ parser.add_option("--noskip", action="store_true", dest="noskip")
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", default=1000)
- parser.add_option("--abitrate", action="store", dest="abitrate", type="int", default=192)
parser.add_option("--chapter", action="store", dest="chapter")
+ parser.add_option("--skipkb", action="store", dest="skipkb", type="int")
parser.add_option("--startpos", action="store", dest="startpos")
parser.add_option("--endpos", action="store", dest="endpos")
parser.add_option("--audioid", action="store", dest="audioid")
parser.add_option("--subtitleid", action="store", dest="subtitleid")
parser.add_option("--profile", action="store", dest="profile_name", default=profile_name)
- parser.add_option("--episode-name", action="store", dest="episode_name")
parser.add_option("--dump", action="store_true", dest="dump")
try:
- opts, (input, output) = parser.parse_args(sys.argv[1:])
+ opts, args = parser.parse_args(sys.argv[1:])
+ if len(args) == 1:
+ input = args[0]
+ output = os.path.splitext(os.path.basename(input))[0]
+ elif len(args) == 2:
+ input, output = args
+ else:
+ raise ValueError
except Exception:
parser.print_usage()
sys.exit(1)
-
- opts.input = input
- opts.output = output
- return opts
-def run(args, dump):
- if dump:
- print "".join(map(commands.mkarg, args))[1:]
+ if "://" not in input:
+ opts.input = os.path.abspath(input)
else:
- return subprocess.Popen(args).wait()
+ if opts.dvd:
+ opts.dvd = os.path.abspath(opts.dvd)
+ opts.input = input
+
+ opts.output = os.path.abspath(output)
+ opts.mplayer_done = False
+
+ return opts
def main():
+ os.nice(1)
+
opts = parse_args()
+
+ # Find our profile
try:
profile = profiles[opts.profile_name]
except KeyError:
- print >>sys.stderr, "Profile '%s' not found!" % profile_name
+ print >>sys.stderr, "Profile '%s' not found!" % opts.profile_name
sys.exit(1)
- cmd = profile.CommandClass(profile, opts)
- if run(cmd.pass1(), opts.dump) == 0 or opts.dump:
- run(cmd.pass2(), opts.dump)
+ # Run in a temp dir so that multiple instances can be run simultaneously
+ tempdir = tempfile.mkdtemp()
+ try:
+ os.chdir(tempdir)
+
+ try:
+ commands = []
+ if opts.fixmux or opts.detelecine:
+ profile.commands.insert(0, MencoderLossless)
+ for CommandClass in profile.commands:
+ if Command in CommandClass.__bases__:
+ command = CommandClass(profile, opts)
+ else:
+ command = CommandClass(commands)
+ commands.append(command)
+ for command in commands:
+ command.run()
+
+ except FatalException, e:
+ print >>sys.stderr, "Error:", str(e)
+ sys.exit(1)
+
+ finally:
+ os.chdir("/")
+ shutil.rmtree(tempdir)
if __name__ == "__main__":
main()