#! /usr/bin/env python

'''
avi2mpg [OPTION]... [FILE]...

Options
    --version
        display current version
    -noauthor
        skip running dvdauthor
    -nomplex
    	skip running mplex
'''

import os
import sys
import getopt

movies = []
options = []
noauthor = 0
nomplex = 0

for file in sys.argv[1:]:
	if file[0] == '-':
		options.append(file)
		continue
	else:
		movies.append(file[:-4])
	
for option in options:	
	if option == '--version': 
		print 'version: 1.0 -- 1/12/07'
		sys.exit()
	elif option == '-noauthor':
		noauthor = 1
	elif option == '-nomplex':
		nomplex = 1
		noauthor = 1
	else:
		print '\n\n\nunrecognized option:', option, 'exiting'
		sys.exit()


def aviConv(movie):
	os.system('transcode -i "' + movie + '.avi" -y ffmpeg --export_prof dvd-ntsc --export_asr 3 -o "' + movie + '" -D0 -s2 -m "' + movie + '.ac3" -J modfps=clonetype=3 --export_fps 29.97')
	print '\n\n'

def mplex(movie):
	os.system('mplex -f 8 -o "' + movie +'.mpg" "' + movie + '.m2v" "' + movie + '.ac3"')
	os.system('rm -f "' + movie + '.m2v"')
	os.system('rm -f "' + movie + '.ac3"')
	print '\n\n'

def dvdAuthor(movies):
	xmlFile = open('author.xml', 'w')
	xmlFile.write('<dvdauthor>\n')
	xmlFile.write('  <vmgm />\n')
	xmlFile.write('  <titleset>\n')
	xmlFile.write('    <titles>\n')
	xmlFile.write('      <pgc pause="inf">\n')
	#write for loop here to construct each .mpg file
	for movie in movies:
		xmlFile.write('        <vob file="' + movie + '.mpg" />\n')  
	xmlFile.write('      </pgc>\n')
	xmlFile.write('    </titles>\n')
	xmlFile.write('  </titleset>\n')
	xmlFile.write('</dvdauthor>\n')
	xmlFile.close()
	os.system('dvdauthor -o ' + os.getcwd() + '/ -x author.xml ')
	for movie in movies:
		os.system('rm "' + movie + '.mpg"')
	os.system('rm -f author.xml')
	print '\n\n'

for movie in movies:
	aviConv(movie)
if not nomplex:
	for movie in movies:
		mplex(movie)
if not noauthor:
	dvdAuthor(movies)

#!/bin/bash
function echolightgray {
	echo -e "\033[37;0m $1 \033[0m"
	}
function echodarkgray {
	echo -e "\033[30;1m $1 \033[0m"
	}
function echolightblue {
	echo -e "\033[34;1m $1 \033[0m"
	}
function echolightgreen {
	echo -e "\033[32;1m $1 \033[0m"
	}
function echolightcyan {
	echo -e "\033[36;1m $1 \033[0m"
	}
function echolightred {
	echo -e "\033[31;1m $1 \033[0m"
	}
function echolightpurple {
	echo -e "\033[35;1m $1 \033[0m"
	}
function echoyellow {
	echo -e "\033[33;1m $1 \033[0m"
	}
function echoyellownobk {
	echo -en "\033[33;1m $1 \033[0m"
	}
function echowhite {
	echo -e "\033[37;1m $1 \033[0m"
	}

function getprojectname
{
	echoyellownobk "Name of the DVD project?"
	read project
	if [[ "$project" == "" ]]; then exit 0; fi
}

function getaviname
{
	let cnt=0
	echoyellow "Which of the following movies do you want to convert? Type blank line to finish"
	echo ""
	ls | grep .avi
	ls | grep .mpg
	ls | grep .wmv
	echo ""
	while ( true)
	do
		read avimovie[$cnt]
		if [[ ${avimovie[$cnt]} == "" ]]; then break; fi
		let cnt=$cnt+1
	done
}

function makedir
{
	mkdir "DVD-$project" 2>/dev/null
}

function convertvid
{
	cd "$workingdir/DVD-$project"
	for (( i=0;i<cnt;i++ ))
	do
		ffmpeg -i ${avimovie[$i]} -y -target pal-dvd video$i.mpg
		echolightgreen "Restoring original videofile: ${avimovie[$i]}"
	mv ${avimovie[$i]} ../
	done
}

function getaspect
{
	echoyellownobk "Enter a custom aspect ratio. Enter for auto"
	read aspect
}

function author
{
	cd "$workingdir/DVD-$project"
	rm dvdauthor.xml 2>/dev/null
	echo -e "<dvdauthor dest=\"DVD\">  
		  <vmgm />
		   <titleset> 
			 <titles>" > dvdauthor.xml 
	if [[ "$aspect" != "" ]]
	then echo -e "               <video aspect=\"$aspect\" />" >> dvdauthor.xml
	fi
			  echo "                 <pgc>" >> dvdauthor.xml
    for ((i=0;i<cnt;i++))
    do
	echo -e "                 <vob file=\"video$i.mpg\"/>" >> dvdauthor.xml
	done
			   
	echo -e	"                </pgc> 
			  </titles> 
		   </titleset> 
		 </dvdauthor>" >> dvdauthor.xml
	echoyellow "Starting dvdauthor..."
	sleep 2
	dvdauthor -x dvdauthor.xml 2>/dev/stdout | grep -v "unknown"
}

function createiso
{
	cd "$workingdir/DVD-$project/DVD"
	echo ""
	echoyellow "Creating temporary image"
	mkisofs -dvd-video -udf ./ > /tmp/tempdvd.iso 2>mkiso.log
}

function burn
{
	cd "$workingdir/DVD-$project/DVD"
	echo ""
	echoyellow "Burning with growisofs"
	sleep 4
	growisofs -Z /dev/hdd=/tmp/tempdvd.iso -use-the-force-luke=notray -use-the-force-luke=tty -dvd-compat -speed=8 -overburn -use-the-force-luke=bufsize:32m
	aplay -q /mnt/str1/sounds/notify.wav
}

reset
workingdir=`pwd`
getprojectname
getaviname
getaspect
makedir
for (( i=0;i<cnt;i++ ))
do
	if [[ ${avimovie[$i]} != "" ]]
		then 
			{
				mv ${avimovie[$i]} DVD-$project
			}
	fi
done
convertvid
author
createiso
a="y"
while [ "$a" == y ] 
do
	burn
	echolightgreen "Another Copy? y/n"
	read a
done
rm /tmp/tempdvd.iso
cd "$workingdir"
rm -r DVD-$project