Merged in the changes from my branch.

[transcoding] / encode.py

#!/usr/bin/env python

import commands, optparse, subprocess, sys, os

class MencoderCommand(object):
        codec2opts = {
                "lavc": "-lavcopts",
                "xvid": "-xvidencopts",
                "x264": "-x264encopts",
                "faac": "-faaccopts",
                "mp3lame": "-lameopts",
        }

        def __init__(self, profile, opts):
                self.profile = profile
                self.opts = opts

        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 pass1(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):
                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]
                cmd += self.profile.extra
                cmd = self.substValues(cmd)
                return cmd

class Profile(object):
        def __init__(self, CommandClass, **kwargs):
                self.CommandClass = CommandClass
                self.__dict__.update(kwargs)

profiles = {
        "qt7" :
        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",
        ),

        "xvid" :
        Profile(
                CommandClass=MencoderCommand,
                vcodec="xvid",
                vopts="bitrate=%(vbitrate)d:vhq=4:autoaspect",
                acodec="mp3lame",
                aopts="abr:br=%(abitrate)d",
                extra=["-ffourcc", "DX50"],
        ),
        "ipodxvid" :
        Profile(
                CommandClass=MencoderCommand,
                vcodec="xvid",
                vopts="bitrate=%(vbitrate)d:vhq=4:autoaspect:max_bframes=0",
                acodec="faac",
                aopts="br=%(abitrate)d:mpeg=4:object=2",
        ),
}




def parse_args():
        for profile_name in profiles.keys():
                if sys.argv[0].find(profile_name) >= 0:
                        break
        else:
                profile_name = "xvid"

        parser = optparse.OptionParser(usage="%prog [options] input output")
        parser.add_option("--dvd", action="store", dest="dvd")
        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("--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("--dump", action="store_true", dest="dump")
        try:
                opts, (input, output) = parser.parse_args(sys.argv[1:])
        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:]
        else:
                return subprocess.Popen(args).wait()

def main():
        opts = parse_args()
        try:
                profile = profiles[opts.profile_name]
        except KeyError:
                print >>sys.stderr, "Profile '%s' not found!" % 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)

if __name__ == "__main__":
        main()