code.delx.au - transcoding/blob - encode.py

   1 #!/usr/bin/env python
   2
   3 import optparse, subprocess, sys
   4
   5 codecs = {
   6 "x264":
   7 [
   8         "mencoder", "%(input)s", "-o", "%(output)s",
   9         "-vf", "%(filters)s",
  10         "-ovc", "x264", "-x264encopts", "pass=%(pass)d:bitrate=%(vbitrate)d:me=umh:partitions=all:trellis=1",
  11         "-oac", "faac", "-faacopts", "br=%(abitrate)d:mpeg=4:object=2", "-channels", "2", "-srate", "48000",
  12 ],
  13
  14 "xvid":
  15 [
  16         "mencoder", "%(input)s", "-o", "%(output)s",
  17         "-ffourcc", "DX50",
  18         "-vf", "%(filters)s",
  19         "-ovc", "xvid", "-xvidencopts", "pass=%(pass)d:bitrate=%(vbitrate)d:vhq=4",
  20         "-oac", "mp3lame", "-lameopts", "abr:br=%(abitrate)d",
  21 ],
  22 }
  23
  24
  25 def parseArgs():
  26         for codec in codecs.keys():
  27                 if sys.argv[0].find(codec) >= 0:
  28                         break
  29         else:
  30                 codec = "x264"
  31
  32         parser = optparse.OptionParser(usage="%prog [options] input output")
  33         parser.add_option("--dvd", action="store", dest="dvd")
  34         parser.add_option("--filters", action="store", dest="filters", default="denoise3d")
  35         parser.add_option("--vbitrate", action="store", dest="vbitrate", type="int", default=600)
  36         parser.add_option("--abitrate", action="store", dest="abitrate", type="int", default=192)
  37         parser.add_option("--chapter", action="store", dest="chapter")
  38         parser.add_option("--startpos", action="store", dest="startpos")
  39         parser.add_option("--endpos", action="store", dest="endpos")
  40         parser.add_option("--audioid", action="store", dest="audioid")
  41         parser.add_option("--codec", action="store", dest="codec", default=codec)
  42         parser.add_option("--dump", action="store_true", dest="dump")
  43         try:
  44                 opts, (input, output) = parser.parse_args(sys.argv[1:])
  45         except:
  46                 parser.print_usage()
  47                 sys.exit(1)
  48
  49         return opts, codec, input, output
  50
  51 def run(args, dump):
  52         if dump:
  53                 print " ".join(args)
  54         else:
  55                 subprocess.Popen(args).wait()
  56
  57 def main():
  58         opts, codec, input, output = parseArgs()
  59         try:
  60                 cmd = codecs[codec]
  61         except:
  62                 print >>sys.stderr, "Codec '%s' not found!" % codec
  63                 sys.exit(1)
  64
  65
  66         def insertOpt(opt, var):
  67                 if var:
  68                         cmd.insert(1, var)
  69                         cmd.insert(1, opt)
  70         insertOpt("-ss", opts.startpos)
  71         insertOpt("-endpos", opts.endpos)
  72         insertOpt("-dvd-device", opts.dvd)
  73         insertOpt("-chapter", opts.chapter)
  74         insertOpt("-aid", opts.audioid)
  75
  76         subst = {
  77                 "vbitrate": opts.vbitrate,
  78                 "abitrate": opts.abitrate,
  79                 "filters": opts.filters,
  80                 "input": input,
  81         }
  82
  83         # Pass 1
  84         subst["pass"] = 1
  85         subst["output"] = "/dev/null"
  86         run([x % subst for x in cmd], opts.dump)
  87
  88         # Pass 2
  89         subst["pass"] = 2
  90         subst["output"] = output
  91         run([x % subst for x in cmd], opts.dump)
  92
  93 if __name__ == "__main__":
  94         main()
  95