Add QP parallelization to rd_collect for daala.
authorJustin Nickelsburg <nickelsburg99@gmail.com>
Thu, 30 May 2019 17:24:45 +0000 (13:24 -0400)
committerNathan E. Egge <negge@dgql.org>
Thu, 30 May 2019 17:47:39 +0000 (13:47 -0400)
Signed-off-by: Nathan E. Egge <negge@dgql.org>
tools/rd_collect.sh
tools/rd_collect_daala.sh

index 7ba24b1..742bd21 100755 (executable)
@@ -59,6 +59,10 @@ case $CODEC in
       exit 1
     fi
 
       exit 1
     fi
 
+    if [ -z "$QPS" ]; then
+      QPS="5 7 11 16 25 37 55 81 122 181 270 400"
+    fi
+
     export RD_COLLECT_SUB=$(dirname "$0")/rd_collect_daala.sh
     ;;
   av1 | av1-rt)
     export RD_COLLECT_SUB=$(dirname "$0")/rd_collect_daala.sh
     ;;
   av1 | av1-rt)
@@ -443,7 +447,7 @@ if [ -z "$CORES" ]; then
 fi
 
 case $CODEC in
 fi
 
 case $CODEC in
-  rav1e)
+  daala | rav1e)
     FILES=$(find -L "$@" -type f -name "*.y4m")
     for f in $FILES; do for q in $QPS; do printf "%s\0" $f $q; done; done | xargs -0 -n2 -P$CORES $RD_COLLECT_SUB
     for f in $FILES; do cat $(basename $f)-*.out | sort -n > $(basename $f)-$CODEC.out && rm $(basename $f)-$CODEC-*.out; done
     FILES=$(find -L "$@" -type f -name "*.y4m")
     for f in $FILES; do for q in $QPS; do printf "%s\0" $f $q; done; done | xargs -0 -n2 -P$CORES $RD_COLLECT_SUB
     for f in $FILES; do cat $(basename $f)-*.out | sort -n > $(basename $f)-$CODEC.out && rm $(basename $f)-$CODEC-*.out; done
index 2f52007..90376ed 100755 (executable)
@@ -7,30 +7,26 @@ if [ -z $RD_COLLECT_SUB ]; then
 fi
 
 FILE=$1
 fi
 
 FILE=$1
+QP=$2
 
 
-BASENAME=$(basename $FILE)-$CODEC
+BASENAME=$(basename $FILE)-$CODEC-$QP
 rm $BASENAME.out 2> /dev/null || true
 echo $BASENAME
 
 WIDTH=$(head -1 $FILE | cut -d\  -f 2 | tr -d 'W')
 HEIGHT=$(head -1 $FILE | cut -d\  -f 3 | tr -d 'H')
 
 rm $BASENAME.out 2> /dev/null || true
 echo $BASENAME
 
 WIDTH=$(head -1 $FILE | cut -d\  -f 2 | tr -d 'W')
 HEIGHT=$(head -1 $FILE | cut -d\  -f 3 | tr -d 'H')
 
-RANGE="5 7 11 16 25 37 55 81 122 181 270 400"
-#RANGE="1 2 3 4 5 6 7 9 11 13 16 20 25 30 37 45 55 67 81 99 122 148 181 221 270 330 400 500"
-
-for x in $RANGE; do
-  $ENCODER_EXAMPLE -k 256 -v $x $EXTRA_OPTS $FILE -o $BASENAME.ogv 2> $BASENAME-$x-enc.out
-  $DUMP_VIDEO $BASENAME.ogv -o $BASENAME.y4m 2> /dev/null
-  SIZE=$(wc -c $BASENAME.ogv | awk '{ print $1 }')
-  $DUMP_PSNR $FILE $BASENAME.y4m > $BASENAME-psnr.out 2> /dev/null
-  FRAMES=$(cat $BASENAME-psnr.out | grep ^0 | wc -l)
-  PIXELS=$(($WIDTH*$HEIGHT*$FRAMES))
-  PSNR=$(cat $BASENAME-psnr.out | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
-  PSNRHVS=$($DUMP_PSNRHVS $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
-  SSIM=$($DUMP_SSIM $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
-  FASTSSIM=$($DUMP_FASTSSIM -c $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
-  CIEDE=$($DUMP_CIEDE $FILE $BASENAME.y4m 2> /dev/null | grep Total | cut -d' ' -f2-)
-  rm $BASENAME.y4m $BASENAME.ogv $BASENAME-$x-enc.out $BASENAME-psnr.out
-  echo $x $PIXELS $SIZE $PSNR $PSNRHVS $SSIM $FASTSSIM $CIEDE >> $BASENAME.out
-  #tail -1 $BASENAME.out
-done
+$ENCODER_EXAMPLE -k 256 -v $QP $EXTRA_OPTS $FILE -o $BASENAME.ogv 2> $BASENAME-enc.out
+$DUMP_VIDEO $BASENAME.ogv -o $BASENAME.y4m 2> /dev/null
+SIZE=$(wc -c $BASENAME.ogv | awk '{ print $1 }')
+$DUMP_PSNR $FILE $BASENAME.y4m > $BASENAME-psnr.out 2> /dev/null
+FRAMES=$(cat $BASENAME-psnr.out | grep ^0 | wc -l)
+PIXELS=$(($WIDTH*$HEIGHT*$FRAMES))
+PSNR=$(cat $BASENAME-psnr.out | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
+PSNRHVS=$($DUMP_PSNRHVS $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
+SSIM=$($DUMP_SSIM $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
+FASTSSIM=$($DUMP_FASTSSIM -c $FILE $BASENAME.y4m 2> /dev/null | grep Total | tr -s ' ' | cut -d\  -f $((4+$PLANE*2)))
+CIEDE=$($DUMP_CIEDE $FILE $BASENAME.y4m 2> /dev/null | grep Total | cut -d' ' -f2-)
+rm $BASENAME.y4m $BASENAME.ogv $BASENAME-enc.out $BASENAME-psnr.out
+echo $QP $PIXELS $SIZE $PSNR $PSNRHVS $SSIM $FASTSSIM $CIEDE >> $BASENAME.out
+#tail -1 $BASENAME.out