X-Git-Url: https://code.delx.au/transcoding/blobdiff_plain/840ae4115dbdf9925d10e60f415e2b0901741335..1ed01efcec794b87acd65726f58dc2101810f6c1:/encode.py diff --git a/encode.py b/encode.py index 2dc94a6..31cc0de 100755 --- a/encode.py +++ b/encode.py @@ -1,94 +1,504 @@ #!/usr/bin/env python -import optparse, subprocess, sys - -codecs = { -"x264": -[ - "mencoder", "%(input)s", "-o", "%(output)s", - "-vf", "%(filters)s", - "-ovc", "x264", "-x264encopts", "pass=%(pass)d:bitrate=%(vbitrate)d:me=umh:partitions=all:trellis=1:subq=7:bframes=1:direct_pred=auto", - "-oac", "faac", "-faacopts", "br=%(abitrate)d:mpeg=4:object=2", "-channels", "2", "-srate", "48000", -], - -"xvid": -[ - "mencoder", "%(input)s", "-o", "%(output)s", - "-ffourcc", "DX50", - "-vf", "%(filters)s", - "-ovc", "xvid", "-xvidencopts", "pass=%(pass)d:bitrate=%(vbitrate)d:vhq=4", - "-oac", "mp3lame", "-lameopts", "abr:br=%(abitrate)d", -], +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): + do_opt = partial(append_cmd, cmd) + + if o.deinterlace: + cmd += ["-vf-add", "yadif"] + if o.noskip: + cmd += ["-noskip"] + if o.skipkb: + cmd += ["-sb", str(o.skipkb * 1024)] + + do_opt("-mc", o.mc) + do_opt("-fps", o.ifps) + 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-add", o.vfilters) + do_opt("-af-add", o.afilters) + + +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 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 MencoderFixRemux(Command): + def init(self): + self.check_command("mencoder") + self.check_no_file("remux.avi") + + orig = self.opts + self.opts = duplicate_opts(orig) + orig.input = "remux.avi" + orig.dvd = orig.chapter = orig.startpos = orig.endpos = None + + def run(self): + o = self.opts + cmd = [ + "mencoder", + "-o", "remux.avi", + "-oac", "copy", "-ovc", "copy", + "-mc", "0.1", + o.input, + ] + do_opt = partial(append_cmd, cmd) + do_opt("-dvd-device", o.dvd) + do_opt("-chapter", o.chapter) + do_opt("-ss", o.startpos) + do_opt("-endpos", o.endpos) + self.do_exec(cmd) + + + + + +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.extra + 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 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) + cmd += self.profile.extra + 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 += ["faac"] + cmd += ["-o", p.audio_tmp] + cmd += p.faacopts + cmd += ["audio.wav"] + self.do_exec(cmd, wait=False) + + +class SwallowAudio(Command): + def run(self): + self.do_exec(["dd", "if=audio.wav", "of=/dev/null"], 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", 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.extra + 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, commands, **kwargs): + self.extra = [] + 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 = { + "x264/lame" : + Profile( + commands=[MPlayer, X264, Lame, Wait, MKVMerge], + x264opts=["--preset", "veryslow", "--crf", "20"], + lameopts=["--preset", "medium"], + ), + + "x264/copyac3" : + Profile( + commands=[MPlayer, X264, SwallowAudio, Wait, MencoderCopyAC3, MKVMerge], + x264opts=["--preset", "veryslow", "--crf", "20"], + lameopts=["--preset", "medium"], + ), + + "xvid/lame" : + Profile( + commands=[Mencoder], + vcodec="xvid", + vopts="fixed_quant=2:vhq=4:autoaspect", + acodec="mp3lame", + aopts="cbr:br=128", + ), + + "apple-quicktime" : + Profile( + commands=[MPlayer, X264, Faac, Wait, MP4Box], + x264opts=["--crf", "20", "--bframes", "1"], + faacopts=["-q", "100", "--mpeg-vers", "4"], + ), + + "nokia-n97" : + Profile( + commands=[Mencoder, MencoderDemux, MP4Box], + vcodec="xvid", + vopts="bitrate=256:vhq=4:autoaspect:max_bframes=0", + acodec="faac", + aopts="br=64:mpeg=4:object=2", + extra=["-vf-add", "scale=640:-10"], + ), } +mappings = { + "x264": "x264/lame", + "xvid": "xvid/lame", +} +for x, y in mappings.iteritems(): + profiles[x] = profiles[y] + + -def parseArgs(): - for codec in codecs.keys(): - if sys.argv[0].find(codec) >= 0: + +def parse_args(): + for profile_name in profiles.keys(): + if sys.argv[0].find(profile_name) >= 0: break else: - codec = "x264" + 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("--filters", action="store", dest="filters", default="denoise3d") - parser.add_option("--vbitrate", action="store", dest="vbitrate", type="int", default=700) - parser.add_option("--abitrate", action="store", dest="abitrate", type="int", default=128) + parser.add_option("--deinterlace", action="store_true", dest="deinterlace") + parser.add_option("--fixmux", action="store_true", dest="fixmux") + parser.add_option("--mc", action="store", dest="mc", type="int") + 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("--chapter", action="store", dest="chapter") + parser.add_option("--ifps", action="store", dest="ifps") + 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("--codec", action="store", dest="codec", default=codec) + parser.add_option("--subtitleid", action="store", dest="subtitleid") + parser.add_option("--profile", action="store", dest="profile_name", default=profile_name) parser.add_option("--dump", action="store_true", dest="dump") try: - opts, (input, output) = parser.parse_args(sys.argv[1:]) - except: + 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) - - return opts, codec, input, output -def run(args, dump): - if dump: - print " ".join(args) + if "://" not in input: + opts.input = os.path.abspath(input) else: - subprocess.Popen(args).wait() + if opts.dvd: + opts.dvd = os.path.abspath(opts.dvd) + opts.input = input + + opts.output = os.path.abspath(output) + + return opts def main(): - opts, codec, input, output = parseArgs() + os.nice(1) + + opts = parse_args() + + # Find our profile try: - cmd = codecs[codec] - except: - print >>sys.stderr, "Codec '%s' not found!" % codec + profile = profiles[opts.profile_name] + except KeyError: + print >>sys.stderr, "Profile '%s' not found!" % opts.profile_name sys.exit(1) + # Run in a temp dir so that multiple instances can be run simultaneously + tempdir = tempfile.mkdtemp() + try: + os.chdir(tempdir) - def insertOpt(opt, var): - if var: - cmd.insert(1, var) - cmd.insert(1, opt) - insertOpt("-ss", opts.startpos) - insertOpt("-endpos", opts.endpos) - insertOpt("-dvd-device", opts.dvd) - insertOpt("-chapter", opts.chapter) - insertOpt("-aid", opts.audioid) - - subst = { - "vbitrate": opts.vbitrate, - "abitrate": opts.abitrate, - "filters": opts.filters, - "input": input, - } + try: + commands = [] + if opts.fixmux: + profile.commands.insert(0, MencoderFixRemux) + 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() - # Pass 1 - subst["pass"] = 1 - subst["output"] = "/dev/null" - run([x % subst for x in cmd], opts.dump) + except FatalException, e: + print >>sys.stderr, "Error:", str(e) + sys.exit(1) - # Pass 2 - subst["pass"] = 2 - subst["output"] = output - run([x % subst for x in cmd], opts.dump) + finally: + os.chdir("/") + shutil.rmtree(tempdir) if __name__ == "__main__": main()