Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(226)

Unified Diff: java/TJUnitTest.java

Issue 1934113002: Update libjpeg_turbo to 1.4.90 from https://github.com/libjpeg-turbo/ (Closed) Base URL: https://chromium.googlesource.com/chromium/deps/libjpeg_turbo.git@master
Patch Set: Created 4 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: java/TJUnitTest.java
diff --git a/java/TJUnitTest.java b/java/TJUnitTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..444e798530ac1634d8981cdebde989c7c3da1057
--- /dev/null
+++ b/java/TJUnitTest.java
@@ -0,0 +1,970 @@
+/*
+ * Copyright (C)2011-2016 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:\n");
+ System.out.println("-yuv = test YUV encoding/decoding support\n");
+ System.out.println("-noyuvpad = do not pad each line of each Y, U, and V plane to the nearest\n");
+ System.out.println(" 4-byte boundary\n");
+ 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[] alphaOffset = {
+ -1, -1, -1, -1, -1, -1, -1, 3, 3, 0, 0, -1
+ };
+
+ 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:
+ if (byteOrder == ByteOrder.BIG_ENDIAN)
+ return TJ.PF_XBGR;
+ else
+ return TJ.PF_RGBX;
+ case BufferedImage.TYPE_INT_RGB:
+ if (byteOrder == ByteOrder.BIG_ENDIAN)
+ return TJ.PF_XRGB;
+ else
+ return TJ.PF_BGRX;
+ case BufferedImage.TYPE_INT_ARGB:
+ case BufferedImage.TYPE_INT_ARGB_PRE:
+ if (byteOrder == ByteOrder.BIG_ENDIAN)
+ return TJ.PF_ARGB;
+ else
+ 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 = alphaOffset[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 = alphaOffset[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 = alphaOffset[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_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 = alphaOffset[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;
+ if (argv[i].equalsIgnoreCase("-noyuvpad"))
+ pad = 1;
+ if (argv[i].substring(0, 1).equalsIgnoreCase("-h") ||
+ argv[i].equalsIgnoreCase("-?"))
+ usage();
+ if (argv[i].equalsIgnoreCase("-bi")) {
+ bi = true;
+ testName = "javabitest";
+ }
+ }
+ 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);
+ }
+}

Powered by Google App Engine
This is Rietveld 408576698