#!/usr/bin/env python
-import commands
import optparse
+import re
import subprocess
import sys
import os
-import tempfile
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
+
class Command(object):
def __init__(self, profile, opts):
self.profile = profile
def do_exec(self, args):
if self.opts.dump:
- print "".join(map(commands.mkarg, args))[1:]
+ print " ".join(map(mkarg, args))
else:
if subprocess.Popen(args).wait() != 0:
raise FatalException("Failure executing command: %s" % args)
def run(self):
p = self.profile
- video = "video.%s" % self.codec2exts[p.vcodec]
- audio = "audio.%s" % self.codec2exts[p.acodec]
input = self.opts.output + ".avi" # From Mencoder command
output = self.opts.output + ".mp4"
+
+ # Check FPS
+ fps = "???"
+ if not self.opts.dump:
+ process = subprocess.Popen(
+ ["mplayer", "-frames", "0", "-identify", input],
+ stdout=subprocess.PIPE,
+ stderr=subprocess.PIPE,
+ )
+ for line in process.stdout:
+ try:
+ key, value = line.split("=")
+ except ValueError:
+ continue
+ if key == "ID_VIDEO_FPS":
+ fps = value
+
+ # Strip out video & audio
+ video = "video.%s" % self.codec2exts[p.vcodec]
+ audio = "audio.%s" % self.codec2exts[p.acodec]
mencoder = ["mencoder", input, "-ovc", "copy", "-oac", "copy", "-of"]
self.do_exec(["rm", "-f", output])
self.do_exec(mencoder + ["rawvideo", "-o", video])
self.do_exec(mencoder + ["rawaudio", "-o", audio])
- self.do_exec(["MP4Box", "-add", video, "-add", audio, output])
+
+ # Mux them back together
+ self.do_exec(["MP4Box", "-add", video, "-add", audio, "-fps", fps, output])
+
+ # Clean up temp files
self.do_exec(["rm", "-f", video, audio, input])
+
+class MKVMerge(Command):
+ def check(self):
+ self.check_command("mkvmerge")
+ self.check_no_file(self.opts.output + ".mkv")
+
+ def run(self):
+ input = self.opts.output + ".avi" # From Mencoder command
+ output = self.opts.output + ".mkv"
+ self.do_exec(["mkvmerge", "-o", output, input])
+ self.do_exec(["rm", "-f", input])
+
+
+
class Mencoder(Command):
codec2opts = {
- "lavc": "-lavcopts",
"xvid": "-xvidencopts",
"x264": "-x264encopts",
"faac": "-faacopts",
cmd.append(var)
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("-aid", self.opts.audioid)
try_opt("-sid", self.opts.subtitleid)
try_opt("-vf-add", self.opts.vfilters)
- try_opt("-af", self.opts.afilters)
+ try_opt("-af-add", self.opts.afilters)
def subst_values(self, cmd, vpass):
subst = {
self.insert_options(cmd)
cmd += ["-ovc", p.vcodec, self.codec2opts[p.vcodec], p.vopts]
cmd += ["-oac", "copy"]
- cmd += self.profile.extra + self.profile.extra1
+ cmd += self.profile.extra1 + self.profile.extra
cmd = self.subst_values(cmd, vpass=1)
return cmd
cmd += ["mencoder", "%(input)s", "-o", "%(output)s"]
self.insert_options(cmd)
cmd += ["-ovc", p.vcodec, self.codec2opts[p.vcodec], p.vopts]
- cmd += ["-oac", p.acodec, self.codec2opts[p.acodec], p.aopts]
+ cmd += ["-oac", p.acodec]
+ if "aopts" in p:
+ cmd += [self.codec2opts[p.acodec], p.aopts]
if self.opts.episode_name:
cmd += ["-info", "name='%s'" % self.opts.episode_name]
- cmd += self.profile.extra + self.profile.extra2
+ cmd += self.profile.extra2 + self.profile.extra
cmd = self.subst_values(cmd, vpass=2)
return cmd
profiles = {
- "qt7" :
+ "x264" :
Profile(
- commands=[Mencoder, MP4Box],
+ commands=[Mencoder, MKVMerge],
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",
+ 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",
+ extra=["-vf-add", "harddup"],
),
"xvid" :
extra2=["-ffourcc", "DX50"],
),
- "ipodxvid" :
+ "apple-quicktime" :
Profile(
commands=[Mencoder, MP4Box],
- vcodec="xvid",
- vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:vhq=4:autoaspect:max_bframes=0",
+ 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",
- extra=["-vf-add", "scale=480:-10"],
+ extra=["-vf-add", "harddup"],
),
- "ipodx264" :
+ "ipod-xvid" :
Profile(
commands=[Mencoder, MP4Box],
- vcodec="x264",
- 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",
+ 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=480:-10"],
- extra2=["-channels", "2", "-srate", "48000"],
+ extra=["-vf-add", "scale=480:-10,harddup"],
),
- "nokiax264" :
+ "ipod-x264" :
Profile(
commands=[Mencoder, MP4Box],
- default_opts={
- "vbitrate": 256,
- "abitrate": 96,
- },
vcodec="x264",
- vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:nocabac:me=umh:partitions=all:trellis=1:subq=7:bframes=0:direct_pred=auto",
+ 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",
- extra=["-vf-add", "scale=320:-10"],
+ extra=["-vf-add", "scale=480:-10,harddup"],
+ extra2=["-channels", "2", "-srate", "48000"],
),
- "n97xvid" :
+ "nokia-n97" :
Profile(
commands=[Mencoder, MP4Box],
default_opts={
- "vbitrate": 1000,
+ "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"],
- ),
-
- "n97x264" :
- Profile(
- commands=[Mencoder, MP4Box],
- default_opts={
- "vbitrate": 1000,
- "abitrate": 96,
- },
- vcodec="x264",
- vopts="pass=%(vpass)d:bitrate=%(vbitrate)d:vbv_maxrate=2000:vbv_bufsize=2000:nocabac:me=umh:partitions=all:trellis=1:subq=7:bframes=0:direct_pred=auto:level_idc=20",
- acodec="faac",
- aopts="br=%(abitrate)d:mpeg=4:object=2",
- extra=["-vf-add", "scale=640:-10"],
+ extra=["-vf-add", "scale=640:-10,harddup"],
),
}
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("--vfilters", action="store", dest="vfilters")
parser.add_option("--afilters", action="store", dest="afilters")
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")
command.run()
except FatalException, e:
- print >>sys.stderr, "Error:", e.message
+ print >>sys.stderr, "Error:", str(e)
sys.exit(1)
finally: