fate: simplify regression-funcs.sh
[ffmpeg.git] / 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 benchfile="$datadir/$this.bench"
27 bench="$datadir/$this.bench.tmp"
28 bench2="$datadir/$this.bench2.tmp"
29 raw_src="${target_path}/$raw_src_dir/%02d.pgm"
30 raw_dst="$datadir/$this.out.yuv"
31 raw_ref="$datadir/$test_ref.ref.yuv"
32 pcm_src="$target_datadir/asynth1.sw"
33 pcm_dst="$datadir/$this.out.wav"
34 pcm_ref="$datadir/$test_ref.ref.wav"
35 crcfile="$datadir/$this.crc"
36 target_crcfile="$target_datadir/$this.crc"
37
38 cleanfiles="$raw_dst $pcm_dst $crcfile $bench $bench2"
39 trap 'rm -f -- $cleanfiles' EXIT
40
41 mkdir -p "$datadir"
42 mkdir -p "$outfile"
43 mkdir -p "$logdir"
44
45 (exec >&3) 2>/dev/null || exec 3>&2
46
47 [ "${V-0}" -gt 0 ] && echov=echov || echov=:
48 [ "${V-0}" -gt 1 ] || exec 2>$errfile
49
50 echov(){
51     echo "$@" >&3
52 }
53
54 . $(dirname $0)/md5.sh
55
56 FFMPEG_OPTS="-v 0 -threads $threads -y -flags +bitexact -dct fastint -idct simple -sws_flags +accurate_rnd+bitexact"
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 -benchmark $* > $bench
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     expr "$(cat $bench)" : '.*utime=\(.*s\)' > $bench2
79     echo $(cat $bench2) $f >> $benchfile
80 }
81
82 do_ffmpeg_nomd5()
83 {
84     f="$1"
85     shift
86     set -- $* ${target_path}/$f
87     run_ffmpeg -benchmark $* > $bench
88     if [ $f = $raw_dst ] ; then
89         $tiny_psnr $f $raw_ref >> $logfile
90     elif [ $f = $pcm_dst ] ; then
91         $tiny_psnr $f $pcm_ref 2 >> $logfile
92     else
93         wc -c $f >> $logfile
94     fi
95     expr "$(cat $bench)" : '.*utime=\(.*s\)' > $bench2
96     echo $(cat $bench2) $f >> $benchfile
97 }
98
99 do_ffmpeg_crc()
100 {
101     f="$1"
102     shift
103     run_ffmpeg $* -f crc "$target_crcfile"
104     echo "$f $(cat $crcfile)" >> $logfile
105 }
106
107 do_ffmpeg_nocheck()
108 {
109     f="$1"
110     shift
111     run_ffmpeg -benchmark $* > $bench
112     expr "$(cat $bench)" : '.*utime=\(.*s\)' > $bench2
113     echo $(cat $bench2) $f >> $benchfile
114 }
115
116 do_video_decoding()
117 {
118     do_ffmpeg $raw_dst $1 -i $target_path/$file -f rawvideo $2
119 }
120
121 do_video_encoding()
122 {
123     file=${outfile}$1
124     do_ffmpeg $file $2 -f image2 -vcodec pgmyuv -i $raw_src $3
125 }
126
127 do_audio_encoding()
128 {
129     file=${outfile}$1
130     do_ffmpeg $file -ab 128k -ac 2 -f s16le -i $pcm_src $3
131 }
132
133 do_audio_decoding()
134 {
135     do_ffmpeg $pcm_dst -i $target_path/$file -sample_fmt s16 -f wav
136 }