1#!/bin/bash
2
3set -u
4set -e
5trap onexit INT
6trap onexit TERM
7trap onexit EXIT
8
9onexit()
10{
11	if [ -d $OUTDIR ]; then
12		rm -rf $OUTDIR
13	fi
14}
15
16runme()
17{
18	echo \*\*\* $*
19	$*
20}
21
22EXT=bmp
23IMAGES="vgl_5674_0098.${EXT} vgl_6434_0018a.${EXT} vgl_6548_0026a.${EXT} nightshot_iso_100.${EXT}"
24IMGDIR=@srcdir@/testimages
25OUTDIR=`mktemp -d /tmp/__tjbenchtest_output.XXXXXX`
26EXEDIR=.
27BMPARG=
28NSARG=
29YUVARG=
30ALLOC=0
31ALLOCARG=
32if [ "$EXT" = "bmp" ]; then BMPARG=-bmp; fi
33
34if [ -d $OUTDIR ]; then
35	rm -rf $OUTDIR
36fi
37mkdir -p $OUTDIR
38
39while [ $# -gt 0 ]; do
40	case "$1" in
41	-yuv)
42		NSARG=-nosmooth
43		YUVARG=-yuv
44
45# NOTE: The combination of tjEncodeYUV*() and tjCompressFromYUV*() does not
46# always produce bitwise-identical results to tjCompress*() if subsampling is
47# enabled.  In both cases, if the image width or height are not evenly
48# divisible by the MCU width/height, then the bottom and/or right edge are
49# expanded.  However, the libjpeg code performs this expansion prior to
50# downsampling, and TurboJPEG performs it in tjCompressFromYUV*(), which is
51# after downsampling.  Thus, the two will agree only if the width/height along
52# each downsampled dimension is an odd number or is evenly divisible by the MCU
53# width/height.  This disagreement basically amounts to a round-off error, but
54# there is no easy way around it, so for now, we just test the only image that
55# works.  (NOTE: nightshot_iso_100 does not suffer from the above issue, but
56# it suffers from an unrelated problem whereby the combination of
57# tjDecompressToYUV*() and tjDecodeYUV*() do not produce bitwise-identical
58# results to tjDecompress*() if decompression scaling is enabled.  This latter
59# phenomenon is not yet fully understood but is also believed to be some sort
60# of round-off error.)
61		IMAGES="vgl_6548_0026a.${EXT}"
62		;;
63	-alloc)
64		ALLOCARG=-alloc
65		ALLOC=1
66		;;
67	esac
68	shift
69done
70
71exec >$EXEDIR/tjbenchtest$YUVARG$ALLOCARG.log
72
73# Standard tests
74for image in $IMAGES; do
75
76	cp $IMGDIR/$image $OUTDIR
77	basename=`basename $image .${EXT}`
78	runme $EXEDIR/cjpeg -quality 95 -dct fast -grayscale -outfile $OUTDIR/${basename}_GRAY_fast_cjpeg.jpg $IMGDIR/${basename}.${EXT}
79	runme $EXEDIR/cjpeg -quality 95 -dct fast -sample 2x2 -outfile $OUTDIR/${basename}_420_fast_cjpeg.jpg $IMGDIR/${basename}.${EXT}
80	runme $EXEDIR/cjpeg -quality 95 -dct fast -sample 2x1 -outfile $OUTDIR/${basename}_422_fast_cjpeg.jpg $IMGDIR/${basename}.${EXT}
81	runme $EXEDIR/cjpeg -quality 95 -dct fast -sample 1x1 -outfile $OUTDIR/${basename}_444_fast_cjpeg.jpg $IMGDIR/${basename}.${EXT}
82	runme $EXEDIR/cjpeg -quality 95 -dct int -grayscale -outfile $OUTDIR/${basename}_GRAY_accurate_cjpeg.jpg $IMGDIR/${basename}.${EXT}
83	runme $EXEDIR/cjpeg -quality 95 -dct int -sample 2x2 -outfile $OUTDIR/${basename}_420_accurate_cjpeg.jpg $IMGDIR/${basename}.${EXT}
84	runme $EXEDIR/cjpeg -quality 95 -dct int -sample 2x1 -outfile $OUTDIR/${basename}_422_accurate_cjpeg.jpg $IMGDIR/${basename}.${EXT}
85	runme $EXEDIR/cjpeg -quality 95 -dct int -sample 1x1 -outfile $OUTDIR/${basename}_444_accurate_cjpeg.jpg $IMGDIR/${basename}.${EXT}
86	for samp in GRAY 420 422 444; do
87		runme $EXEDIR/djpeg -rgb $NSARG $BMPARG -outfile $OUTDIR/${basename}_${samp}_default_djpeg.${EXT} $OUTDIR/${basename}_${samp}_fast_cjpeg.jpg
88		runme $EXEDIR/djpeg -dct fast -rgb $NSARG $BMPARG -outfile $OUTDIR/${basename}_${samp}_fast_djpeg.${EXT} $OUTDIR/${basename}_${samp}_fast_cjpeg.jpg
89		runme $EXEDIR/djpeg -dct int -rgb $NSARG $BMPARG -outfile $OUTDIR/${basename}_${samp}_accurate_djpeg.${EXT} $OUTDIR/${basename}_${samp}_accurate_cjpeg.jpg
90	done
91	for samp in 420 422; do
92		runme $EXEDIR/djpeg -nosmooth $BMPARG -outfile $OUTDIR/${basename}_${samp}_default_nosmooth_djpeg.${EXT} $OUTDIR/${basename}_${samp}_fast_cjpeg.jpg
93		runme $EXEDIR/djpeg -dct fast -nosmooth $BMPARG -outfile $OUTDIR/${basename}_${samp}_fast_nosmooth_djpeg.${EXT} $OUTDIR/${basename}_${samp}_fast_cjpeg.jpg
94		runme $EXEDIR/djpeg -dct int -nosmooth $BMPARG -outfile $OUTDIR/${basename}_${samp}_accurate_nosmooth_djpeg.${EXT} $OUTDIR/${basename}_${samp}_accurate_cjpeg.jpg
95	done
96
97	# Compression
98	for dct in accurate fast; do
99		runme $EXEDIR/tjbench $OUTDIR/$image 95 -rgb -quiet -benchtime 0.01 -warmup 0 -${dct}dct $YUVARG $ALLOCARG
100		for samp in GRAY 420 422 444; do
101			runme cmp $OUTDIR/${basename}_${samp}_Q95.jpg $OUTDIR/${basename}_${samp}_${dct}_cjpeg.jpg
102		done
103	done
104
105	for dct in fast accurate default; do
106		dctarg=-${dct}dct
107		if [ "${dct}" = "default" ]; then
108			dctarg=
109		fi
110
111		# Tiled compression & decompression
112		runme $EXEDIR/tjbench $OUTDIR/$image 95 -rgb -tile -quiet -benchtime 0.01 -warmup 0 ${dctarg} $YUVARG $ALLOCARG
113		for samp in GRAY 444; do
114			if [ $ALLOC = 1 ]; then
115				runme cmp $OUTDIR/${basename}_${samp}_Q95_full.${EXT} $OUTDIR/${basename}_${samp}_${dct}_djpeg.${EXT}
116				rm $OUTDIR/${basename}_${samp}_Q95_full.${EXT}
117			else
118				for i in $OUTDIR/${basename}_${samp}_Q95_[0-9]*[0-9]x[0-9]*[0-9].${EXT} \
119					$OUTDIR/${basename}_${samp}_Q95_full.${EXT}; do
120					runme cmp $i $OUTDIR/${basename}_${samp}_${dct}_djpeg.${EXT}
121					rm $i
122				done
123			fi
124		done
125		runme $EXEDIR/tjbench $OUTDIR/$image 95 -rgb -tile -quiet -benchtime 0.01 -warmup 0 -fastupsample ${dctarg} $YUVARG $ALLOCARG
126		for samp in 420 422; do
127			if [ $ALLOC = 1 ]; then
128				runme cmp $OUTDIR/${basename}_${samp}_Q95_full.${EXT} $OUTDIR/${basename}_${samp}_${dct}_nosmooth_djpeg.${EXT}
129				rm $OUTDIR/${basename}_${samp}_Q95_full.${EXT}
130			else
131				for i in $OUTDIR/${basename}_${samp}_Q95_[0-9]*[0-9]x[0-9]*[0-9].${EXT} \
132					$OUTDIR/${basename}_${samp}_Q95_full.${EXT}; do
133					runme cmp $i $OUTDIR/${basename}_${samp}_${dct}_nosmooth_djpeg.${EXT}
134					rm $i
135				done
136			fi
137		done
138
139		# Tiled decompression
140		for samp in GRAY 444; do
141			runme $EXEDIR/tjbench $OUTDIR/${basename}_${samp}_Q95.jpg $BMPARG -tile -quiet -benchtime 0.01 -warmup 0 ${dctarg} $YUVARG $ALLOCARG
142			if [ $ALLOC = 1 ]; then
143				runme cmp $OUTDIR/${basename}_${samp}_Q95_full.${EXT} $OUTDIR/${basename}_${samp}_${dct}_djpeg.${EXT}
144				rm $OUTDIR/${basename}_${samp}_Q95_full.${EXT}
145			else
146				for i in $OUTDIR/${basename}_${samp}_Q95_[0-9]*[0-9]x[0-9]*[0-9].${EXT} \
147					$OUTDIR/${basename}_${samp}_Q95_full.${EXT}; do
148					runme cmp $i $OUTDIR/${basename}_${samp}_${dct}_djpeg.${EXT}
149					rm $i
150				done
151			fi
152		done
153		for samp in 420 422; do
154			runme $EXEDIR/tjbench $OUTDIR/${basename}_${samp}_Q95.jpg $BMPARG -tile -quiet -benchtime 0.01 -warmup 0 -fastupsample ${dctarg} $YUVARG $ALLOCARG
155			if [ $ALLOC = 1 ]; then
156				runme cmp $OUTDIR/${basename}_${samp}_Q95_full.${EXT} $OUTDIR/${basename}_${samp}_${dct}_nosmooth_djpeg.${EXT}
157				rm $OUTDIR/${basename}_${samp}_Q95_full.${EXT}
158			else
159				for i in $OUTDIR/${basename}_${samp}_Q95_[0-9]*[0-9]x[0-9]*[0-9].${EXT} \
160					$OUTDIR/${basename}_${samp}_Q95_full.${EXT}; do
161					runme cmp $i $OUTDIR/${basename}_${samp}_${dct}_nosmooth_djpeg.${EXT}
162					rm $i
163				done
164			fi
165		done
166	done
167
168	# Scaled decompression
169	for scale in 2_1 15_8 7_4 13_8 3_2 11_8 5_4 9_8 7_8 3_4 5_8 1_2 3_8 1_4 1_8; do
170		scalearg=`echo $scale | sed s@_@/@g`
171		for samp in GRAY 420 422 444; do
172			runme $EXEDIR/djpeg -rgb -scale ${scalearg} $NSARG $BMPARG -outfile $OUTDIR/${basename}_${samp}_${scale}_djpeg.${EXT} $OUTDIR/${basename}_${samp}_fast_cjpeg.jpg
173			runme $EXEDIR/tjbench $OUTDIR/${basename}_${samp}_Q95.jpg $BMPARG -scale ${scalearg} -quiet -benchtime 0.01 -warmup 0 $YUVARG $ALLOCARG
174			runme cmp $OUTDIR/${basename}_${samp}_Q95_${scale}.${EXT} $OUTDIR/${basename}_${samp}_${scale}_djpeg.${EXT}
175			rm $OUTDIR/${basename}_${samp}_Q95_${scale}.${EXT}
176		done
177	done
178
179	# Transforms
180	for samp in GRAY 420 422 444; do
181		runme $EXEDIR/jpegtran -flip horizontal -trim -outfile $OUTDIR/${basename}_${samp}_hflip_jpegtran.jpg $OUTDIR/${basename}_${samp}_Q95.jpg
182		runme $EXEDIR/jpegtran -flip vertical -trim -outfile $OUTDIR/${basename}_${samp}_vflip_jpegtran.jpg $OUTDIR/${basename}_${samp}_Q95.jpg
183		runme $EXEDIR/jpegtran -transpose -trim -outfile $OUTDIR/${basename}_${samp}_transpose_jpegtran.jpg $OUTDIR/${basename}_${samp}_Q95.jpg
184		runme $EXEDIR/jpegtran -transverse -trim -outfile $OUTDIR/${basename}_${samp}_transverse_jpegtran.jpg $OUTDIR/${basename}_${samp}_Q95.jpg
185		runme $EXEDIR/jpegtran -rotate 90 -trim -outfile $OUTDIR/${basename}_${samp}_rot90_jpegtran.jpg $OUTDIR/${basename}_${samp}_Q95.jpg
186		runme $EXEDIR/jpegtran -rotate 180 -trim -outfile $OUTDIR/${basename}_${samp}_rot180_jpegtran.jpg $OUTDIR/${basename}_${samp}_Q95.jpg
187		runme $EXEDIR/jpegtran -rotate 270 -trim -outfile $OUTDIR/${basename}_${samp}_rot270_jpegtran.jpg $OUTDIR/${basename}_${samp}_Q95.jpg
188	done
189	for xform in hflip vflip transpose transverse rot90 rot180 rot270; do
190		for samp in GRAY 444; do
191			runme $EXEDIR/djpeg -rgb $BMPARG -outfile $OUTDIR/${basename}_${samp}_${xform}_jpegtran.${EXT} $OUTDIR/${basename}_${samp}_${xform}_jpegtran.jpg
192			runme $EXEDIR/tjbench $OUTDIR/${basename}_${samp}_Q95.jpg $BMPARG -$xform -tile -quiet -benchtime 0.01 -warmup 0 $YUVARG $ALLOCARG
193			if [ $ALLOC = 1 ]; then
194				runme cmp $OUTDIR/${basename}_${samp}_Q95_full.${EXT} $OUTDIR/${basename}_${samp}_${xform}_jpegtran.${EXT}
195				rm $OUTDIR/${basename}_${samp}_Q95_full.${EXT}
196			else
197				for i in $OUTDIR/${basename}_${samp}_Q95_[0-9]*[0-9]x[0-9]*[0-9].${EXT} \
198					$OUTDIR/${basename}_${samp}_Q95_full.${EXT}; do
199					runme cmp $i $OUTDIR/${basename}_${samp}_${xform}_jpegtran.${EXT}
200					rm $i
201				done
202			fi
203		done
204		for samp in 420 422; do
205			runme $EXEDIR/djpeg -nosmooth -rgb $BMPARG -outfile $OUTDIR/${basename}_${samp}_${xform}_jpegtran.${EXT} $OUTDIR/${basename}_${samp}_${xform}_jpegtran.jpg
206			runme $EXEDIR/tjbench $OUTDIR/${basename}_${samp}_Q95.jpg $BMPARG -$xform -tile -quiet -benchtime 0.01 -warmup 0 -fastupsample $YUVARG $ALLOCARG
207			if [ $ALLOC = 1 ]; then
208				runme cmp $OUTDIR/${basename}_${samp}_Q95_full.${EXT} $OUTDIR/${basename}_${samp}_${xform}_jpegtran.${EXT}
209				rm $OUTDIR/${basename}_${samp}_Q95_full.${EXT}
210			else
211				for i in $OUTDIR/${basename}_${samp}_Q95_[0-9]*[0-9]x[0-9]*[0-9].${EXT} \
212					$OUTDIR/${basename}_${samp}_Q95_full.${EXT}; do
213					runme cmp $i $OUTDIR/${basename}_${samp}_${xform}_jpegtran.${EXT}
214					rm $i
215				done
216			fi
217		done
218	done
219
220	# Grayscale transform
221	for xform in hflip vflip transpose transverse rot90 rot180 rot270; do
222		for samp in GRAY 444 422 420; do
223			runme $EXEDIR/tjbench $OUTDIR/${basename}_${samp}_Q95.jpg $BMPARG -$xform -tile -quiet -benchtime 0.01 -warmup 0 -grayscale $YUVARG $ALLOCARG
224			if [ $ALLOC = 1 ]; then
225				runme cmp $OUTDIR/${basename}_${samp}_Q95_full.${EXT} $OUTDIR/${basename}_GRAY_${xform}_jpegtran.${EXT}
226				rm $OUTDIR/${basename}_${samp}_Q95_full.${EXT}
227			else
228				for i in $OUTDIR/${basename}_${samp}_Q95_[0-9]*[0-9]x[0-9]*[0-9].${EXT} \
229					$OUTDIR/${basename}_${samp}_Q95_full.${EXT}; do
230					runme cmp $i $OUTDIR/${basename}_GRAY_${xform}_jpegtran.${EXT}
231					rm $i
232				done
233			fi
234		done
235	done
236
237	# Transforms with scaling
238	for xform in hflip vflip transpose transverse rot90 rot180 rot270; do
239		for samp in GRAY 444 422 420; do
240			for scale in 2_1 15_8 7_4 13_8 3_2 11_8 5_4 9_8 7_8 3_4 5_8 1_2 3_8 1_4 1_8; do
241				scalearg=`echo $scale | sed s@_@/@g`
242				runme $EXEDIR/djpeg -rgb -scale ${scalearg} $NSARG $BMPARG -outfile $OUTDIR/${basename}_${samp}_${xform}_${scale}_jpegtran.${EXT} $OUTDIR/${basename}_${samp}_${xform}_jpegtran.jpg
243				runme $EXEDIR/tjbench $OUTDIR/${basename}_${samp}_Q95.jpg $BMPARG -$xform -scale ${scalearg} -quiet -benchtime 0.01 -warmup 0 $YUVARG $ALLOCARG
244				runme cmp $OUTDIR/${basename}_${samp}_Q95_${scale}.${EXT} $OUTDIR/${basename}_${samp}_${xform}_${scale}_jpegtran.${EXT}
245				rm $OUTDIR/${basename}_${samp}_Q95_${scale}.${EXT}
246			done
247		done
248	done
249
250done
251
252echo SUCCESS!
253