Merge branch 'upstream' into tizen_base
[platform/upstream/libjpeg-turbo.git] / tjunittest.c
index c543196..f95b51a 100644 (file)
@@ -1,5 +1,6 @@
 /*
- * Copyright (C)2009-2014, 2017-2018 D. R. Commander.  All Rights Reserved.
+ * Copyright (C)2009-2014, 2017-2019, 2022-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:
  * This program tests the various code paths in the TurboJPEG C Wrapper
  */
 
+#ifdef _MSC_VER
+#define _CRT_SECURE_NO_DEPRECATE
+#endif
+
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <limits.h>
 #include <errno.h>
 #include "tjutil.h"
 #include "turbojpeg.h"
 #include "md5/md5.h"
-#include "cmyk.h"
+#include "jconfigint.h"
 #ifdef _WIN32
 #include <time.h>
+#include <process.h>
 #define random()  rand()
+#define getpid()  _getpid()
 #else
 #include <unistd.h>
 #endif
 
 
-void usage(char *progName)
+static void usage(char *progName)
 {
   printf("\nUSAGE: %s [options]\n\n", progName);
   printf("Options:\n");
-  printf("-yuv = test YUV encoding/decoding support\n");
-  printf("-noyuvpad = do not pad each line of each Y, U, and V plane to the nearest\n");
-  printf("            4-byte boundary\n");
-  printf("-alloc = test automatic buffer allocation\n");
-  printf("-bmp = tjLoadImage()/tjSaveImage() unit test\n\n");
+  printf("-yuv = test YUV encoding/compression/decompression/decoding\n");
+  printf("       (8-bit data precision only)\n");
+  printf("-noyuvpad = do not pad each row in each Y, U, and V plane to the nearest\n");
+  printf("            multiple of 4 bytes\n");
+  printf("-precision N = test N-bit data precision (N is 8, 12, or 16; default is 8; if N\n");
+  printf("               is 16, then -lossless is implied)\n");
+  printf("-lossless = test lossless JPEG compression/decompression\n");
+  printf("-alloc = test automatic JPEG buffer allocation\n");
+  printf("-bmp = test packed-pixel image I/O\n");
   exit(1);
 }
 
 
-#define _throwtj() { \
-  printf("TurboJPEG ERROR:\n%s\n", tjGetErrorStr()); \
-  bailout() \
+#define THROW_TJ(handle) { \
+  printf("TurboJPEG ERROR:\n%s\n", tj3GetErrorStr(handle)); \
+  BAILOUT() \
 }
-#define _tj(f) { if ((f) == -1) _throwtj(); }
-#define _throw(m) { printf("ERROR: %s\n", m);  bailout() }
-#define _throwmd5(filename, md5sum, ref) { \
+#define TRY_TJ(handle, f) { if ((f) == -1) THROW_TJ(handle); }
+#define THROW(m) { printf("ERROR: %s\n", m);  BAILOUT() }
+#define THROW_MD5(filename, md5sum, ref) { \
   printf("\n%s has an MD5 sum of %s.\n   Should be %s.\n", filename, md5sum, \
          ref); \
-  bailout() \
+  BAILOUT() \
 }
 
 const char *subNameLong[TJ_NUMSAMP] = {
-  "4:4:4", "4:2:2", "4:2:0", "GRAY", "4:4:0", "4:1:1"
+  "4:4:4", "4:2:2", "4:2:0", "GRAY", "4:4:0", "4:1:1", "4:4:1"
 };
 const char *subName[TJ_NUMSAMP] = {
-  "444", "422", "420", "GRAY", "440", "411"
+  "444", "422", "420", "GRAY", "440", "411", "441"
 };
 
 const char *pixFormatStr[TJ_NUMPF] = {
@@ -83,67 +95,79 @@ const char *pixFormatStr[TJ_NUMPF] = {
   "RGBA", "BGRA", "ABGR", "ARGB", "CMYK"
 };
 
-const int _3byteFormats[] = { TJPF_RGB, TJPF_BGR };
-const int _4byteFormats[] = {
+const int _3sampleFormats[] = { TJPF_RGB, TJPF_BGR };
+const int _4sampleFormats[] = {
   TJPF_RGBX, TJPF_BGRX, TJPF_XBGR, TJPF_XRGB, TJPF_CMYK
 };
 const int _onlyGray[] = { TJPF_GRAY };
 const int _onlyRGB[] = { TJPF_RGB };
 
-int doYUV = 0, alloc = 0, pad = 4;
+int doYUV = 0, lossless = 0, psv = 1, alloc = 0, yuvAlign = 4;
+int precision = 8, sampleSize, maxSample, tolerance, redToY, yellowToY;
 
 int exitStatus = 0;
-#define bailout() { exitStatus = -1;  goto bailout; }
+#define BAILOUT() { exitStatus = -1;  goto bailout; }
+
 
+static void setVal(void *buf, int index, int value)
+{
+  if (precision == 8)
+    ((unsigned char *)buf)[index] = (unsigned char)value;
+  else if (precision == 12)
+    ((short *)buf)[index] = (short)value;
+  else
+    ((unsigned short *)buf)[index] = (unsigned short)value;
+}
 
-void initBuf(unsigned char *buf, int w, int h, int pf, int flags)
+static void initBuf(void *buf, int w, int h, int pf, int bottomUp)
 {
   int roffset = tjRedOffset[pf];
   int goffset = tjGreenOffset[pf];
   int boffset = tjBlueOffset[pf];
   int ps = tjPixelSize[pf];
-  int index, row, col, halfway = 16;
+  int i, index, row, col, halfway = 16;
 
   if (pf == TJPF_GRAY) {
-    memset(buf, 0, w * h * ps);
+    memset(buf, 0, w * h * ps * sampleSize);
     for (row = 0; row < h; row++) {
       for (col = 0; col < w; col++) {
-        if (flags & TJFLAG_BOTTOMUP) index = (h - row - 1) * w + col;
+        if (bottomUp) index = (h - row - 1) * w + col;
         else index = row * w + col;
         if (((row / 8) + (col / 8)) % 2 == 0)
-          buf[index] = (row < halfway) ? 255 : 0;
-        else buf[index] = (row < halfway) ? 76 : 226;
+          setVal(buf, index, (row < halfway) ? maxSample : 0);
+        else setVal(buf, index, (row < halfway) ? redToY : yellowToY);
       }
     }
   } else if (pf == TJPF_CMYK) {
-    memset(buf, 255, w * h * ps);
+    for (i = 0; i < w * h * ps; i++)
+      setVal(buf, i, maxSample);
     for (row = 0; row < h; row++) {
       for (col = 0; col < w; col++) {
-        if (flags & TJFLAG_BOTTOMUP) index = (h - row - 1) * w + col;
+        if (bottomUp) index = (h - row - 1) * w + col;
         else index = row * w + col;
         if (((row / 8) + (col / 8)) % 2 == 0) {
-          if (row >= halfway) buf[index * ps + 3] = 0;
+          if (row >= halfway) setVal(buf, index * ps + 3, 0);
         } else {
-          buf[index * ps + 2] = 0;
-          if (row < halfway) buf[index * ps + 1] = 0;
+          setVal(buf, index * ps + 2, 0);
+          if (row < halfway) setVal(buf, index * ps + 1, 0);
         }
       }
     }
   } else {
-    memset(buf, 0, w * h * ps);
+    memset(buf, 0, w * h * ps * sampleSize);
     for (row = 0; row < h; row++) {
       for (col = 0; col < w; col++) {
-        if (flags & TJFLAG_BOTTOMUP) index = (h - row - 1) * w + col;
+        if (bottomUp) index = (h - row - 1) * w + col;
         else index = row * w + col;
         if (((row / 8) + (col / 8)) % 2 == 0) {
           if (row < halfway) {
-            buf[index * ps + roffset] = 255;
-            buf[index * ps + goffset] = 255;
-            buf[index * ps + boffset] = 255;
+            setVal(buf, index * ps + roffset, maxSample);
+            setVal(buf, index * ps + goffset, maxSample);
+            setVal(buf, index * ps + boffset, maxSample);
           }
         } else {
-          buf[index * ps + roffset] = 255;
-          if (row >= halfway) buf[index * ps + goffset] = 255;
+          setVal(buf, index * ps + roffset, maxSample);
+          if (row >= halfway) setVal(buf, index * ps + goffset, maxSample);
         }
       }
     }
@@ -151,31 +175,42 @@ void initBuf(unsigned char *buf, int w, int h, int pf, int flags)
 }
 
 
-#define checkval(v, cv) { \
-  if (v < cv - 1 || v > cv + 1) { \
+#define CHECKVAL(v, cv) { \
+  if (v < cv - tolerance || v > cv + tolerance) { \
     printf("\nComp. %s at %d,%d should be %d, not %d\n", #v, row, col, cv, \
            v); \
     retval = 0;  exitStatus = -1;  goto bailout; \
   } \
 }
 
-#define checkval0(v) { \
-  if (v > 1) { \
+#define CHECKVAL0(v) { \
+  if (v > tolerance) { \
     printf("\nComp. %s at %d,%d should be 0, not %d\n", #v, row, col, v); \
     retval = 0;  exitStatus = -1;  goto bailout; \
   } \
 }
 
-#define checkval255(v) { \
-  if (v < 254) { \
-    printf("\nComp. %s at %d,%d should be 255, not %d\n", #v, row, col, v); \
+#define CHECKVALMAX(v) { \
+  if (v < maxSample - tolerance) { \
+    printf("\nComp. %s at %d,%d should be %d, not %d\n", #v, row, col, \
+           maxSample, v); \
     retval = 0;  exitStatus = -1;  goto bailout; \
   } \
 }
 
 
-int checkBuf(unsigned char *buf, int w, int h, int pf, int subsamp,
-             tjscalingfactor sf, int flags)
+static int getVal(void *buf, int index)
+{
+  if (precision == 8)
+    return ((unsigned char *)buf)[index];
+  else if (precision == 12)
+    return ((short *)buf)[index];
+  else
+    return ((unsigned short *)buf)[index];
+}
+
+static int checkBuf(void *buf, int w, int h, int pf,  int subsamp,
+                    tjscalingfactor sf, int bottomUp)
 {
   int roffset = tjRedOffset[pf];
   int goffset = tjGreenOffset[pf];
@@ -191,22 +226,22 @@ int checkBuf(unsigned char *buf, int w, int h, int pf, int subsamp,
   if (pf == TJPF_CMYK) {
     for (row = 0; row < h; row++) {
       for (col = 0; col < w; col++) {
-        unsigned char c, m, y, k;
+        int c, m, y, k;
 
-        if (flags & TJFLAG_BOTTOMUP) index = (h - row - 1) * w + col;
+        if (bottomUp) index = (h - row - 1) * w + col;
         else index = row * w + col;
-        c = buf[index * ps];
-        m = buf[index * ps + 1];
-        y = buf[index * ps + 2];
-        k = buf[index * ps + 3];
+        c = getVal(buf, index * ps);
+        m = getVal(buf, index * ps + 1);
+        y = getVal(buf, index * ps + 2);
+        k = getVal(buf, index * ps + 3);
         if (((row / blocksize) + (col / blocksize)) % 2 == 0) {
-          checkval255(c);  checkval255(m);  checkval255(y);
-          if (row < halfway) checkval255(k)
-          else checkval0(k)
+          CHECKVALMAX(c);  CHECKVALMAX(m);  CHECKVALMAX(y);
+          if (row < halfway) CHECKVALMAX(k)
+          else CHECKVAL0(k)
         } else {
-          checkval255(c);  checkval0(y);  checkval255(k);
-          if (row < halfway) checkval0(m)
-          else checkval255(m)
+          CHECKVALMAX(c);  CHECKVAL0(y);  CHECKVALMAX(k);
+          if (row < halfway) CHECKVAL0(m)
+          else CHECKVALMAX(m)
         }
       }
     }
@@ -215,36 +250,37 @@ int checkBuf(unsigned char *buf, int w, int h, int pf, int subsamp,
 
   for (row = 0; row < h; row++) {
     for (col = 0; col < w; col++) {
-      unsigned char r, g, b, a;
+      int r, g, b, a;
 
-      if (flags & TJFLAG_BOTTOMUP) index = (h - row - 1) * w + col;
+      if (bottomUp) index = (h - row - 1) * w + col;
       else index = row * w + col;
-      r = buf[index * ps + roffset];
-      g = buf[index * ps + goffset];
-      b = buf[index * ps + boffset];
-      a = aoffset >= 0 ? buf[index * ps + aoffset] : 0xFF;
+      r = getVal(buf, index * ps + roffset);
+      g = getVal(buf, index * ps + goffset);
+      b = getVal(buf, index * ps + boffset);
+      a = aoffset >= 0 ? getVal(buf, index * ps + aoffset) : maxSample;
       if (((row / blocksize) + (col / blocksize)) % 2 == 0) {
         if (row < halfway) {
-          checkval255(r);  checkval255(g);  checkval255(b);
+          CHECKVALMAX(r);  CHECKVALMAX(g);  CHECKVALMAX(b);
         } else {
-          checkval0(r);  checkval0(g);  checkval0(b);
+          CHECKVAL0(r);  CHECKVAL0(g);  CHECKVAL0(b);
         }
       } else {
         if (subsamp == TJSAMP_GRAY) {
           if (row < halfway) {
-            checkval(r, 76);  checkval(g, 76);  checkval(b, 76);
+            CHECKVAL(r, redToY);  CHECKVAL(g, redToY);  CHECKVAL(b, redToY);
           } else {
-            checkval(r, 226);  checkval(g, 226);  checkval(b, 226);
+            CHECKVAL(r, yellowToY);  CHECKVAL(g, yellowToY);
+            CHECKVAL(b, yellowToY);
           }
         } else {
           if (row < halfway) {
-            checkval255(r);  checkval0(g);  checkval0(b);
+            CHECKVALMAX(r);  CHECKVAL0(g);  CHECKVAL0(b);
           } else {
-            checkval255(r);  checkval255(g);  checkval0(b);
+            CHECKVALMAX(r);  CHECKVALMAX(g);  CHECKVAL0(b);
           }
         }
       }
-      checkval255(a);
+      CHECKVALMAX(a);
     }
   }
 
@@ -253,13 +289,15 @@ bailout:
     for (row = 0; row < h; row++) {
       for (col = 0; col < w; col++) {
         if (pf == TJPF_CMYK)
-          printf("%.3d/%.3d/%.3d/%.3d ", buf[(row * w + col) * ps],
-                 buf[(row * w + col) * ps + 1], buf[(row * w + col) * ps + 2],
-                 buf[(row * w + col) * ps + 3]);
+          printf("%.3d/%.3d/%.3d/%.3d ", getVal(buf, (row * w + col) * ps),
+                 getVal(buf, (row * w + col) * ps + 1),
+                 getVal(buf, (row * w + col) * ps + 2),
+                 getVal(buf, (row * w + col) * ps + 3));
         else
-          printf("%.3d/%.3d/%.3d ", buf[(row * w + col) * ps + roffset],
-                 buf[(row * w + col) * ps + goffset],
-                 buf[(row * w + col) * ps + boffset]);
+          printf("%.3d/%.3d/%.3d ",
+                 getVal(buf, (row * w + col) * ps + roffset),
+                 getVal(buf, (row * w + col) * ps + goffset),
+                 getVal(buf, (row * w + col) * ps + boffset));
       }
       printf("\n");
     }
@@ -270,14 +308,14 @@ bailout:
 
 #define PAD(v, p)  ((v + (p) - 1) & (~((p) - 1)))
 
-int checkBufYUV(unsigned char *buf, int w, int h, int subsamp,
-                tjscalingfactor sf)
+static int checkBufYUV(unsigned char *buf, int w, int h, int subsamp,
+                       tjscalingfactor sf)
 {
   int row, col;
   int hsf = tjMCUWidth[subsamp] / 8, vsf = tjMCUHeight[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 ypitch = PAD(pw, yuvAlign), uvpitch = PAD(cw, yuvAlign);
   int retval = 1;
   int halfway = 16 * sf.num / sf.denom;
   int blocksize = 8 * sf.num / sf.denom;
@@ -287,16 +325,16 @@ int checkBufYUV(unsigned char *buf, int w, int h, int subsamp,
       unsigned char y = buf[ypitch * row + col];
 
       if (((row / blocksize) + (col / blocksize)) % 2 == 0) {
-        if (row < halfway) checkval255(y)
-        else checkval0(y);
+        if (row < halfway) CHECKVALMAX(y)
+        else CHECKVAL0(y);
       } else {
-        if (row < halfway) checkval(y, 76)
-        else checkval(y, 226);
+        if (row < halfway) CHECKVAL(y, 76)
+        else CHECKVAL(y, 225);
       }
     }
   }
   if (subsamp != TJSAMP_GRAY) {
-    int halfway = 16 / vsf * sf.num / sf.denom;
+    halfway = 16 / vsf * sf.num / sf.denom;
 
     for (row = 0; row < ch; row++) {
       for (col = 0; col < cw; col++) {
@@ -304,12 +342,12 @@ int checkBufYUV(unsigned char *buf, int w, int h, int subsamp,
           v = buf[ypitch * ph + uvpitch * ch + (uvpitch * row + col)];
 
         if (((row * vsf / blocksize) + (col * hsf / blocksize)) % 2 == 0) {
-          checkval(u, 128);  checkval(v, 128);
+          CHECKVAL(u, 128);  CHECKVAL(v, 128);
         } else {
           if (row < halfway) {
-            checkval(u, 85);  checkval255(v);
+            CHECKVAL(u, 85);  CHECKVALMAX(v);
           } else {
-            checkval0(u);  checkval(v, 149);
+            CHECKVAL0(u);  CHECKVAL(v, 149);
           }
         }
       }
@@ -342,7 +380,7 @@ bailout:
 }
 
 
-void writeJPEG(unsigned char *jpegBuf, unsigned long jpegSize, char *filename)
+static void writeJPEG(unsigned char *jpegBuf, size_t jpegSize, char *filename)
 {
   FILE *file = fopen(filename, "wb");
 
@@ -354,7 +392,7 @@ void writeJPEG(unsigned char *jpegBuf, unsigned long jpegSize, char *filename)
 #else
     printf("ERROR: Could not write to %s.\n%s\n", filename, strerror(errno));
 #endif
-    bailout()
+    BAILOUT()
   }
 
 bailout:
@@ -362,148 +400,195 @@ bailout:
 }
 
 
-void compTest(tjhandle handle, unsigned char **dstBuf, unsigned long *dstSize,
-              int w, int h, int pf, char *basename, int subsamp, int jpegQual,
-              int flags)
+static void compTest(tjhandle handle, unsigned char **dstBuf, size_t *dstSize,
+                     int w, int h, int pf, char *basename)
 {
   char tempStr[1024];
-  unsigned char *srcBuf = NULL, *yuvBuf = NULL;
+  void *srcBuf = NULL;
+  unsigned char *yuvBuf = NULL;
   const char *pfStr = pixFormatStr[pf];
-  const char *buStrLong =
-    (flags & TJFLAG_BOTTOMUP) ? "Bottom-Up" : "Top-Down ";
-  const char *buStr = (flags & TJFLAG_BOTTOMUP) ? "BU" : "TD";
+  int bottomUp = tj3Get(handle, TJPARAM_BOTTOMUP);
+  int subsamp = tj3Get(handle, TJPARAM_SUBSAMP);
+  int jpegPSV = tj3Get(handle, TJPARAM_LOSSLESSPSV);
+  int jpegQual = tj3Get(handle, TJPARAM_QUALITY);
+  const char *buStrLong = bottomUp ? "Bottom-Up" : "Top-Down ";
+  const char *buStr = bottomUp ? "BU" : "TD";
 
-  if ((srcBuf = (unsigned char *)malloc(w * h * tjPixelSize[pf])) == NULL)
-    _throw("Memory allocation failure");
-  initBuf(srcBuf, w, h, pf, flags);
+  if ((srcBuf = malloc(w * h * tjPixelSize[pf] * sampleSize)) == NULL)
+      THROW("Memory allocation failure");
+  initBuf(srcBuf, w, h, pf, bottomUp);
 
   if (*dstBuf && *dstSize > 0) memset(*dstBuf, 0, *dstSize);
 
-  if (!alloc) flags |= TJFLAG_NOREALLOC;
   if (doYUV) {
-    unsigned long yuvSize = tjBufSizeYUV2(w, pad, h, subsamp);
+    size_t yuvSize = tj3YUVBufSize(w, yuvAlign, h, subsamp);
     tjscalingfactor sf = { 1, 1 };
-    tjhandle handle2 = tjInitCompress();
+    tjhandle handle2 = NULL;
 
-    if (!handle2) _throwtj();
+    if ((handle2 = tj3Init(TJINIT_COMPRESS)) == NULL)
+      THROW_TJ(NULL);
+    TRY_TJ(handle2, tj3Set(handle2, TJPARAM_BOTTOMUP, bottomUp));
+    TRY_TJ(handle2, tj3Set(handle2, TJPARAM_SUBSAMP, subsamp));
 
     if ((yuvBuf = (unsigned char *)malloc(yuvSize)) == NULL)
-      _throw("Memory allocation failure");
+      THROW("Memory allocation failure");
     memset(yuvBuf, 0, yuvSize);
 
     printf("%s %s -> YUV %s ... ", pfStr, buStrLong, subNameLong[subsamp]);
-    _tj(tjEncodeYUV3(handle2, srcBuf, w, 0, h, pf, yuvBuf, pad, subsamp,
-                     flags));
-    tjDestroy(handle2);
+    TRY_TJ(handle2, tj3EncodeYUV8(handle2, (unsigned char *)srcBuf, w, 0, h,
+                                  pf, yuvBuf, yuvAlign));
+    tj3Destroy(handle2);
     if (checkBufYUV(yuvBuf, w, h, subsamp, sf)) printf("Passed.\n");
     else printf("FAILED!\n");
 
     printf("YUV %s %s -> JPEG Q%d ... ", subNameLong[subsamp], buStrLong,
            jpegQual);
-    _tj(tjCompressFromYUV(handle, yuvBuf, w, pad, h, subsamp, dstBuf, dstSize,
-                          jpegQual, flags));
+    TRY_TJ(handle, tj3CompressFromYUV8(handle, yuvBuf, w, yuvAlign, h, dstBuf,
+                                       dstSize));
   } else {
-    printf("%s %s -> %s Q%d ... ", pfStr, buStrLong, subNameLong[subsamp],
-           jpegQual);
-    _tj(tjCompress2(handle, srcBuf, w, 0, h, pf, dstBuf, dstSize, subsamp,
-                    jpegQual, flags));
+    if (lossless)
+      printf("%s %s -> LOSSLESS PSV%d ... ", pfStr, buStrLong, jpegPSV);
+    else
+      printf("%s %s -> %s Q%d ... ", pfStr, buStrLong, subNameLong[subsamp],
+             jpegQual);
+    if (precision == 8) {
+      TRY_TJ(handle, tj3Compress8(handle, (unsigned char *)srcBuf, w, 0, h, pf,
+                                  dstBuf, dstSize));
+    } else if (precision == 12) {
+      TRY_TJ(handle, tj3Compress12(handle, (short *)srcBuf, w, 0, h, pf,
+                                   dstBuf, dstSize));
+    } else {
+      TRY_TJ(handle, tj3Compress16(handle, (unsigned short *)srcBuf, w, 0, h,
+                                   pf, dstBuf, dstSize));
+    }
   }
 
-  snprintf(tempStr, 1024, "%s_enc_%s_%s_%s_Q%d.jpg", basename, pfStr, buStr,
-           subName[subsamp], jpegQual);
+  if (lossless)
+    SNPRINTF(tempStr, 1024, "%s_enc%d_%s_%s_LOSSLESS_PSV%d.jpg", basename,
+             precision, pfStr, buStr, jpegPSV);
+  else
+    SNPRINTF(tempStr, 1024, "%s_enc%d_%s_%s_%s_Q%d.jpg", basename, precision,
+             pfStr, buStr, subName[subsamp], jpegQual);
   writeJPEG(*dstBuf, *dstSize, tempStr);
   printf("Done.\n  Result in %s\n", tempStr);
 
 bailout:
-  if (yuvBuf) free(yuvBuf);
-  if (srcBuf) free(srcBuf);
+  free(yuvBuf);
+  free(srcBuf);
 }
 
 
-void _decompTest(tjhandle handle, unsigned char *jpegBuf,
-                 unsigned long jpegSize, int w, int h, int pf, char *basename,
-                 int subsamp, int flags, tjscalingfactor sf)
+static void _decompTest(tjhandle handle, unsigned char *jpegBuf,
+                        size_t jpegSize, int w, int h, int pf, char *basename,
+                        int subsamp, tjscalingfactor sf)
 {
-  unsigned char *dstBuf = NULL, *yuvBuf = NULL;
-  int _hdrw = 0, _hdrh = 0, _hdrsubsamp = -1;
+  void *dstBuf = NULL;
+  unsigned char *yuvBuf = NULL;
+  int _hdrw = 0, _hdrh = 0, _hdrsubsamp;
   int scaledWidth = TJSCALED(w, sf);
   int scaledHeight = TJSCALED(h, sf);
-  unsigned long dstSize = 0;
+  size_t dstSize = 0;
+  int bottomUp = tj3Get(handle, TJPARAM_BOTTOMUP);
 
-  _tj(tjDecompressHeader2(handle, jpegBuf, jpegSize, &_hdrw, &_hdrh,
-                          &_hdrsubsamp));
+  TRY_TJ(handle, tj3SetScalingFactor(handle, sf));
+
+  TRY_TJ(handle, tj3DecompressHeader(handle, jpegBuf, jpegSize));
+  _hdrw = tj3Get(handle, TJPARAM_JPEGWIDTH);
+  _hdrh = tj3Get(handle, TJPARAM_JPEGHEIGHT);
+  _hdrsubsamp = tj3Get(handle, TJPARAM_SUBSAMP);
+  if (lossless && subsamp != TJSAMP_444 && subsamp != TJSAMP_GRAY)
+    subsamp = TJSAMP_444;
   if (_hdrw != w || _hdrh != h || _hdrsubsamp != subsamp)
-    _throw("Incorrect JPEG header");
+    THROW("Incorrect JPEG header");
 
   dstSize = scaledWidth * scaledHeight * tjPixelSize[pf];
-  if ((dstBuf = (unsigned char *)malloc(dstSize)) == NULL)
-    _throw("Memory allocation failure");
-  memset(dstBuf, 0, dstSize);
+  if ((dstBuf = malloc(dstSize * sampleSize)) == NULL)
+    THROW("Memory allocation failure");
+  memset(dstBuf, 0, dstSize * sampleSize);
 
   if (doYUV) {
-    unsigned long yuvSize = tjBufSizeYUV2(scaledWidth, pad, scaledHeight,
-                                          subsamp);
-    tjhandle handle2 = tjInitDecompress();
+    size_t yuvSize = tj3YUVBufSize(scaledWidth, yuvAlign, scaledHeight,
+                                   subsamp);
+    tjhandle handle2 = NULL;
 
-    if (!handle2) _throwtj();
+    if ((handle2 = tj3Init(TJINIT_DECOMPRESS)) == NULL)
+      THROW_TJ(NULL);
+    TRY_TJ(handle2, tj3Set(handle2, TJPARAM_BOTTOMUP, bottomUp));
+    TRY_TJ(handle2, tj3Set(handle2, TJPARAM_SUBSAMP, subsamp));
 
     if ((yuvBuf = (unsigned char *)malloc(yuvSize)) == NULL)
-      _throw("Memory allocation failure");
+      THROW("Memory allocation failure");
     memset(yuvBuf, 0, yuvSize);
 
     printf("JPEG -> YUV %s ", subNameLong[subsamp]);
     if (sf.num != 1 || sf.denom != 1)
       printf("%d/%d ... ", sf.num, sf.denom);
     else printf("... ");
-    _tj(tjDecompressToYUV2(handle, jpegBuf, jpegSize, yuvBuf, scaledWidth, pad,
-                           scaledHeight, flags));
+    TRY_TJ(handle, tj3DecompressToYUV8(handle, jpegBuf, jpegSize, yuvBuf,
+                                       yuvAlign));
     if (checkBufYUV(yuvBuf, scaledWidth, scaledHeight, subsamp, sf))
       printf("Passed.\n");
     else printf("FAILED!\n");
 
     printf("YUV %s -> %s %s ... ", subNameLong[subsamp], pixFormatStr[pf],
-           (flags & TJFLAG_BOTTOMUP) ? "Bottom-Up" : "Top-Down ");
-    _tj(tjDecodeYUV(handle2, yuvBuf, pad, subsamp, dstBuf, scaledWidth, 0,
-                    scaledHeight, pf, flags));
-    tjDestroy(handle2);
+           bottomUp ? "Bottom-Up" : "Top-Down ");
+    TRY_TJ(handle2, tj3DecodeYUV8(handle2, yuvBuf, yuvAlign,
+                                  (unsigned char *)dstBuf, scaledWidth, 0,
+                                  scaledHeight, pf));
+    tj3Destroy(handle2);
   } else {
     printf("JPEG -> %s %s ", pixFormatStr[pf],
-           (flags & TJFLAG_BOTTOMUP) ? "Bottom-Up" : "Top-Down ");
+           bottomUp ? "Bottom-Up" : "Top-Down ");
     if (sf.num != 1 || sf.denom != 1)
       printf("%d/%d ... ", sf.num, sf.denom);
     else printf("... ");
-    _tj(tjDecompress2(handle, jpegBuf, jpegSize, dstBuf, scaledWidth, 0,
-                      scaledHeight, pf, flags));
+    if (precision == 8) {
+      TRY_TJ(handle, tj3Decompress8(handle, jpegBuf, jpegSize,
+                                    (unsigned char *)dstBuf, 0, pf));
+    } else if (precision == 12) {
+      TRY_TJ(handle, tj3Decompress12(handle, jpegBuf, jpegSize,
+                                     (short *)dstBuf, 0, pf));
+    } else {
+      TRY_TJ(handle, tj3Decompress16(handle, jpegBuf, jpegSize,
+                                     (unsigned short *)dstBuf, 0, pf));
+    }
   }
 
-  if (checkBuf(dstBuf, scaledWidth, scaledHeight, pf, subsamp, sf, flags))
+  if (checkBuf(dstBuf, scaledWidth, scaledHeight, pf, subsamp, sf, bottomUp))
     printf("Passed.");
   else printf("FAILED!");
   printf("\n");
 
 bailout:
-  if (yuvBuf) free(yuvBuf);
-  if (dstBuf) free(dstBuf);
+  free(yuvBuf);
+  free(dstBuf);
 }
 
 
-void decompTest(tjhandle handle, unsigned char *jpegBuf,
-                unsigned long jpegSize, int w, int h, int pf, char *basename,
-                int subsamp, int flags)
+static void decompTest(tjhandle handle, unsigned char *jpegBuf,
+                       size_t jpegSize, int w, int h, int pf, char *basename,
+                       int subsamp)
 {
   int i, n = 0;
-  tjscalingfactor *sf = tjGetScalingFactors(&n);
+  tjscalingfactor *sf = NULL;
 
-  if (!sf || !n) _throwtj();
+  if (lossless) {
+    _decompTest(handle, jpegBuf, jpegSize, w, h, pf, basename, subsamp,
+                TJUNSCALED);
+    return;
+  }
+
+  sf = tj3GetScalingFactors(&n);
+  if (!sf || !n) THROW_TJ(NULL);
 
   for (i = 0; i < n; i++) {
     if (subsamp == TJSAMP_444 || subsamp == TJSAMP_GRAY ||
-        (subsamp == TJSAMP_411 && sf[i].num == 1 &&
+        ((subsamp == TJSAMP_411 || subsamp == TJSAMP_441) && sf[i].num == 1 &&
          (sf[i].denom == 2 || sf[i].denom == 1)) ||
-        (subsamp != TJSAMP_411 && sf[i].num == 1 &&
+        (subsamp != TJSAMP_411 && subsamp != TJSAMP_441 && sf[i].num == 1 &&
          (sf[i].denom == 4 || sf[i].denom == 2 || sf[i].denom == 1)))
       _decompTest(handle, jpegBuf, jpegSize, w, h, pf, basename, subsamp,
-                  flags, sf[i]);
+                  sf[i]);
   }
 
 bailout:
@@ -511,40 +596,52 @@ bailout:
 }
 
 
-void doTest(int w, int h, const int *formats, int nformats, int subsamp,
-            char *basename)
+static void doTest(int w, int h, const int *formats, int nformats, int subsamp,
+                   char *basename)
 {
   tjhandle chandle = NULL, dhandle = NULL;
   unsigned char *dstBuf = NULL;
-  unsigned long size = 0;
+  size_t size = 0;
   int pfi, pf, i;
 
+  if (lossless && subsamp != TJSAMP_GRAY)
+    subsamp = TJSAMP_444;
+
   if (!alloc)
-    size = tjBufSize(w, h, subsamp);
+    size = tj3JPEGBufSize(w, h, subsamp);
   if (size != 0)
-    if ((dstBuf = (unsigned char *)tjAlloc(size)) == NULL)
-      _throw("Memory allocation failure.");
-
-  if ((chandle = tjInitCompress()) == NULL ||
-      (dhandle = tjInitDecompress()) == NULL)
-    _throwtj();
+    if ((dstBuf = (unsigned char *)tj3Alloc(size)) == NULL)
+      THROW("Memory allocation failure.");
+
+  if ((chandle = tj3Init(TJINIT_COMPRESS)) == NULL ||
+      (dhandle = tj3Init(TJINIT_DECOMPRESS)) == NULL)
+    THROW_TJ(NULL);
+
+  TRY_TJ(chandle, tj3Set(chandle, TJPARAM_NOREALLOC, !alloc));
+  if (lossless) {
+    TRY_TJ(chandle, tj3Set(chandle, TJPARAM_LOSSLESS, lossless));
+    TRY_TJ(chandle, tj3Set(chandle, TJPARAM_LOSSLESSPSV,
+                           ((psv++ - 1) % 7) + 1));
+  } else {
+    TRY_TJ(chandle, tj3Set(chandle, TJPARAM_QUALITY, 100));
+    if (subsamp == TJSAMP_422 || subsamp == TJSAMP_420 ||
+        subsamp == TJSAMP_440 || subsamp == TJSAMP_411 ||
+        subsamp == TJSAMP_441)
+      TRY_TJ(dhandle, tj3Set(dhandle, TJPARAM_FASTUPSAMPLE, 1));
+  }
+  TRY_TJ(chandle, tj3Set(chandle, TJPARAM_SUBSAMP, subsamp));
 
   for (pfi = 0; pfi < nformats; pfi++) {
     for (i = 0; i < 2; i++) {
-      int flags = 0;
-
-      if (subsamp == TJSAMP_422 || subsamp == TJSAMP_420 ||
-          subsamp == TJSAMP_440 || subsamp == TJSAMP_411)
-        flags |= TJFLAG_FASTUPSAMPLE;
-      if (i == 1) flags |= TJFLAG_BOTTOMUP;
+      TRY_TJ(chandle, tj3Set(chandle, TJPARAM_BOTTOMUP, i == 1));
+      TRY_TJ(dhandle, tj3Set(dhandle, TJPARAM_BOTTOMUP, i == 1));
       pf = formats[pfi];
-      compTest(chandle, &dstBuf, &size, w, h, pf, basename, subsamp, 100,
-               flags);
-      decompTest(dhandle, dstBuf, size, w, h, pf, basename, subsamp, flags);
+      compTest(chandle, &dstBuf, &size, w, h, pf, basename);
+      decompTest(dhandle, dstBuf, size, w, h, pf, basename, subsamp);
       if (pf >= TJPF_RGBX && pf <= TJPF_XRGB) {
         printf("\n");
         decompTest(dhandle, dstBuf, size, w, h, pf + (TJPF_RGBA - TJPF_RGBX),
-                   basename, subsamp, flags);
+                   basename, subsamp);
       }
       printf("\n");
     }
@@ -552,80 +649,175 @@ void doTest(int w, int h, const int *formats, int nformats, int subsamp,
   printf("--------------------\n\n");
 
 bailout:
-  if (chandle) tjDestroy(chandle);
-  if (dhandle) tjDestroy(dhandle);
-  if (dstBuf) tjFree(dstBuf);
+  tj3Destroy(chandle);
+  tj3Destroy(dhandle);
+  tj3Free(dstBuf);
 }
 
 
-void bufSizeTest(void)
+#if SIZEOF_SIZE_T == 8
+#define CHECKSIZE(function) { \
+  if (size && size < (size_t)0xFFFFFFFF) \
+    THROW(#function " overflow"); \
+}
+#define CHECKSIZEUL(function) { \
+  if ((unsigned long long)ulsize < (unsigned long long)0xFFFFFFFF) \
+    THROW(#function " overflow"); \
+}
+#else
+#define CHECKSIZE(function) { \
+  if (size != 0 || !strcmp(tj3GetErrorStr(NULL), "No error")) \
+    THROW(#function " overflow"); \
+}
+#define CHECKSIZEUL(function) { \
+  if (ulsize != (unsigned long)(-1) || \
+      !strcmp(tj3GetErrorStr(NULL), "No error")) \
+    THROW(#function " overflow"); \
+}
+#endif
+#define CHECKSIZEINT(function) { \
+  if (intsize != 0 || !strcmp(tj3GetErrorStr(NULL), "No error")) \
+    THROW(#function " overflow"); \
+}
+
+static void overflowTest(void)
+{
+  /* Ensure that the various buffer size functions don't overflow */
+  size_t size;
+  unsigned long ulsize;
+  int intsize;
+
+  size = tj3JPEGBufSize(26755, 26755, TJSAMP_444);
+  CHECKSIZE(tj3JPEGBufSize());
+  ulsize = tjBufSize(26755, 26755, TJSAMP_444);
+  CHECKSIZEUL(tjBufSize());
+  ulsize = TJBUFSIZE(26755, 26755);
+  CHECKSIZEUL(TJBUFSIZE());
+  size = tj3YUVBufSize(37838, 1, 37838, TJSAMP_444);
+  CHECKSIZE(tj3YUVBufSize());
+  size = tj3YUVBufSize(37837, 3, 37837, TJSAMP_444);
+  CHECKSIZE(tj3YUVBufSize());
+  size = tj3YUVBufSize(37837, -1, 37837, TJSAMP_444);
+  CHECKSIZE(tj3YUVBufSize());
+  ulsize = tjBufSizeYUV2(37838, 1, 37838, TJSAMP_444);
+  CHECKSIZEUL(tjBufSizeYUV2());
+  ulsize = tjBufSizeYUV2(37837, 3, 37837, TJSAMP_444);
+  CHECKSIZEUL(tjBufSizeYUV2());
+  ulsize = tjBufSizeYUV2(37837, -1, 37837, TJSAMP_444);
+  CHECKSIZEUL(tjBufSizeYUV2());
+  ulsize = TJBUFSIZEYUV(37838, 37838, TJSAMP_444);
+  CHECKSIZEUL(TJBUFSIZEYUV());
+  ulsize = tjBufSizeYUV(37838, 37838, TJSAMP_444);
+  CHECKSIZEUL(tjBufSizeYUV());
+  size = tj3YUVPlaneSize(0, 65536, 0, 65536, TJSAMP_444);
+  CHECKSIZE(tj3YUVPlaneSize());
+  ulsize = tjPlaneSizeYUV(0, 65536, 0, 65536, TJSAMP_444);
+  CHECKSIZEUL(tjPlaneSizeYUV());
+  intsize = tj3YUVPlaneWidth(0, INT_MAX, TJSAMP_420);
+  CHECKSIZEINT(tj3YUVPlaneWidth());
+  intsize = tj3YUVPlaneHeight(0, INT_MAX, TJSAMP_420);
+  CHECKSIZEINT(tj3YUVPlaneHeight());
+
+bailout:
+  return;
+}
+
+
+static void bufSizeTest(void)
 {
   int w, h, i, subsamp;
-  unsigned char *srcBuf = NULL, *dstBuf = NULL;
+  void *srcBuf = NULL;
+  unsigned char *dstBuf = NULL;
   tjhandle handle = NULL;
-  unsigned long dstSize = 0;
+  size_t dstSize = 0;
+  int numSamp = TJ_NUMSAMP;
 
-  if ((handle = tjInitCompress()) == NULL) _throwtj();
+  if ((handle = tj3Init(TJINIT_COMPRESS)) == NULL)
+    THROW_TJ(NULL);
+
+  TRY_TJ(handle, tj3Set(handle, TJPARAM_NOREALLOC, !alloc));
+  if (lossless) {
+    TRY_TJ(handle, tj3Set(handle, TJPARAM_LOSSLESS, lossless));
+    TRY_TJ(handle, tj3Set(handle, TJPARAM_LOSSLESSPSV,
+                          ((psv++ - 1) % 7) + 1));
+    numSamp = 1;
+  } else
+    TRY_TJ(handle, tj3Set(handle, TJPARAM_QUALITY, 100));
 
   printf("Buffer size regression test\n");
-  for (subsamp = 0; subsamp < TJ_NUMSAMP; subsamp++) {
+  for (subsamp = 0; subsamp < numSamp; subsamp++) {
+    TRY_TJ(handle, tj3Set(handle, TJPARAM_SUBSAMP, subsamp));
     for (w = 1; w < 48; w++) {
       int maxh = (w == 1) ? 2048 : 48;
 
       for (h = 1; h < maxh; h++) {
         if (h % 100 == 0) printf("%.4d x %.4d\b\b\b\b\b\b\b\b\b\b\b", w, h);
-        if ((srcBuf = (unsigned char *)malloc(w * h * 4)) == NULL)
-          _throw("Memory allocation failure");
+        if ((srcBuf = malloc(w * h * 4 * sampleSize)) == NULL)
+          THROW("Memory allocation failure");
         if (!alloc || doYUV) {
-          if (doYUV) dstSize = tjBufSizeYUV2(w, pad, h, subsamp);
-          else dstSize = tjBufSize(w, h, subsamp);
-          if ((dstBuf = (unsigned char *)tjAlloc(dstSize)) == NULL)
-            _throw("Memory allocation failure");
+          if (doYUV) dstSize = tj3YUVBufSize(w, yuvAlign, h, subsamp);
+          else dstSize = tj3JPEGBufSize(w, h, subsamp);
+          if ((dstBuf = (unsigned char *)tj3Alloc(dstSize)) == NULL)
+            THROW("Memory allocation failure");
         }
 
         for (i = 0; i < w * h * 4; i++) {
-          if (random() < RAND_MAX / 2) srcBuf[i] = 0;
-          else srcBuf[i] = 255;
+          if (random() < RAND_MAX / 2) setVal(srcBuf, i, 0);
+          else setVal(srcBuf, i, maxSample);
         }
 
         if (doYUV) {
-          _tj(tjEncodeYUV3(handle, srcBuf, w, 0, h, TJPF_BGRX, dstBuf, pad,
-                           subsamp, 0));
+          TRY_TJ(handle, tj3EncodeYUV8(handle, (unsigned char *)srcBuf, w, 0,
+                                       h, TJPF_BGRX, dstBuf, yuvAlign));
         } else {
-          _tj(tjCompress2(handle, srcBuf, w, 0, h, TJPF_BGRX, &dstBuf,
-                          &dstSize, subsamp, 100,
-                          alloc ? 0 : TJFLAG_NOREALLOC));
+          if (precision == 8) {
+            TRY_TJ(handle, tj3Compress8(handle, (unsigned char *)srcBuf, w, 0,
+                                        h, TJPF_BGRX, &dstBuf, &dstSize));
+          } else if (precision == 12) {
+            TRY_TJ(handle, tj3Compress12(handle, (short *)srcBuf, w, 0, h,
+                                         TJPF_BGRX, &dstBuf, &dstSize));
+          } else {
+            TRY_TJ(handle, tj3Compress16(handle, (unsigned short *)srcBuf, w,
+                                         0, h, TJPF_BGRX, &dstBuf, &dstSize));
+          }
         }
         free(srcBuf);  srcBuf = NULL;
         if (!alloc || doYUV) {
-          tjFree(dstBuf);  dstBuf = NULL;
+          tj3Free(dstBuf);  dstBuf = NULL;
         }
 
-        if ((srcBuf = (unsigned char *)malloc(h * w * 4)) == NULL)
-          _throw("Memory allocation failure");
+        if ((srcBuf = malloc(h * w * 4 * sampleSize)) == NULL)
+          THROW("Memory allocation failure");
         if (!alloc || doYUV) {
-          if (doYUV) dstSize = tjBufSizeYUV2(h, pad, w, subsamp);
-          else dstSize = tjBufSize(h, w, subsamp);
-          if ((dstBuf = (unsigned char *)tjAlloc(dstSize)) == NULL)
-            _throw("Memory allocation failure");
+          if (doYUV) dstSize = tj3YUVBufSize(h, yuvAlign, w, subsamp);
+          else dstSize = tj3JPEGBufSize(h, w, subsamp);
+          if ((dstBuf = (unsigned char *)tj3Alloc(dstSize)) == NULL)
+            THROW("Memory allocation failure");
         }
 
         for (i = 0; i < h * w * 4; i++) {
-          if (random() < RAND_MAX / 2) srcBuf[i] = 0;
-          else srcBuf[i] = 255;
+          if (random() < RAND_MAX / 2) setVal(srcBuf, i, 0);
+          else setVal(srcBuf, i, maxSample);
         }
 
         if (doYUV) {
-          _tj(tjEncodeYUV3(handle, srcBuf, h, 0, w, TJPF_BGRX, dstBuf, pad,
-                           subsamp, 0));
+          TRY_TJ(handle, tj3EncodeYUV8(handle, (unsigned char *)srcBuf, h, 0,
+                                       w, TJPF_BGRX, dstBuf, yuvAlign));
         } else {
-          _tj(tjCompress2(handle, srcBuf, h, 0, w, TJPF_BGRX, &dstBuf,
-                          &dstSize, subsamp, 100,
-                          alloc ? 0 : TJFLAG_NOREALLOC));
+          if (precision == 8) {
+            TRY_TJ(handle, tj3Compress8(handle, (unsigned char *)srcBuf, h, 0,
+                                        w, TJPF_BGRX, &dstBuf, &dstSize));
+          } else if (precision == 12) {
+            TRY_TJ(handle, tj3Compress12(handle, (short *)srcBuf, h, 0, w,
+                                         TJPF_BGRX, &dstBuf, &dstSize));
+          } else {
+            TRY_TJ(handle, tj3Compress16(handle, (unsigned short *)srcBuf, h,
+                                         0, w, TJPF_BGRX, &dstBuf, &dstSize));
+          }
         }
         free(srcBuf);  srcBuf = NULL;
         if (!alloc || doYUV) {
-          tjFree(dstBuf);  dstBuf = NULL;
+          tj3Free(dstBuf);  dstBuf = NULL;
         }
       }
     }
@@ -633,48 +825,79 @@ void bufSizeTest(void)
   printf("Done.      \n");
 
 bailout:
-  if (srcBuf) free(srcBuf);
-  if (dstBuf) tjFree(dstBuf);
-  if (handle) tjDestroy(handle);
+  free(srcBuf);
+  tj3Free(dstBuf);
+  tj3Destroy(handle);
 }
 
 
-void initBitmap(unsigned char *buf, int width, int pitch, int height, int pf,
-                int flags)
+static void rgb_to_cmyk(int r, int g, int b, int *c, int *m, int *y, int *k)
+{
+  double ctmp = 1.0 - ((double)r / (double)maxSample);
+  double mtmp = 1.0 - ((double)g / (double)maxSample);
+  double ytmp = 1.0 - ((double)b / (double)maxSample);
+  double ktmp = min(min(ctmp, mtmp), ytmp);
+
+  if (ktmp == 1.0) ctmp = mtmp = ytmp = 0.0;
+  else {
+    ctmp = (ctmp - ktmp) / (1.0 - ktmp);
+    mtmp = (mtmp - ktmp) / (1.0 - ktmp);
+    ytmp = (ytmp - ktmp) / (1.0 - ktmp);
+  }
+  *c = (int)((double)maxSample - ctmp * (double)maxSample + 0.5);
+  *m = (int)((double)maxSample - mtmp * (double)maxSample + 0.5);
+  *y = (int)((double)maxSample - ytmp * (double)maxSample + 0.5);
+  *k = (int)((double)maxSample - ktmp * (double)maxSample + 0.5);
+}
+
+static void initBitmap(void *buf, int width, int pitch, int height, int pf,
+                       int bottomUp)
 {
   int roffset = tjRedOffset[pf];
   int goffset = tjGreenOffset[pf];
   int boffset = tjBlueOffset[pf];
   int ps = tjPixelSize[pf];
-  int i, j;
+  int i, j, ci;
 
   for (j = 0; j < height; j++) {
-    int row = (flags & TJFLAG_BOTTOMUP) ? height - j - 1 : j;
+    int row = bottomUp ? height - j - 1 : j;
 
     for (i = 0; i < width; i++) {
-      unsigned char r = (i * 256 / width) % 256;
-      unsigned char g = (j * 256 / height) % 256;
-      unsigned char b = (j * 256 / height + i * 256 / width) % 256;
-
-      memset(&buf[row * pitch + i * ps], 0, ps);
-      if (pf == TJPF_GRAY) buf[row * pitch + i * ps] = b;
-      else if (pf == TJPF_CMYK)
-        rgb_to_cmyk(r, g, b, &buf[row * pitch + i * ps + 0],
-                    &buf[row * pitch + i * ps + 1],
-                    &buf[row * pitch + i * ps + 2],
-                    &buf[row * pitch + i * ps + 3]);
-      else {
-        buf[row * pitch + i * ps + roffset] = r;
-        buf[row * pitch + i * ps + goffset] = g;
-        buf[row * pitch + i * ps + boffset] = b;
+      int r = (i * (maxSample + 1) / width) % (maxSample + 1);
+      int g = (j * (maxSample + 1) / height) % (maxSample + 1);
+      int b = (j * (maxSample + 1) / height +
+               i * (maxSample + 1) / width) % (maxSample + 1);
+
+      for (ci = 0; ci < ps; ci++)
+        setVal(buf, row * pitch + i * ps + ci, 0);
+      if (pf == TJPF_GRAY) setVal(buf, row * pitch + i * ps, b);
+      else if (pf == TJPF_CMYK) {
+        int c, m, y, k;
+
+        rgb_to_cmyk(r, g, b, &c, &m, &y, &k);
+        setVal(buf, row * pitch + i * ps + 0, c);
+        setVal(buf, row * pitch + i * ps + 1, m);
+        setVal(buf, row * pitch + i * ps + 2, y);
+        setVal(buf, row * pitch + i * ps + 3, k);
+      } else {
+        setVal(buf, row * pitch + i * ps + roffset, r);
+        setVal(buf, row * pitch + i * ps + goffset, g);
+        setVal(buf, row * pitch + i * ps + boffset, b);
       }
     }
   }
 }
 
 
-int cmpBitmap(unsigned char *buf, int width, int pitch, int height, int pf,
-              int flags, int gray2rgb)
+static void cmyk_to_rgb(int c, int m, int y, int k, int *r, int *g, int *b)
+{
+  *r = (int)((double)c * (double)k / (double)maxSample + 0.5);
+  *g = (int)((double)m * (double)k / (double)maxSample + 0.5);
+  *b = (int)((double)y * (double)k / (double)maxSample + 0.5);
+}
+
+static int cmpBitmap(void *buf, int width, int pitch, int height, int pf,
+                     int bottomUp, int gray2rgb)
 {
   int roffset = tjRedOffset[pf];
   int goffset = tjGreenOffset[pf];
@@ -684,38 +907,40 @@ int cmpBitmap(unsigned char *buf, int width, int pitch, int height, int pf,
   int i, j;
 
   for (j = 0; j < height; j++) {
-    int row = (flags & TJFLAG_BOTTOMUP) ? height - j - 1 : j;
+    int row = bottomUp ? height - j - 1 : j;
 
     for (i = 0; i < width; i++) {
-      unsigned char r = (i * 256 / width) % 256;
-      unsigned char g = (j * 256 / height) % 256;
-      unsigned char b = (j * 256 / height + i * 256 / width) % 256;
+      int r = (i * (maxSample + 1) / width) % (maxSample + 1);
+      int g = (j * (maxSample + 1) / height) % (maxSample + 1);
+      int b = (j * (maxSample + 1) / height +
+               i * (maxSample + 1) / width) % (maxSample + 1);
 
       if (pf == TJPF_GRAY) {
-        if (buf[row * pitch + i * ps] != b)
+        if (getVal(buf, row * pitch + i * ps) != b)
           return 0;
       } else if (pf == TJPF_CMYK) {
-        unsigned char rf, gf, bf;
+        int rf, gf, bf;
 
-        cmyk_to_rgb(buf[row * pitch + i * ps + 0],
-                    buf[row * pitch + i * ps + 1],
-                    buf[row * pitch + i * ps + 2],
-                    buf[row * pitch + i * ps + 3], &rf, &gf, &bf);
+        cmyk_to_rgb(getVal(buf, row * pitch + i * ps + 0),
+                    getVal(buf, row * pitch + i * ps + 1),
+                    getVal(buf, row * pitch + i * ps + 2),
+                    getVal(buf, row * pitch + i * ps + 3), &rf, &gf, &bf);
         if (gray2rgb) {
           if (rf != b || gf != b || bf != b)
             return 0;
         } else if (rf != r || gf != g || bf != b) return 0;
       } else {
         if (gray2rgb) {
-          if (buf[row * pitch + i * ps + roffset] != b ||
-              buf[row * pitch + i * ps + goffset] != b ||
-              buf[row * pitch + i * ps + boffset] != b)
+          if (getVal(buf, row * pitch + i * ps + roffset) != b ||
+              getVal(buf, row * pitch + i * ps + goffset) != b ||
+              getVal(buf, row * pitch + i * ps + boffset) != b)
             return 0;
-        } else if (buf[row * pitch + i * ps + roffset] != r ||
-                   buf[row * pitch + i * ps + goffset] != g ||
-                   buf[row * pitch + i * ps + boffset] != b)
+        } else if (getVal(buf, row * pitch + i * ps + roffset) != r ||
+                   getVal(buf, row * pitch + i * ps + goffset) != g ||
+                   getVal(buf, row * pitch + i * ps + boffset) != b)
           return 0;
-        if (aoffset >= 0 && buf[row * pitch + i * ps + aoffset] != 0xFF)
+        if (aoffset >= 0 &&
+            getVal(buf, row * pitch + i * ps + aoffset) != maxSample)
           return 0;
       }
     }
@@ -724,124 +949,195 @@ int cmpBitmap(unsigned char *buf, int width, int pitch, int height, int pf,
 }
 
 
-int doBmpTest(const char *ext, int width, int align, int height, int pf,
-              int flags)
+static int doBmpTest(const char *ext, int width, int align, int height, int pf,
+                     int bottomUp)
 {
+  tjhandle handle = NULL;
   char filename[80], *md5sum, md5buf[65];
   int ps = tjPixelSize[pf], pitch = PAD(width * ps, align), loadWidth = 0,
     loadHeight = 0, retval = 0, pixelFormat = pf;
-  unsigned char *buf = NULL;
+  void *buf = NULL;
   char *md5ref;
 
+  if ((handle = tj3Init(TJINIT_TRANSFORM)) == NULL)
+    THROW_TJ(NULL);
+  TRY_TJ(handle, tj3Set(handle, TJPARAM_BOTTOMUP, bottomUp));
+
   if (pf == TJPF_GRAY) {
-    md5ref = !strcasecmp(ext, "ppm") ? "112c682e82ce5de1cca089e20d60000b" :
-                                       "51976530acf75f02beddf5d21149101d";
+    if (precision == 8)
+      md5ref = !strcasecmp(ext, "ppm") ? "112c682e82ce5de1cca089e20d60000b" :
+                                         "51976530acf75f02beddf5d21149101d";
+    else if (precision == 12)
+      md5ref = "0d1895c7e6f2b2c9af6e821a655c239c";
+    else
+      md5ref = "64f3320b226ea37fb58080713b4df1b2";
   } else {
-    md5ref = !strcasecmp(ext, "ppm") ? "c0c9f772b464d1896326883a5c79c545" :
-                                       "6d659071b9bfcdee2def22cb58ddadca";
+    if (precision == 8)
+      md5ref = !strcasecmp(ext, "ppm") ? "c0c9f772b464d1896326883a5c79c545" :
+                                         "6d659071b9bfcdee2def22cb58ddadca";
+    else if (precision == 12)
+      md5ref = "2ff5299287017502832c99718450c90a";
+    else
+      md5ref = "623f54661b928d170bd2324bc3620565";
   }
 
-  if ((buf = (unsigned char *)tjAlloc(pitch * height)) == NULL)
-    _throw("Could not allocate memory");
-  initBitmap(buf, width, pitch, height, pf, flags);
-
-  snprintf(filename, 80, "test_bmp_%s_%d_%s.%s", pixFormatStr[pf], align,
-           (flags & TJFLAG_BOTTOMUP) ? "bu" : "td", ext);
-  _tj(tjSaveImage(filename, buf, width, pitch, height, pf, flags));
+  if ((buf = tj3Alloc(pitch * height * sampleSize)) == NULL)
+    THROW("Could not allocate memory");
+  initBitmap(buf, width, pitch, height, pf, bottomUp);
+
+  SNPRINTF(filename, 80, "test_bmp%d_%s_%d_%s_%d.%s", precision, pixFormatStr[pf],
+           align, bottomUp ? "bu" : "td", getpid(), ext);
+  if (precision == 8) {
+    TRY_TJ(handle, tj3SaveImage8(handle, filename, (unsigned char *)buf, width,
+                                 pitch, height, pf));
+  } else if (precision == 12) {
+    TRY_TJ(handle, tj3SaveImage12(handle, filename, (short *)buf, width, pitch,
+                                  height, pf));
+  } else {
+    TRY_TJ(handle, tj3SaveImage16(handle, filename, (unsigned short *)buf,
+                                  width, pitch, height, pf));
+  }
   md5sum = MD5File(filename, md5buf);
+  if (!md5sum) {
+    printf("\n   Could not determine MD5 sum of %s\n", filename);
+    retval = -1;  goto bailout;
+  }
   if (strcasecmp(md5sum, md5ref))
-    _throwmd5(filename, md5sum, md5ref);
-
-  tjFree(buf);  buf = NULL;
-  if ((buf = tjLoadImage(filename, &loadWidth, align, &loadHeight, &pf,
-                         flags)) == NULL)
-    _throwtj();
+    THROW_MD5(filename, md5sum, md5ref);
+
+  tj3Free(buf);  buf = NULL;
+  if (precision == 8) {
+    if ((buf = tj3LoadImage8(handle, filename, &loadWidth, align, &loadHeight,
+                             &pf)) == NULL)
+      THROW_TJ(handle);
+  } else if (precision == 12) {
+    if ((buf = tj3LoadImage12(handle, filename, &loadWidth, align, &loadHeight,
+                              &pf)) == NULL)
+      THROW_TJ(handle);
+  } else {
+    if ((buf = tj3LoadImage16(handle, filename, &loadWidth, align, &loadHeight,
+                              &pf)) == NULL)
+      THROW_TJ(handle);
+  }
   if (width != loadWidth || height != loadHeight) {
     printf("\n   Image dimensions of %s are bogus\n", filename);
     retval = -1;  goto bailout;
   }
-  if (!cmpBitmap(buf, width, pitch, height, pf, flags, 0)) {
+  if (!cmpBitmap(buf, width, pitch, height, pf, bottomUp, 0)) {
     printf("\n   Pixel data in %s is bogus\n", filename);
     retval = -1;  goto bailout;
   }
   if (pf == TJPF_GRAY) {
-    tjFree(buf);  buf = NULL;
+    tj3Free(buf);  buf = NULL;
     pf = TJPF_XBGR;
-    if ((buf = tjLoadImage(filename, &loadWidth, align, &loadHeight, &pf,
-                           flags)) == NULL)
-      _throwtj();
+    if (precision == 8) {
+      if ((buf = tj3LoadImage8(handle, filename, &loadWidth, align,
+                               &loadHeight, &pf)) == NULL)
+        THROW_TJ(handle);
+    } else if (precision == 12) {
+      if ((buf = tj3LoadImage12(handle, filename, &loadWidth, align,
+                                &loadHeight, &pf)) == NULL)
+        THROW_TJ(handle);
+    } else {
+      if ((buf = tj3LoadImage16(handle, filename, &loadWidth, align,
+                                &loadHeight, &pf)) == NULL)
+        THROW_TJ(handle);
+    }
     pitch = PAD(width * tjPixelSize[pf], align);
-    if (!cmpBitmap(buf, width, pitch, height, pf, flags, 1)) {
+    if (!cmpBitmap(buf, width, pitch, height, pf, bottomUp, 1)) {
       printf("\n   Converting %s to RGB failed\n", filename);
       retval = -1;  goto bailout;
     }
 
-    tjFree(buf);  buf = NULL;
+    tj3Free(buf);  buf = NULL;
     pf = TJPF_CMYK;
-    if ((buf = tjLoadImage(filename, &loadWidth, align, &loadHeight, &pf,
-                           flags)) == NULL)
-      _throwtj();
+    if (precision == 8) {
+      if ((buf = tj3LoadImage8(handle, filename, &loadWidth, align,
+                               &loadHeight, &pf)) == NULL)
+        THROW_TJ(handle);
+    } else if (precision == 12) {
+      if ((buf = tj3LoadImage12(handle, filename, &loadWidth, align,
+                                &loadHeight, &pf)) == NULL)
+        THROW_TJ(handle);
+    } else {
+      if ((buf = tj3LoadImage16(handle, filename, &loadWidth, align,
+                                &loadHeight, &pf)) == NULL)
+        THROW_TJ(handle);
+    }
     pitch = PAD(width * tjPixelSize[pf], align);
-    if (!cmpBitmap(buf, width, pitch, height, pf, flags, 1)) {
+    if (!cmpBitmap(buf, width, pitch, height, pf, bottomUp, 1)) {
       printf("\n   Converting %s to CMYK failed\n", filename);
       retval = -1;  goto bailout;
     }
   }
-  /* Verify that tjLoadImage() returns the proper "preferred" pixel format for
-     the file type. */
-  tjFree(buf);  buf = NULL;
+  /* Verify that tj3LoadImage*() returns the proper "preferred" pixel format
+     for the file type. */
+  tj3Free(buf);  buf = NULL;
   pf = pixelFormat;
   pixelFormat = TJPF_UNKNOWN;
-  if ((buf = tjLoadImage(filename, &loadWidth, align, &loadHeight,
-                         &pixelFormat, flags)) == NULL)
-    _throwtj();
+  if (precision == 8) {
+    if ((buf = tj3LoadImage8(handle, filename, &loadWidth, align, &loadHeight,
+                             &pixelFormat)) == NULL)
+      THROW_TJ(handle);
+  } else if (precision == 12) {
+    if ((buf = tj3LoadImage12(handle, filename, &loadWidth, align, &loadHeight,
+                              &pixelFormat)) == NULL)
+      THROW_TJ(handle);
+  } else {
+    if ((buf = tj3LoadImage16(handle, filename, &loadWidth, align, &loadHeight,
+                              &pixelFormat)) == NULL)
+      THROW_TJ(handle);
+  }
   if ((pf == TJPF_GRAY && pixelFormat != TJPF_GRAY) ||
       (pf != TJPF_GRAY && !strcasecmp(ext, "bmp") &&
        pixelFormat != TJPF_BGR) ||
       (pf != TJPF_GRAY && !strcasecmp(ext, "ppm") &&
        pixelFormat != TJPF_RGB)) {
-    printf("\n   tjLoadImage() returned unexpected pixel format: %s\n",
+    printf("\n   tj3LoadImage8() returned unexpected pixel format: %s\n",
            pixFormatStr[pixelFormat]);
     retval = -1;
   }
   unlink(filename);
 
 bailout:
-  if (buf) tjFree(buf);
+  tj3Destroy(handle);
+  tj3Free(buf);
   if (exitStatus < 0) return exitStatus;
   return retval;
 }
 
 
-int bmpTest(void)
+static int bmpTest(void)
 {
   int align, width = 35, height = 39, format;
 
   for (align = 1; align <= 8; align *= 2) {
     for (format = 0; format < TJ_NUMPF; format++) {
-      printf("%s Top-Down BMP (row alignment = %d bytes)  ...  ",
-             pixFormatStr[format], align);
-      if (doBmpTest("bmp", width, align, height, format, 0) == -1)
-        return -1;
-      printf("OK.\n");
+      if (precision == 8) {
+        printf("%s Top-Down BMP (row alignment = %d samples)  ...  ",
+               pixFormatStr[format], align);
+        if (doBmpTest("bmp", width, align, height, format, 0) == -1)
+          return -1;
+        printf("OK.\n");
+      }
 
-      printf("%s Top-Down PPM (row alignment = %d bytes)  ...  ",
+      printf("%s Top-Down PPM (row alignment = %d samples)  ...  ",
              pixFormatStr[format], align);
-      if (doBmpTest("ppm", width, align, height, format,
-                    TJFLAG_BOTTOMUP) == -1)
+      if (doBmpTest("ppm", width, align, height, format, 1) == -1)
         return -1;
       printf("OK.\n");
 
-      printf("%s Bottom-Up BMP (row alignment = %d bytes)  ...  ",
-             pixFormatStr[format], align);
-      if (doBmpTest("bmp", width, align, height, format, 0) == -1)
-        return -1;
-      printf("OK.\n");
+      if (precision == 8) {
+        printf("%s Bottom-Up BMP (row alignment = %d samples)  ...  ",
+               pixFormatStr[format], align);
+        if (doBmpTest("bmp", width, align, height, format, 0) == -1)
+          return -1;
+        printf("OK.\n");
+      }
 
-      printf("%s Bottom-Up PPM (row alignment = %d bytes)  ...  ",
+      printf("%s Bottom-Up PPM (row alignment = %d samples)  ...  ",
              pixFormatStr[format], align);
-      if (doBmpTest("ppm", width, align, height, format,
-                    TJFLAG_BOTTOMUP) == -1)
+      if (doBmpTest("ppm", width, align, height, format, 1) == -1)
         return -1;
       printf("OK.\n");
     }
@@ -853,7 +1149,7 @@ int bmpTest(void)
 
 int main(int argc, char *argv[])
 {
-  int i, num4bf = 5;
+  int i, bmp = 0, num4bf = 5;
 
 #ifdef _WIN32
   srand((unsigned int)time(NULL));
@@ -861,27 +1157,56 @@ int main(int argc, char *argv[])
   if (argc > 1) {
     for (i = 1; i < argc; i++) {
       if (!strcasecmp(argv[i], "-yuv")) doYUV = 1;
-      else if (!strcasecmp(argv[i], "-noyuvpad")) pad = 1;
+      else if (!strcasecmp(argv[i], "-noyuvpad")) yuvAlign = 1;
+      else if (!strcasecmp(argv[i], "-lossless")) lossless = 1;
       else if (!strcasecmp(argv[i], "-alloc")) alloc = 1;
-      else if (!strcasecmp(argv[i], "-bmp")) return bmpTest();
-      else usage(argv[0]);
+      else if (!strcasecmp(argv[i], "-bmp")) bmp = 1;
+      else if (!strcasecmp(argv[i], "-precision") && i < argc - 1) {
+        int tempi = atoi(argv[++i]);
+
+        if (tempi != 8 && tempi != 12 && tempi != 16)
+          usage(argv[0]);
+        precision = tempi;
+        if (precision == 16) lossless = 1;
+      } else
+        usage(argv[0]);
     }
   }
+  if (lossless && doYUV)
+    THROW("Lossless JPEG and YUV encoding/decoding are incompatible.");
+  if (precision != 8 && doYUV)
+    THROW("YUV encoding/decoding requires 8-bit data precision.");
+
+  printf("Testing %d-bit precision\n", precision);
+  sampleSize = (precision == 8 ? sizeof(unsigned char) : sizeof(short));
+  maxSample = (1 << precision) - 1;
+  tolerance = (lossless ? 0 : (precision > 8 ? 2 : 1));
+  redToY = (19595U * maxSample) >> 16;
+  yellowToY = (58065U * maxSample) >> 16;
+
+  if (bmp) return bmpTest();
   if (alloc) printf("Testing automatic buffer allocation\n");
   if (doYUV) num4bf = 4;
-  doTest(35, 39, _3byteFormats, 2, TJSAMP_444, "test");
-  doTest(39, 41, _4byteFormats, num4bf, TJSAMP_444, "test");
-  doTest(41, 35, _3byteFormats, 2, TJSAMP_422, "test");
-  doTest(35, 39, _4byteFormats, num4bf, TJSAMP_422, "test");
-  doTest(39, 41, _3byteFormats, 2, TJSAMP_420, "test");
-  doTest(41, 35, _4byteFormats, num4bf, TJSAMP_420, "test");
-  doTest(35, 39, _3byteFormats, 2, TJSAMP_440, "test");
-  doTest(39, 41, _4byteFormats, num4bf, TJSAMP_440, "test");
-  doTest(41, 35, _3byteFormats, 2, TJSAMP_411, "test");
-  doTest(35, 39, _4byteFormats, num4bf, TJSAMP_411, "test");
+  overflowTest();
+  doTest(35, 39, _3sampleFormats, 2, TJSAMP_444, "test");
+  doTest(39, 41, _4sampleFormats, num4bf, TJSAMP_444, "test");
+  doTest(41, 35, _3sampleFormats, 2, TJSAMP_422, "test");
+  if (!lossless) {
+    doTest(35, 39, _4sampleFormats, num4bf, TJSAMP_422, "test");
+    doTest(39, 41, _3sampleFormats, 2, TJSAMP_420, "test");
+    doTest(41, 35, _4sampleFormats, num4bf, TJSAMP_420, "test");
+    doTest(35, 39, _3sampleFormats, 2, TJSAMP_440, "test");
+    doTest(39, 41, _4sampleFormats, num4bf, TJSAMP_440, "test");
+    doTest(41, 35, _3sampleFormats, 2, TJSAMP_411, "test");
+    doTest(35, 39, _4sampleFormats, num4bf, TJSAMP_411, "test");
+    doTest(39, 41, _3sampleFormats, 2, TJSAMP_441, "test");
+    doTest(41, 35, _4sampleFormats, num4bf, TJSAMP_441, "test");
+  }
   doTest(39, 41, _onlyGray, 1, TJSAMP_GRAY, "test");
-  doTest(41, 35, _3byteFormats, 2, TJSAMP_GRAY, "test");
-  doTest(35, 39, _4byteFormats, 4, TJSAMP_GRAY, "test");
+  if (!lossless) {
+    doTest(41, 35, _3sampleFormats, 2, TJSAMP_GRAY, "test");
+    doTest(35, 39, _4sampleFormats, 4, TJSAMP_GRAY, "test");
+  }
   bufSizeTest();
   if (doYUV) {
     printf("\n--------------------\n\n");
@@ -890,9 +1215,11 @@ int main(int argc, char *argv[])
     doTest(48, 48, _onlyRGB, 1, TJSAMP_420, "test_yuv0");
     doTest(48, 48, _onlyRGB, 1, TJSAMP_440, "test_yuv0");
     doTest(48, 48, _onlyRGB, 1, TJSAMP_411, "test_yuv0");
+    doTest(48, 48, _onlyRGB, 1, TJSAMP_441, "test_yuv0");
     doTest(48, 48, _onlyRGB, 1, TJSAMP_GRAY, "test_yuv0");
     doTest(48, 48, _onlyGray, 1, TJSAMP_GRAY, "test_yuv0");
   }
 
+  bailout:
   return exitStatus;
 }