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