breeze/libjpeg-turbo
1
0
Fork 0
mirror of https://github.com/libjpeg-turbo/libjpeg-turbo.git synced 2026-01-12 00:04:17 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
2858783db53540804094801bc8cb4bef18fd3e96
libjpeg-turbo/java/TJUnitTest.java

1413 lines
51 KiB
Java
Raw Blame History

/*
* Copyright (C)2011-2018, 2022-2024 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 java.nio.*;
import java.nio.file.*;
import org.libjpegturbo.turbojpeg.*;
@SuppressWarnings("checkstyle:JavadocType")
final class TJUnitTest {
private TJUnitTest() {}
static final String CLASS_NAME =
new TJUnitTest().getClass().getName();
static void usage() {
System.out.println("\nUSAGE: java " + CLASS_NAME + " [options]\n");
System.out.println("Options:");
System.out.println("-yuv = test YUV encoding/compression/decompression/decoding");
System.out.println(" (8-bit data precision only)");
System.out.println("-noyuvpad = do not pad each row in each Y, U, and V plane to the nearest");
System.out.println(" multiple of 4 bytes");
System.out.println("-precision N = test N-bit data precision (N=2..16; default is 8; if N is not 8");
System.out.println(" or 12, then -lossless is implied)");
System.out.println("-lossless = test lossless JPEG compression/decompression");
System.out.println("-bi = test BufferedImage I/O (8-bit data precision only)\n");
System.exit(1);
}
static final String[] SUBNAME_LONG = {
"4:4:4", "4:2:2", "4:2:0", "GRAY", "4:4:0", "4:1:1", "4:4:1"
};
static final String[] SUBNAME = {
"444", "422", "420", "GRAY", "440", "411", "441"
};
static final String[] PIXFORMATSTR = {
"RGB", "BGR", "RGBX", "BGRX", "XBGR", "XRGB", "Grayscale",
"RGBA", "BGRA", "ABGR", "ARGB", "CMYK"
};
static final int[] FORMATS_3SAMPLE = {
TJ.PF_RGB, TJ.PF_BGR
};
static final int[] FORMATS_3BYTEBI = {
BufferedImage.TYPE_3BYTE_BGR
};
static final int[] FORMATS_4SAMPLE = {
TJ.PF_RGBX, TJ.PF_BGRX, TJ.PF_XBGR, TJ.PF_XRGB, TJ.PF_CMYK
};
static final int[] FORMATS_4BYTEBI = {
BufferedImage.TYPE_INT_BGR, BufferedImage.TYPE_INT_RGB,
BufferedImage.TYPE_4BYTE_ABGR, BufferedImage.TYPE_4BYTE_ABGR_PRE,
BufferedImage.TYPE_INT_ARGB, BufferedImage.TYPE_INT_ARGB_PRE
};
static final int[] FORMATS_GRAY = {
TJ.PF_GRAY
};
static final int[] FORMATS_GRAYBI = {
BufferedImage.TYPE_BYTE_GRAY
};
static final int[] FORMATS_RGB = {
TJ.PF_RGB
};
private static boolean doYUV = false;
private static boolean lossless = false;
private static int psv = 1;
private static int yuvAlign = 4;
private static int precision = 8;
private static int sampleSize, maxSample, tolerance, redToY, yellowToY;
private static boolean bi = false;
private static int exitStatus = 0;
private static String uniqueID =
java.util.UUID.randomUUID().toString().replace("-", "");
static int biTypePF(int biType) {
ByteOrder byteOrder = ByteOrder.nativeOrder();
switch (biType) {
case BufferedImage.TYPE_3BYTE_BGR:
return TJ.PF_BGR;
case BufferedImage.TYPE_4BYTE_ABGR:
case BufferedImage.TYPE_4BYTE_ABGR_PRE:
return TJ.PF_ABGR;
case BufferedImage.TYPE_BYTE_GRAY:
return TJ.PF_GRAY;
case BufferedImage.TYPE_INT_BGR:
return TJ.PF_RGBX;
case BufferedImage.TYPE_INT_RGB:
return TJ.PF_BGRX;
case BufferedImage.TYPE_INT_ARGB:
case BufferedImage.TYPE_INT_ARGB_PRE:
return TJ.PF_BGRA;
default:
return 0;
}
}
static String biTypeStr(int biType) {
switch (biType) {
case BufferedImage.TYPE_3BYTE_BGR:
return "3BYTE_BGR";
case BufferedImage.TYPE_4BYTE_ABGR:
return "4BYTE_ABGR";
case BufferedImage.TYPE_4BYTE_ABGR_PRE:
return "4BYTE_ABGR_PRE";
case BufferedImage.TYPE_BYTE_GRAY:
return "BYTE_GRAY";
case BufferedImage.TYPE_INT_BGR:
return "INT_BGR";
case BufferedImage.TYPE_INT_RGB:
return "INT_RGB";
case BufferedImage.TYPE_INT_ARGB:
return "INT_ARGB";
case BufferedImage.TYPE_INT_ARGB_PRE:
return "INT_ARGB_PRE";
default:
return "Unknown";
}
}
static void fillArray(Object buf, int val) {
if (precision <= 8)
Arrays.fill((byte[])buf, (byte)val);
else
Arrays.fill((short[])buf, (short)val);
}
static void setVal(Object buf, int index, int value) {
if (precision <= 8)
((byte[])buf)[index] = (byte)value;
else
((short[])buf)[index] = (short)value;
}
static void initBuf(Object buf, int w, int pitch, int h, int pf,
boolean bottomUp) throws Exception {
int roffset = TJ.getRedOffset(pf);
int goffset = TJ.getGreenOffset(pf);
int boffset = TJ.getBlueOffset(pf);
int aoffset = TJ.getAlphaOffset(pf);
int ps = TJ.getPixelSize(pf);
int index, row, col, halfway = 16;
if (pf == TJ.PF_GRAY) {
fillArray(buf, 0);
for (row = 0; row < h; row++) {
for (col = 0; col < w; col++) {
if (bottomUp)
index = pitch * (h - row - 1) + col;
else
index = pitch * row + col;
if (((row / 8) + (col / 8)) % 2 == 0)
setVal(buf, index, (row < halfway) ? maxSample : 0);
else
setVal(buf, index, (row < halfway) ? redToY : yellowToY);
}
}
return;
}
if (pf == TJ.PF_CMYK) {
fillArray(buf, maxSample);
for (row = 0; row < h; row++) {
for (col = 0; col < w; col++) {
if (bottomUp)
index = (h - row - 1) * w + col;
else
index = row * w + col;
if (((row / 8) + (col / 8)) % 2 == 0) {
if (row >= halfway) setVal(buf, index * ps + 3, 0);
} else {
setVal(buf, index * ps + 2, 0);
if (row < halfway)
setVal(buf, index * ps + 1, 0);
}
}
}
return;
}
fillArray(buf, 0);
for (row = 0; row < h; row++) {
for (col = 0; col < w; col++) {
if (bottomUp)
index = pitch * (h - row - 1) + col * ps;
else
index = pitch * row + col * ps;
if (((row / 8) + (col / 8)) % 2 == 0) {
if (row < halfway) {
setVal(buf, index + roffset, maxSample);
setVal(buf, index + goffset, maxSample);
setVal(buf, index + boffset, maxSample);
}
} else {
setVal(buf, index + roffset, maxSample);
if (row >= halfway)
setVal(buf, index + goffset, maxSample);
}
if (aoffset >= 0)
setVal(buf, index + aoffset, maxSample);
}
}
}
static void initIntBuf(int[] buf, int w, int pitch, int h, int pf,
boolean bottomUp) throws Exception {
int rshift = TJ.getRedOffset(pf) * 8;
int gshift = TJ.getGreenOffset(pf) * 8;
int bshift = TJ.getBlueOffset(pf) * 8;
int ashift = TJ.getAlphaOffset(pf) * 8;
int index, row, col, halfway = 16;
Arrays.fill(buf, 0);
for (row = 0; row < h; row++) {
for (col = 0; col < w; col++) {
if (bottomUp)
index = pitch * (h - row - 1) + col;
else
index = pitch * row + col;
if (((row / 8) + (col / 8)) % 2 == 0) {
if (row < halfway) {
buf[index] |= (255 << rshift);
buf[index] |= (255 << gshift);
buf[index] |= (255 << bshift);
}
} else {
buf[index] |= (255 << rshift);
if (row >= halfway)
buf[index] |= (255 << gshift);
}
if (ashift >= 0)
buf[index] |= (255 << ashift);
}
}
}
static void initImg(BufferedImage img, int pf, boolean bottomUp)
throws Exception {
WritableRaster wr = img.getRaster();
int imgType = img.getType();
if (imgType == BufferedImage.TYPE_INT_RGB ||
imgType == BufferedImage.TYPE_INT_BGR ||
imgType == BufferedImage.TYPE_INT_ARGB ||
imgType == BufferedImage.TYPE_INT_ARGB_PRE) {
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, bottomUp);
} 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, bottomUp);
}
}
static void checkVal(int row, int col, int v, String vname, int cv)
throws Exception {
v = (v < 0) ? v + 256 : v;
if (v < cv - tolerance || v > cv + tolerance) {
throw new Exception("Comp. " + vname + " at " + row + "," + col +
" should be " + cv + ", not " + v);
}
}
static void checkVal0(int row, int col, int v, String vname)
throws Exception {
v = (v < 0) ? v + 256 : v;
if (v > tolerance) {
throw new Exception("Comp. " + vname + " at " + row + "," + col +
" should be 0, not " + v);
}
}
static void checkValMax(int row, int col, int v, String vname)
throws Exception {
v = (v < 0) ? v + 256 : v;
if (v < maxSample - tolerance) {
throw new Exception("Comp. " + vname + " at " + row + "," + col +
" should be " + maxSample + ", not " + v);
}
}
static int getVal(Object buf, int index) {
int v;
if (precision <= 8)
v = (int)(((byte[])buf)[index]);
else
v = (int)(((short[])buf)[index]);
if (v < 0)
v += maxSample + 1;
return v;
}
static boolean checkBuf(Object buf, int w, int pitch, int h, int pf,
int subsamp, TJScalingFactor sf, boolean bottomUp)
throws Exception {
int roffset = TJ.getRedOffset(pf);
int goffset = TJ.getGreenOffset(pf);
int boffset = TJ.getBlueOffset(pf);
int aoffset = TJ.getAlphaOffset(pf);
int ps = TJ.getPixelSize(pf);
int index, row, col;
boolean retval = true;
int halfway = 16 * sf.getNum() / sf.getDenom();
int blockSize = 8 * sf.getNum() / sf.getDenom();
try {
if (pf == TJ.PF_GRAY)
roffset = goffset = boffset = 0;
if (pf == TJ.PF_CMYK) {
for (row = 0; row < h; row++) {
for (col = 0; col < w; col++) {
if (bottomUp)
index = (h - row - 1) * w + col;
else
index = row * w + col;
int c = getVal(buf, index * ps);
int m = getVal(buf, index * ps + 1);
int y = getVal(buf, index * ps + 2);
int k = getVal(buf, index * ps + 3);
checkValMax(row, col, c, "C");
if (((row / blockSize) + (col / blockSize)) % 2 == 0) {
checkValMax(row, col, m, "M");
checkValMax(row, col, y, "Y");
if (row < halfway)
checkValMax(row, col, k, "K");
else
checkVal0(row, col, k, "K");
} else {
checkVal0(row, col, y, "Y");
checkValMax(row, col, k, "K");
if (row < halfway)
checkVal0(row, col, m, "M");
else
checkValMax(row, col, m, "M");
}
}
}
return true;
}
for (row = 0; row < halfway; row++) {
for (col = 0; col < w; col++) {
if (bottomUp)
index = pitch * (h - row - 1) + col * ps;
else
index = pitch * row + col * ps;
int r = getVal(buf, index + roffset);
int g = getVal(buf, index + goffset);
int b = getVal(buf, index + boffset);
int a = aoffset >= 0 ? getVal(buf, index + aoffset) : maxSample;
if (((row / blockSize) + (col / blockSize)) % 2 == 0) {
if (row < halfway) {
checkValMax(row, col, r, "R");
checkValMax(row, col, g, "G");
checkValMax(row, col, b, "B");
} else {
checkVal0(row, col, r, "R");
checkVal0(row, col, g, "G");
checkVal0(row, col, b, "B");
}
} else {
if (subsamp == TJ.SAMP_GRAY) {
if (row < halfway) {
checkVal(row, col, r, "R", redToY);
checkVal(row, col, g, "G", redToY);
checkVal(row, col, b, "B", redToY);
} else {
checkVal(row, col, r, "R", yellowToY);
checkVal(row, col, g, "G", yellowToY);
checkVal(row, col, b, "B", yellowToY);
}
} else {
checkValMax(row, col, r, "R");
if (row < halfway) {
checkVal0(row, col, g, "G");
} else {
checkValMax(row, col, g, "G");
}
checkVal0(row, col, b, "B");
}
}
checkValMax(row, col, a, "A");
}
}
} catch (Exception e) {
System.out.println("\n" + e.getMessage());
retval = false;
}
if (!retval) {
for (row = 0; row < h; row++) {
for (col = 0; col < w; col++) {
if (pf == TJ.PF_CMYK) {
int c = getVal(buf, pitch * row + col * ps);
int m = getVal(buf, pitch * row + col * ps + 1);
int y = getVal(buf, pitch * row + col * ps + 2);
int k = getVal(buf, pitch * row + col * ps + 3);
System.out.format("%3d/%3d/%3d/%3d ", c, m, y, k);
} else {
int r = getVal(buf, pitch * row + col * ps + roffset);
int g = getVal(buf, pitch * row + col * ps + goffset);
int b = getVal(buf, pitch * row + col * ps + boffset);
System.out.format("%3d/%3d/%3d ", r, g, b);
}
}
System.out.print("\n");
}
}
return retval;
}
static boolean checkIntBuf(int[] buf, int w, int pitch, int h, int pf,
int subsamp, TJScalingFactor sf, boolean bottomUp)
throws Exception {
int rshift = TJ.getRedOffset(pf) * 8;
int gshift = TJ.getGreenOffset(pf) * 8;
int bshift = TJ.getBlueOffset(pf) * 8;
int ashift = TJ.getAlphaOffset(pf) * 8;
int index, row, col;
boolean retval = true;
int halfway = 16 * sf.getNum() / sf.getDenom();
int blockSize = 8 * sf.getNum() / sf.getDenom();
try {
for (row = 0; row < halfway; row++) {
for (col = 0; col < w; col++) {
if (bottomUp)
index = pitch * (h - row - 1) + col;
else
index = pitch * row + col;
int r = (buf[index] >> rshift) & 0xFF;
int g = (buf[index] >> gshift) & 0xFF;
int b = (buf[index] >> bshift) & 0xFF;
int a = ashift >= 0 ? (buf[index] >> ashift) & 0xFF : 255;
if (((row / blockSize) + (col / blockSize)) % 2 == 0) {
if (row < halfway) {
checkValMax(row, col, r, "R");
checkValMax(row, col, g, "G");
checkValMax(row, col, b, "B");
} else {
checkVal0(row, col, r, "R");
checkVal0(row, col, g, "G");
checkVal0(row, col, b, "B");
}
} else {
if (subsamp == TJ.SAMP_GRAY) {
if (row < halfway) {
checkVal(row, col, r, "R", 76);
checkVal(row, col, g, "G", 76);
checkVal(row, col, b, "B", 76);
} else {
checkVal(row, col, r, "R", 226);
checkVal(row, col, g, "G", 226);
checkVal(row, col, b, "B", 226);
}
} else {
checkValMax(row, col, r, "R");
if (row < halfway) {
checkVal0(row, col, g, "G");
} else {
checkValMax(row, col, g, "G");
}
checkVal0(row, col, b, "B");
}
}
checkValMax(row, col, a, "A");
}
}
} catch (Exception e) {
System.out.println("\n" + e.getMessage());
retval = false;
}
if (!retval) {
for (row = 0; row < h; row++) {
for (col = 0; col < w; col++) {
int r = (buf[pitch * row + col] >> rshift) & 0xFF;
int g = (buf[pitch * row + col] >> gshift) & 0xFF;
int b = (buf[pitch * row + col] >> 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;
}
static boolean checkImg(BufferedImage img, int pf, int subsamp,
TJScalingFactor sf, boolean bottomUp)
throws Exception {
WritableRaster wr = img.getRaster();
int imgType = img.getType();
if (imgType == BufferedImage.TYPE_INT_RGB ||
imgType == BufferedImage.TYPE_INT_BGR ||
imgType == BufferedImage.TYPE_INT_ARGB ||
imgType == BufferedImage.TYPE_INT_ARGB_PRE) {
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, sf, bottomUp);
} 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,
sf, bottomUp);
}
}
static int pad(int v, int p) {
return ((v + (p) - 1) & (~((p) - 1)));
}
static boolean checkBufYUV(byte[] buf, int size, int w, int h, int subsamp,
TJScalingFactor sf) throws Exception {
int row, col;
int hsf = TJ.getMCUWidth(subsamp) / 8, vsf = TJ.getMCUHeight(subsamp) / 8;
int pw = pad(w, hsf), ph = pad(h, vsf);
int cw = pw / hsf, ch = ph / vsf;
int ypitch = pad(pw, yuvAlign), uvpitch = pad(cw, yuvAlign);
boolean retval = true;
int correctsize = ypitch * ph +
(subsamp == TJ.SAMP_GRAY ? 0 : uvpitch * ch * 2);
int halfway = 16 * sf.getNum() / sf.getDenom();
int blockSize = 8 * sf.getNum() / sf.getDenom();
try {
if (size != correctsize)
throw new Exception("Incorrect size " + size + ". Should be " +
correctsize);
for (row = 0; row < ph; row++) {
for (col = 0; col < pw; col++) {
byte y = buf[ypitch * row + col];
if (((row / blockSize) + (col / blockSize)) % 2 == 0) {
if (row < halfway)
checkValMax(row, col, y, "Y");
else
checkVal0(row, col, y, "Y");
} else {
if (row < halfway)
checkVal(row, col, y, "Y", 76);
else
checkVal(row, col, y, "Y", 226);
}
}
}
if (subsamp != TJ.SAMP_GRAY) {
halfway = 16 / vsf * sf.getNum() / sf.getDenom();
for (row = 0; row < ch; row++) {
for (col = 0; col < cw; col++) {
byte u = buf[ypitch * ph + (uvpitch * row + col)],
v = buf[ypitch * ph + uvpitch * ch + (uvpitch * row + col)];
if (((row * vsf / blockSize) + (col * hsf / blockSize)) % 2 == 0) {
checkVal(row, col, u, "U", 128);
checkVal(row, col, v, "V", 128);
} else {
if (row < halfway) {
checkVal(row, col, u, "U", 85);
checkValMax(row, col, v, "V");
} else {
checkVal0(row, col, u, "U");
checkVal(row, col, v, "V", 149);
}
}
}
}
}
} catch (Exception e) {
System.out.println("\n" + e.getMessage());
retval = false;
}
if (!retval) {
for (row = 0; row < ph; row++) {
for (col = 0; col < pw; col++) {
int y = buf[ypitch * row + col];
if (y < 0) y += 256;
System.out.format("%3d ", y);
}
System.out.print("\n");
}
System.out.print("\n");
for (row = 0; row < ch; row++) {
for (col = 0; col < cw; col++) {
int u = buf[ypitch * ph + (uvpitch * row + col)];
if (u < 0) u += 256;
System.out.format("%3d ", u);
}
System.out.print("\n");
}
System.out.print("\n");
for (row = 0; row < ch; row++) {
for (col = 0; col < cw; col++) {
int v = buf[ypitch * ph + uvpitch * ch + (uvpitch * row + col)];
if (v < 0) v += 256;
System.out.format("%3d ", v);
}
System.out.print("\n");
}
}
return retval;
}
static void writeJPEG(byte[] jpegBuf, int jpegBufSize, String filename)
throws Exception {
File file = new File(filename);
FileOutputStream fos = new FileOutputStream(file);
fos.write(jpegBuf, 0, jpegBufSize);
fos.close();
}
static int compTest(TJCompressor tjc, byte[] dstBuf, int w, int h, int pf,
String baseName) throws Exception {
String tempStr;
Object srcBuf = null;
BufferedImage img = null;
String pfStr, pfStrLong;
boolean bottomUp = (tjc.get(TJ.PARAM_BOTTOMUP) == 1);
int subsamp = tjc.get(TJ.PARAM_SUBSAMP);
int jpegQual = tjc.get(TJ.PARAM_QUALITY);
int jpegPSV = tjc.get(TJ.PARAM_LOSSLESSPSV);
String buStr = bottomUp ? "BU" : "TD";
String buStrLong = bottomUp ? "Bottom-Up" : "Top-Down ";
int size = 0, ps, imgType = pf;
if (bi) {
pf = biTypePF(imgType);
pfStr = biTypeStr(imgType);
pfStrLong = pfStr + " (" + PIXFORMATSTR[pf] + ")";
} else {
pfStr = PIXFORMATSTR[pf];
pfStrLong = pfStr;
}
ps = TJ.getPixelSize(pf);
if (bi) {
img = new BufferedImage(w, h, imgType);
initImg(img, pf, bottomUp);
tempStr = baseName + "_enc" + precision + "_" + pfStr + "_" + buStr +
"_" + SUBNAME[subsamp] + "_Q" + jpegQual + ".png";
File file = new File(tempStr);
ImageIO.write(img, "png", file);
tjc.setSourceImage(img, 0, 0, 0, 0);
} else {
if (precision <= 8)
srcBuf = new byte[w * h * ps + 1];
else
srcBuf = new short[w * h * ps + 1];
initBuf(srcBuf, w, w * ps, h, pf, bottomUp);
if (precision <= 8)
tjc.setSourceImage((byte[])srcBuf, 0, 0, w, 0, h, pf);
else if (precision <= 12)
tjc.setSourceImage12((short[])srcBuf, 0, 0, w, 0, h, pf);
else
tjc.setSourceImage16((short[])srcBuf, 0, 0, w, 0, h, pf);
}
Arrays.fill(dstBuf, (byte)0);
if (doYUV) {
System.out.format("%s %s -> YUV %s ... ", pfStrLong, buStrLong,
SUBNAME_LONG[subsamp]);
YUVImage yuvImage = tjc.encodeYUV(yuvAlign);
if (checkBufYUV(yuvImage.getBuf(), yuvImage.getSize(), w, h, subsamp,
new TJScalingFactor(1, 1)))
System.out.print("Passed.\n");
else {
System.out.print("FAILED!\n");
exitStatus = -1;
}
System.out.format("YUV %s %s -> JPEG Q%d ... ", SUBNAME_LONG[subsamp],
buStrLong, jpegQual);
tjc.setSourceImage(yuvImage);
} else {
if (lossless) {
tjc.set(TJ.PARAM_PRECISION, precision);
System.out.format("%s %s -> LOSSLESS PSV%d ... ", pfStrLong, buStrLong,
jpegPSV);
} else
System.out.format("%s %s -> %s Q%d ... ", pfStrLong, buStrLong,
SUBNAME_LONG[subsamp], jpegQual);
}
tjc.compress(dstBuf);
size = tjc.getCompressedSize();
if (lossless)
tempStr = baseName + "_enc" + precision + "_" + pfStr + "_" + buStr +
"_LOSSLESS_PSV" + jpegPSV + ".jpg";
else
tempStr = baseName + "_enc" + precision + "_" + pfStr + "_" + buStr +
"_" + SUBNAME[subsamp] + "_Q" + jpegQual + ".jpg";
writeJPEG(dstBuf, size, tempStr);
System.out.println("Done.\n Result in " + tempStr);
return size;
}
static void decompTest(TJDecompressor tjd, byte[] jpegBuf, int jpegSize,
int w, int h, int pf, String baseName, int subsamp,
TJScalingFactor sf) throws Exception {
String pfStr, pfStrLong, tempStr;
boolean bottomUp = (tjd.get(TJ.PARAM_BOTTOMUP) == 1);
String buStrLong = bottomUp ? "Bottom-Up" : "Top-Down ";
int scaledWidth = sf.getScaled(w);
int scaledHeight = sf.getScaled(h);
int temp1, temp2, imgType = pf;
BufferedImage img = null;
Object dstBuf = null;
if (bi) {
pf = biTypePF(imgType);
pfStr = biTypeStr(imgType);
pfStrLong = pfStr + " (" + PIXFORMATSTR[pf] + ")";
} else {
pfStr = PIXFORMATSTR[pf];
pfStrLong = pfStr;
}
tjd.setSourceImage(jpegBuf, jpegSize);
tjd.setScalingFactor(sf);
if (lossless && subsamp != TJ.SAMP_444 && subsamp != TJ.SAMP_GRAY)
subsamp = TJ.SAMP_444;
if (tjd.getWidth() != w || tjd.getHeight() != h ||
tjd.get(TJ.PARAM_SUBSAMP) != subsamp)
throw new Exception("Incorrect JPEG header");
if (doYUV) {
System.out.format("JPEG -> YUV %s ", SUBNAME_LONG[subsamp]);
if (!sf.isOne())
System.out.format("%d/%d ... ", sf.getNum(), sf.getDenom());
else System.out.print("... ");
YUVImage yuvImage = tjd.decompressToYUV(yuvAlign);
if (checkBufYUV(yuvImage.getBuf(), yuvImage.getSize(), scaledWidth,
scaledHeight, subsamp, sf))
System.out.print("Passed.\n");
else {
System.out.print("FAILED!\n"); exitStatus = -1;
}
System.out.format("YUV %s -> %s %s ... ", SUBNAME_LONG[subsamp],
pfStrLong, buStrLong);
tjd.setSourceImage(yuvImage);
} else {
System.out.format("JPEG -> %s %s ", pfStrLong, buStrLong);
if (!sf.isOne())
System.out.format("%d/%d ... ", sf.getNum(), sf.getDenom());
else System.out.print("... ");
}
if (bi)
img = tjd.decompress8(imgType);
else {
if (precision <= 8)
dstBuf = tjd.decompress8(0, pf);
else if (precision <= 12)
dstBuf = tjd.decompress12(0, pf);
else
dstBuf = tjd.decompress16(0, pf);
}
if (bi) {
tempStr = baseName + "_dec_" + pfStr + "_" + (bottomUp ? "BU" : "TD") +
"_" + SUBNAME[subsamp] + "_" +
(double)sf.getNum() / (double)sf.getDenom() + "x" + ".png";
File file = new File(tempStr);
ImageIO.write(img, "png", file);
}
if ((bi && checkImg(img, pf, subsamp, sf, bottomUp)) ||
(!bi && checkBuf(dstBuf, scaledWidth,
scaledWidth * TJ.getPixelSize(pf), scaledHeight, pf,
subsamp, sf, bottomUp)))
System.out.print("Passed.\n");
else {
System.out.print("FAILED!\n");
exitStatus = -1;
}
}
static void decompTest(TJDecompressor tjd, byte[] jpegBuf, int jpegSize,
int w, int h, int pf, String baseName, int subsamp)
throws Exception {
int i;
if (lossless) {
decompTest(tjd, jpegBuf, jpegSize, w, h, pf, baseName, subsamp,
TJ.UNSCALED);
return;
}
TJScalingFactor[] sf = TJ.getScalingFactors();
for (i = 0; i < sf.length; i++) {
int num = sf[i].getNum();
int denom = sf[i].getDenom();
if (subsamp == TJ.SAMP_444 || subsamp == TJ.SAMP_GRAY ||
((subsamp == TJ.SAMP_411 || subsamp == TJ.SAMP_441) && num == 1 &&
(denom == 2 || denom == 1)) ||
(subsamp != TJ.SAMP_411 && subsamp != TJ.SAMP_441 && num == 1 &&
(denom == 4 || denom == 2 || denom == 1)))
decompTest(tjd, jpegBuf, jpegSize, w, h, pf, baseName, subsamp, sf[i]);
}
}
static void doTest(int w, int h, int[] formats, int subsamp, String baseName)
throws Exception {
TJCompressor tjc = null;
TJDecompressor tjd = null;
int size;
byte[] dstBuf;
if (lossless && subsamp != TJ.SAMP_GRAY)
subsamp = TJ.SAMP_444;
dstBuf = new byte[TJ.bufSize(w, h, subsamp)];
try {
tjc = new TJCompressor();
tjd = new TJDecompressor();
if (lossless) {
tjc.set(TJ.PARAM_LOSSLESS, 1);
tjc.set(TJ.PARAM_LOSSLESSPSV, ((psv++ - 1) % 7) + 1);
} else {
tjc.set(TJ.PARAM_QUALITY, 100);
if (subsamp == TJ.SAMP_422 || subsamp == TJ.SAMP_420 ||
subsamp == TJ.SAMP_440 || subsamp == TJ.SAMP_411 ||
subsamp == TJ.SAMP_441)
tjd.set(TJ.PARAM_FASTUPSAMPLE, 1);
}
tjc.set(TJ.PARAM_SUBSAMP, subsamp);
for (int pf : formats) {
if (pf < 0) continue;
for (int i = 0; i < 2; i++) {
tjc.set(TJ.PARAM_BOTTOMUP, i == 1 ? 1 : 0);
tjd.set(TJ.PARAM_BOTTOMUP, i == 1 ? 1 : 0);
size = compTest(tjc, dstBuf, w, h, pf, baseName);
decompTest(tjd, dstBuf, size, w, h, pf, baseName, subsamp);
if (pf >= TJ.PF_RGBX && pf <= TJ.PF_XRGB && !bi) {
System.out.print("\n");
decompTest(tjd, dstBuf, size, w, h, pf + (TJ.PF_RGBA - TJ.PF_RGBX),
baseName, subsamp);
}
System.out.print("\n");
}
}
System.out.print("--------------------\n\n");
} 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();
}
static void overflowTest() throws Exception {
/* Ensure that the various buffer size methods don't overflow */
int size = 0;
boolean exception = false;
try {
exception = false;
size = TJ.bufSize(18919, 18919, TJ.SAMP_444);
} catch (Exception e) { exception = true; }
if (!exception || size != 0)
throw new Exception("TJ.bufSize() overflow");
try {
exception = false;
size = TJ.bufSizeYUV(26755, 1, 26755, TJ.SAMP_444);
} catch (Exception e) { exception = true; }
if (!exception || size != 0)
throw new Exception("TJ.bufSizeYUV() overflow");
try {
exception = false;
size = TJ.bufSizeYUV(26754, 3, 26754, TJ.SAMP_444);
} catch (Exception e) { exception = true; }
if (!exception || size != 0)
throw new Exception("TJ.bufSizeYUV() overflow");
try {
exception = false;
size = TJ.bufSizeYUV(26754, -1, 26754, TJ.SAMP_444);
} catch (Exception e) { exception = true; }
if (!exception || size != 0)
throw new Exception("TJ.bufSizeYUV() overflow");
try {
exception = false;
size = TJ.planeSizeYUV(0, 46341, 0, 46341, TJ.SAMP_444);
} catch (Exception e) { exception = true; }
if (!exception || size != 0)
throw new Exception("TJ.planeSizeYUV() overflow");
try {
exception = false;
size = TJ.planeWidth(0, Integer.MAX_VALUE, TJ.SAMP_420);
} catch (Exception e) { exception = true; }
if (!exception || size != 0)
throw new Exception("TJ.planeWidth() overflow");
try {
exception = false;
size = TJ.planeHeight(0, Integer.MAX_VALUE, TJ.SAMP_420);
} catch (Exception e) { exception = true; }
if (!exception || size != 0)
throw new Exception("TJ.planeHeight() overflow");
}
static void bufSizeTest() throws Exception {
int w, h, i, subsamp, numSamp = TJ.NUMSAMP;
byte[] srcBuf, dstBuf = null;
YUVImage dstImage = null;
TJCompressor tjc = null;
Random r = new Random();
try {
tjc = new TJCompressor();
if (lossless) {
tjc.set(TJ.PARAM_PRECISION, precision);
tjc.set(TJ.PARAM_LOSSLESS, 1);
tjc.set(TJ.PARAM_LOSSLESSPSV, ((psv++ - 1) % 7) + 1);
numSamp = 1;
} else
tjc.set(TJ.PARAM_QUALITY, 100);
System.out.println("Buffer size regression test");
for (subsamp = 0; subsamp < numSamp; subsamp++) {
tjc.set(TJ.PARAM_SUBSAMP, subsamp);
for (w = 1; w < 48; w++) {
int maxh = (w == 1) ? 2048 : 48;
for (h = 1; h < maxh; h++) {
if (h % 100 == 0)
System.out.format("%04d x %04d\b\b\b\b\b\b\b\b\b\b\b", w, h);
srcBuf = new byte[w * h * 4];
if (doYUV)
dstImage = new YUVImage(w, yuvAlign, h, subsamp);
else
dstBuf = new byte[TJ.bufSize(w, h, subsamp)];
for (i = 0; i < w * h * 4; i++) {
srcBuf[i] = (byte)(r.nextInt(2) * 255);
}
tjc.setSourceImage(srcBuf, 0, 0, w, 0, h, TJ.PF_BGRX);
if (doYUV)
tjc.encodeYUV(dstImage);
else
tjc.compress(dstBuf);
srcBuf = new byte[h * w * 4];
if (doYUV)
dstImage = new YUVImage(h, yuvAlign, w, subsamp);
else
dstBuf = new byte[TJ.bufSize(h, w, subsamp)];
for (i = 0; i < h * w * 4; i++) {
srcBuf[i] = (byte)(r.nextInt(2) * 255);
}
tjc.setSourceImage(srcBuf, 0, 0, h, 0, w, TJ.PF_BGRX);
if (doYUV)
tjc.encodeYUV(dstImage);
else
tjc.compress(dstBuf);
}
dstImage = null;
dstBuf = null;
System.gc();
}
}
System.out.println("Done. ");
} catch (Exception e) {
if (tjc != null) tjc.close();
throw e;
}
if (tjc != null) tjc.close();
}
static void rgbToCMYK(int r, int g, int b, int[] c, int[] m, int[] y,
int[] k) {
double ctmp = 1.0 - ((double)r / (double)maxSample);
double mtmp = 1.0 - ((double)g / (double)maxSample);
double ytmp = 1.0 - ((double)b / (double)maxSample);
double ktmp = Math.min(Math.min(ctmp, mtmp), ytmp);
if (ktmp == 1.0)
ctmp = mtmp = ytmp = 0.0;
else {
ctmp = (ctmp - ktmp) / (1.0 - ktmp);
mtmp = (mtmp - ktmp) / (1.0 - ktmp);
ytmp = (ytmp - ktmp) / (1.0 - ktmp);
}
c[0] = (int)((double)maxSample - ctmp * (double)maxSample + 0.5);
m[0] = (int)((double)maxSample - mtmp * (double)maxSample + 0.5);
y[0] = (int)((double)maxSample - ytmp * (double)maxSample + 0.5);
k[0] = (int)((double)maxSample - ktmp * (double)maxSample + 0.5);
}
static void initBitmap(Object buf, int width, int pitch, int height, int pf,
boolean bottomUp) {
int roffset = TJ.getRedOffset(pf);
int goffset = TJ.getGreenOffset(pf);
int boffset = TJ.getBlueOffset(pf);
int ps = TJ.getPixelSize(pf);
int i, j, ci;
for (j = 0; j < height; j++) {
int row = bottomUp ? height - j - 1 : j;
for (i = 0; i < width; i++) {
int r = (i * (maxSample + 1) / width) % (maxSample + 1);
int g = (j * (maxSample + 1) / height) % (maxSample + 1);
int b = (j * (maxSample + 1) / height +
i * (maxSample + 1) / width) % (maxSample + 1);
for (ci = 0; ci < ps; ci++)
setVal(buf, row * pitch + i * ps + ci, 0);
if (pf == TJ.PF_GRAY)
setVal(buf, row * pitch + i * ps, b);
else if (pf == TJ.PF_CMYK) {
int[] c = new int[1], m = new int[1], y = new int[1], k = new int[1];
rgbToCMYK(r, g, b, c, m, y, k);
setVal(buf, row * pitch + i * ps + 0, c[0]);
setVal(buf, row * pitch + i * ps + 1, m[0]);
setVal(buf, row * pitch + i * ps + 2, y[0]);
setVal(buf, row * pitch + i * ps + 3, k[0]);
} else {
setVal(buf, row * pitch + i * ps + roffset, r);
setVal(buf, row * pitch + i * ps + goffset, g);
setVal(buf, row * pitch + i * ps + boffset, b);
}
}
}
}
static void cmykToRGB(int c, int m, int y, int k, int[] r, int[] g,
int[] b) {
r[0] = (int)((double)c * (double)k / (double)maxSample + 0.5);
g[0] = (int)((double)m * (double)k / (double)maxSample + 0.5);
b[0] = (int)((double)y * (double)k / (double)maxSample + 0.5);
}
static boolean cmpBitmap(Object buf, int width, int pitch, int height,
int pf, boolean bottomUp, boolean gray2rgb) {
int roffset = TJ.getRedOffset(pf);
int goffset = TJ.getGreenOffset(pf);
int boffset = TJ.getBlueOffset(pf);
int aoffset = TJ.getAlphaOffset(pf);
int ps = TJ.getPixelSize(pf);
int i, j;
for (j = 0; j < height; j++) {
int row = bottomUp ? height - j - 1 : j;
for (i = 0; i < width; i++) {
int r = (i * (maxSample + 1) / width) % (maxSample + 1);
int g = (j * (maxSample + 1) / height) % (maxSample + 1);
int b = (j * (maxSample + 1) / height +
i * (maxSample + 1) / width) % (maxSample + 1);
if (pf == TJ.PF_GRAY) {
if (getVal(buf, row * pitch + i * ps) != b)
return false;
} else if (pf == TJ.PF_CMYK) {
int[] rf = new int[1], gf = new int[1], bf = new int[1];
cmykToRGB(getVal(buf, row * pitch + i * ps + 0),
getVal(buf, row * pitch + i * ps + 1),
getVal(buf, row * pitch + i * ps + 2),
getVal(buf, row * pitch + i * ps + 3), rf, gf, bf);
if (gray2rgb) {
if (rf[0] != b || gf[0] != b || bf[0] != b)
return false;
} else if (rf[0] != r || gf[0] != g || bf[0] != b)
return false;
} else {
if (gray2rgb) {
if (getVal(buf, row * pitch + i * ps + roffset) != b ||
getVal(buf, row * pitch + i * ps + goffset) != b ||
getVal(buf, row * pitch + i * ps + boffset) != b)
return false;
} else if (getVal(buf, row * pitch + i * ps + roffset) != r ||
getVal(buf, row * pitch + i * ps + goffset) != g ||
getVal(buf, row * pitch + i * ps + boffset) != b)
return false;
if (aoffset >= 0 &&
getVal(buf, row * pitch + i * ps + aoffset) != maxSample)
return false;
}
}
}
return true;
}
static String getMD5Sum(String filename) throws Exception {
byte[] bytes = Files.readAllBytes(Paths.get(filename));
byte[] md5sum =
java.security.MessageDigest.getInstance("MD5").digest(bytes);
return new java.math.BigInteger(1, md5sum).toString(16);
}
static void doBmpTest(String ext, int width, int align, int height, int pf,
boolean bottomUp) throws Exception {
TJCompressor tjc = null;
TJDecompressor tjd = null;
String filename, md5sum;
int ps = TJ.getPixelSize(pf), pitch = pad(width * ps, align),
loadWidth = 0, loadPitch = 0, loadHeight = 0, pixelFormat = pf;
Object buf = null;
String md5ref;
String[] colorPPMRefs = new String[] {
"", "", "bad09d9ef38eda566848fb7c0b7fd0a",
"7ef2c87261a8bd6838303b541563cf27", "28a37cf9636ff6bb9ed6b206bdac60db",
"723307791d42e0b5f9e91625c7636086", "d729c4bcd3addc14abc16b656c6bbc98",
"5d7636eedae3cf579b6de13078227548", "c0c9f772b464d1896326883a5c79c545",
"fcf6490e0445569427f1d95baf5f8fcb", "5cbc3b0ccba23f5781d950a72e0ccc83",
"d4e26d6d16d7bfee380f6feb10f7e53", "2ff5299287017502832c99718450c90a",
"44ae6cd70c798ea583ab0c8c03621092", "697b2fe03892bc9a75396ad3e73d9203",
"599732f973eb7c0849a888e783bbe27e", "623f54661b928d170bd2324bc3620565"
};
String[] grayPPMRefs = new String[] {
"", "", "7565be35a2ce909cae016fa282af8efa",
"e86b9ea57f7d53f6b5497653740992b5", "8924d4d81fe0220c684719294f93407a",
"e2e69ba70efcfae317528c91651c7ae2", "e6154aafc1eb9e4333d68ce7ad9df051",
"3d7fe831d6fbe55d3fa12f52059c15d3", "112c682e82ce5de1cca089e20d60000b",
"5a7ce86c649dda86d6fed185ab78a67", "b723c0bc087592816523fbc906b7c3a",
"5da422b1ddfd44c7659094d42ba5580c", "d1895c7e6f2b2c9af6e821a655c239c",
"fc2803bca103ff75785ea0dca992aa", "d8c91fac522c16b029e514d331a22bc4",
"e50cff0b3562ed7e64dbfc093440e333", "64f3320b226ea37fb58080713b4df1b2"
};
try {
tjc = new TJCompressor();
tjc.set(TJ.PARAM_BOTTOMUP, bottomUp ? 1 : 0);
tjc.set(TJ.PARAM_PRECISION, precision);
tjd = new TJDecompressor();
tjd.set(TJ.PARAM_BOTTOMUP, bottomUp ? 1 : 0);
tjd.set(TJ.PARAM_PRECISION, precision);
if (precision == 8 && ext.equalsIgnoreCase("bmp"))
md5ref = (pf == TJ.PF_GRAY ? "51976530acf75f02beddf5d21149101d" :
"6d659071b9bfcdee2def22cb58ddadca");
else
md5ref = (pf == TJ.PF_GRAY ? grayPPMRefs[precision] :
colorPPMRefs[precision]);
if (precision <= 8)
buf = new byte[pitch * height];
else
buf = new short[pitch * height];
initBitmap(buf, width, pitch, height, pf, bottomUp);
filename = String.format("test_bmp%d_%s_%d_%s_%s.%s", precision,
PIXFORMATSTR[pf], align, bottomUp ? "bu" : "td",
uniqueID, ext);
tjd.saveImage(filename, buf, 0, 0, width, pitch, height, pf);
md5sum = getMD5Sum(filename);
if (md5sum == null)
throw new Exception("Could not determine MD5 sum of " + filename);
if (!md5sum.equalsIgnoreCase(md5ref))
throw new Exception(filename + " has an MD5 sum of " + md5sum +
". Should be " + md5ref);
tjc.loadSourceImage(filename, align, pf);
loadWidth = tjc.getWidth();
loadPitch = tjc.getPitch();
loadHeight = tjc.getHeight();
pf = tjc.getPixelFormat();
buf = tjc.getSourceBuf();
if (width != loadWidth || pitch != loadPitch || height != loadHeight)
throw new Exception("Image dimensions of " + filename + " are bogus");
if (!cmpBitmap(buf, width, pitch, height, pf, bottomUp, false))
throw new Exception("Pixel data in " + filename + " is bogus");
if (pf == TJ.PF_GRAY) {
pf = TJ.PF_XBGR;
tjc.loadSourceImage(filename, align, pf);
loadWidth = tjc.getWidth();
loadPitch = tjc.getPitch();
loadHeight = tjc.getHeight();
pf = tjc.getPixelFormat();
buf = tjc.getSourceBuf();
pitch = pad(width * TJ.getPixelSize(pf), align);
if (width != loadWidth || pitch != loadPitch || height != loadHeight)
throw new Exception("Image dimensions of " + filename +
" are bogus");
if (!cmpBitmap(buf, width, pitch, height, pf, bottomUp, true))
throw new Exception("Converting " + filename + " to RGB failed");
pf = TJ.PF_CMYK;
tjc.loadSourceImage(filename, align, pf);
loadWidth = tjc.getWidth();
loadPitch = tjc.getPitch();
loadHeight = tjc.getHeight();
pf = tjc.getPixelFormat();
buf = tjc.getSourceBuf();
pitch = pad(width * TJ.getPixelSize(pf), align);
if (width != loadWidth || pitch != loadPitch || height != loadHeight)
throw new Exception("Image dimensions of " + filename +
" are bogus");
if (!cmpBitmap(buf, width, pitch, height, pf, bottomUp, true))
throw new Exception("Converting " + filename + " to CMYK failed");
}
/* Verify that TJCompressor.loadSourceImage() returns the proper
"preferred" pixel format for the file type. */
pf = pixelFormat;
pixelFormat = TJ.PF_UNKNOWN;
tjc.loadSourceImage(filename, align, pixelFormat);
pixelFormat = tjc.getPixelFormat();
if ((pf == TJ.PF_GRAY && pixelFormat != TJ.PF_GRAY) ||
(pf != TJ.PF_GRAY && ext.equalsIgnoreCase("bmp") &&
pixelFormat != TJ.PF_BGR) ||
(pf != TJ.PF_GRAY && ext.equalsIgnoreCase("ppm") &&
pixelFormat != TJ.PF_RGB))
throw new Exception("TJCompressor.loadImage() returned unexpected " +
"pixel format: " + PIXFORMATSTR[pixelFormat]);
File file = new File(filename);
file.delete();
} 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();
}
static void bmpTest() throws Exception {
int align, width = 35, height = 39, format;
for (align = 1; align <= 8; align *= 2) {
for (format = 0; format < TJ.NUMPF; format++) {
if (precision == 8) {
System.out.format("%s Top-Down BMP (row alignment = %d samples) ... ",
PIXFORMATSTR[format], align);
doBmpTest("bmp", width, align, height, format, false);
System.out.println("OK.");
}
System.out.format("%s Top-Down PPM (row alignment = %d samples) ... ",
PIXFORMATSTR[format], align);
doBmpTest("ppm", width, align, height, format, true);
System.out.println("OK.");
if (precision == 8) {
System.out.format("%s Bottom-Up BMP (row alignment = %d samples) ... ",
PIXFORMATSTR[format], align);
doBmpTest("bmp", width, align, height, format, false);
System.out.println("OK.");
}
System.out.format("%s Bottom-Up PPM (row alignment = %d samples) ... ",
PIXFORMATSTR[format], align);
doBmpTest("ppm", width, align, height, format, true);
System.out.println("OK.");
}
}
}
public static void main(String[] argv) {
try {
String testName = "javatest";
boolean bmp = false;
for (int i = 0; i < argv.length; i++) {
if (argv[i].equalsIgnoreCase("-yuv"))
doYUV = true;
else if (argv[i].equalsIgnoreCase("-noyuvpad"))
yuvAlign = 1;
else if (argv[i].equalsIgnoreCase("-lossless"))
lossless = true;
else if (argv[i].equalsIgnoreCase("-bi")) {
bi = true;
testName = "javabitest";
} else if (argv[i].equalsIgnoreCase("-bmp"))
bmp = true;
else if (argv[i].equalsIgnoreCase("-precision") &&
i < argv.length - 1) {
int tempi = -1;
try {
tempi = Integer.parseInt(argv[++i]);
} catch (NumberFormatException e) {}
if (tempi < 2 || tempi > 16)
usage();
precision = tempi;
if (precision != 8 && precision != 12)
lossless = true;
} else
usage();
}
if (lossless && doYUV)
throw new Exception("Lossless JPEG and YUV encoding/decoding are incompatible.");
if (precision != 8 && doYUV)
throw new Exception("YUV encoding/decoding requires 8-bit data precision.");
if (precision != 8 && bi)
throw new Exception("BufferedImage support requires 8-bit data precision.");
System.out.format("Testing %d-bit precision\n", precision);
sampleSize = (precision <= 8 ? 1 : 2);
maxSample = (1 << precision) - 1;
tolerance = (lossless ? 0 : (precision > 8 ? 2 : 1));
redToY = (19595 * maxSample) >> 16;
yellowToY = (58065 * maxSample) >> 16;
if (bmp) {
bmpTest();
System.exit(exitStatus);
}
if (doYUV)
FORMATS_4SAMPLE[4] = -1;
overflowTest();
doTest(35, 39, bi ? FORMATS_3BYTEBI : FORMATS_3SAMPLE, TJ.SAMP_444,
testName);
doTest(39, 41, bi ? FORMATS_4BYTEBI : FORMATS_4SAMPLE, TJ.SAMP_444,
testName);
doTest(41, 35, bi ? FORMATS_3BYTEBI : FORMATS_3SAMPLE, TJ.SAMP_422,
testName);
if (!lossless) {
doTest(35, 39, bi ? FORMATS_4BYTEBI : FORMATS_4SAMPLE, TJ.SAMP_422,
testName);
doTest(39, 41, bi ? FORMATS_3BYTEBI : FORMATS_3SAMPLE, TJ.SAMP_420,
testName);
doTest(41, 35, bi ? FORMATS_4BYTEBI : FORMATS_4SAMPLE, TJ.SAMP_420,
testName);
doTest(35, 39, bi ? FORMATS_3BYTEBI : FORMATS_3SAMPLE, TJ.SAMP_440,
testName);
doTest(39, 41, bi ? FORMATS_4BYTEBI : FORMATS_4SAMPLE, TJ.SAMP_440,
testName);
doTest(41, 35, bi ? FORMATS_3BYTEBI : FORMATS_3SAMPLE, TJ.SAMP_411,
testName);
doTest(35, 39, bi ? FORMATS_4BYTEBI : FORMATS_4SAMPLE, TJ.SAMP_411,
testName);
doTest(39, 41, bi ? FORMATS_3BYTEBI : FORMATS_3SAMPLE, TJ.SAMP_441,
testName);
doTest(41, 35, bi ? FORMATS_4BYTEBI : FORMATS_4SAMPLE, TJ.SAMP_441,
testName);
}
doTest(39, 41, bi ? FORMATS_GRAYBI : FORMATS_GRAY, TJ.SAMP_GRAY,
testName);
if (!lossless) {
doTest(41, 35, bi ? FORMATS_3BYTEBI : FORMATS_3SAMPLE, TJ.SAMP_GRAY,
testName);
FORMATS_4SAMPLE[4] = -1;
doTest(35, 39, bi ? FORMATS_4BYTEBI : FORMATS_4SAMPLE, TJ.SAMP_GRAY,
testName);
}
if (!bi)
bufSizeTest();
if (doYUV && !bi) {
System.out.print("\n--------------------\n\n");
doTest(48, 48, FORMATS_RGB, TJ.SAMP_444, "javatest_yuv0");
doTest(48, 48, FORMATS_RGB, TJ.SAMP_422, "javatest_yuv0");
doTest(48, 48, FORMATS_RGB, TJ.SAMP_420, "javatest_yuv0");
doTest(48, 48, FORMATS_RGB, TJ.SAMP_440, "javatest_yuv0");
doTest(48, 48, FORMATS_RGB, TJ.SAMP_411, "javatest_yuv0");
doTest(48, 48, FORMATS_RGB, TJ.SAMP_441, "javatest_yuv0");
doTest(48, 48, FORMATS_RGB, TJ.SAMP_GRAY, "javatest_yuv0");
doTest(48, 48, FORMATS_GRAY, TJ.SAMP_GRAY, "javatest_yuv0");
}
} catch (Exception e) {
e.printStackTrace();
exitStatus = -1;
}
System.exit(exitStatus);
}
}
Reference in New Issue View Git Blame Copy Permalink