tests/regression-funcs.sh

   1 #!/bin/sh
   2 #
   3 # common regression functions for avconv
   4 #
   5 #
   6
   7 test="${1#regtest-}"
   8 test_ref=$2
   9 raw_src_dir=$3
  10 target_exec=$4
  11 target_path=$5
  12 threads=${6:-1}
  13 tool=$8
  14
  15 datadir="./tests/data"
  16 target_datadir="${target_path}/${datadir}"
  17
  18 this="$test.$test_ref"
  19 outfile="$datadir/$test_ref/"
  20
  21 # various files
  22 avconv="$target_exec ${target_path}/${tool}"
  23 tiny_psnr="tests/tiny_psnr"
  24 raw_src="${target_path}/$raw_src_dir/%02d.pgm"
  25 raw_dst="$datadir/$this.out.yuv"
  26 raw_ref="$datadir/$test_ref.ref.yuv"
  27 pcm_src="$target_datadir/asynth1.sw"
  28 pcm_dst="$datadir/$this.out.wav"
  29 pcm_ref="$datadir/$test_ref.ref.wav"
  30 crcfile="$datadir/$this.crc"
  31 target_crcfile="$target_datadir/$this.crc"
  32
  33 cleanfiles="$raw_dst $pcm_dst $crcfile"
  34 trap 'rm -f -- $cleanfiles' EXIT
  35
  36 mkdir -p "$datadir"
  37 mkdir -p "$outfile"
  38
  39 [ "${V-0}" -gt 0 ] && echov=echov || echov=:
  40
  41 echov(){
  42     echo "$@" >&3
  43 }
  44
  45 . $(dirname $0)/md5.sh
  46
  47 AVCONV_OPTS="-v 0 -y"
  48 COMMON_OPTS="-flags +bitexact -idct simple -sws_flags +accurate_rnd+bitexact"
  49 DEC_OPTS="$COMMON_OPTS -threads $threads"
  50 ENC_OPTS="$COMMON_OPTS -threads 1 -dct fastint"
  51
  52 run_avconv()
  53 {
  54     $echov $avconv $AVCONV_OPTS $*
  55     $avconv $AVCONV_OPTS $*
  56 }
  57
  58 do_avconv()
  59 {
  60     f="$1"
  61     shift
  62     set -- $* ${target_path}/$f
  63     run_avconv $*
  64     do_md5sum $f
  65     if [ $f = $raw_dst ] ; then
  66         $tiny_psnr $f $raw_ref
  67     elif [ $f = $pcm_dst ] ; then
  68         $tiny_psnr $f $pcm_ref 2
  69     else
  70         wc -c $f
  71     fi
  72 }
  73
  74 do_avconv_nomd5()
  75 {
  76     f="$1"
  77     shift
  78     set -- $* ${target_path}/$f
  79     run_avconv $*
  80     if [ $f = $raw_dst ] ; then
  81         $tiny_psnr $f $raw_ref
  82     elif [ $f = $pcm_dst ] ; then
  83         $tiny_psnr $f $pcm_ref 2
  84     else
  85         wc -c $f
  86     fi
  87 }
  88
  89 do_avconv_crc()
  90 {
  91     f="$1"
  92     shift
  93     run_avconv $* -f crc "$target_crcfile"
  94     echo "$f $(cat $crcfile)"
  95 }
  96
  97 do_video_decoding()
  98 {
  99     do_avconv $raw_dst $DEC_OPTS $1 -i $target_path/$file -f rawvideo $ENC_OPTS -vsync 0 $2
 100 }
 101
 102 do_video_encoding()
 103 {
 104     file=${outfile}$1
 105     do_avconv $file $DEC_OPTS -f image2 -vcodec pgmyuv -i $raw_src $ENC_OPTS $2
 106 }
 107
 108 do_audio_encoding()
 109 {
 110     file=${outfile}$1
 111     do_avconv $file $DEC_OPTS -ac 2 -ar 44100 -f s16le -i $pcm_src -ab 128k $ENC_OPTS $2
 112 }
 113
 114 do_audio_decoding()
 115 {
 116     do_avconv $pcm_dst $DEC_OPTS -i $target_path/$file -sample_fmt s16 -f wav
 117 }