Add QP parallelization to rd_collect for daala.
[daala.git] / tools / rd_collect_daala.sh
1 #!/bin/bash
2 set -e
3
4 if [ -z $RD_COLLECT_SUB ]; then
5   echo "Please use: $(dirname $0)/rd_collect.sh daala *.y4m"
6   exit 1
7 fi
8
9 FILE=$1
10 QP=$2
11
12 BASENAME=$(basename $FILE)-$CODEC-$QP
13 rm $BASENAME.out 2> /dev/null || true
14 echo $BASENAME
15
16 WIDTH=$(head -1 $FILE | cut -d\  -f 2 | tr -d 'W')
17 HEIGHT=$(head -1 $FILE | cut -d\  -f 3 | tr -d 'H')
18
19 $ENCODER_EXAMPLE -k 256 -v $QP $EXTRA_OPTS $FILE -o $BASENAME.ogv 2> $BASENAME-enc.out
20 $DUMP_VIDEO $BASENAME.ogv -o $BASENAME.y4m 2> /dev/null
21 SIZE=$(wc -c $BASENAME.ogv | awk '{ print $1 }')
22 $DUMP_PSNR $FILE $BASENAME.y4m > $BASENAME-psnr.out 2> /dev/null
23 FRAMES=$(cat $BASENAME-psnr.out | grep ^0 | wc -l)
24 PIXELS=$(($WIDTH*$HEIGHT*$FRAMES))
25 PSNR=$(cat $BASENAME-psnr.out | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
26 PSNRHVS=$($DUMP_PSNRHVS $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
27 SSIM=$($DUMP_SSIM $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
28 FASTSSIM=$($DUMP_FASTSSIM -c $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
29 CIEDE=$($DUMP_CIEDE $FILE $BASENAME.y4m 2> /dev/null | grep Total | cut -d' ' -f2-)
30 rm $BASENAME.y4m $BASENAME.ogv $BASENAME-enc.out $BASENAME-psnr.out
31 echo $QP $PIXELS $SIZE $PSNR $PSNRHVS $SSIM $FASTSSIM $CIEDE >> $BASENAME.out
32 #tail -1 $BASENAME.out