X-Git-Url: https://code.delx.au/transcoding/blobdiff_plain/34aafc95b0d851b45421e50a3005bd0aac44d834..4515ace8b9c4721c427a163b63341e8b2a5a374c:/py-encode

diff --git a/py-encode b/py-encode
new file mode 100755
index 0000000..e92d89d
--- /dev/null
+++ b/py-encode
@@ -0,0 +1,508 @@
+#!/usr/bin/env python
+
+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):
+	if o.mplayer_done:
+		return
+
+	do_opt = partial(append_cmd, cmd)
+	do_opt("-mc", o.mc)
+	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", o.vfilters)
+	do_opt("-af", o.afilters)
+
+	if o.deinterlace:
+		cmd += ["-vf-pre", "yadif"]
+	if o.detelecine:
+		cmd += ["-vf-pre", "pullup,softskip", "-ofps", "24000/1001"]
+	if o.noskip:
+		cmd += ["-noskip"]
+	if o.skipkb:
+		cmd += ["-sb", str(o.skipkb * 1024)]
+
+
+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 MencoderLossless(Command):
+	def init(self):
+		self.check_command("mencoder")
+		self.check_no_file("lossless.avi")
+
+		ofut = self.opts
+		self.opts = duplicate_opts(ofut)
+		ofut.input = "lossless.avi"
+		ofut.mplayer_done = True
+
+	def run(self):
+		fifo = False
+		if fifo:
+			os.mkfifo("lossless.avi")
+		o = self.opts
+		cmd = []
+		cmd += ["mencoder", self.opts.input, "-o", "lossless.avi"]
+		cmd += ["-noconfig", "all"]
+		cmd += ["-oac", "copy", "-ovc", "lavc", "-lavcopts", "vcodec=ffv1:autoaspect"]
+		insert_mplayer_options(cmd, self.opts)
+		cmd += ["-vf-add", "harddup"]
+		self.do_exec(cmd, wait=not fifo)
+
+
+
+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.mplayeropts
+		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)
+		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 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.mplayeropts
+		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.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],
+		mplayeropts=[],
+		x264opts=["--preset", "veryslow", "--crf", "20"],
+		lameopts=["--preset", "medium"],
+	),
+
+	"x264/copyac3" :
+	Profile(
+		commands=[MPlayer, X264, Wait, MencoderCopyAC3, MKVMerge],
+		mplayeropts=["-nosound"],
+		x264opts=["--preset", "veryslow", "--crf", "20"],
+	),
+
+	"x264/lame/fast" :
+	Profile(
+		commands=[MPlayer, X264, Lame, Wait, MKVMerge],
+		mplayeropts=[],
+		x264opts=["--preset", "fast", "--crf", "23"],
+		lameopts=["--preset", "medium"],
+	),
+
+	"xvid/lame" :
+	Profile(
+		commands=[Mencoder],
+		mplayeropts=["-ffourcc", "DX50"],
+		vcodec="xvid",
+		vopts="fixed_quant=2:vhq=4:autoaspect",
+		acodec="mp3lame",
+		aopts="cbr:br=128",
+	),
+
+	"apple-quicktime" :
+	Profile(
+		commands=[MPlayer, X264, Faac, Wait, MP4Box],
+		mplayeropts=[],
+		x264opts=["--crf", "20", "--bframes", "1"],
+		faacopts=["-q", "100", "--mpeg-vers", "4"],
+	),
+
+	"nokia-n97" :
+	Profile(
+		commands=[Mencoder, MencoderDemux, MP4Box],
+		mplayeropts=["-vf-add", "scale=640:-10"],
+		vcodec="xvid",
+		vopts="bitrate=384:vhq=4:autoaspect:max_bframes=0",
+		acodec="faac",
+		aopts="br=64:mpeg=4:object=2",
+	),
+}
+
+mappings = {
+	"x264": "x264/lame",
+	"xvid": "xvid/lame",
+}
+for x, y in mappings.iteritems():
+	profiles[x] = profiles[y]
+
+
+
+
+def parse_args():
+	for profile_name in profiles.keys():
+		if sys.argv[0].find(profile_name) >= 0:
+			break
+	else:
+		profile_name = "xvid/lame"
+
+	parser = optparse.OptionParser(usage="%prog [options] input [output]")
+	parser.add_option("--dvd", action="store", dest="dvd")
+	parser.add_option("--fixmux", action="store_true", dest="fixmux")
+	parser.add_option("--deinterlace", action="store_true", dest="deinterlace")
+	parser.add_option("--detelecine", action="store_true", dest="detelecine")
+	parser.add_option("--mc", action="store", dest="mc", type="float")
+	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("--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("--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, 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)
+
+	if "://" not in input:
+		opts.input = os.path.abspath(input)
+	else:
+		if opts.dvd:
+			opts.dvd = os.path.abspath(opts.dvd)
+		opts.input = input
+
+	opts.output = os.path.abspath(output)
+	opts.mplayer_done = False
+
+	return opts
+
+def main():
+	os.nice(1)
+
+	opts = parse_args()
+
+	# Find our profile
+	try:
+		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)
+
+		try:
+			commands = []
+			if opts.fixmux or opts.detelecine:
+				profile.commands.insert(0, MencoderLossless)
+			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()
+
+		except FatalException, e:
+			print >>sys.stderr, "Error:", str(e)
+			sys.exit(1)
+
+	finally:
+		os.chdir("/")
+		shutil.rmtree(tempdir)
+
+if __name__ == "__main__":
+	main()
+