#!/bin/sh
FILE=$1
FRAMES=$2

usage="USAGE: compare.sh <FILENAME> <FRAMECOUNT>"

vqmt=/tmp/VQMT/build/bin/Release/vqmt 
height=2160
width=4096
chroma=2 # 0 = YUV400, 1 = YUV420, 2 = YUV422, 3 = YUV444

metrics="PSNR SSIM MSSSIM VIFP PSNRHVS PSNRHVSM"
bitrates="30000k 60000k 100000k 150000k 200000k 300000k 400000k 500000k 600000k"

if [ -z $FILE ]; then
	echo $usage
	exit
fi

if [ -z $FRAMES ]; then
	echo $usage
	exit
fi

if [ ! -f $FILE.yuv ]; then
	echo "Generating master YUV file from $FILE"
	ffmpeg -i $FILE $FILE.yuv
fi

echo "Encoding different targets"
for bitrate in $bitrates; do
	echo "Encoding with $bitrate"
	ffmpeg -i $FILE -c:v libx264 -b:v $bitrate $FILE.$bitrate.mov
	ffmpeg -i $FILE.$bitrate.mov $FILE.$bitrate.yuv
	$vqmt $FILE.yuv $FILE.$bitrate.yuv $height $width $FRAMES $chroma results $metrics
	rm $FILE.$bitrate.yuv
done