X-Git-Url: https://code.delx.au/transcoding/blobdiff_plain/bcc820a76ab96fbb0f9529510ddd0453f3383614..dfdf1ac02ed71f25d4bce552aeaae6aea86c5f96:/encode.py diff --git a/encode.py b/encode.py index fc3b7c8..6e35071 100755 --- a/encode.py +++ b/encode.py @@ -1,111 +1,303 @@ #!/usr/bin/env python -import commands, optparse, subprocess, sys, os +import optparse +import re +import subprocess +import sys +import os +import shutil +import tempfile -class MencoderCommand(object): +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 = { - "lavc": "-lavcopts", "xvid": "-xvidencopts", "x264": "-x264encopts", "faac": "-faacopts", "mp3lame": "-lameopts", } - def __init__(self, profile, opts): - self.profile = profile - self.opts = opts - - def insertOptions(self, cmd): - def tryOpt(opt, var): + def insert_options(self, cmd): + def try_opt(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): + 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, - "input": self.opts.input, - "output": self.opts.output, + "vpass": vpass, } return [x % subst for x in cmd] - - def pass1(self): + + 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", "%(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) + 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): - 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) + 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, CommandClass, **kwargs): + def __init__(self, commands, **kwargs): + self.default_opts = { + "vbitrate": 1000, + "abitrate": 192, + } self.extra = [] - - self.CommandClass = CommandClass + self.extra1 = [] + self.extra2 = [] + self.commands = commands self.__dict__.update(kwargs) + def __contains__(self, keyname): return hasattr(self, keyname) + profiles = { - "qt7" : + "x264" : Profile( - CommandClass=MencoderCommand, + commands=[Mencoder, MKVMerge], 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", + 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( - CommandClass=MencoderCommand, + commands=[Mencoder, MencoderMux], vcodec="xvid", - vopts="bitrate=%(vbitrate)d:vhq=4:autoaspect", + vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:vhq=4:autoaspect", acodec="mp3lame", aopts="abr:br=%(abitrate)d", - extra=["-ffourcc", "DX50"], ), - "ipodxvid" : + + "apple-quicktime" : Profile( - CommandClass=MencoderCommand, + 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="bitrate=%(vbitrate)d:vhq=4:autoaspect:max_bframes=0", + 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"], ), - "ipod264" : + + "ipod-x264" : Profile( - CommandClass=MencoderCommand, + commands=[Mencoder, MP4Box], vcodec="x264", - vopts="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", - extra=['-of', 'lavf', '-lavfopts', 'format=mp4', '-channels', '2', '-srate', '48000'] + 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": 800, + "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"], ), } @@ -119,47 +311,84 @@ def parse_args(): else: profile_name = "xvid" - 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("--deinterlace", action="store_true", dest="deinterlace") + parser.add_option("--detelecine", action="store_true", dest="detelecine") + parser.add_option("--copyac3", action="store_true", dest="copyac3") 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("--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") 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) + + return opts def main(): 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) + # Pull in default option values from the profile + for key, value in profile.default_opts.iteritems(): + if getattr(opts, key) is None: + setattr(opts, key, value) + + # Run in a temp dir so that multiple instances can be run simultaneously + tempdir = tempfile.mkdtemp() + try: + os.chdir(tempdir) + + try: + commands = [] + for CommandClass in profile.commands: + command = CommandClass(profile, opts) + commands.append(command) + command.check() + 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()