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