Include low-level unit tests borrowed from VirtualGL
git-svn-id: svn://svn.code.sf.net/p/tigervnc/code/trunk@3745 3789f03b-4d11-0410-bbf8-ca57d06f2519
diff --git a/common/jpeg/jpegut.c b/common/jpeg/jpegut.c
new file mode 100644
index 0000000..777cd3d
--- /dev/null
+++ b/common/jpeg/jpegut.c
@@ -0,0 +1,384 @@
+/* Copyright (C)2004 Landmark Graphics Corporation
+ * Copyright (C)2005 Sun Microsystems, Inc.
+ * Copyright (C)2009 D. R. Commander
+ *
+ * This library is free software and may be redistributed and/or modified under
+ * the terms of the wxWindows Library License, Version 3.1 or (at your option)
+ * any later version. The full license is in the LICENSE.txt file included
+ * with this distribution.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * wxWindows Library License for more details.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include "./rrtimer.h"
+#include "./turbojpeg.h"
+
+#define _catch(f) {if((f)==-1) {printf("TJPEG: %s\n", tjGetErrorStr()); goto finally;}}
+
+const char *_subnamel[NUMSUBOPT]={"4:4:4", "4:2:2", "4:1:1", "GRAY"};
+const char *_subnames[NUMSUBOPT]={"444", "422", "411", "GRAY"};
+
+int pixels[9][3]=
+{
+ {0, 255, 0},
+ {255, 0, 255},
+ {255, 255, 0},
+ {0, 0, 255},
+ {0, 255, 255},
+ {255, 0, 0},
+ {255, 255, 255},
+ {0, 0, 0},
+ {255, 0, 0}
+};
+
+void initbuf(unsigned char *buf, int w, int h, int ps, int flags)
+{
+ int roffset=(flags&TJ_BGR)?2:0, goffset=1, boffset=(flags&TJ_BGR)?0:2, i,
+ _i, j;
+ if(flags&TJ_ALPHAFIRST) {roffset++; goffset++; boffset++;}
+ memset(buf, 0, w*h*ps);
+ for(_i=0; _i<16; _i++)
+ {
+ if(flags&TJ_BOTTOMUP) i=h-_i-1; else i=_i;
+ for(j=0; j<w; j++)
+ {
+ buf[(w*i+j)*ps+roffset]=255;
+ if(((_i/8)+(j/8))%2==0)
+ {
+ buf[(w*i+j)*ps+goffset]=255;
+ buf[(w*i+j)*ps+boffset]=255;
+ }
+ }
+ }
+ for(_i=16; _i<h; _i++)
+ {
+ if(flags&TJ_BOTTOMUP) i=h-_i-1; else i=_i;
+ for(j=0; j<w; j++)
+ {
+ if(((_i/8)+(j/8))%2!=0)
+ {
+ buf[(w*i+j)*ps+roffset]=255;
+ buf[(w*i+j)*ps+goffset]=255;
+ }
+ }
+ }
+}
+
+int dumpbuf(unsigned char *buf, int w, int h, int ps, int flags)
+{
+ int roffset=(flags&TJ_BGR)?2:0, goffset=1, boffset=(flags&TJ_BGR)?0:2, i,
+ j;
+ for(i=0; i<h; i++)
+ {
+ for(j=0; j<w; j++)
+ {
+ printf("%.3d/%.3d/%.3d ", buf[(w*i+j)*ps+roffset],
+ buf[(w*i+j)*ps+roffset], buf[(w*i+j)*ps+roffset]);
+ }
+ printf("\n");
+ }
+}
+
+int checkbuf(unsigned char *buf, int w, int h, int ps, int subsamp, int flags)
+{
+ int roffset=(flags&TJ_BGR)?2:0, goffset=1, boffset=(flags&TJ_BGR)?0:2, i,
+ _i, j;
+ if(flags&TJ_ALPHAFIRST) {roffset++; goffset++; boffset++;}
+ if(subsamp==TJ_GRAYSCALE)
+ {
+ for(_i=0; _i<16; _i++)
+ {
+ if(flags&TJ_BOTTOMUP) i=h-_i-1; else i=_i;
+ for(j=0; j<w; j++)
+ {
+ unsigned char r=buf[(w*i+j)*ps+roffset],
+ g=buf[(w*i+j)*ps+goffset],
+ b=buf[(w*i+j)*ps+boffset];
+ if(((_i/8)+(j/8))%2==0)
+ {
+ if(r<253 || g<253 || b<253) return 0;
+ }
+ else
+ {
+ if(r<74 || r>78 || g<74 || g>78 || b<74 || b>78) return 0;
+ }
+ }
+ }
+ for(_i=16; _i<h; _i++)
+ {
+ if(flags&TJ_BOTTOMUP) i=h-_i-1; else i=_i;
+ for(j=0; j<w; j++)
+ {
+ unsigned char r=buf[(w*i+j)*ps+roffset],
+ g=buf[(w*i+j)*ps+goffset],
+ b=buf[(w*i+j)*ps+boffset];
+ if(((_i/8)+(j/8))%2==0)
+ {
+ if(r>2 || g>2 || b>2) return 0;
+ }
+ else
+ {
+ if(r<224 || r>228 || g<224 || g>228 || b<224 || b>228) return 0;
+ }
+ }
+ }
+ }
+ else
+ {
+ for(_i=0; _i<16; _i++)
+ {
+ if(flags&TJ_BOTTOMUP) i=h-_i-1; else i=_i;
+ for(j=0; j<w; j++)
+ {
+ if(buf[(w*i+j)*ps+roffset]<253) return 0;
+ if(((_i/8)+(j/8))%2==0)
+ {
+ if(buf[(w*i+j)*ps+goffset]<253) return 0;
+ if(buf[(w*i+j)*ps+boffset]<253) return 0;
+ }
+ else
+ {
+ if(buf[(w*i+j)*ps+goffset]>2) return 0;
+ if(buf[(w*i+j)*ps+boffset]>2) return 0;
+ }
+ }
+ }
+ for(_i=16; _i<h; _i++)
+ {
+ if(flags&TJ_BOTTOMUP) i=h-_i-1; else i=_i;
+ for(j=0; j<w; j++)
+ {
+ if(buf[(w*i+j)*ps+boffset]>2) return 0;
+ if(((_i/8)+(j/8))%2==0)
+ {
+ if(buf[(w*i+j)*ps+roffset]>2) return 0;
+ if(buf[(w*i+j)*ps+goffset]>2) return 0;
+ }
+ else
+ {
+ if(buf[(w*i+j)*ps+roffset]<253) return 0;
+ if(buf[(w*i+j)*ps+goffset]<253) return 0;
+ }
+ }
+ }
+ }
+ return 1;
+}
+
+void writejpeg(unsigned char *jpegbuf, unsigned long jpgbufsize, char *filename)
+{
+ FILE *outfile=NULL;
+ if((outfile=fopen(filename, "wb"))==NULL)
+ {
+ printf("ERROR: Could not open %s for writing.\n", filename);
+ goto finally;
+ }
+ if(fwrite(jpegbuf, jpgbufsize, 1, outfile)!=1)
+ {
+ printf("ERROR: Could not write to %s.\n", filename);
+ goto finally;
+ }
+
+ finally:
+ if(outfile) fclose(outfile);
+}
+
+void gentestjpeg(tjhandle hnd, unsigned char *jpegbuf, unsigned long *size,
+ int w, int h, int ps, char *basefilename, int subsamp, int qual, int flags)
+{
+ char tempstr[1024]; unsigned char *bmpbuf=NULL;
+ const char *pixformat; double t;
+
+ if(flags&TJ_BGR)
+ {
+ if(ps==3) pixformat="BGR";
+ else {if(flags&TJ_ALPHAFIRST) pixformat="ABGR"; else pixformat="BGRA";}
+ }
+ else
+ {
+ if(ps==3) pixformat="RGB";
+ else {if(flags&TJ_ALPHAFIRST) pixformat="ARGB"; else pixformat="RGBA";}
+ }
+ printf("%s %s -> %s Q%d ... ", pixformat,
+ (flags&TJ_BOTTOMUP)?"Bottom-Up":"Top-Down ", _subnamel[subsamp], qual);
+
+ if((bmpbuf=(unsigned char *)malloc(w*h*ps+1))==NULL)
+ {
+ printf("ERROR: Could not allocate buffer\n"); goto finally;
+ }
+ initbuf(bmpbuf, w, h, ps, flags);
+ memset(jpegbuf, 0, TJBUFSIZE(w, h));
+
+ t=rrtime();
+ _catch(tjCompress(hnd, bmpbuf, w, 0, h, ps, jpegbuf, size, subsamp, qual, flags));
+ t=rrtime()-t;
+
+ sprintf(tempstr, "%s_enc_%s_%s_%sQ%d.jpg", basefilename, pixformat,
+ (flags&TJ_BOTTOMUP)? "BU":"TD", _subnames[subsamp], qual);
+ writejpeg(jpegbuf, *size, tempstr);
+ printf("Done. %f ms\n Result in %s\n", t*1000., tempstr);
+
+ finally:
+ if(bmpbuf) free(bmpbuf);
+}
+
+void gentestbmp(tjhandle hnd, unsigned char *jpegbuf, unsigned long jpegsize,
+ int w, int h, int ps, char *basefilename, int subsamp, int qual, int flags)
+{
+ unsigned char *bmpbuf=NULL;
+ const char *pixformat; int _w=0, _h=0; double t;
+
+ if(flags&TJ_BGR)
+ {
+ if(ps==3) pixformat="BGR";
+ else {if(flags&TJ_ALPHAFIRST) pixformat="ABGR"; else pixformat="BGRA";}
+ }
+ else
+ {
+ if(ps==3) pixformat="RGB";
+ else {if(flags&TJ_ALPHAFIRST) pixformat="ARGB"; else pixformat="RGBA";}
+ }
+ printf("JPEG -> %s %s ... ", pixformat, (flags&TJ_BOTTOMUP)?"Bottom-Up":"Top-Down ");
+
+ _catch(tjDecompressHeader(hnd, jpegbuf, jpegsize, &_w, &_h));
+ if(_w!=w || _h!=h)
+ {
+ printf("Incorrect JPEG header\n"); goto finally;
+ }
+
+ if((bmpbuf=(unsigned char *)malloc(w*h*ps+1))==NULL)
+ {
+ printf("ERROR: Could not allocate buffer\n"); goto finally;
+ }
+ memset(bmpbuf, 0, w*ps*h);
+
+ t=rrtime();
+ _catch(tjDecompress(hnd, jpegbuf, jpegsize, bmpbuf, w, w*ps, h, ps, flags));
+ t=rrtime()-t;
+
+ if(checkbuf(bmpbuf, w, h, ps, subsamp, flags)) printf("Passed.");
+ else {printf("FAILED!"); dumpbuf(bmpbuf, w, h, ps, flags);}
+
+ printf(" %f ms\n\n", t*1000.);
+
+ finally:
+ if(bmpbuf) free(bmpbuf);
+}
+
+void dotest(int w, int h, int ps, int subsamp, char *basefilename)
+{
+ tjhandle hnd=NULL, dhnd=NULL; unsigned char *jpegbuf=NULL;
+ unsigned long size;
+
+ if((jpegbuf=(unsigned char *)malloc(TJBUFSIZE(w, h))) == NULL)
+ {
+ puts("ERROR: Could not allocate buffer."); goto finally;
+ }
+
+ if((hnd=tjInitCompress())==NULL)
+ {printf("Error in tjInitCompress():\n%s\n", tjGetErrorStr()); goto finally;}
+ if((dhnd=tjInitDecompress())==NULL)
+ {printf("Error in tjInitDecompress():\n%s\n", tjGetErrorStr()); goto finally;}
+
+ gentestjpeg(hnd, jpegbuf, &size, w, h, ps, basefilename, subsamp, 100, 0);
+ gentestbmp(dhnd, jpegbuf, size, w, h, ps, basefilename, subsamp, 100, 0);
+
+ gentestjpeg(hnd, jpegbuf, &size, w, h, ps, basefilename, subsamp, 100, TJ_BGR);
+ gentestbmp(dhnd, jpegbuf, size, w, h, ps, basefilename, subsamp, 100, TJ_BGR);
+
+ gentestjpeg(hnd, jpegbuf, &size, w, h, ps, basefilename, subsamp, 100, TJ_BOTTOMUP);
+ gentestbmp(dhnd, jpegbuf, size, w, h, ps, basefilename, subsamp, 100, TJ_BOTTOMUP);
+
+ gentestjpeg(hnd, jpegbuf, &size, w, h, ps, basefilename, subsamp, 100, TJ_BGR|TJ_BOTTOMUP);
+ gentestbmp(dhnd, jpegbuf, size, w, h, ps, basefilename, subsamp, 100, TJ_BGR|TJ_BOTTOMUP);
+
+ if(ps==4)
+ {
+ gentestjpeg(hnd, jpegbuf, &size, w, h, ps, basefilename, subsamp, 100, TJ_ALPHAFIRST);
+ gentestbmp(dhnd, jpegbuf, size, w, h, ps, basefilename, subsamp, 100, TJ_ALPHAFIRST);
+
+ gentestjpeg(hnd, jpegbuf, &size, w, h, ps, basefilename, subsamp, 100, TJ_ALPHAFIRST|TJ_BGR);
+ gentestbmp(dhnd, jpegbuf, size, w, h, ps, basefilename, subsamp, 100, TJ_ALPHAFIRST|TJ_BGR);
+
+ gentestjpeg(hnd, jpegbuf, &size, w, h, ps, basefilename, subsamp, 100, TJ_ALPHAFIRST|TJ_BOTTOMUP);
+ gentestbmp(dhnd, jpegbuf, size, w, h, ps, basefilename, subsamp, 100, TJ_ALPHAFIRST|TJ_BOTTOMUP);
+
+ gentestjpeg(hnd, jpegbuf, &size, w, h, ps, basefilename, subsamp, 100, TJ_ALPHAFIRST|TJ_BGR|TJ_BOTTOMUP);
+ gentestbmp(dhnd, jpegbuf, size, w, h, ps, basefilename, subsamp, 100, TJ_ALPHAFIRST|TJ_BGR|TJ_BOTTOMUP);
+ }
+
+ finally:
+ if(hnd) tjDestroy(hnd);
+ if(dhnd) tjDestroy(dhnd);
+
+ if(jpegbuf) free(jpegbuf);
+}
+
+#define MAXLENGTH 2048
+
+void dotest1(void)
+{
+ int i, j, i2; unsigned char *bmpbuf=NULL, *jpgbuf=NULL;
+ tjhandle hnd=NULL; unsigned long size;
+ if((hnd=tjInitCompress())==NULL)
+ {printf("Error in tjInitCompress():\n%s\n", tjGetErrorStr()); goto finally;}
+ printf("Buffer size regression test\n");
+ for(j=1; j<48; j++)
+ {
+ for(i=1; i<(j==1?MAXLENGTH:48); i++)
+ {
+ if(i%100==0) printf("%.4d x %.4d\b\b\b\b\b\b\b\b\b\b\b", i, j);
+ if((bmpbuf=(unsigned char *)malloc(i*j*4))==NULL
+ || (jpgbuf=(unsigned char *)malloc(TJBUFSIZE(i, j)))==NULL)
+ {
+ printf("Memory allocation failure\n"); goto finally;
+ }
+ memset(bmpbuf, 0, i*j*4);
+ for(i2=0; i2<i*j; i2++)
+ {
+ bmpbuf[i2*4]=pixels[i2%9][2];
+ bmpbuf[i2*4+1]=pixels[i2%9][1];
+ bmpbuf[i2*2+2]=pixels[i2%9][0];
+ }
+ _catch(tjCompress(hnd, bmpbuf, i, i*4, j, 4,
+ jpgbuf, &size, TJ_444, 100, TJ_BGR));
+ free(bmpbuf); bmpbuf=NULL; free(jpgbuf); jpgbuf=NULL;
+
+ if((bmpbuf=(unsigned char *)malloc(j*i*4))==NULL
+ || (jpgbuf=(unsigned char *)malloc(TJBUFSIZE(j, i)))==NULL)
+ {
+ printf("Memory allocation failure\n"); goto finally;
+ }
+ for(i2=0; i2<j*i*4; i2++)
+ {
+ if(i2%2==0) bmpbuf[i2]=0xFF;
+ else bmpbuf[i2]=0;
+ }
+ _catch(tjCompress(hnd, bmpbuf, j, j*4, i, 4,
+ jpgbuf, &size, TJ_444, 100, TJ_BGR));
+ free(bmpbuf); bmpbuf=NULL; free(jpgbuf); jpgbuf=NULL;
+ }
+ }
+ printf("Done. \n");
+
+ finally:
+ if(bmpbuf) free(bmpbuf); if(jpgbuf) free(jpgbuf);
+ if(hnd) tjDestroy(hnd);
+}
+
+int main(int argc, char *argv[])
+{
+ dotest(35, 41, 3, TJ_444, "test");
+ dotest(35, 41, 4, TJ_444, "test");
+ dotest(35, 41, 3, TJ_GRAYSCALE, "test");
+ dotest(35, 41, 4, TJ_GRAYSCALE, "test");
+ dotest1();
+
+ return 0;
+}