/*
 * Copyright (C)2011-2012, 2014-2015, 2017-2018, 2022-2024 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 demonstrates how to use the TurboJPEG C API to approximate the
 * functionality of the IJG's djpeg program.  djpeg features that are not
 * covered:
 *
 * - OS/2 BMP, GIF, and Targa output file formats [legacy feature]
 * - Color quantization and dithering [legacy feature]
 * - The floating-point IDCT method [legacy feature]
 * - Extracting an ICC color management profile
 * - Progress reporting
 * - Skipping rows (i.e. exclusive rather than inclusive partial decompression)
 * - Debug output
 */

import java.io.*;
import java.util.*;
import org.libjpegturbo.turbojpeg.*;


@SuppressWarnings("checkstyle:JavadocType")
final class TJDecomp {

  private TJDecomp() {}

  static final String CLASS_NAME =
    new TJDecomp().getClass().getName();


  private static boolean isCropped(java.awt.Rectangle cr) {
    return (cr.x != 0 || cr.y != 0 || cr.width != 0 || cr.height != 0);
  }


  private static String tjErrorMsg;
  private static int tjErrorCode = -1;

  static void handleTJException(TJException e, int stopOnWarning)
                                throws TJException {
    String errorMsg = e.getMessage();
    int errorCode = e.getErrorCode();

    if (stopOnWarning != 1 && errorCode == TJ.ERR_WARNING) {
      if (tjErrorMsg == null || !tjErrorMsg.equals(errorMsg) ||
          tjErrorCode != errorCode) {
        tjErrorMsg = errorMsg;
        tjErrorCode = errorCode;
        System.out.println("WARNING: " + errorMsg);
      }
    } else
      throw e;
  }


  static void usage() {
    int i;
    TJScalingFactor[] scalingFactors = TJ.getScalingFactors();
    int numScalingFactors = scalingFactors.length;

    System.out.println("\nUSAGE: java [Java options] " + CLASS_NAME +
                       " [options] <JPEG image> <Output image>\n");

    System.out.println("The output image will be in Windows BMP or PBMPLUS (PPM/PGM) format, depending");
    System.out.println("on the file extension.\n");

    System.out.println("GENERAL OPTIONS (CAN BE ABBREVBIATED)");
    System.out.println("-------------------------------------");
    System.out.println("-icc FILE");
    System.out.println("    Extract the ICC (International Color Consortium) color profile from the");
    System.out.println("    JPEG image to the specified file");
    System.out.println("-strict");
    System.out.println("    Treat all warnings as fatal; abort immediately if incomplete or corrupt");
    System.out.println("    data is encountered in the JPEG image, rather than trying to salvage the");
    System.out.println("    rest of the image\n");

    System.out.println("LOSSY JPEG OPTIONS (CAN BE ABBREVIATED)");
    System.out.println("---------------------------------------");
    System.out.println("-crop WxH+X+Y");
    System.out.println("    Decompress only the specified region of the JPEG image.  (W, H, X, and Y");
    System.out.println("    are the width, height, left boundary, and upper boundary of the region, all");
    System.out.println("    specified relative to the scaled image dimensions.)  If necessary, X will");
    System.out.println("    be shifted left to the nearest iMCU boundary, and W will be increased");
    System.out.println("    accordingly.");
    System.out.println("-dct fast");
    System.out.println("    Use less accurate IDCT algorithm [legacy feature]");
    System.out.println("-dct int");
    System.out.println("    Use more accurate IDCT algorithm [default]");
    System.out.println("-grayscale");
    System.out.println("    Decompress a full-color JPEG image into a grayscale output image");
    System.out.println("-maxmemory N");
    System.out.println("    Memory limit (in megabytes) for intermediate buffers used with progressive");
    System.out.println("    JPEG decompression [default = no limit]");
    System.out.println("-maxscans N");
    System.out.println("    Refuse to decompress progressive JPEG images that have more than N scans");
    System.out.println("-nosmooth");
    System.out.println("    Use the fastest chrominance upsampling algorithm available");
    System.out.println("-rgb");
    System.out.println("    Decompress a grayscale JPEG image into a full-color output image");
    System.out.println("-scale M/N");
    System.out.println("    Scale the width/height of the JPEG image by a factor of M/N when");
    System.out.print("    decompressing it (M/N = ");
    for (i = 0; i < numScalingFactors; i++) {
      System.out.format("%d/%d", scalingFactors[i].getNum(),
                        scalingFactors[i].getDenom());
      if (numScalingFactors == 2 && i != numScalingFactors - 1)
        System.out.print(" or ");
      else if (numScalingFactors > 2) {
        if (i != numScalingFactors - 1)
          System.out.print(", ");
        if (i == numScalingFactors - 2)
          System.out.print("or ");
      }
      if (i % 8 == 0 && i != 0) System.out.print("\n    ");
    }
    System.out.println(")\n");

    System.exit(1);
  }


  static boolean matchArg(String arg, String string, int minChars) {
    if (arg.length() > string.length() || arg.length() < minChars)
      return false;

    int cmpChars = Math.max(arg.length(), minChars);
    string = string.substring(0, cmpChars);

    return arg.equalsIgnoreCase(string);
  }


  public static void main(String[] argv) {
    int exitStatus = 0;
    TJDecompressor tjd = null;
    FileInputStream fis = null;
    FileOutputStream fos = null;

    try {

      int i;
      int colorspace, fastDCT = -1, fastUpsample = -1, maxMemory = -1,
        maxScans = -1, pixelFormat = TJ.PF_UNKNOWN, precision,
        stopOnWarning = -1, subsamp;
      java.awt.Rectangle croppingRegion = TJ.UNCROPPED;
      String iccFilename = null;
      TJScalingFactor scalingFactor = TJ.UNSCALED;
      int width, height;
      byte[] jpegBuf, iccBuf = null;
      Object dstBuf = null;

      for (i = 0; i < argv.length; i++) {
        if (matchArg(argv[i], "-crop", 2) && i < argv.length - 1) {
          int tempWidth = -1, tempHeight = -1, tempX = -1, tempY = -1;
          Scanner scanner = new Scanner(argv[++i]).useDelimiter("x|X|\\+");

          try {
            tempWidth = scanner.nextInt();
            tempHeight = scanner.nextInt();
            tempX = scanner.nextInt();
            tempY = scanner.nextInt();
          } catch (Exception e) {}

          if (tempWidth < 1 || tempHeight < 1 || tempX < 0 || tempY < 0)
            usage();
          croppingRegion.width = tempWidth;
          croppingRegion.height = tempHeight;
          croppingRegion.x = tempX;
          croppingRegion.y = tempY;
        } else if (matchArg(argv[i], "-dct", 2) && i < argv.length - 1) {
          i++;
          if (matchArg(argv[i], "fast", 1))
            fastDCT = 1;
          else if (!matchArg(argv[i], "int", 1))
            usage();
        } else if (matchArg(argv[i], "-grayscale", 2) ||
                   matchArg(argv[i], "-greyscale", 2))
          pixelFormat = TJ.PF_GRAY;
        else if (matchArg(argv[i], "-icc", 2) && i < argv.length - 1)
          iccFilename = argv[++i];
        else if (matchArg(argv[i], "-maxscans", 5) && i < argv.length - 1) {
          int temp = -1;

          try {
            temp = Integer.parseInt(argv[++i]);
          } catch (NumberFormatException e) {}
          if (temp < 0)
            usage();
          maxScans = temp;
        } else if (matchArg(argv[i], "-maxmemory", 2) && i < argv.length - 1) {
          int temp = -1;

          try {
            temp = Integer.parseInt(argv[++i]);
          } catch (NumberFormatException e) {}
          if (temp < 0)
            usage();
          maxMemory = temp;
        } else if (matchArg(argv[i], "-nosmooth", 2))
          fastUpsample = 1;
        else if (matchArg(argv[i], "-rgb", 2))
          pixelFormat = TJ.PF_RGB;
        else if (matchArg(argv[i], "-strict", 3))
          stopOnWarning = 1;
        else if (matchArg(argv[i], "-scale", 2) && i < argv.length - 1) {
          int tempNum = 0, tempDenom = 0;
          boolean match = false, scanned = true;
          Scanner scanner = new Scanner(argv[++i]).useDelimiter("/");

          try {
            tempNum = scanner.nextInt();
            tempDenom = scanner.nextInt();
          } catch (Exception e) {}

          if (tempNum < 1 || tempDenom < 1)
            usage();

          TJScalingFactor[] scalingFactors = TJ.getScalingFactors();
          for (int j = 0; j < scalingFactors.length; j++) {
            if ((double)tempNum / (double)tempDenom ==
                (double)scalingFactors[j].getNum() /
                (double)scalingFactors[j].getDenom()) {
              scalingFactor = scalingFactors[j];
              match = true;  break;
            }
          }
          if (!match) usage();
        } else break;
      }

      if (i != argv.length - 2)
        usage();

      tjd = new TJDecompressor();

      if (stopOnWarning >= 0)
        tjd.set(TJ.PARAM_STOPONWARNING, stopOnWarning);
      if (fastUpsample >= 0)
        tjd.set(TJ.PARAM_FASTUPSAMPLE, fastUpsample);
      if (fastDCT >= 0)
        tjd.set(TJ.PARAM_FASTDCT, fastDCT);
      if (maxScans >= 0)
        tjd.set(TJ.PARAM_SCANLIMIT, maxScans);
      if (maxMemory >= 0)
        tjd.set(TJ.PARAM_MAXMEMORY, maxMemory);

      File jpegFile = new File(argv[i++]);
      fis = new FileInputStream(jpegFile);
      int jpegSize = fis.available();
      if (jpegSize < 1)
        throw new Exception("Input file contains no data");
      jpegBuf = new byte[jpegSize];
      fis.read(jpegBuf);
      fis.close();  fis = null;

      try {
        tjd.setSourceImage(jpegBuf, jpegSize);
      } catch (TJException e) { handleTJException(e, stopOnWarning); }
      subsamp = tjd.get(TJ.PARAM_SUBSAMP);
      width = tjd.get(TJ.PARAM_JPEGWIDTH);
      height = tjd.get(TJ.PARAM_JPEGHEIGHT);
      precision = tjd.get(TJ.PARAM_PRECISION);
      colorspace = tjd.get(TJ.PARAM_COLORSPACE);

      if (iccFilename != null) {
        try {
          iccBuf = tjd.getICCProfile();
        } catch (TJException e) { handleTJException(e, stopOnWarning); }
        if (iccBuf != null) {
          File iccFile = new File(iccFilename);
          fos = new FileOutputStream(iccFile);
          fos.write(iccBuf, 0, iccBuf.length);
        }
      }

      if (pixelFormat == TJ.PF_UNKNOWN) {
        if (colorspace == TJ.CS_GRAY)
          pixelFormat = TJ.PF_GRAY;
        else if (colorspace == TJ.CS_CMYK || colorspace == TJ.CS_YCCK)
          pixelFormat = TJ.PF_CMYK;
        else
          pixelFormat = TJ.PF_RGB;
      }

      if (tjd.get(TJ.PARAM_LOSSLESS) == 0) {
        tjd.setScalingFactor(scalingFactor);
        width = scalingFactor.getScaled(width);
        height = scalingFactor.getScaled(height);

        if (isCropped(croppingRegion)) {
          int adjustment;

          if (subsamp == TJ.SAMP_UNKNOWN)
            throw new Exception("Could not determine subsampling level of JPEG image");
          adjustment = croppingRegion.x %
                       scalingFactor.getScaled(TJ.getMCUWidth(subsamp));
          croppingRegion.x -= adjustment;
          croppingRegion.width += adjustment;
          tjd.setCroppingRegion(croppingRegion);
          width = croppingRegion.width;
          height = croppingRegion.height;
        }
      }

      if (precision <= 8)
        dstBuf = new byte[width * height * TJ.getPixelSize(pixelFormat)];
      else
        dstBuf = new short[width * height * TJ.getPixelSize(pixelFormat)];

      try {
        if (precision <= 8)
          tjd.decompress8((byte[])dstBuf, 0, 0, 0, pixelFormat);
        else if (precision <= 12)
          tjd.decompress12((short[])dstBuf, 0, 0, 0, pixelFormat);
        else
          tjd.decompress16((short[])dstBuf, 0, 0, 0, pixelFormat);
      } catch (TJException e) { handleTJException(e, stopOnWarning); }

      tjd.saveImage(argv[i], dstBuf, 0, 0, width, 0, height, pixelFormat);
    } catch (Exception e) {
      e.printStackTrace();
      exitStatus = -1;
    }

    try {
      if (tjd != null) tjd.close();
      if (fis != null) fis.close();
      if (fos != null) fos.close();
    } catch (Exception e) {}
    System.exit(exitStatus);
  }
};