X-Git-Url: https://code.delx.au/transcoding/blobdiff_plain/0573196acbaf26157aa22042000942bd9b54a7cd..13b524fc4eb8ab1243a3fd6ba2b1625b4a2e37ed:/encode.py

diff --git a/encode.py b/encode.py
index dbe8c7e..2cc7d00 100755
--- a/encode.py
+++ b/encode.py
@@ -1,13 +1,13 @@
 #!/usr/bin/env python
 
-import commands, optparse, subprocess, sys, os
+import commands, optparse, subprocess, sys, os, tempfile, shutil
 
 class MencoderCommand(object):
 	codec2opts = {
 		"lavc": "-lavcopts",
 		"xvid": "-xvidencopts",
 		"x264": "-x264encopts",
-		"faac": "-faaccopts",
+		"faac": "-faacopts",
 		"mp3lame": "-lameopts",
 	}
 
@@ -58,14 +58,20 @@ class MencoderCommand(object):
 		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)
 		return cmd
 
 class Profile(object):
 	def __init__(self, CommandClass, **kwargs):
+		self.extra = []
+
 		self.CommandClass = CommandClass
 		self.__dict__.update(kwargs)
+	def __contains__(self, keyname):
+		return hasattr(self, keyname)
 
 profiles = {
 	"qt7" :
@@ -94,6 +100,15 @@ profiles = {
 		acodec="faac",
 		aopts="br=%(abitrate)d:mpeg=4:object=2",
 	),
+	"ipod264" :
+	Profile(
+		CommandClass=MencoderCommand,
+		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",
+		acodec="faac",
+		aopts="br=%(abitrate)d:mpeg=4:object=2:raw",
+		extra=['-of', 'lavf', '-lavfopts', 'format=mp4', '-channels', '2', '-srate', '48000']
+	),
 }
 
 
@@ -119,6 +134,7 @@ def parse_args():
 	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:])
@@ -126,8 +142,14 @@ def parse_args():
 		parser.print_usage()
 		sys.exit(1)
 	
-	opts.input = input
-	opts.output = output
+	if '://' not in input:
+		opts.input = os.path.join(os.getcwd(), input)
+	else:
+		if opts.dvd:
+			opts.dvd = os.path.join(os.getcwd(), opts.dvd)
+		opts.input = input
+
+	opts.output = os.path.join(os.getcwd(), output)
 	return opts
 
 def run(args, dump):
@@ -144,9 +166,15 @@ def main():
 		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)
+	tempdir = tempfile.mkdtemp()
+	try:
+		os.chdir(tempdir)
+		cmd = profile.CommandClass(profile, opts)
+		if run(cmd.pass1(), opts.dump) == 0 or opts.dump:
+			run(cmd.pass2(), opts.dump)
+	finally:
+		os.chdir('/')
+		shutil.rmtree(tempdir)
 
 if __name__ == "__main__":
 	main()