#!/bin/sh
#
#  bitrate is 2500 by default for SVCD      (lower is worse)
#  quality factor is 8 by default for SVCD  (higher is worse)
#  The peak bit-rate and average bit-rate should differ by 20-25%
#  Suggested -b2300 -q10 / -b2400 -q9 / -b2500 -q8
#
#  .mov files from mobo are top interlaced first
#  .dv files are bottom interlaced first
#
def_interlace=1
def_ratio=1
def_sound=1
def_signal=1
def_tools=0
def_quality=2

help(){
    cat <<END
Usage: $0 [options] <filename.[avi|dv|mov|mpg|qt]>
Options:
    -i n      n=0 progressive, n=1 top, n=2 bottom  ($def_interlace)
    -q n      quality, n=1 highest ... n=4 lowest   ($def_quality)
    -r n      aspect ratio, n=1 4:3, n=2 letterbox  ($def_ratio)
    -s n      audio n=0 mp2enc, n=1 toolame         ($def_sound)
    -t n      video, n=0 mjpegtools, n=1 ffmpeg     ($def_tools)
    -v n      norm, n=1 ntsc n=2 pal                ($def_signal)
    -h        Print this help message
END
    exit 0
}

interlace=$def_interlace
ratio=$def_ratio
sound=$def_sound
signal=$def_signal
mjpegtools=$def_mjpegtools
quality=$def_quality

while getopts i:q:r:s:t:v:h name "$@"
do
    case $name in
i)
    interlace=$OPTARG ;;
q)
    quality=$OPTARG ;;
r)
    ratio=$OPTARG ;;
s)
    sound=$OPTARG ;;
t)
    tools=$OPTARG ;;
v)
    signal=$OPTARG ;;
*)
    help ;;
    esac
done
let shiftind=$OPTIND-1
shift $shiftind
if test "$#" != "1"
then
    help
fi
case $1 in
*.qt)
    input="$1"
    base=`echo $input | sed "s/\.qt$//"` ;;
*.mov)
    input="$1"
    base=`echo $input | sed "s/\.mov$//"` ;;
*.avi)
    input="$1"
    base=`echo $input | sed "s/\.avi$//"` ;;
*.dv)
    input="$1"
    base=`echo $input | sed "s/\.dv$//"` ;;
*.mpg)
    input="$1"
    base=`echo $input | sed "s/\.mpg$//"`-svcd ;;
*)
    help ;;
esac

case $quality in
1)
    abitrate=224
    bitrate=2500
    quant_m=8
    quant_f=4 ;;
3)
    abitrate=192
    bitrate=2200
    quant_m=10
    quant_f=5 ;;
4)
    abitrate=96
    bitrate=2000
    quant_m=12
    quant_f=6 ;;
*)
    abitrate=192
    bitrate=2400
    quant_m=9
    quant_f=4 ;;
esac
case $signal in
2)
    nflag_f="-r pal"
    nflag_m="-np" ;;
*)
    nflag_f="-r ntsc"
    nflag_m="-nn" ;;
esac
case $interlace in
0)
    iflag_f="-interlace 0"
    iflag_m=NOT_INTERLACED ;;
1)
    iflag_f="-interlace 1 -ildct -ilme -top 1"
    iflag_m=INTERLACED_TOP_FIRST ;;
*)
    iflag_f="-interlace 1 -ildct -ilme -top 0"
    iflag_m=INTERLACED_BOTTOM_FIRST ;;
esac
case $ratio in
2)
    case $interlace in
0)
        iflag_f="-interlace 0" ;;
*)
        iflag_f="-deinterlace" ;;
esac
    aflag_f="-s480x360 -padtop 60 -padbottom 60 -aspect 4:3"
    aflag_m="-a3" ;;
*)
    aflag_f="-aspect 4:3"
    aflag_m="-a2"  ;;
esac

echo "Encoding $input to $base.mpg..."
echo "    video bitrate $bitrate"
echo "    audio bitrate $abitrate"
sleep 5

case $tools in
1)
    ffmpeg -i $input -target svcd $aflag_f $iflag_f $nflag_f \
        -flags scan_offset \
        -qmin $quant_f -b $bitrate -ab $abitrate -y $base.mpg ;;
*)
#    ffmpeg -i $input -f wav -y $base.wav
    lav2wav $input > $base.wav
case $sound in
1)
        sox -V $base.wav -r44100 $base-44.wav
        toolame -b$abitrate -s44.1 $base-44.wav $base.m2a ;;
*)
        mp2enc < $base.wav -r44100 -b$abitrate -o $base.m2a ;;
esac
#    ffmpeg -i $input -f yuv4mpegpipe -pix_fmt yuv420p -y /dev/stdout |
#    yuvdenoise -g3,4,4 -t8,12,12 -M4,6,6 |
    lav2yuv $input |
    yuvcorrect -T $iflag_m |
    yuvscaler $nflag_m -O SVCD |
    yuvdenoise -g0,0,0 -t8,12,12 -M0,0,0 |
    mpeg2enc -M0 $nflag_m $aflag_m -f5 -G18 -b$bitrate \
        -V230 -q$quant_m -o $base.m2v 
    mplex -f4 $base.m2v $base.m2a -o $base.mpg ;;
esac