org.jpedal.parser.image.utils.JPegImageDecoder Maven / Gradle / Ivy

/*
* ===========================================
* Java Pdf Extraction Decoding Access Library
* ===========================================
*
* Project Info:  http://www.idrsolutions.com
* Help section for developers at http://www.idrsolutions.com/support/
*
* (C) Copyright 1997-2017 IDRsolutions and Contributors.
*
* This file is part of JPedal/JPDF2HTML5
*
@LICENSE@
*
* ---------------
* JPegImageDecoder.java
* ---------------
*/

package org.jpedal.parser.image.utils;

import java.awt.image.BufferedImage;

import org.jpedal.color.ColorSpaces;
import org.jpedal.color.DeviceCMYKColorSpace;
import org.jpedal.color.DeviceRGBColorSpace;
import org.jpedal.color.GenericColorSpace;
import org.jpedal.external.ErrorTracker;
import org.jpedal.parser.ParserOptions;
import org.jpedal.parser.image.data.ImageData;
import org.jpedal.utils.LogWriter;

/**
 * @author markee
 */
public class JPegImageDecoder {


    public static BufferedImage decode(final int w, final int h,
                                       final GenericColorSpace decodeColorData, final byte[] data, final ImageData imageData,
                                       final ErrorTracker errorTracker, final ParserOptions parserOptions) {

        GenericColorSpace jpegDecodeColorData = decodeColorData;

        BufferedImage image = null;

        //handle JPEGS
        LogWriter.writeLog("JPeg Image " + w + "W * " + h + 'H');
        
        /*
        try {
            java.io.FileOutputStream a =new java.io.FileOutputStream("/Users/markee/Desktop/jpg.jpg");
            
            a.write(data);
            a.flush();
            a.close();
            
        } catch (Exception e) {
            LogWriter.writeLog("Unable to save jpeg " + name);
            
        }  /**/

        //if ICC with Alt RGB, use alternative first
        boolean decodedOnAltColorspace = false;
        if (decodeColorData.getID() == ColorSpaces.ICC) {


            //try first and catch any error
            final int alt = decodeColorData.getAlternateColorSpace();

            GenericColorSpace altDecodeColorData = null;

            if (alt == ColorSpaces.DeviceRGB) {
                altDecodeColorData = new DeviceRGBColorSpace();
            } else if (alt == ColorSpaces.DeviceCMYK) {
                altDecodeColorData = new DeviceCMYKColorSpace();
            }

            //try if any alt found
            if (altDecodeColorData != null) {

                try {
                    image = altDecodeColorData.JPEGToRGBImage(data, w, h, imageData.getpX(), imageData.getpY());

                    //if it returns image it worked flag and switch over
                    if (image != null) {
                        decodedOnAltColorspace = true;
                        jpegDecodeColorData = altDecodeColorData;

                        //flag if YCCK
                        if (jpegDecodeColorData.isImageYCCK()) {
                            parserOptions.hasYCCKimages = true;
                        }
                    }
                } catch (final Exception e) {
                    errorTracker.addPageFailureMessage("Unable to use alt colorspace to JPEG");

                    LogWriter.writeLog("Exception: " + e.getMessage());

                    if (image != null) {
                        image.flush();
                    }
                    image = null;
                }
            }
        }
        /*decode if not done above*/
        if (!decodedOnAltColorspace) {
            //separation, renderer
            try {
                image = jpegDecodeColorData.JPEGToRGBImage(data, w, h, imageData.getpX(), imageData.getpY());

                //flag if YCCK
                if (jpegDecodeColorData.isImageYCCK()) {
                    parserOptions.hasYCCKimages = true;
                }

                //image=simulateOP(image);
            } catch (final Exception e) {
                errorTracker.addPageFailureMessage("Problem converting to JPEG");

                LogWriter.writeLog("Exception: " + e.getMessage());

                image.flush();
                image = null;
            }
        }
        return image;
    }

}