diff options
Diffstat (limited to 'TOOLS/psnr-video.sh')
| -rwxr-xr-x | TOOLS/psnr-video.sh | 150 |
1 files changed, 150 insertions, 0 deletions
diff --git a/TOOLS/psnr-video.sh b/TOOLS/psnr-video.sh new file mode 100755 index 0000000000..1fdc8c106e --- /dev/null +++ b/TOOLS/psnr-video.sh @@ -0,0 +1,150 @@ +#!/bin/sh +# Helper script to ease comparing two video files +# Copyleft 2001 by Matthias Wieser +# This file comes under GPL, see http://www.gnu.org/copyleft/gpl.html for more +# information on it's licensing. + +TEMPDIR="/tmp/psnr_video" +WORKDIR=`pwd`/ +if [ $# -le 1 ]; then + echo + echo "Usage: `basename $0` <file1> <file2> [<frames>] [<options1>] [<options2>]" + echo + echo " <file1> and <file2> are the video files for which the PSNR should be calculated." + echo " [<frames>] is the number of frames to process, starting from frame 1." + echo " [<options1>] are additional MPlayer options for <file1>" + echo " [<options2>] are additional MPlayer options for <file2>" + echo + echo " Be aware that `basename $0` needs a lot of temporal space inside /temp/." + echo + echo "Example:" + echo " ./`basename $0` ./orig.avi ./test.avi 250 \"\" \"-vf pp=ac\"" + echo + + exit 1 +fi + +FILE1=$1 +FILE2=$2 + +LastFrame=-1 +if [ $# -ge 3 ]; then + LastFrame=$3 + echo + echo "Will process $LastFrame frames" +fi + +if [ $# -ge 4 ]; then + FILE1Opts=$4 + echo "Mplayer options for ${FILE1}: $FILE1Opts" +fi + +if [ $# -ge 5 ]; then + FILE2Opts=$5 + echo "Mplayer options for ${FILE2}: $FILE2Opts" +fi + +mkdir -p ${TEMPDIR}/FILE1 +mkdir -p ${TEMPDIR}/FILE2 + +### File 1 +echo +echo "############## $FILE1 #################" + +cd ${TEMPDIR}/FILE1 + +rm *ppm 2> /dev/null +rm *del 2> /dev/null + +if [ $LastFrame -ge 0 ]; then + mplayer $FILE1Opts -frames $LastFrame -nosound -vo pnm ${WORKDIR}$FILE1 >/dev/null +else + mplayer $FILE1Opts -nosound -vo pnm ${WORKDIR}$FILE1 >/dev/null +fi +### File 2 + +echo +echo "############## $FILE2 #################" + +cd ${TEMPDIR}/FILE2 + +rm *ppm 2> /dev/null + +if [ $LastFrame -ge 0 ]; then + mplayer $FILE2Opts -frames $LastFrame -nosound -vo pnm ${WORKDIR}$FILE2 >/dev/null +else + mplayer $FILE2Opts -nosound -vo pnm ${WORKDIR}$FILE2 >/dev/null +fi + + +### PSNR + +echo +echo "############## Calculation PSNR #################" + +if [[ `ls -1 ${TEMPDIR}/FILE1/*ppm|wc -l` = `ls -1 ${TEMPDIR}/FILE2/*ppm|wc -l` ]] +then + echo +else + echo "Files have differing numbers of frames!" + echo "$FILE1 has `ls -1 ${TEMPDIR}/FILE1/*ppm|wc -l` frames," + echo "$FILE2 has `ls -1 ${TEMPDIR}/FILE2/*ppm|wc -l` frames." + echo "Processing the first `ls -1 ${TEMPDIR}/FILE2/*ppm|wc -l` frames." + echo +fi + + +cd ${TEMPDIR}/FILE2 +#rm ../psnr.dat +echo "File;Y;Cb;Cr" >../psnr.dat +echo "0" > errorsum.del +i=0 +for FILE in `ls -1 *.ppm` + do + echo $FILE + echo -n "$FILE">>../psnr.dat + echo -n ";">>../psnr.dat + pnmpsnr ../FILE1/$FILE $FILE 2> del.del + grep "Y" del.del | dd bs=1c count=5 skip=29 of=del2.del 2>/dev/null + Y=`cat del2.del` + echo -n "$Y;">>../psnr.dat + grep "Cb" del.del | dd bs=1c count=5 skip=29 of=del2.del 2>/dev/null + CB=`cat del2.del` + echo -n "$CB;">>../psnr.dat + grep "Cr" del.del | dd bs=1c count=5 skip=29 of=del2.del 2>/dev/null + CR=`cat del2.del` + echo -n "$CR;">>../psnr.dat + ALL=`echo "(-10)*l((e(-$Y/10*l(10))+e(-$CB/10*l(10))/4+e(-$CR/10*l(10))/4)/1.5)/l(10)"|bc -l` + echo "$ALL">>../psnr.dat + ERROR=`echo "scale=30; (e(-1*$Y/10*l(10))+e(-1*$CB/10*l(10))/4+e(-1*$CR/10*l(10))/4)/1.5"|bc -l` + ERRORSUM=`cat errorsum.del` + echo `echo "scale=30; $ERROR + $ERRORSUM"|bc -l` > errorsum.del + i=$(($i+1)) + if [[ $i = $LastFrame ]] + then + break + fi +done +ERRORSUM=`cat errorsum.del` +PSNR=`echo "-10*l($ERRORSUM/$i)/l(10)"|bc -l` +echo "PSNR:;$PSNR">>../psnr.dat +cd .. +mv psnr.dat ${WORKDIR} + +if [[ `ls -1 ${TEMPDIR}/FILE1/*ppm|wc -l` = `ls -1 ${TEMPDIR}/FILE2/*ppm|wc -l` ]] +then + echo +else + echo "Files have differing numbers of frames!" + echo "$FILE1 has `ls -1 ${TEMPDIR}/FILE1/*ppm|wc -l` frames," + echo "$FILE2 has `ls -1 ${TEMPDIR}/FILE2/*ppm|wc -l` frames." + echo "Processed the first `ls -1 ${TEMPDIR}/FILE2/*ppm|wc -l` frames." + echo +fi + +cd .. +rm -r ${TEMPDIR} + +echo "Created ${WORKDIR}psnr.dat" +echo + |