OLD | NEW |
| (Empty) |
1 /* Copyright (C)2004 Landmark Graphics Corporation | |
2 * Copyright (C)2005, 2006 Sun Microsystems, Inc. | |
3 * Copyright (C)2009 D. R. Commander | |
4 * | |
5 * This library is free software and may be redistributed and/or modified under | |
6 * the terms of the wxWindows Library License, Version 3.1 or (at your option) | |
7 * any later version. The full license is in the LICENSE.txt file included | |
8 * with this distribution. | |
9 * | |
10 * This library is distributed in the hope that it will be useful, | |
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
13 * wxWindows Library License for more details. | |
14 */ | |
15 | |
16 #include <stdio.h> | |
17 #include <stdlib.h> | |
18 #include <string.h> | |
19 #include <math.h> | |
20 #include "./bmp.h" | |
21 #include "./rrutil.h" | |
22 #include "./rrtimer.h" | |
23 #include "./turbojpeg.h" | |
24 | |
25 #define _catch(f) {if((f)==-1) {printf("Error in %s:\n%s\n", #f, tjGetErrorStr()
); goto bailout;}} | |
26 | |
27 int forcemmx=0, forcesse=0, forcesse2=0, forcesse3=0, fastupsample=0; | |
28 const int _ps[BMPPIXELFORMATS]={3, 4, 3, 4, 4, 4}; | |
29 const int _flags[BMPPIXELFORMATS]={0, 0, TJ_BGR, TJ_BGR, | |
30 TJ_BGR|TJ_ALPHAFIRST, TJ_ALPHAFIRST}; | |
31 const int _rindex[BMPPIXELFORMATS]={0, 0, 2, 2, 3, 1}; | |
32 const int _gindex[BMPPIXELFORMATS]={1, 1, 1, 1, 2, 2}; | |
33 const int _bindex[BMPPIXELFORMATS]={2, 2, 0, 0, 1, 3}; | |
34 const char *_pfname[]={"RGB", "RGBA", "BGR", "BGRA", "ABGR", "ARGB"}; | |
35 const char *_subnamel[NUMSUBOPT]={"4:4:4", "4:2:2", "4:2:0", "GRAY"}; | |
36 const char *_subnames[NUMSUBOPT]={"444", "422", "420", "GRAY"}; | |
37 | |
38 void printsigfig(double val, int figs) | |
39 { | |
40 char format[80]; | |
41 double _l=log10(val); int l; | |
42 if(_l<0.) | |
43 { | |
44 l=(int)fabs(_l); | |
45 sprintf(format, "%%%d.%df", figs+l+2, figs+l); | |
46 } | |
47 else | |
48 { | |
49 l=(int)_l+1; | |
50 if(figs<=l) sprintf(format, "%%.0f"); | |
51 else sprintf(format, "%%%d.%df", figs+1, figs-l); | |
52 } | |
53 printf(format, val); | |
54 } | |
55 | |
56 void dotest(unsigned char *srcbuf, int w, int h, BMPPIXELFORMAT pf, int bu, | |
57 int jpegsub, int qual, char *filename, int dotile, int useppm, int quiet
) | |
58 { | |
59 char tempstr[1024]; | |
60 FILE *outfile; tjhandle hnd; | |
61 unsigned char **jpegbuf=NULL, *rgbbuf=NULL; | |
62 rrtimer timer; double elapsed; | |
63 int jpgbufsize=0, i, j, tilesizex, tilesizey, numtilesx, numtilesy, ITER
; | |
64 unsigned long *comptilesize=NULL; | |
65 int flags=(forcemmx?TJ_FORCEMMX:0)|(forcesse?TJ_FORCESSE:0) | |
66 |(forcesse2?TJ_FORCESSE2:0)|(forcesse3?TJ_FORCESSE3:0) | |
67 |(fastupsample?TJ_FASTUPSAMPLE:0); | |
68 int ps=_ps[pf]; | |
69 int pitch=w*ps; | |
70 | |
71 flags |= _flags[pf]; | |
72 if(bu) flags |= TJ_BOTTOMUP; | |
73 | |
74 if((rgbbuf=(unsigned char *)malloc(pitch*h)) == NULL) | |
75 { | |
76 puts("ERROR: Could not allocate image buffer."); | |
77 exit(1); | |
78 } | |
79 | |
80 if(!quiet) printf("\n>>>>> %s (%s) <--> JPEG %s Q%d <<<<<\n", _pfname[
pf], | |
81 bu?"Bottom-up":"Top-down", _subnamel[jpegsub], qual); | |
82 if(dotile) {tilesizex=tilesizey=4;} else {tilesizex=w; tilesizey=h;} | |
83 | |
84 do | |
85 { | |
86 tilesizex*=2; if(tilesizex>w) tilesizex=w; | |
87 tilesizey*=2; if(tilesizey>h) tilesizey=h; | |
88 numtilesx=(w+tilesizex-1)/tilesizex; | |
89 numtilesy=(h+tilesizey-1)/tilesizey; | |
90 if((comptilesize=(unsigned long *)malloc(sizeof(unsigned long)*n
umtilesx*numtilesy)) == NULL | |
91 || (jpegbuf=(unsigned char **)malloc(sizeof(unsigned char *)*num
tilesx*numtilesy)) == NULL) | |
92 { | |
93 puts("ERROR: Could not allocate image buffers."); | |
94 goto bailout; | |
95 } | |
96 memset(jpegbuf, 0, sizeof(unsigned char *)*numtilesx*numtilesy); | |
97 for(i=0; i<numtilesx*numtilesy; i++) | |
98 { | |
99 if((jpegbuf[i]=(unsigned char *)malloc(TJBUFSIZE(tilesiz
ex, tilesizey))) == NULL) | |
100 { | |
101 puts("ERROR: Could not allocate image buffers.")
; | |
102 goto bailout; | |
103 } | |
104 } | |
105 | |
106 // Compression test | |
107 if(quiet) printf("%s\t%s\t%s\t%d\t", _pfname[pf], bu?"BU":"TD", | |
108 _subnamel[jpegsub], qual); | |
109 for(i=0; i<h; i++) memcpy(&rgbbuf[pitch*i], &srcbuf[w*ps*i], w*p
s); | |
110 if((hnd=tjInitCompress())==NULL) | |
111 { | |
112 printf("Error in tjInitCompress():\n%s\n", tjGetErrorStr
()); | |
113 goto bailout; | |
114 } | |
115 _catch(tjCompress(hnd, rgbbuf, tilesizex, pitch, tilesizey, ps, | |
116 jpegbuf[0], &comptilesize[0], jpegsub, qual, flags)); | |
117 ITER=0; | |
118 timer.start(); | |
119 do | |
120 { | |
121 jpgbufsize=0; int tilen=0; | |
122 for(i=0; i<h; i+=tilesizey) | |
123 { | |
124 for(j=0; j<w; j+=tilesizex) | |
125 { | |
126 int tempw=min(tilesizex, w-j), temph=min
(tilesizey, h-i); | |
127 _catch(tjCompress(hnd, &rgbbuf[pitch*i+j
*ps], tempw, pitch, | |
128 temph, ps, jpegbuf[tilen], &comp
tilesize[tilen], jpegsub, qual, | |
129 flags)); | |
130 jpgbufsize+=comptilesize[tilen]; | |
131 tilen++; | |
132 } | |
133 } | |
134 ITER++; | |
135 } while((elapsed=timer.elapsed())<5.); | |
136 _catch(tjDestroy(hnd)); | |
137 if(quiet) | |
138 { | |
139 if(tilesizex==w && tilesizey==h) printf("Full \t"); | |
140 else printf("%-4d %-4d\t", tilesizex, tilesizey); | |
141 printsigfig((double)(w*h)/1000000.*(double)ITER/elapsed,
4); | |
142 printf("\t"); | |
143 printsigfig((double)(w*h*ps)/(double)jpgbufsize, 4); | |
144 printf("\t"); | |
145 } | |
146 else | |
147 { | |
148 if(tilesizex==w && tilesizey==h) printf("\nFull image\n"
); | |
149 else printf("\nTile size: %d x %d\n", tilesizex, tilesiz
ey); | |
150 printf("C--> Frame rate: %f fps\n", (double)IT
ER/elapsed); | |
151 printf(" Output image size: %d bytes\n", jpgbufsi
ze); | |
152 printf(" Compression ratio: %f:1\n", | |
153 (double)(w*h*ps)/(double)jpgbufsize); | |
154 printf(" Source throughput: %f Megapixels/sec\n", | |
155 (double)(w*h)/1000000.*(double)ITER/elapsed); | |
156 printf(" Output bit stream: %f Megabits/sec\n", | |
157 (double)jpgbufsize*8./1000000.*(double)ITER/elap
sed); | |
158 } | |
159 if(tilesizex==w && tilesizey==h) | |
160 { | |
161 sprintf(tempstr, "%s_%sQ%d.jpg", filename, _subnames[jpe
gsub], qual); | |
162 if((outfile=fopen(tempstr, "wb"))==NULL) | |
163 { | |
164 puts("ERROR: Could not open reference image"); | |
165 exit(1); | |
166 } | |
167 if(fwrite(jpegbuf[0], jpgbufsize, 1, outfile)!=1) | |
168 { | |
169 puts("ERROR: Could not write reference image"); | |
170 exit(1); | |
171 } | |
172 fclose(outfile); | |
173 if(!quiet) printf("Reference image written to %s\n", tem
pstr); | |
174 } | |
175 | |
176 // Decompression test | |
177 memset(rgbbuf, 127, pitch*h); // Grey image means decompressor
did nothing | |
178 if((hnd=tjInitDecompress())==NULL) | |
179 { | |
180 printf("Error in tjInitDecompress():\n%s\n", tjGetErrorS
tr()); | |
181 goto bailout; | |
182 } | |
183 _catch(tjDecompress(hnd, jpegbuf[0], jpgbufsize, rgbbuf, tilesiz
ex, pitch, | |
184 tilesizey, ps, flags)); | |
185 ITER=0; | |
186 timer.start(); | |
187 do | |
188 { | |
189 int tilen=0; | |
190 for(i=0; i<h; i+=tilesizey) | |
191 { | |
192 for(j=0; j<w; j+=tilesizex) | |
193 { | |
194 int tempw=min(tilesizex, w-j), temph=min
(tilesizey, h-i); | |
195 _catch(tjDecompress(hnd, jpegbuf[tilen],
comptilesize[tilen], | |
196 &rgbbuf[pitch*i+ps*j], tempw, pi
tch, temph, ps, flags)); | |
197 tilen++; | |
198 } | |
199 } | |
200 ITER++; | |
201 } while((elapsed=timer.elapsed())<5.); | |
202 _catch(tjDestroy(hnd)); | |
203 if(quiet) | |
204 { | |
205 printsigfig((double)(w*h)/1000000.*(double)ITER/elapsed,
4); | |
206 printf("\n"); | |
207 } | |
208 else | |
209 { | |
210 printf("D--> Frame rate: %f fps\n", (double)IT
ER/elapsed); | |
211 printf(" Dest. throughput: %f Megapixels/sec\n", | |
212 (double)(w*h)/1000000.*(double)ITER/elapsed); | |
213 } | |
214 if(tilesizex==w && tilesizey==h) | |
215 sprintf(tempstr, "%s_%sQ%d_full.%s", filename, _subnames
[jpegsub], qual, | |
216 useppm?"ppm":"bmp"); | |
217 else sprintf(tempstr, "%s_%sQ%d_%dx%d.%s", filename, _subnames[j
pegsub], | |
218 qual, tilesizex, tilesizey, useppm?"ppm":"bmp"); | |
219 if(savebmp(tempstr, rgbbuf, w, h, pf, pitch, bu)==-1) | |
220 { | |
221 printf("ERROR saving bitmap: %s\n", bmpgeterr()); | |
222 goto bailout; | |
223 } | |
224 sprintf(strrchr(tempstr, '.'), "-err.%s", useppm?"ppm":"bmp"); | |
225 if(!quiet) | |
226 printf("Computing compression error and saving to %s.\n"
, tempstr); | |
227 if(jpegsub==TJ_GRAYSCALE) | |
228 { | |
229 for(j=0; j<h; j++) | |
230 { | |
231 for(i=0; i<w*ps; i+=ps) | |
232 { | |
233 int y=(int)((double)srcbuf[w*ps*j+i+_rin
dex[pf]]*0.299 | |
234 + (double)srcbuf[w*ps*j+i+_ginde
x[pf]]*0.587 | |
235 + (double)srcbuf[w*ps*j+i+_binde
x[pf]]*0.114 + 0.5); | |
236 if(y>255) y=255; if(y<0) y=0; | |
237 rgbbuf[pitch*j+i+_rindex[pf]]=abs(rgbbuf
[pitch*j+i+_rindex[pf]]-y); | |
238 rgbbuf[pitch*j+i+_gindex[pf]]=abs(rgbbuf
[pitch*j+i+_gindex[pf]]-y); | |
239 rgbbuf[pitch*j+i+_bindex[pf]]=abs(rgbbuf
[pitch*j+i+_bindex[pf]]-y); | |
240 } | |
241 } | |
242 } | |
243 else | |
244 { | |
245 for(j=0; j<h; j++) for(i=0; i<w*ps; i++) | |
246 rgbbuf[pitch*j+i]=abs(rgbbuf[pitch*j+i]-srcbuf[w
*ps*j+i]); | |
247 } | |
248 if(savebmp(tempstr, rgbbuf, w, h, pf, pitch, bu)==-1) | |
249 { | |
250 printf("ERROR saving bitmap: %s\n", bmpgeterr()); | |
251 goto bailout; | |
252 } | |
253 | |
254 // Cleanup | |
255 if(jpegbuf) | |
256 { | |
257 for(i=0; i<numtilesx*numtilesy; i++) | |
258 {if(jpegbuf[i]) free(jpegbuf[i]); jpegbuf[i]=NU
LL;} | |
259 free(jpegbuf); jpegbuf=NULL; | |
260 } | |
261 if(comptilesize) {free(comptilesize); comptilesize=NULL;} | |
262 } while(tilesizex<w || tilesizey<h); | |
263 | |
264 if(rgbbuf) {free(rgbbuf); rgbbuf=NULL;} | |
265 return; | |
266 | |
267 bailout: | |
268 if(jpegbuf) | |
269 { | |
270 for(i=0; i<numtilesx*numtilesy; i++) | |
271 {if(jpegbuf[i]) free(jpegbuf[i]); jpegbuf[i]=NULL;} | |
272 free(jpegbuf); jpegbuf=NULL; | |
273 } | |
274 if(comptilesize) {free(comptilesize); comptilesize=NULL;} | |
275 if(rgbbuf) {free(rgbbuf); rgbbuf=NULL;} | |
276 return; | |
277 } | |
278 | |
279 | |
280 int main(int argc, char *argv[]) | |
281 { | |
282 unsigned char *bmpbuf=NULL; int w, h, i, useppm=0; | |
283 int qual, dotile=0, quiet=0, hiqual=-1; char *temp; | |
284 BMPPIXELFORMAT pf=BMP_BGR; | |
285 int bu=0; | |
286 | |
287 printf("\n"); | |
288 | |
289 if(argc<3) | |
290 { | |
291 printf("USAGE: %s <Inputfile (BMP|PPM)> <%% Quality>\n\n", argv[
0]); | |
292 printf(" [-tile]\n"); | |
293 printf(" Test performance of the codec when the image is e
ncoded\n"); | |
294 printf(" as separate tiles of varying sizes.\n\n"); | |
295 printf(" [-forcemmx] [-forcesse] [-forcesse2] [-forcesse3]
\n"); | |
296 printf(" Force MMX, SSE, or SSE2 code paths in Intel codec
\n\n"); | |
297 printf(" [-rgb | -bgr | -rgba | -bgra | -abgr | -argb]\n")
; | |
298 printf(" Test the specified color conversion path in the c
odec (default: BGR)\n\n"); | |
299 printf(" [-fastupsample]\n"); | |
300 printf(" Use fast, inaccurate upsampling code to perform 4
:2:2 and 4:2:0\n"); | |
301 printf(" YUV decoding in libjpeg decompressor\n\n"); | |
302 printf(" [-quiet]\n"); | |
303 printf(" Output in tabular rather than verbose format\n\n"
); | |
304 printf(" NOTE: If the quality is specified as a range, i.e
. 90-100, a separate\n"); | |
305 printf(" test will be performed for all quality values in
the range.\n"); | |
306 exit(1); | |
307 } | |
308 if((qual=atoi(argv[2]))<1 || qual>100) | |
309 { | |
310 puts("ERROR: Quality must be between 1 and 100."); | |
311 exit(1); | |
312 } | |
313 if((temp=strchr(argv[2], '-'))!=NULL && strlen(temp)>1 | |
314 && sscanf(&temp[1], "%d", &hiqual)==1 && hiqual>qual && hiqual>=
1 | |
315 && hiqual<=100) {} | |
316 else hiqual=qual; | |
317 | |
318 if(argc>3) | |
319 { | |
320 for(i=3; i<argc; i++) | |
321 { | |
322 if(!stricmp(argv[i], "-tile")) dotile=1; | |
323 if(!stricmp(argv[i], "-forcesse3")) | |
324 { | |
325 printf("Using SSE3 code\n"); | |
326 forcesse3=1; | |
327 } | |
328 if(!stricmp(argv[i], "-forcesse2")) | |
329 { | |
330 printf("Using SSE2 code\n"); | |
331 forcesse2=1; | |
332 } | |
333 if(!stricmp(argv[i], "-forcesse")) | |
334 { | |
335 printf("Using SSE code\n"); | |
336 forcesse=1; | |
337 } | |
338 if(!stricmp(argv[i], "-forcemmx")) | |
339 { | |
340 printf("Using MMX code\n"); | |
341 forcemmx=1; | |
342 } | |
343 if(!stricmp(argv[i], "-fastupsample")) | |
344 { | |
345 printf("Using fast upsampling code\n"); | |
346 fastupsample=1; | |
347 } | |
348 if(!stricmp(argv[i], "-rgb")) pf=BMP_RGB; | |
349 if(!stricmp(argv[i], "-rgba")) pf=BMP_RGBA; | |
350 if(!stricmp(argv[i], "-bgr")) pf=BMP_BGR; | |
351 if(!stricmp(argv[i], "-bgra")) pf=BMP_BGRA; | |
352 if(!stricmp(argv[i], "-abgr")) pf=BMP_ABGR; | |
353 if(!stricmp(argv[i], "-argb")) pf=BMP_ARGB; | |
354 if(!stricmp(argv[i], "-bottomup")) bu=1; | |
355 if(!stricmp(argv[i], "-quiet")) quiet=1; | |
356 } | |
357 } | |
358 | |
359 if(loadbmp(argv[1], &bmpbuf, &w, &h, pf, 1, bu)==-1) | |
360 { | |
361 printf("ERROR loading bitmap: %s\n", bmpgeterr()); exit(1); | |
362 } | |
363 | |
364 temp=strrchr(argv[1], '.'); | |
365 if(temp!=NULL) | |
366 { | |
367 if(!stricmp(temp, ".ppm")) useppm=1; | |
368 *temp='\0'; | |
369 } | |
370 | |
371 if(quiet) | |
372 { | |
373 printf("All performance values in Mpixels/sec\n\n"); | |
374 printf("Bitmap\tBitmap\tJPEG\tJPEG\tTile Size\tCompr\tCompr\tDec
omp\n"); | |
375 printf("Format\tOrder\tFormat\tQual\t X Y \tPerf \tRatio\tPe
rf\n\n"); | |
376 } | |
377 | |
378 for(i=hiqual; i>=qual; i--) | |
379 dotest(bmpbuf, w, h, pf, bu, TJ_GRAYSCALE, i, argv[1], dotile, u
seppm, quiet); | |
380 if(quiet) printf("\n"); | |
381 for(i=hiqual; i>=qual; i--) | |
382 dotest(bmpbuf, w, h, pf, bu, TJ_420, i, argv[1], dotile, useppm,
quiet); | |
383 if(quiet) printf("\n"); | |
384 for(i=hiqual; i>=qual; i--) | |
385 dotest(bmpbuf, w, h, pf, bu, TJ_422, i, argv[1], dotile, useppm,
quiet); | |
386 if(quiet) printf("\n"); | |
387 for(i=hiqual; i>=qual; i--) | |
388 dotest(bmpbuf, w, h, pf, bu, TJ_444, i, argv[1], dotile, useppm,
quiet); | |
389 | |
390 if(bmpbuf) free(bmpbuf); | |
391 return 0; | |
392 } | |
OLD | NEW |