ryan/mozjpeg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
109a578e89ea8cd2c39d50b012698148dd11dedb
mozjpeg/java/TJUnitTest.java
DRC 109a578e89 tjGetScaledSize() would never be able to accommodate scaling factors > 1, so replace it with a function that returns a list of fractional scaling factors that TurboJPEG supports. 
git-svn-id: svn+ssh://svn.code.sf.net/p/libjpeg-turbo/code/trunk@477 632fc199-4ca6-4c93-a231-07263d6284db
2011-03-01 09:53:07 +00:00

922 lines
24 KiB
Java
Raw Blame History

/*
* Copyright (C)2011 D. R. Commander. All Rights Reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* - Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* - Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
* - Neither the name of the libjpeg-turbo Project nor the names of its
* contributors may be used to endorse or promote products derived from this
* software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS",
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
/*
* This program tests the various code paths in the TurboJPEG JNI Wrapper
*/
import java.io.*;
import java.util.*;
import java.awt.image.*;
import javax.imageio.*;
import org.libjpegturbo.turbojpeg.*;
public class TJUnitTest {
private static final String classname=new TJUnitTest().getClass().getName();
private static void usage() {
System.out.println("\nUSAGE: java "+classname+" [options]\n");
System.out.println("Options:\n");
System.out.println("-yuv = test YUV encoding/decoding support\n");
System.out.println("-bi = test BufferedImage support\n");
System.exit(1);
}
private final static String _subnamel[]=
{"4:4:4", "4:2:2", "4:2:0", "GRAY"};
private final static String _subnames[]=
{"444", "422", "420", "GRAY"};
private final static int _hsf[]={1, 2, 2, 1};
private final static int _vsf[]={1, 1, 2, 1};
private final static String _pixformatstr[]=
{"RGB", "BGR", "RGBX", "BGRX", "XBGR", "XRGB", "Grayscale"};
private final static int biType[]=
{0, BufferedImage.TYPE_3BYTE_BGR, BufferedImage.TYPE_INT_BGR,
BufferedImage.TYPE_INT_RGB, 0, 0, BufferedImage.TYPE_BYTE_GRAY};
private final static int _3byteFormats[]=
{TJ.PF_RGB, TJ.PF_BGR};
private final static int _3byteFormatsBI[]=
{TJ.PF_BGR};
private final static int _4byteFormats[]=
{TJ.PF_RGBX, TJ.PF_BGRX, TJ.PF_XBGR, TJ.PF_XRGB};
private final static int _4byteFormatsBI[]=
{TJ.PF_RGBX, TJ.PF_BGRX};
private final static int _onlyGray[]=
{TJ.PF_GRAY};
private final static int _onlyRGB[]=
{TJ.PF_RGB};
private final static int YUVENCODE=1, YUVDECODE=2;
private static int yuv=0;
private static boolean bi=false;
private static int exitstatus=0;
private static double gettime()
{
return (double)System.nanoTime()/1.0e9;
}
private final static byte pixels[][]=
{
{0, (byte)255, 0},
{(byte)255, 0, (byte)255},
{(byte)255, (byte)255, 0},
{0, 0, (byte)255},
{0, (byte)255, (byte)255},
{(byte)255, 0, 0},
{(byte)255, (byte)255, (byte)255},
{0, 0, 0},
{(byte)255, 0, 0}
};
private static void initbuf(byte [] buf, int w, int pitch, int h, int pf,
int flags) throws Exception
{
int roffset=TJ.getRedShift(pf)/8;
int goffset=TJ.getGreenShift(pf)/8;
int boffset=TJ.getBlueShift(pf)/8;
int ps=TJ.getPixelSize(pf);
int i, _i, j;
Arrays.fill(buf, (byte)0);
if(pf==TJ.PF_GRAY)
{
for(_i=0; _i<16; _i++)
{
if((flags&TJ.BOTTOMUP)!=0) i=h-_i-1; else i=_i;
for(j=0; j<w; j++)
{
if(((_i/8)+(j/8))%2==0) buf[pitch*i+j]=(byte)255;
else buf[pitch*i+j]=76;
}
}
for(_i=16; _i<h; _i++)
{
if((flags&TJ.BOTTOMUP)!=0) i=h-_i-1; else i=_i;
for(j=0; j<w; j++)
{
if(((_i/8)+(j/8))%2==0) buf[pitch*i+j]=0;
else buf[pitch*i+j]=(byte)226;
}
}
return;
}
for(_i=0; _i<16; _i++)
{
if((flags&TJ.BOTTOMUP)!=0) i=h-_i-1; else i=_i;
for(j=0; j<w; j++)
{
buf[pitch*i+j*ps+roffset]=(byte)255;
if(((_i/8)+(j/8))%2==0)
{
buf[pitch*i+j*ps+goffset]=(byte)255;
buf[pitch*i+j*ps+boffset]=(byte)255;
}
}
}
for(_i=16; _i<h; _i++)
{
if((flags&TJ.BOTTOMUP)!=0) i=h-_i-1; else i=_i;
for(j=0; j<w; j++)
{
if(((_i/8)+(j/8))%2!=0)
{
buf[pitch*i+j*ps+roffset]=(byte)255;
buf[pitch*i+j*ps+goffset]=(byte)255;
}
}
}
}
private static void initintbuf(int [] buf, int w, int pitch, int h, int pf,
int flags) throws Exception
{
int rshift=TJ.getRedShift(pf);
int gshift=TJ.getGreenShift(pf);
int bshift=TJ.getBlueShift(pf);
int i, _i, j;
Arrays.fill(buf, 0);
for(_i=0; _i<16; _i++)
{
if((flags&TJ.BOTTOMUP)!=0) i=h-_i-1; else i=_i;
for(j=0; j<w; j++)
{
buf[pitch*i+j]=(255 << rshift);
if(((_i/8)+(j/8))%2==0)
{
buf[pitch*i+j]|=(255 << gshift);
buf[pitch*i+j]|=(255 << bshift);
}
}
}
for(_i=16; _i<h; _i++)
{
if((flags&TJ.BOTTOMUP)!=0) i=h-_i-1; else i=_i;
for(j=0; j<w; j++)
{
if(((_i/8)+(j/8))%2!=0)
{
buf[pitch*i+j]=(255 << rshift);
buf[pitch*i+j]|=(255 << gshift);
}
}
}
}
private static void initimg(BufferedImage img, int pf, int flags)
throws Exception
{
WritableRaster wr=img.getRaster();
int imgtype=img.getType();
if(imgtype==BufferedImage.TYPE_INT_RGB
|| imgtype==BufferedImage.TYPE_INT_BGR)
{
SinglePixelPackedSampleModel sm=
(SinglePixelPackedSampleModel)img.getSampleModel();
int pitch=sm.getScanlineStride();
DataBufferInt db=(DataBufferInt)wr.getDataBuffer();
int [] buf = db.getData();
initintbuf(buf, img.getWidth(), pitch, img.getHeight(), pf, flags);
}
else
{
ComponentSampleModel sm=
(ComponentSampleModel)img.getSampleModel();
int pitch=sm.getScanlineStride();
DataBufferByte db=(DataBufferByte)wr.getDataBuffer();
byte [] buf = db.getData();
initbuf(buf, img.getWidth(), pitch, img.getHeight(), pf, flags);
}
}
private static void checkval(int i, int j, int v, String vname, int cv)
throws Exception
{
v=(v<0)? v+256:v;
if(v<cv-1 || v>cv+1)
{
throw new Exception("\nComp. "+vname+" at "+i+","+j+" should be "+cv
+", not "+v+"\n");
}
}
private static void checkval0(int i, int j, int v, String vname)
throws Exception
{
v=(v<0)? v+256:v;
if(v>1)
{
throw new Exception("\nComp. "+vname+" at "+i+","+j+" should be 0, not "
+v+"\n");
}
}
private static void checkval255(int i, int j, int v, String vname)
throws Exception
{
v=(v<0)? v+256:v;
if(v<254 && !(v==217 && i==0 && j==21))
{
throw new Exception("\nComp. "+vname+" at "+i+","+j+" should be 255, not "
+v+"\n");
}
}
private static int checkbuf(byte [] buf, int w, int pitch, int h, int pf,
int subsamp, int scale_num, int scale_denom, int flags) throws Exception
{
int roffset=TJ.getRedShift(pf)/8;
int goffset=TJ.getGreenShift(pf)/8;
int boffset=TJ.getBlueShift(pf)/8;
int ps=TJ.getPixelSize(pf);
int i, _i, j, retval=1;
int halfway=16*scale_num/scale_denom, blocksize=8*scale_num/scale_denom;
try
{
for(_i=0; _i<halfway; _i++)
{
if((flags&TJ.BOTTOMUP)!=0) i=h-_i-1; else i=_i;
for(j=0; j<w; j++)
{
byte r=buf[pitch*i+j*ps+roffset], g=buf[pitch*i+j*ps+goffset],
b=buf[pitch*i+j*ps+boffset];
if(((_i/blocksize)+(j/blocksize))%2==0)
{
checkval255(_i, j, r, "R");
checkval255(_i, j, g, "G");
checkval255(_i, j, b, "B");
}
else
{
if(subsamp==TJ.SAMP_GRAY)
{
checkval(_i, j, r, "R", 76);
checkval(_i, j, g, "G", 76);
checkval(_i, j, b, "B", 76);
}
else
{
checkval255(_i, j, r, "R");
checkval0(_i, j, g, "G");
checkval0(_i, j, b, "B");
}
}
}
}
for(_i=halfway; _i<h; _i++)
{
if((flags&TJ.BOTTOMUP)!=0) i=h-_i-1; else i=_i;
for(j=0; j<w; j++)
{
byte r=buf[pitch*i+j*ps+roffset], g=buf[pitch*i+j*ps+goffset],
b=buf[pitch*i+j*ps+boffset];
if(((_i/blocksize)+(j/blocksize))%2==0)
{
checkval0(_i, j, r, "R");
checkval0(_i, j, g, "G");
}
else
{
if(subsamp==TJ.SAMP_GRAY)
{
checkval(_i, j, r, "R", 226);
checkval(_i, j, g, "G", 226);
checkval(_i, j, b, "B", 226);
}
else
{
checkval255(_i, j, r, "R");
checkval255(_i, j, g, "G");
checkval0(_i, j, b, "B");
}
}
}
}
}
catch(Exception e)
{
System.out.println(e);
retval=0;
}
if(retval==0)
{
System.out.print("\n");
for(i=0; i<h; i++)
{
for(j=0; j<w; j++)
{
int r=buf[pitch*i+j*ps+roffset];
int g=buf[pitch*i+j*ps+goffset];
int b=buf[pitch*i+j*ps+boffset];
if(r<0) r+=256; if(g<0) g+=256; if(b<0) b+=256;
System.out.format("%3d/%3d/%3d ", r, g, b);
}
System.out.print("\n");
}
}
return retval;
}
private static int checkintbuf(int [] buf, int w, int pitch, int h, int pf,
int subsamp, int scale_num, int scale_denom, int flags) throws Exception
{
int rshift=TJ.getRedShift(pf);
int gshift=TJ.getGreenShift(pf);
int bshift=TJ.getBlueShift(pf);
int i, _i, j, retval=1;
int halfway=16*scale_num/scale_denom, blocksize=8*scale_num/scale_denom;
try
{
for(_i=0; _i<halfway; _i++)
{
if((flags&TJ.BOTTOMUP)!=0) i=h-_i-1; else i=_i;
for(j=0; j<w; j++)
{
int r=(buf[pitch*i+j] >> rshift) & 0xFF;
int g=(buf[pitch*i+j] >> gshift) & 0xFF;
int b=(buf[pitch*i+j] >> bshift) & 0xFF;
if(((_i/blocksize)+(j/blocksize))%2==0)
{
checkval255(_i, j, r, "R");
checkval255(_i, j, g, "G");
checkval255(_i, j, b, "B");
}
else
{
if(subsamp==TJ.SAMP_GRAY)
{
checkval(_i, j, r, "R", 76);
checkval(_i, j, g, "G", 76);
checkval(_i, j, b, "B", 76);
}
else
{
checkval255(_i, j, r, "R");
checkval0(_i, j, g, "G");
checkval0(_i, j, b, "B");
}
}
}
}
for(_i=halfway; _i<h; _i++)
{
if((flags&TJ.BOTTOMUP)!=0) i=h-_i-1; else i=_i;
for(j=0; j<w; j++)
{
int r=(buf[pitch*i+j] >> rshift) & 0xFF;
int g=(buf[pitch*i+j] >> gshift) & 0xFF;
int b=(buf[pitch*i+j] >> bshift) & 0xFF;
if(((_i/blocksize)+(j/blocksize))%2==0)
{
checkval0(_i, j, r, "R");
checkval0(_i, j, g, "G");
}
else
{
if(subsamp==TJ.SAMP_GRAY)
{
checkval(_i, j, r, "R", 226);
checkval(_i, j, g, "G", 226);
checkval(_i, j, b, "B", 226);
}
else
{
checkval255(_i, j, r, "R");
checkval255(_i, j, g, "G");
checkval0(_i, j, b, "B");
}
}
}
}
}
catch(Exception e)
{
System.out.println(e);
retval=0;
}
if(retval==0)
{
System.out.print("\n");
for(i=0; i<h; i++)
{
for(j=0; j<w; j++)
{
int r=(buf[pitch*i+j] >> rshift) & 0xFF;
int g=(buf[pitch*i+j] >> gshift) & 0xFF;
int b=(buf[pitch*i+j] >> bshift) & 0xFF;
if(r<0) r+=256; if(g<0) g+=256; if(b<0) b+=256;
System.out.format("%3d/%3d/%3d ", r, g, b);
}
System.out.print("\n");
}
}
return retval;
}
private static int checkimg(BufferedImage img, int pf,
int subsamp, int scale_num, int scale_denom, int flags) throws Exception
{
WritableRaster wr=img.getRaster();
int imgtype=img.getType();
if(imgtype==BufferedImage.TYPE_INT_RGB
|| imgtype==BufferedImage.TYPE_INT_BGR)
{
SinglePixelPackedSampleModel sm=
(SinglePixelPackedSampleModel)img.getSampleModel();
int pitch=sm.getScanlineStride();
DataBufferInt db=(DataBufferInt)wr.getDataBuffer();
int [] buf = db.getData();
return checkintbuf(buf, img.getWidth(), pitch, img.getHeight(), pf,
subsamp, scale_num, scale_denom, flags);
}
else
{
ComponentSampleModel sm=
(ComponentSampleModel)img.getSampleModel();
int pitch=sm.getScanlineStride();
DataBufferByte db=(DataBufferByte)wr.getDataBuffer();
byte [] buf = db.getData();
return checkbuf(buf, img.getWidth(), pitch, img.getHeight(), pf, subsamp,
scale_num, scale_denom, flags);
}
}
private static int PAD(int v, int p)
{
return ((v+(p)-1)&(~((p)-1)));
}
private static int checkbufyuv(byte [] buf, int size, int w, int h,
int subsamp)
{
int i, j;
int hsf=_hsf[subsamp], vsf=_vsf[subsamp];
int pw=PAD(w, hsf), ph=PAD(h, vsf);
int cw=pw/hsf, ch=ph/vsf;
int ypitch=PAD(pw, 4), uvpitch=PAD(cw, 4);
int retval=1;
int correctsize=ypitch*ph + (subsamp==TJ.SAMP_GRAY? 0:uvpitch*ch*2);
try
{
if(size!=correctsize)
throw new Exception("\nIncorrect size "+size+". Should be "
+correctsize);
for(i=0; i<16; i++)
{
for(j=0; j<pw; j++)
{
byte y=buf[ypitch*i+j];
if(((i/8)+(j/8))%2==0) checkval255(i, j, y, "Y");
else checkval(i, j, y, "Y", 76);
}
}
for(i=16; i<ph; i++)
{
for(j=0; j<pw; j++)
{
byte y=buf[ypitch*i+j];
if(((i/8)+(j/8))%2==0) checkval0(i, j, y, "Y");
else checkval(i, j, y, "Y", 226);
}
}
if(subsamp!=TJ.SAMP_GRAY)
{
for(i=0; i<16/vsf; i++)
{
for(j=0; j<cw; j++)
{
byte u=buf[ypitch*ph + (uvpitch*i+j)],
v=buf[ypitch*ph + uvpitch*ch + (uvpitch*i+j)];
if(((i*vsf/8)+(j*hsf/8))%2==0)
{
checkval(i, j, u, "U", 128); checkval(i, j, v, "V", 128);
}
else
{
checkval(i, j, u, "U", 85); checkval255(i, j, v, "V");
}
}
}
for(i=16/vsf; i<ch; i++)
{
for(j=0; j<cw; j++)
{
byte u=buf[ypitch*ph + (uvpitch*i+j)],
v=buf[ypitch*ph + uvpitch*ch + (uvpitch*i+j)];
if(((i*vsf/8)+(j*hsf/8))%2==0)
{
checkval(i, j, u, "U", 128); checkval(i, j, v, "V", 128);
}
else
{
checkval0(i, j, u, "U"); checkval(i, j, v, "V", 149);
}
}
}
}
}
catch(Exception e)
{
System.out.println(e);
retval=0;
}
if(retval==0)
{
for(i=0; i<ph; i++)
{
for(j=0; j<pw; j++)
{
int y=buf[ypitch*i+j];
if(y<0) y+=256;
System.out.format("%3d ", y);
}
System.out.print("\n");
}
System.out.print("\n");
for(i=0; i<ch; i++)
{
for(j=0; j<cw; j++)
{
int u=buf[ypitch*ph + (uvpitch*i+j)];
if(u<0) u+=256;
System.out.format("%3d ", u);
}
System.out.print("\n");
}
System.out.print("\n");
for(i=0; i<ch; i++)
{
for(j=0; j<cw; j++)
{
int v=buf[ypitch*ph + uvpitch*ch + (uvpitch*i+j)];
if(v<0) v+=256;
System.out.format("%3d ", v);
}
System.out.print("\n");
}
System.out.print("\n");
}
return retval;
}
private static void writejpeg(byte [] jpegbuf, int jpgbufsize,
String filename) throws Exception
{
File file=new File(filename);
FileOutputStream fos=new FileOutputStream(file);
fos.write(jpegbuf, 0, jpgbufsize);
fos.close();
}
private static int gentestjpeg(TJCompressor tjc, byte [] jpegbuf, int w,
int h, int pf, String basefilename, int subsamp, int qual,
int flags) throws Exception
{
String tempstr; byte [] bmpbuf=null; BufferedImage img=null;
String pixformat; double t;
int size=0, ps=TJ.getPixelSize(pf);
pixformat=_pixformatstr[pf];
System.out.print(pixformat+" ");
if((flags&TJ.BOTTOMUP)!=0) System.out.print("Bottom-Up");
else System.out.print("Top-Down ");
System.out.print(" -> "+_subnamel[subsamp]+" ");
if(yuv==YUVENCODE) System.out.print("YUV ... ");
else System.out.print("Q"+qual+" ... ");
if(bi)
{
img=new BufferedImage(w, h, biType[pf]);
initimg(img, pf, flags);
tempstr=basefilename+"_enc_"+pixformat+"_"
+(((flags&TJ.BOTTOMUP)!=0)? "BU":"TD")+"_"+_subnames[subsamp]
+"_Q"+qual+".png";
File file=new File(tempstr);
ImageIO.write(img, "png", file);
}
else
{
bmpbuf=new byte[w*h*ps+1];
initbuf(bmpbuf, w, w*ps, h, pf, flags);
}
Arrays.fill(jpegbuf, (byte)0);
t=gettime();
tjc.setSubsamp(subsamp);
tjc.setJPEGQuality(qual);
if(bi)
{
if(yuv==YUVENCODE) tjc.encodeYUV(img, jpegbuf, flags);
else tjc.compress(img, jpegbuf, flags);
}
else
{
tjc.setBitmapBuffer(bmpbuf, w, 0, h, pf);
if(yuv==YUVENCODE) tjc.encodeYUV(jpegbuf, flags);
else tjc.compress(jpegbuf, flags);
}
size=tjc.getCompressedSize();
t=gettime()-t;
if(yuv==YUVENCODE)
tempstr=basefilename+"_enc_"+pixformat+"_"
+(((flags&TJ.BOTTOMUP)!=0)? "BU":"TD")+"_"+_subnames[subsamp]+".yuv";
else
tempstr=basefilename+"_enc_"+pixformat+"_"
+(((flags&TJ.BOTTOMUP)!=0)? "BU":"TD")+"_"+_subnames[subsamp]
+"_Q"+qual+".jpg";
writejpeg(jpegbuf, size, tempstr);
if(yuv==YUVENCODE)
{
if(checkbufyuv(jpegbuf, size, w, h, subsamp)==1)
System.out.print("Passed.");
else {System.out.print("FAILED!"); exitstatus=-1;}
}
else System.out.print("Done.");
System.out.format(" %.6f ms\n", t*1000.);
System.out.println(" Result in "+tempstr);
return size;
}
private static void _gentestbmp(TJDecompressor tjd, byte [] jpegbuf,
int jpegsize, int w, int h, int pf, String basefilename, int subsamp,
int flags, int scale_num, int scale_denom) throws Exception
{
String pixformat, tempstr; int _hdrw=0, _hdrh=0, _hdrsubsamp=-1;
double t;
int scaledw=(w*scale_num+scale_denom-1)/scale_denom;
int scaledh=(h*scale_num+scale_denom-1)/scale_denom;
int temp1, temp2;
BufferedImage img=null; byte [] bmpbuf=null;
if(yuv==YUVENCODE) return;
pixformat=_pixformatstr[pf];
System.out.print("JPEG -> ");
if(yuv==YUVDECODE)
System.out.print("YUV "+_subnames[subsamp]+" ... ");
else
{
System.out.print(pixformat+" ");
if((flags&TJ.BOTTOMUP)!=0) System.out.print("Bottom-Up ");
else System.out.print("Top-Down ");
if(scale_num!=1 || scale_denom!=1)
System.out.print(scale_num+"/"+scale_denom+" ... ");
else System.out.print("... ");
}
t=gettime();
tjd.setJPEGBuffer(jpegbuf, jpegsize);
if(tjd.getWidth()!=w || tjd.getHeight()!=h || tjd.getSubsamp()!=subsamp)
throw new Exception("Incorrect JPEG header");
temp1=scaledw; temp2=scaledh;
temp1=tjd.getScaledWidth(temp1, temp2);
temp2=tjd.getScaledHeight(temp1, temp2);
if(temp1!=scaledw || temp2!=scaledh)
throw new Exception("Scaled size mismatch");
if(yuv==YUVDECODE) bmpbuf=tjd.decompressToYUV(flags);
else
{
if(bi) img=tjd.decompress(scaledw, scaledh, biType[pf], flags);
else bmpbuf=tjd.decompress(scaledw, 0, scaledh, pf, flags);
}
t=gettime()-t;
if(bi)
{
tempstr=basefilename+"_dec_"+pixformat+"_"
+(((flags&TJ.BOTTOMUP)!=0)? "BU":"TD")+"_"+_subnames[subsamp]
+"_"+(double)scale_num/(double)scale_denom+"x"+".png";
File file=new File(tempstr);
ImageIO.write(img, "png", file);
}
if(yuv==YUVDECODE)
{
if(checkbufyuv(bmpbuf, bmpbuf.length, w, h, subsamp)==1)
System.out.print("Passed.");
else {System.out.print("FAILED!"); exitstatus=-1;}
}
else
{
if((bi && checkimg(img, pf, subsamp, scale_num, scale_denom, flags)==1)
|| (!bi && checkbuf(bmpbuf, scaledw, scaledw*TJ.getPixelSize(pf),
scaledh, pf, subsamp, scale_num, scale_denom, flags)==1))
System.out.print("Passed.");
else
{
System.out.print("FAILED!"); exitstatus=-1;
}
}
System.out.format(" %.6f ms\n", t*1000.);
}
private static void gentestbmp(TJDecompressor tjd, byte [] jpegbuf,
int jpegsize, int w, int h, int pf, String basefilename, int subsamp,
int flags) throws Exception
{
int i;
if((subsamp==TJ.SAMP_444 || subsamp==TJ.SAMP_GRAY) && yuv==0)
{
TJ.ScalingFactor sf []=TJ.getScalingFactors();
for(i=0; i<sf.length; i++)
_gentestbmp(tjd, jpegbuf, jpegsize, w, h, pf, basefilename, subsamp,
flags, sf[i].num, sf[i].denom);
}
else
_gentestbmp(tjd, jpegbuf, jpegsize, w, h, pf, basefilename, subsamp,
flags, 1, 1);
System.out.print("\n");
}
private static void dotest(int w, int h, int [] formats, int subsamp,
String basefilename) throws Exception
{
TJCompressor tjc=null; TJDecompressor tjd=null;
int size; int pfstart, pfend;
byte [] jpegbuf;
if(yuv==YUVENCODE) jpegbuf=new byte[TJ.bufSizeYUV(w, h, subsamp)];
else jpegbuf=new byte[TJ.bufSize(w, h)];
try
{
tjc=new TJCompressor();
tjd=new TJDecompressor();
for(int pf : formats)
{
for(int i=0; i<2; i++)
{
int flags=0;
if(i==1)
{
if(yuv==YUVDECODE)
{
tjc.close(); tjd.close(); return;
}
else flags|=TJ.BOTTOMUP;
}
size=gentestjpeg(tjc, jpegbuf, w, h, pf, basefilename, subsamp, 100,
flags);
gentestbmp(tjd, jpegbuf, size, w, h, pf, basefilename, subsamp,
flags);
}
}
}
catch(Exception e)
{
if(tjc!=null) tjc.close();
if(tjd!=null) tjd.close();
throw e;
}
if(tjc!=null) tjc.close();
if(tjd!=null) tjd.close();
}
private final static int MAXLENGTH=2048;
private static void dotest1() throws Exception
{
int i, j, i2; byte [] bmpbuf, jpgbuf;
TJCompressor tjc=null;
try
{
tjc=new TJCompressor();
System.out.println("Buffer size regression test");
for(j=1; j<48; j++)
{
for(i=1; i<(j==1?MAXLENGTH:48); i++)
{
if(i%100==0) System.out.format("%04d x %04d\b\b\b\b\b\b\b\b\b\b\b",
i, j);
bmpbuf=new byte[i*j*4];
jpgbuf=new byte[TJ.bufSize(i, j)];
Arrays.fill(bmpbuf, (byte)0);
for(i2=0; i2<i*j; i2++)
{
bmpbuf[i2*4]=pixels[i2%9][2];
bmpbuf[i2*4+1]=pixels[i2%9][1];
bmpbuf[i2*4+2]=pixels[i2%9][0];
}
tjc.setBitmapBuffer(bmpbuf, i, 0, j, TJ.PF_BGRX);
tjc.setSubsamp(TJ.SAMP_444);
tjc.setJPEGQuality(100);
tjc.compress(jpgbuf, 0);
bmpbuf=new byte[j*i*4];
jpgbuf=new byte[TJ.bufSize(j, i)];
for(i2=0; i2<j*i; i2++)
{
if(i2%2==0) bmpbuf[i2*4]=bmpbuf[i2*4+1]=bmpbuf[i2*4+2]=(byte)0xFF;
else bmpbuf[i2*4]=bmpbuf[i2*4+1]=bmpbuf[i2*4+2]=0;
}
tjc.setBitmapBuffer(bmpbuf, j, 0, i, TJ.PF_BGRX);
tjc.compress(jpgbuf, 0);
}
}
System.out.println("Done. ");
}
catch(Exception e)
{
if(tjc!=null) tjc.close();
throw e;
}
if(tjc!=null) tjc.close();
}
public static void main(String argv[])
{
try
{
boolean doyuv=false;
for(int i=0; i<argv.length; i++)
{
if(argv[i].equalsIgnoreCase("-yuv")) doyuv=true;
if(argv[i].substring(0, 1).equalsIgnoreCase("-h")
|| argv[i].equalsIgnoreCase("-?"))
usage();
if(argv[i].equalsIgnoreCase("-bi")) bi=true;
}
if(doyuv) yuv=YUVENCODE;
dotest(35, 39, bi? _3byteFormatsBI:_3byteFormats, TJ.SAMP_444, "test");
dotest(39, 41, bi? _4byteFormatsBI:_4byteFormats, TJ.SAMP_444, "test");
if(doyuv)
{
dotest(41, 35, bi? _3byteFormatsBI:_3byteFormats, TJ.SAMP_422, "test");
dotest(35, 39, bi? _4byteFormatsBI:_4byteFormats, TJ.SAMP_422, "test");
dotest(39, 41, bi? _3byteFormatsBI:_3byteFormats, TJ.SAMP_420, "test");
dotest(41, 35, bi? _4byteFormatsBI:_4byteFormats, TJ.SAMP_420, "test");
}
dotest(35, 39, _onlyGray, TJ.SAMP_GRAY, "test");
dotest(39, 41, bi? _3byteFormatsBI:_3byteFormats, TJ.SAMP_GRAY, "test");
dotest(41, 35, bi? _4byteFormatsBI:_4byteFormats, TJ.SAMP_GRAY, "test");
if(!doyuv && !bi) dotest1();
if(doyuv && !bi)
{
yuv=YUVDECODE;
dotest(48, 48, _onlyRGB, TJ.SAMP_444, "test");
dotest(35, 39, _onlyRGB, TJ.SAMP_444, "test");
dotest(48, 48, _onlyRGB, TJ.SAMP_422, "test");
dotest(39, 41, _onlyRGB, TJ.SAMP_422, "test");
dotest(48, 48, _onlyRGB, TJ.SAMP_420, "test");
dotest(41, 35, _onlyRGB, TJ.SAMP_420, "test");
dotest(48, 48, _onlyRGB, TJ.SAMP_GRAY, "test");
dotest(35, 39, _onlyRGB, TJ.SAMP_GRAY, "test");
dotest(48, 48, _onlyGray, TJ.SAMP_GRAY, "test");
dotest(39, 41, _onlyGray, TJ.SAMP_GRAY, "test");
}
}
catch(Exception e)
{
System.out.println(e);
exitstatus=-1;
}
System.exit(exitstatus);
}
}
Reference in New Issue View Git Blame Copy Permalink