git.sesse.net Git - ffmpeg/blob - tests/regression-funcs.sh

   1 #!/bin/sh
   2 #
   3 # common regression functions for ffmpeg
   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 ffmpeg="$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 pcm_src_1ch="$target_datadir/asynth-16000-1.wav"
  32 pcm_ref_1ch="$datadir/$test_ref-16000-1.ref.wav"
  33 crcfile="$datadir/$this.crc"
  34 target_crcfile="$target_datadir/$this.crc"
  35
  36 cleanfiles="$raw_dst $pcm_dst $crcfile"
  37 trap 'rm -f -- $cleanfiles' EXIT
  38
  39 mkdir -p "$datadir"
  40 mkdir -p "$outfile"
  41
  42 [ "${V-0}" -gt 0 ] && echov=echov || echov=:
  43
  44 echov(){
  45     echo "$@" >&3
  46 }
  47
  48 . $(dirname $0)/md5.sh
  49
  50 AVCONV_OPTS="-nostats -y -cpuflags $cpuflags"
  51 COMMON_OPTS="-flags +bitexact -idct simple -sws_flags +accurate_rnd+bitexact"
  52 DEC_OPTS="$COMMON_OPTS -threads $threads"
  53 ENC_OPTS="$COMMON_OPTS -threads 1 -dct fastint"
  54
  55 run_avconv()
  56 {
  57     $echov $ffmpeg $AVCONV_OPTS $*
  58     $ffmpeg $AVCONV_OPTS $*
  59 }
  60
  61 do_avconv()
  62 {
  63     f="$1"
  64     shift
  65     set -- $* ${target_path}/$f
  66     run_avconv $*
  67     do_md5sum $f
  68     if [ $f = $raw_dst ] ; then
  69         $tiny_psnr $f $raw_ref
  70     elif [ $f = $pcm_dst ] ; then
  71         $tiny_psnr $f $pcm_ref 2
  72     else
  73         echo $(wc -c $f)
  74     fi
  75 }
  76
  77 do_avconv_nomd5()
  78 {
  79     f="$1"
  80     shift
  81     set -- $* ${target_path}/$f
  82     run_avconv $*
  83     if [ $f = $raw_dst ] ; then
  84         $tiny_psnr $f $raw_ref
  85     elif [ $f = $pcm_dst ] ; then
  86         $tiny_psnr $f $pcm_ref 2
  87     else
  88         echo $(wc -c $f)
  89     fi
  90 }
  91
  92 do_avconv_crc()
  93 {
  94     f="$1"
  95     shift
  96     run_avconv $* -f crc "$target_crcfile"
  97     echo "$f $(cat $crcfile)"
  98 }
  99
 100 do_video_decoding()
 101 {
 102     do_avconv $raw_dst $DEC_OPTS $1 -i $target_path/$file -f rawvideo $ENC_OPTS -vsync 0 $2
 103 }
 104
 105 do_video_encoding()
 106 {
 107     file=${outfile}$1
 108     do_avconv $file $DEC_OPTS -f image2 -vcodec pgmyuv -i $raw_src $ENC_OPTS $2
 109 }
 110
 111 do_video_encoding_nomd5()
 112 {
 113     file=${outfile}$1
 114     do_avconv_nomd5 $file $DEC_OPTS -f image2 -vcodec pgmyuv -i $raw_src $ENC_OPTS $2
 115 }
 116
 117 do_audio_encoding()
 118 {
 119     file=${outfile}$1
 120     do_avconv $file $DEC_OPTS -ac 2 -ar 44100 -f s16le -i $pcm_src -ab 128k $ENC_OPTS $2
 121 }
 122
 123 do_audio_decoding()
 124 {
 125     do_avconv $pcm_dst $DEC_OPTS -i $target_path/$file -sample_fmt s16 -f wav $1
 126 }
 127
 128 do_audio_decoding_nomd5()
 129 {
 130     do_avconv_nomd5 $pcm_dst $DEC_OPTS -i $target_path/$file -sample_fmt s16 -f wav $1
 131 }