summaryrefslogtreecommitdiff
path: root/java/TJUnitTest.java
diff options
context:
space:
mode:
authorElliott Hughes <enh@google.com>2020-12-02 18:09:57 -0800
committerElliott Hughes <enh@google.com>2020-12-02 18:24:38 -0800
commit98e581f8227b9846b7adc92c0c63f5ed2384ff4b (patch)
tree26effec72dbace7671158aeddc9fde1e3e9cd1a9 /java/TJUnitTest.java
parentff78ee5a324a7d37e0bfffd6152ea37056f29931 (diff)
parentd5148db386ceb4a608058320071cbed890bd6ad2 (diff)
Switch to chromium upstream.
Bug: https://issuetracker.google.com/135180511 Test: treehugger Change-Id: I0c78ec9b07debdb501a96df0d6cb2f9a24b5fc84
Diffstat (limited to 'java/TJUnitTest.java')
-rw-r--r--java/TJUnitTest.java960
1 files changed, 0 insertions, 960 deletions
diff --git a/java/TJUnitTest.java b/java/TJUnitTest.java
deleted file mode 100644
index 91ad5fd..0000000
--- a/java/TJUnitTest.java
+++ /dev/null
@@ -1,960 +0,0 @@
-/*
- * Copyright (C)2011-2018 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(" 4-byte boundary");
- 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 pad = 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, 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;
- }
-
- 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(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 ... ", 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, 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 ... ", 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, 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)
- 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);
- }
-}