#!/usr/bin/python
#
# @ rbl/2006
#
#     Requires: mencoder, gpac
#

import sys, os, getopt, md5

def usage():
    sys.stderr.write('Usage: movie-to-3-mp4 [-s <subtitle>]' \
        '\n\t\t     [--sub-font=<fontconfig name>]' \
	'\n\t\t     [--sub-size=<font size>]' \
	'\n\t\t     [--video-bitrate=<video bitrate (kbps)>]' \
	'\n\t\t     [--video-fps=<frames per second>]' \
	'\n\t\t     [--video-passes=<number of passes (1+)>]' \
	'\n\t\t     [--audio-quality=<audio quality (10-500)>]' \
	'\n\t\t     [--audio-frequency=<audio frequency (Hz)>]' \
	'\n\t\t     [--audio-volume=<audio volume (dB)>] <source file>\n')
    sys.exit(1)

try:
    opts, args = getopt.gnu_getopt(sys.argv[1:], 's:', ['sub-font=',
        'sub-size=', 'video-bitrate=', 'video-fps=', 'video-passes=',
	'audio-quality=', 'audio-frequency=', 'audio-volume='])
except getopt.GetoptError:
    usage()

if len(args) != 1:
    usage()

subt = None
subfont = 'Bitstream Vera Sans'
subscale = '14'
videobr = '384'
videofps = '25'
videops = '2'
audioq = '64'
audiofreq = '44100'
audiovol = '0'

try:
    for o, a in opts:
        if o == '-s':
            subt = a
        elif o == '--sub-font':
            subfont = a
	elif o == '--sub-size':
	    subscale = str(int(a))
	elif o == '--video-bitrate':
	    videobr = str(int(a))
	elif o == '--video-fps':
	    videofps = str(int(a))
	elif o == '--video-passes':
	    videops = str(int(a))
	elif o == '--audio-quality':
	    audioq = str(int(a))
	elif o == '--audio-frequency':
	    audiofreq = str(int(a))
	elif o == '--audio-volume':
	    audiovol = str(int(a))
        else:
            usage()
except ValueError:
    usage()

src = args[0]
srcb, srce = os.path.splitext(src)
if srce.lower() == '.mp4':
    srcb += '.2'
dst = '%s.%s' % (srcb, 'mp4')
tmp = os.path.join(os.path.split(src)[0], md5.new(src).hexdigest())

saud = '-oac pcm'
svid = ''
outp = '-o /dev/null -of rawvideo'
for i in range(1, int(videops) + 1):
    if i == int(videops):
	saud = '-oac faac -faacopts mpeg=4:object=2:raw:quality=%s ' \
	    '-channels 2 -srate %s -af volume=%sdb' % (audioq, audiofreq,
	    audiovol)
	svid = ':mbd=2'
	outp = '-o \"%s.mp4\" -of lavf -lavfopts format=mp4:' \
	    'i_certify_that_my_video_stream_does_not_use_b_frames' % tmp
    if subt:
	os.system('mencoder -subfont-autoscale 0 ' \
	    '-subfont-text-scale %s -fontconfig -font \"%s\" -sub \"%s\" ' \
	    '%s -ovc lavc -lavcopts ' \
	    'vcodec=mpeg4:vglobal=1:vbitrate=%s:vpass=%s%s ' \
	    '-vf scale=352:-2,expand=352:288:::1,harddup -ofps %s ' \
	    '%s \"%s\"' % (subscale, subfont, subt, saud, videobr, str(i),
	    svid, videofps, outp, src))
    else:
	os.system('mencoder %s -ovc lavc -lavcopts ' \
	    'vcodec=mpeg4:vglobal=1:vbitrate=%s:vpass=%s%s ' \
	    '-vf scale=352:-2,expand=352:288:::1,harddup -ofps %s ' \
	    '%s \"%s\"' % (saud, videobr, str(i), svid, videofps, outp, src))

try:
    os.remove('divx2pass.log')
except OSError:
    pass
try:
    os.remove(dst)
except OSError:
    pass

retv = os.system('MP4Box -add \"%s.mp4\":fps=%s \"%s\"' % (tmp, videofps, dst))

os.remove('%s.mp4' % tmp)

sys.exit(retv >> 8)