19c791cdac
With rare exceptions ...
- Always separate line continuation characters by one space from
preceding code.
- Always use two-space indentation. Never use tabs.
- Always use K&R-style conditional blocks.
- Always surround operators with spaces, except in raw assembly code.
- Always put a space after, but not before, a comma.
- Never put a space between type casts and variables/function calls.
- Never put a space between the function name and the argument list in
function declarations and prototypes.
- Always surround braces ('{' and '}') with spaces.
- Always surround statements (if, for, else, catch, while, do, switch)
with spaces.
- Always attach pointer symbols ('*' and '**') to the variable or
function name.
- Always precede pointer symbols ('*' and '**') by a space in type
casts.
- Use the MIN() macro from jpegint.h within the libjpeg and TurboJPEG
API libraries (using min() from tjutil.h is still necessary for
TJBench.)
- Where it makes sense (particularly in the TurboJPEG code), put a blank
line after variable declaration blocks.
- Always separate statements in one-liners by two spaces.
The purpose of this was to ease maintenance on my part and also to make
it easier for contributors to figure out how to format patch
submissions. This was admittedly confusing (even to me sometimes) when
we had 3 or 4 different style conventions in the same source tree. The
new convention is more consistent with the formatting of other OSS code
bases.
This commit corrects deviations from the chosen formatting style in the
libjpeg API code and reformats the TurboJPEG API code such that it
conforms to the same standard.
NOTES:
- Although it is no longer necessary for the function name in function
declarations to begin in Column 1 (this was historically necessary
because of the ansi2knr utility, which allowed libjpeg to be built
with non-ANSI compilers), we retain that formatting for the libjpeg
code because it improves readability when using libjpeg's function
attribute macros (GLOBAL(), etc.)
- This reformatting project was accomplished with the help of AStyle and
Uncrustify, although neither was completely up to the task, and thus
a great deal of manual tweaking was required. Note to developers of
code formatting utilities: the libjpeg-turbo code base is an
excellent test bed, because AFAICT, it breaks every single one of the
utilities that are currently available.
- The legacy (MMX, SSE, 3DNow!) assembly code for i386 has been
formatted to match the SSE2 code (refer to
ff5685d5344273df321eb63a005eaae19d2496e3.) I hadn't intended to
bother with this, but the Loongson MMI implementation demonstrated
that there is still academic value to the MMX implementation, as an
algorithmic model for other 64-bit vector implementations. Thus, it
is desirable to improve its readability in the same manner as that of
the SSE2 implementation.
960 lines
33 KiB
Java
960 lines
33 KiB
Java
/*
|
|
* Copyright (C)2011-2017 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.*;
|
|
|
|
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:");
|
|
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(" 4-byte boundary");
|
|
System.out.println("-bi = test BufferedImage support\n");
|
|
System.exit(1);
|
|
}
|
|
|
|
private static final String[] subNameLong = {
|
|
"4:4:4", "4:2:2", "4:2:0", "GRAY", "4:4:0", "4:1:1"
|
|
};
|
|
private static final String[] subName = {
|
|
"444", "422", "420", "GRAY", "440", "411"
|
|
};
|
|
|
|
private static final String[] pixFormatStr = {
|
|
"RGB", "BGR", "RGBX", "BGRX", "XBGR", "XRGB", "Grayscale",
|
|
"RGBA", "BGRA", "ABGR", "ARGB", "CMYK"
|
|
};
|
|
|
|
private static final int[] _3byteFormats = {
|
|
TJ.PF_RGB, TJ.PF_BGR
|
|
};
|
|
private static final int[] _3byteFormatsBI = {
|
|
BufferedImage.TYPE_3BYTE_BGR
|
|
};
|
|
private static final int[] _4byteFormats = {
|
|
TJ.PF_RGBX, TJ.PF_BGRX, TJ.PF_XBGR, TJ.PF_XRGB, TJ.PF_CMYK
|
|
};
|
|
private static final int[] _4byteFormatsBI = {
|
|
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
|
|
};
|
|
private static final int[] onlyGray = {
|
|
TJ.PF_GRAY
|
|
};
|
|
private static final int[] onlyGrayBI = {
|
|
BufferedImage.TYPE_BYTE_GRAY
|
|
};
|
|
private static final int[] onlyRGB = {
|
|
TJ.PF_RGB
|
|
};
|
|
|
|
private static boolean doYUV = false;
|
|
private static int pad = 4;
|
|
private static boolean bi = false;
|
|
|
|
private static int exitStatus = 0;
|
|
|
|
private 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;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
private 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";
|
|
}
|
|
return "Unknown";
|
|
}
|
|
|
|
private 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;
|
|
}
|
|
}
|
|
}
|
|
|
|
private 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);
|
|
}
|
|
}
|
|
}
|
|
|
|
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 ||
|
|
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);
|
|
}
|
|
}
|
|
|
|
private 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);
|
|
}
|
|
}
|
|
|
|
private 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);
|
|
}
|
|
}
|
|
|
|
private 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);
|
|
}
|
|
}
|
|
|
|
private 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;
|
|
}
|
|
|
|
private 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;
|
|
}
|
|
|
|
private 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);
|
|
}
|
|
}
|
|
|
|
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, 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, pad), uvpitch = PAD(cw, pad);
|
|
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;
|
|
}
|
|
|
|
private 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();
|
|
}
|
|
|
|
private 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,
|
|
subNameLong[subsamp]);
|
|
YUVImage yuvImage = tjc.encodeYUV(pad, 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 ... ", subNameLong[subsamp],
|
|
buStrLong, jpegQual);
|
|
tjc.setSourceImage(yuvImage);
|
|
} else {
|
|
System.out.format("%s %s -> %s Q%d ... ", pfStrLong, buStrLong,
|
|
subNameLong[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;
|
|
}
|
|
|
|
private 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 ", subNameLong[subsamp]);
|
|
if (!sf.isOne())
|
|
System.out.format("%d/%d ... ", sf.getNum(), sf.getDenom());
|
|
else System.out.print("... ");
|
|
YUVImage yuvImage = tjd.decompressToYUV(scaledWidth, pad, 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 ... ", subNameLong[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;
|
|
}
|
|
}
|
|
|
|
private 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]);
|
|
}
|
|
}
|
|
|
|
private 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();
|
|
}
|
|
|
|
private 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, pad, 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, pad, 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"))
|
|
pad = 1;
|
|
else if (argv[i].equalsIgnoreCase("-bi")) {
|
|
bi = true;
|
|
testName = "javabitest";
|
|
} else
|
|
usage();
|
|
}
|
|
if (doYUV)
|
|
_4byteFormats[4] = -1;
|
|
doTest(35, 39, bi ? _3byteFormatsBI : _3byteFormats, TJ.SAMP_444,
|
|
testName);
|
|
doTest(39, 41, bi ? _4byteFormatsBI : _4byteFormats, TJ.SAMP_444,
|
|
testName);
|
|
doTest(41, 35, bi ? _3byteFormatsBI : _3byteFormats, TJ.SAMP_422,
|
|
testName);
|
|
doTest(35, 39, bi ? _4byteFormatsBI : _4byteFormats, TJ.SAMP_422,
|
|
testName);
|
|
doTest(39, 41, bi ? _3byteFormatsBI : _3byteFormats, TJ.SAMP_420,
|
|
testName);
|
|
doTest(41, 35, bi ? _4byteFormatsBI : _4byteFormats, TJ.SAMP_420,
|
|
testName);
|
|
doTest(35, 39, bi ? _3byteFormatsBI : _3byteFormats, TJ.SAMP_440,
|
|
testName);
|
|
doTest(39, 41, bi ? _4byteFormatsBI : _4byteFormats, TJ.SAMP_440,
|
|
testName);
|
|
doTest(41, 35, bi ? _3byteFormatsBI : _3byteFormats, TJ.SAMP_411,
|
|
testName);
|
|
doTest(35, 39, bi ? _4byteFormatsBI : _4byteFormats, TJ.SAMP_411,
|
|
testName);
|
|
doTest(39, 41, bi ? onlyGrayBI : onlyGray, TJ.SAMP_GRAY, testName);
|
|
doTest(41, 35, bi ? _3byteFormatsBI : _3byteFormats, TJ.SAMP_GRAY,
|
|
testName);
|
|
_4byteFormats[4] = -1;
|
|
doTest(35, 39, bi ? _4byteFormatsBI : _4byteFormats, TJ.SAMP_GRAY,
|
|
testName);
|
|
if (!bi)
|
|
bufSizeTest();
|
|
if (doYUV && !bi) {
|
|
System.out.print("\n--------------------\n\n");
|
|
doTest(48, 48, onlyRGB, TJ.SAMP_444, "javatest_yuv0");
|
|
doTest(48, 48, onlyRGB, TJ.SAMP_422, "javatest_yuv0");
|
|
doTest(48, 48, onlyRGB, TJ.SAMP_420, "javatest_yuv0");
|
|
doTest(48, 48, onlyRGB, TJ.SAMP_440, "javatest_yuv0");
|
|
doTest(48, 48, onlyRGB, TJ.SAMP_411, "javatest_yuv0");
|
|
doTest(48, 48, onlyRGB, TJ.SAMP_GRAY, "javatest_yuv0");
|
|
doTest(48, 48, onlyGray, TJ.SAMP_GRAY, "javatest_yuv0");
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
exitStatus = -1;
|
|
}
|
|
System.exit(exitStatus);
|
|
}
|
|
}
|