9a146f0f23
- Wordsmithing, formatting, and grammar tweaks
- Various clarifications and corrections, including specifying whether
a particular buffer or image is used as a source or destination
- Accommodate/mention features that were introduced since the API
documentation was created.
- For clarity, use "packed-pixel" to describe uncompressed
source/destination images that are not planar YUV.
- Use "row" rather than "line" to refer to a single horizontal group of
pixels or component values, for consistency with the libjpeg API
documentation. (libjpeg also uses "scanline", which is a more archaic
term.)
- Use "alignment" rather than "padding" to refer to the number of bytes
by which a row's width is evenly divisible. This consistifies the
documention of the YUV functions and tjLoadImage(). ("Padding"
typically refers to the number of bytes added to each row, which is
not the same thing.)
- Remove all references to "the underlying codec." Although the
TurboJPEG API originated as a cross-platform wrapper for the Intel
Integrated Performance Primitives, Sun mediaLib, QuickTime, and
libjpeg, none of those TurboJPEG implementations has been maintained
since 2009. Nothing would prevent someone from implementing the
TurboJPEG API without libjpeg-turbo, but such an implementation would
not necessarily have an "underlying codec." (It could be fully
self-contained.)
- Use "destination image" rather than "output image", for consistency,
or describe the type of image that will be output.
- Avoid the term "image buffer" and instead use "byte buffer" to
refer to buffers that will hold JPEG images, or describe the type of
image that will be contained in the buffer. (The Java documentation
doesn't use "byte buffer", because the buffer arrays literally have
"byte" in front of them, and since Java doesn't have pointers, it is
not possible for mere mortals to store any other type of data in those
arrays.)
- C: Use "unified" to describe YUV images stored in a single buffer, for
consistency with the Java documentation.
- Use "planar YUV" rather than "YUV planar". Is is our convention to
describe images using {component layout} {colorspace/pixel format}
{image function}, e.g. "packed-pixel RGB source image" or "planar YUV
destination image."
- C: Document the TurboJPEG API version in which a particular function
or macro was introduced, and reorder the backward compatibility
function stubs in turbojpeg.h alphabetically by API version.
- C: Use Markdown rather than HTML tags, where possible, in the Doxygen
comments.
961 lines
33 KiB
Java
961 lines
33 KiB
Java
/*
|
|
* Copyright (C)2011-2018, 2023 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 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/decoding support");
|
|
System.out.println("-noyuvpad = do not pad each line of each Y, U, and V plane to the nearest");
|
|
System.out.println(" multiple of 4 bytes");
|
|
System.out.println("-bi = test BufferedImage support\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"
|
|
};
|
|
static final String[] SUBNAME = {
|
|
"444", "422", "420", "GRAY", "440", "411"
|
|
};
|
|
|
|
static final String[] PIXFORMATSTR = {
|
|
"RGB", "BGR", "RGBX", "BGRX", "XBGR", "XRGB", "Grayscale",
|
|
"RGBA", "BGRA", "ABGR", "ARGB", "CMYK"
|
|
};
|
|
|
|
static final int[] FORMATS_3BYTE = {
|
|
TJ.PF_RGB, TJ.PF_BGR
|
|
};
|
|
static final int[] FORMATS_3BYTEBI = {
|
|
BufferedImage.TYPE_3BYTE_BGR
|
|
};
|
|
static final int[] FORMATS_4BYTE = {
|
|
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 int yuvAlign = 4;
|
|
private static boolean bi = false;
|
|
|
|
private static int exitStatus = 0;
|
|
|
|
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 initBuf(byte[] buf, int w, int pitch, int h, int pf, int flags)
|
|
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) {
|
|
Arrays.fill(buf, (byte)0);
|
|
for (row = 0; row < h; row++) {
|
|
for (col = 0; col < w; col++) {
|
|
if ((flags & TJ.FLAG_BOTTOMUP) != 0)
|
|
index = pitch * (h - row - 1) + col;
|
|
else
|
|
index = pitch * row + col;
|
|
if (((row / 8) + (col / 8)) % 2 == 0)
|
|
buf[index] = (row < halfway) ? (byte)255 : 0;
|
|
else
|
|
buf[index] = (row < halfway) ? 76 : (byte)226;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
if (pf == TJ.PF_CMYK) {
|
|
Arrays.fill(buf, (byte)255);
|
|
for (row = 0; row < h; row++) {
|
|
for (col = 0; col < w; col++) {
|
|
if ((flags & TJ.FLAG_BOTTOMUP) != 0)
|
|
index = (h - row - 1) * w + col;
|
|
else
|
|
index = row * w + col;
|
|
if (((row / 8) + (col / 8)) % 2 == 0) {
|
|
if (row >= halfway) buf[index * ps + 3] = 0;
|
|
} else {
|
|
buf[index * ps + 2] = 0;
|
|
if (row < halfway)
|
|
buf[index * ps + 1] = 0;
|
|
}
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
Arrays.fill(buf, (byte)0);
|
|
for (row = 0; row < h; row++) {
|
|
for (col = 0; col < w; col++) {
|
|
if ((flags & TJ.FLAG_BOTTOMUP) != 0)
|
|
index = pitch * (h - row - 1) + col * ps;
|
|
else
|
|
index = pitch * row + col * ps;
|
|
if (((row / 8) + (col / 8)) % 2 == 0) {
|
|
if (row < halfway) {
|
|
buf[index + roffset] = (byte)255;
|
|
buf[index + goffset] = (byte)255;
|
|
buf[index + boffset] = (byte)255;
|
|
}
|
|
} else {
|
|
buf[index + roffset] = (byte)255;
|
|
if (row >= halfway)
|
|
buf[index + goffset] = (byte)255;
|
|
}
|
|
if (aoffset >= 0)
|
|
buf[index + aoffset] = (byte)255;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void initIntBuf(int[] buf, int w, int pitch, int h, int pf, int flags)
|
|
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 ((flags & TJ.FLAG_BOTTOMUP) != 0)
|
|
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, int flags) 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, 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);
|
|
}
|
|
}
|
|
|
|
static void checkVal(int row, int col, int v, String vname, int cv)
|
|
throws Exception {
|
|
v = (v < 0) ? v + 256 : v;
|
|
if (v < cv - 1 || v > cv + 1) {
|
|
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 > 1) {
|
|
throw new Exception("Comp. " + vname + " at " + row + "," + col +
|
|
" should be 0, not " + v);
|
|
}
|
|
}
|
|
|
|
static void checkVal255(int row, int col, int v, String vname)
|
|
throws Exception {
|
|
v = (v < 0) ? v + 256 : v;
|
|
if (v < 254) {
|
|
throw new Exception("Comp. " + vname + " at " + row + "," + col +
|
|
" should be 255, not " + v);
|
|
}
|
|
}
|
|
|
|
static int checkBuf(byte[] buf, int w, int pitch, int h, int pf, int subsamp,
|
|
TJScalingFactor sf, int flags) 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, retval = 1;
|
|
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 ((flags & TJ.FLAG_BOTTOMUP) != 0)
|
|
index = (h - row - 1) * w + col;
|
|
else
|
|
index = row * w + col;
|
|
byte c = buf[index * ps];
|
|
byte m = buf[index * ps + 1];
|
|
byte y = buf[index * ps + 2];
|
|
byte k = buf[index * ps + 3];
|
|
checkVal255(row, col, c, "C");
|
|
if (((row / blockSize) + (col / blockSize)) % 2 == 0) {
|
|
checkVal255(row, col, m, "M");
|
|
checkVal255(row, col, y, "Y");
|
|
if (row < halfway)
|
|
checkVal255(row, col, k, "K");
|
|
else
|
|
checkVal0(row, col, k, "K");
|
|
} else {
|
|
checkVal0(row, col, y, "Y");
|
|
checkVal255(row, col, k, "K");
|
|
if (row < halfway)
|
|
checkVal0(row, col, m, "M");
|
|
else
|
|
checkVal255(row, col, m, "M");
|
|
}
|
|
}
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
for (row = 0; row < halfway; row++) {
|
|
for (col = 0; col < w; col++) {
|
|
if ((flags & TJ.FLAG_BOTTOMUP) != 0)
|
|
index = pitch * (h - row - 1) + col * ps;
|
|
else
|
|
index = pitch * row + col * ps;
|
|
byte r = buf[index + roffset];
|
|
byte g = buf[index + goffset];
|
|
byte b = buf[index + boffset];
|
|
byte a = aoffset >= 0 ? buf[index + aoffset] : (byte)255;
|
|
if (((row / blockSize) + (col / blockSize)) % 2 == 0) {
|
|
if (row < halfway) {
|
|
checkVal255(row, col, r, "R");
|
|
checkVal255(row, col, g, "G");
|
|
checkVal255(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 {
|
|
checkVal255(row, col, r, "R");
|
|
if (row < halfway) {
|
|
checkVal0(row, col, g, "G");
|
|
} else {
|
|
checkVal255(row, col, g, "G");
|
|
}
|
|
checkVal0(row, col, b, "B");
|
|
}
|
|
}
|
|
checkVal255(row, col, a, "A");
|
|
}
|
|
}
|
|
} catch (Exception e) {
|
|
System.out.println("\n" + e.getMessage());
|
|
retval = 0;
|
|
}
|
|
|
|
if (retval == 0) {
|
|
for (row = 0; row < h; row++) {
|
|
for (col = 0; col < w; col++) {
|
|
if (pf == TJ.PF_CMYK) {
|
|
int c = buf[pitch * row + col * ps];
|
|
int m = buf[pitch * row + col * ps + 1];
|
|
int y = buf[pitch * row + col * ps + 2];
|
|
int k = buf[pitch * row + col * ps + 3];
|
|
if (c < 0) c += 256;
|
|
if (m < 0) m += 256;
|
|
if (y < 0) y += 256;
|
|
if (k < 0) k += 256;
|
|
System.out.format("%3d/%3d/%3d/%3d ", c, m, y, k);
|
|
} else {
|
|
int r = buf[pitch * row + col * ps + roffset];
|
|
int g = buf[pitch * row + col * ps + goffset];
|
|
int b = buf[pitch * row + col * 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;
|
|
}
|
|
|
|
static int checkIntBuf(int[] buf, int w, int pitch, int h, int pf,
|
|
int subsamp, TJScalingFactor sf, int flags)
|
|
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, retval = 1;
|
|
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 ((flags & TJ.FLAG_BOTTOMUP) != 0)
|
|
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) {
|
|
checkVal255(row, col, r, "R");
|
|
checkVal255(row, col, g, "G");
|
|
checkVal255(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 {
|
|
checkVal255(row, col, r, "R");
|
|
if (row < halfway) {
|
|
checkVal0(row, col, g, "G");
|
|
} else {
|
|
checkVal255(row, col, g, "G");
|
|
}
|
|
checkVal0(row, col, b, "B");
|
|
}
|
|
}
|
|
checkVal255(row, col, a, "A");
|
|
}
|
|
}
|
|
} catch (Exception e) {
|
|
System.out.println("\n" + e.getMessage());
|
|
retval = 0;
|
|
}
|
|
|
|
if (retval == 0) {
|
|
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 int checkImg(BufferedImage img, int pf, int subsamp,
|
|
TJScalingFactor sf, int flags) 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, 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,
|
|
sf, flags);
|
|
}
|
|
}
|
|
|
|
static int pad(int v, int p) {
|
|
return ((v + (p) - 1) & (~((p) - 1)));
|
|
}
|
|
|
|
static int 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);
|
|
int retval = 1;
|
|
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)
|
|
checkVal255(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);
|
|
checkVal255(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 = 0;
|
|
}
|
|
|
|
if (retval == 0) {
|
|
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, int subsamp, int jpegQual, int flags)
|
|
throws Exception {
|
|
String tempStr;
|
|
byte[] srcBuf = null;
|
|
BufferedImage img = null;
|
|
String pfStr, pfStrLong;
|
|
String buStr = (flags & TJ.FLAG_BOTTOMUP) != 0 ? "BU" : "TD";
|
|
String buStrLong = (flags & TJ.FLAG_BOTTOMUP) != 0 ?
|
|
"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, flags);
|
|
tempStr = baseName + "_enc_" + 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 {
|
|
srcBuf = new byte[w * h * ps + 1];
|
|
initBuf(srcBuf, w, w * ps, h, pf, flags);
|
|
tjc.setSourceImage(srcBuf, 0, 0, w, 0, h, pf);
|
|
}
|
|
Arrays.fill(dstBuf, (byte)0);
|
|
|
|
tjc.setSubsamp(subsamp);
|
|
tjc.setJPEGQuality(jpegQual);
|
|
if (doYUV) {
|
|
System.out.format("%s %s -> YUV %s ... ", pfStrLong, buStrLong,
|
|
SUBNAME_LONG[subsamp]);
|
|
YUVImage yuvImage = tjc.encodeYUV(yuvAlign, flags);
|
|
if (checkBufYUV(yuvImage.getBuf(), yuvImage.getSize(), w, h, subsamp,
|
|
new TJScalingFactor(1, 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 {
|
|
System.out.format("%s %s -> %s Q%d ... ", pfStrLong, buStrLong,
|
|
SUBNAME_LONG[subsamp], jpegQual);
|
|
}
|
|
tjc.compress(dstBuf, flags);
|
|
size = tjc.getCompressedSize();
|
|
|
|
tempStr = baseName + "_enc_" + 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,
|
|
int flags, TJScalingFactor sf) throws Exception {
|
|
String pfStr, pfStrLong, tempStr;
|
|
String buStrLong = (flags & TJ.FLAG_BOTTOMUP) != 0 ?
|
|
"Bottom-Up" : "Top-Down ";
|
|
int scaledWidth = sf.getScaled(w);
|
|
int scaledHeight = sf.getScaled(h);
|
|
int temp1, temp2, imgType = pf;
|
|
BufferedImage img = null;
|
|
byte[] dstBuf = null;
|
|
|
|
if (bi) {
|
|
pf = biTypePF(imgType);
|
|
pfStr = biTypeStr(imgType);
|
|
pfStrLong = pfStr + " (" + PIXFORMATSTR[pf] + ")";
|
|
} else {
|
|
pfStr = PIXFORMATSTR[pf];
|
|
pfStrLong = pfStr;
|
|
}
|
|
|
|
tjd.setSourceImage(jpegBuf, jpegSize);
|
|
if (tjd.getWidth() != w || tjd.getHeight() != h ||
|
|
tjd.getSubsamp() != subsamp)
|
|
throw new Exception("Incorrect JPEG header");
|
|
|
|
temp1 = scaledWidth;
|
|
temp2 = scaledHeight;
|
|
temp1 = tjd.getScaledWidth(temp1, temp2);
|
|
temp2 = tjd.getScaledHeight(temp1, temp2);
|
|
if (temp1 != scaledWidth || temp2 != scaledHeight)
|
|
throw new Exception("Scaled size mismatch");
|
|
|
|
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(scaledWidth, yuvAlign,
|
|
scaledHeight, flags);
|
|
if (checkBufYUV(yuvImage.getBuf(), yuvImage.getSize(), scaledWidth,
|
|
scaledHeight, subsamp, sf) == 1)
|
|
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.decompress(scaledWidth, scaledHeight, imgType, flags);
|
|
else
|
|
dstBuf = tjd.decompress(scaledWidth, 0, scaledHeight, pf, flags);
|
|
|
|
if (bi) {
|
|
tempStr = baseName + "_dec_" + pfStr + "_" +
|
|
(((flags & TJ.FLAG_BOTTOMUP) != 0) ? "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, flags) == 1) ||
|
|
(!bi && checkBuf(dstBuf, scaledWidth,
|
|
scaledWidth * TJ.getPixelSize(pf), scaledHeight, pf,
|
|
subsamp, sf, flags) == 1))
|
|
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,
|
|
int flags) throws Exception {
|
|
int i;
|
|
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 && num == 1 &&
|
|
(denom == 2 || denom == 1)) ||
|
|
(subsamp != TJ.SAMP_411 && num == 1 &&
|
|
(denom == 4 || denom == 2 || denom == 1)))
|
|
decompTest(tjd, jpegBuf, jpegSize, w, h, pf, baseName, subsamp,
|
|
flags, 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;
|
|
|
|
dstBuf = new byte[TJ.bufSize(w, h, subsamp)];
|
|
|
|
try {
|
|
tjc = new TJCompressor();
|
|
tjd = new TJDecompressor();
|
|
|
|
for (int pf : formats) {
|
|
if (pf < 0) continue;
|
|
for (int i = 0; i < 2; i++) {
|
|
int flags = 0;
|
|
if (subsamp == TJ.SAMP_422 || subsamp == TJ.SAMP_420 ||
|
|
subsamp == TJ.SAMP_440 || subsamp == TJ.SAMP_411)
|
|
flags |= TJ.FLAG_FASTUPSAMPLE;
|
|
if (i == 1)
|
|
flags |= TJ.FLAG_BOTTOMUP;
|
|
size = compTest(tjc, dstBuf, w, h, pf, baseName, subsamp, 100,
|
|
flags);
|
|
decompTest(tjd, dstBuf, size, w, h, pf, baseName, subsamp, flags);
|
|
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, flags);
|
|
}
|
|
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 bufSizeTest() throws Exception {
|
|
int w, h, i, subsamp;
|
|
byte[] srcBuf, dstBuf = null;
|
|
YUVImage dstImage = null;
|
|
TJCompressor tjc = null;
|
|
Random r = new Random();
|
|
|
|
try {
|
|
tjc = new TJCompressor();
|
|
System.out.println("Buffer size regression test");
|
|
for (subsamp = 0; subsamp < TJ.NUMSAMP; 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);
|
|
tjc.setSubsamp(subsamp);
|
|
tjc.setJPEGQuality(100);
|
|
if (doYUV)
|
|
tjc.encodeYUV(dstImage, 0);
|
|
else
|
|
tjc.compress(dstBuf, 0);
|
|
|
|
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, 0);
|
|
else
|
|
tjc.compress(dstBuf, 0);
|
|
}
|
|
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();
|
|
}
|
|
|
|
public static void main(String[] argv) {
|
|
try {
|
|
String testName = "javatest";
|
|
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("-bi")) {
|
|
bi = true;
|
|
testName = "javabitest";
|
|
} else
|
|
usage();
|
|
}
|
|
if (doYUV)
|
|
FORMATS_4BYTE[4] = -1;
|
|
doTest(35, 39, bi ? FORMATS_3BYTEBI : FORMATS_3BYTE, TJ.SAMP_444,
|
|
testName);
|
|
doTest(39, 41, bi ? FORMATS_4BYTEBI : FORMATS_4BYTE, TJ.SAMP_444,
|
|
testName);
|
|
doTest(41, 35, bi ? FORMATS_3BYTEBI : FORMATS_3BYTE, TJ.SAMP_422,
|
|
testName);
|
|
doTest(35, 39, bi ? FORMATS_4BYTEBI : FORMATS_4BYTE, TJ.SAMP_422,
|
|
testName);
|
|
doTest(39, 41, bi ? FORMATS_3BYTEBI : FORMATS_3BYTE, TJ.SAMP_420,
|
|
testName);
|
|
doTest(41, 35, bi ? FORMATS_4BYTEBI : FORMATS_4BYTE, TJ.SAMP_420,
|
|
testName);
|
|
doTest(35, 39, bi ? FORMATS_3BYTEBI : FORMATS_3BYTE, TJ.SAMP_440,
|
|
testName);
|
|
doTest(39, 41, bi ? FORMATS_4BYTEBI : FORMATS_4BYTE, TJ.SAMP_440,
|
|
testName);
|
|
doTest(41, 35, bi ? FORMATS_3BYTEBI : FORMATS_3BYTE, TJ.SAMP_411,
|
|
testName);
|
|
doTest(35, 39, bi ? FORMATS_4BYTEBI : FORMATS_4BYTE, TJ.SAMP_411,
|
|
testName);
|
|
doTest(39, 41, bi ? FORMATS_GRAYBI : FORMATS_GRAY, TJ.SAMP_GRAY,
|
|
testName);
|
|
doTest(41, 35, bi ? FORMATS_3BYTEBI : FORMATS_3BYTE, TJ.SAMP_GRAY,
|
|
testName);
|
|
FORMATS_4BYTE[4] = -1;
|
|
doTest(35, 39, bi ? FORMATS_4BYTEBI : FORMATS_4BYTE, 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_GRAY, "javatest_yuv0");
|
|
doTest(48, 48, FORMATS_GRAY, TJ.SAMP_GRAY, "javatest_yuv0");
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
exitStatus = -1;
|
|
}
|
|
System.exit(exitStatus);
|
|
}
|
|
}
|