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