enc_dec_pcm: add bitexact flag
[ffmpeg.git] / tests / fate-run.sh
1 #! /bin/sh
2
3 export LC_ALL=C
4
5 base=$(dirname $0)
6 . "${base}/md5.sh"
7
8 base64=tests/base64
9
10 test="${1#fate-}"
11 samples=$2
12 target_exec=$3
13 target_path=$4
14 command=$5
15 cmp=${6:-diff}
16 ref=${7:-"${base}/ref/fate/${test}"}
17 fuzz=${8:-1}
18 threads=${9:-1}
19 thread_type=${10:-frame+slice}
20 cpuflags=${11:-all}
21 cmp_shift=${12:-0}
22 cmp_target=${13:-0}
23 size_tolerance=${14:-0}
24 cmp_unit=${15:-2}
25
26 outdir="tests/data/fate"
27 outfile="${outdir}/${test}"
28 errfile="${outdir}/${test}.err"
29 cmpfile="${outdir}/${test}.diff"
30 repfile="${outdir}/${test}.rep"
31
32 target_path(){
33     test ${1} = ${1#/} && p=${target_path}/
34     echo ${p}${1}
35 }
36
37 # $1=value1, $2=value2, $3=threshold
38 # prints 0 if absolute difference between value1 and value2 is <= threshold
39 compare(){
40     echo "scale=2; v = $1 - $2; if (v < 0) v = -v; if (v > $3) r = 1; r" | bc
41 }
42
43 do_tiny_psnr(){
44     psnr=$(tests/tiny_psnr "$1" "$2" $cmp_unit $cmp_shift 0)
45     val=$(expr "$psnr" : ".*$3: *\([0-9.]*\)")
46     size1=$(expr "$psnr" : '.*bytes: *\([0-9]*\)')
47     size2=$(expr "$psnr" : '.*bytes:[ 0-9]*/ *\([0-9]*\)')
48     val_cmp=$(compare $val $cmp_target $fuzz)
49     size_cmp=$(compare $size1 $size2 $size_tolerance)
50     if [ "$val_cmp" != 0 ] || [ "$size_cmp" != 0 ]; then
51         echo "$psnr"
52         return 1
53     fi
54 }
55
56 oneoff(){
57     do_tiny_psnr "$1" "$2" MAXDIFF
58 }
59
60 stddev(){
61     do_tiny_psnr "$1" "$2" stddev
62 }
63
64 oneline(){
65     printf '%s\n' "$1" | diff -u -b - "$2"
66 }
67
68 run(){
69     test "${V:-0}" -gt 0 && echo "$target_exec" $target_path/"$@" >&3
70     $target_exec $target_path/"$@"
71 }
72
73 probefmt(){
74     run ffprobe -show_format_entry format_name -print_format default=nw=1:nk=1 -v 0 "$@"
75 }
76
77 ffmpeg(){
78     run ffmpeg -nostats -threads $threads -thread_type $thread_type -cpuflags $cpuflags "$@"
79 }
80
81 framecrc(){
82     ffmpeg "$@" -f framecrc -
83 }
84
85 framemd5(){
86     ffmpeg "$@" -f framemd5 -
87 }
88
89 crc(){
90     ffmpeg "$@" -f crc -
91 }
92
93 md5(){
94     ffmpeg "$@" md5:
95 }
96
97 pcm(){
98     ffmpeg "$@" -vn -f s16le -
99 }
100
101 enc_dec_pcm(){
102     out_fmt=$1
103     dec_fmt=$2
104     pcm_fmt=$3
105     src_file=$(target_path $4)
106     shift 4
107     encfile="${outdir}/${test}.${out_fmt}"
108     cleanfiles=$encfile
109     encfile=$(target_path ${encfile})
110     ffmpeg -i $src_file "$@" -f $out_fmt -y ${encfile} || return
111     ffmpeg -flags +bitexact -i ${encfile} -c:a pcm_${pcm_fmt} -f ${dec_fmt} -
112 }
113
114 FLAGS="-flags +bitexact -sws_flags +accurate_rnd+bitexact"
115 DEC_OPTS="-threads $threads -idct simple $FLAGS"
116 ENC_OPTS="-threads 1        -idct simple -dct fastint"
117
118 enc_dec(){
119     src_fmt=$1
120     srcfile=$2
121     enc_fmt=$3
122     enc_opt=$4
123     dec_fmt=$5
124     dec_opt=$6
125     encfile="${outdir}/${test}.${enc_fmt}"
126     decfile="${outdir}/${test}.out.${dec_fmt}"
127     cleanfiles="$cleanfiles $decfile"
128     test "$7" = -keep || cleanfiles="$cleanfiles $encfile"
129     tsrcfile=$(target_path $srcfile)
130     tencfile=$(target_path $encfile)
131     tdecfile=$(target_path $decfile)
132     ffmpeg -f $src_fmt $DEC_OPTS -i $tsrcfile $ENC_OPTS $enc_opt $FLAGS \
133         -f $enc_fmt -y $tencfile || return
134     do_md5sum $encfile
135     echo $(wc -c $encfile)
136     ffmpeg $8 $DEC_OPTS -i $tencfile $ENC_OPTS $dec_opt $FLAGS \
137         -f $dec_fmt -y $tdecfile || return
138     do_md5sum $decfile
139     tests/tiny_psnr $srcfile $decfile $cmp_unit $cmp_shift
140 }
141
142 regtest(){
143     t="${test#$2-}"
144     ref=${base}/ref/$2/$t
145     ${base}/${1}-regression.sh $t $2 $3 "$target_exec" "$target_path" "$threads" "$thread_type" "$cpuflags" "$samples"
146 }
147
148 lavffatetest(){
149     regtest lavf lavf-fate tests/vsynth1
150 }
151
152 lavftest(){
153     regtest lavf lavf tests/vsynth1
154 }
155
156 lavfitest(){
157     cleanfiles="tests/data/lavfi/${test#lavfi-}.nut"
158     regtest lavfi lavfi tests/vsynth1
159 }
160
161 seektest(){
162     t="${test#seek-}"
163     ref=${base}/ref/seek/$t
164     case $t in
165         image_*) file="tests/data/images/${t#image_}/%02d.${t#image_}" ;;
166         *)       file=$(echo $t | tr _ '?')
167                  for d in fate/acodec- fate/vsynth2- lavf/; do
168                      test -f tests/data/$d$file && break
169                  done
170                  file=$(echo tests/data/$d$file)
171                  ;;
172     esac
173     run libavformat/seek-test $target_path/$file
174 }
175
176 mkdir -p "$outdir"
177
178 exec 3>&2
179 eval $command >"$outfile" 2>$errfile
180 err=$?
181
182 if [ $err -gt 128 ]; then
183     sig=$(kill -l $err 2>/dev/null)
184     test "${sig}" = "${sig%[!A-Za-z]*}" || unset sig
185 fi
186
187 if test -e "$ref" || test $cmp = "oneline" ; then
188     case $cmp in
189         diff)   diff -u -b "$ref" "$outfile"            >$cmpfile ;;
190         oneoff) oneoff     "$ref" "$outfile"            >$cmpfile ;;
191         stddev) stddev     "$ref" "$outfile"            >$cmpfile ;;
192         oneline)oneline    "$ref" "$outfile"            >$cmpfile ;;
193         null)   cat               "$outfile"            >$cmpfile ;;
194     esac
195     cmperr=$?
196     test $err = 0 && err=$cmperr
197     test $err = 0 || cat $cmpfile
198 else
199     echo "reference file '$ref' not found"
200     err=1
201 fi
202
203 echo "${test}:${sig:-$err}:$($base64 <$cmpfile):$($base64 <$errfile)" >$repfile
204
205 if test $err = 0; then
206     rm -f $outfile $errfile $cmpfile $cleanfiles
207 else
208     echo "Test $test failed. Look at $errfile for details."
209 fi
210 exit $err