/*

 * Copyright (c) 2007, 2011, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

#include <stdio.h>

#include <stdlib.h>

#include <memory.h>

#include "sun_java2d_cmm_lcms_LCMS.h"

#include "jni_util.h"

#include "Trace.h"

#include "Disposer.h"

#include "lcms2.h"

#define ALIGNLONG(x) (((x)+3) & ~(3))         // Aligns to DWORD boundary

#ifdef USE_BIG_ENDIAN

#define AdjustEndianess32(a)

#else

static

void AdjustEndianess32(cmsUInt8Number *pByte)

{

    cmsUInt8Number temp1;

    cmsUInt8Number temp2;

    temp1 = *pByte++;

    temp2 = *pByte++;

    *(pByte-1) = *pByte;

    *pByte++ = temp2;

    *(pByte-3) = *pByte;

    *pByte = temp1;

}

#endif

// Transports to properly encoded values - note that icc profiles does use

// big endian notation.

static

cmsInt32Number TransportValue32(cmsInt32Number Value)

{

    cmsInt32Number Temp = Value;

    AdjustEndianess32((cmsUInt8Number*) &Temp);

    return Temp;

}

#define SigMake(a,b,c,d) \

                    ( ( ((int) ((unsigned char) (a))) << 24) | \

                      ( ((int) ((unsigned char) (b))) << 16) | \

                      ( ((int) ((unsigned char) (c))) <<  8) | \

                          (int) ((unsigned char) (d)))

#define TagIdConst(a, b, c, d) \

                ((int) SigMake ((a), (b), (c), (d)))

#define SigHead TagIdConst('h','e','a','d')

#define DT_BYTE     0

#define DT_SHORT    1

#define DT_INT      2

#define DT_DOUBLE   3

/* Default temp profile list size */

#define DF_ICC_BUF_SIZE 32

#define ERR_MSG_SIZE 256

#ifdef _MSC_VER

# ifndef snprintf

#       define snprintf  _snprintf

# endif

#endif

typedef struct lcmsProfile_s {

    cmsHPROFILE pf;

} lcmsProfile_t, *lcmsProfile_p;

typedef union storeID_s {    /* store SProfile stuff in a Java Long */

    lcmsProfile_p lcmsPf;

    cmsHTRANSFORM xf;

    jobject jobj;

    jlong j;

} storeID_t, *storeID_p;

typedef union {

    cmsTagSignature cms;

    jint j;

} TagSignature_t, *TagSignature_p;

static jfieldID Trans_renderType_fID;

static jfieldID Trans_ID_fID;

static jfieldID IL_isIntPacked_fID;

static jfieldID IL_dataType_fID;

static jfieldID IL_pixelType_fID;

static jfieldID IL_dataArray_fID;

static jfieldID IL_offset_fID;

static jfieldID IL_nextRowOffset_fID;

static jfieldID IL_width_fID;

static jfieldID IL_height_fID;

static jfieldID IL_imageAtOnce_fID;

JavaVM *javaVM;

void errorHandler(cmsContext ContextID, cmsUInt32Number errorCode,

                  const char *errorText) {

    JNIEnv *env;

    char errMsg[ERR_MSG_SIZE];

    int count = snprintf(errMsg, ERR_MSG_SIZE,

                          "LCMS error %d: %s", errorCode, errorText);

    if (count < 0 || count >= ERR_MSG_SIZE) {

        count = ERR_MSG_SIZE - 1;

    }

    errMsg[count] = 0;

    (*javaVM)->AttachCurrentThread(javaVM, (void**)&env, NULL);

    JNU_ThrowByName(env, "java/awt/color/CMMException", errMsg);

}

JNIEXPORT jint JNICALL JNI_OnLoad(JavaVM *jvm, void *reserved) {

    javaVM = jvm;

    cmsSetLogErrorHandler(errorHandler);

    return JNI_VERSION_1_6;

}

void LCMS_freeProfile(JNIEnv *env, jlong ptr) {

    storeID_t sProfile;

    sProfile.j = ptr;

    if (sProfile.lcmsPf != NULL) {

        if (sProfile.lcmsPf->pf != NULL) {

            cmsCloseProfile(sProfile.lcmsPf->pf);

        }

        free(sProfile.lcmsPf);

    }

}

void LCMS_freeTransform(JNIEnv *env, jlong ID)

{

    storeID_t sTrans;

    sTrans.j = ID;

    /* Passed ID is always valid native ref so there is no check for zero */

    cmsDeleteTransform(sTrans.xf);

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    createNativeTransform

 * Signature: ([JI)J

 */

JNIEXPORT jlong JNICALL Java_sun_java2d_cmm_lcms_LCMS_createNativeTransform

  (JNIEnv *env, jclass cls, jlongArray profileIDs, jint renderType,

   jint inFormatter, jboolean isInIntPacked,

   jint outFormatter, jboolean isOutIntPacked, jobject disposerRef)

{

    cmsHPROFILE _iccArray[DF_ICC_BUF_SIZE];

    cmsHPROFILE *iccArray = &_iccArray[0];

    storeID_t sTrans;

    int i, j, size;

    jlong* ids;

    size = (*env)->GetArrayLength (env, profileIDs);

    ids = (*env)->GetLongArrayElements(env, profileIDs, 0);

#ifdef _LITTLE_ENDIAN

    /* Reversing data packed into int for LE archs */

    if (isInIntPacked) {

        inFormatter ^= DOSWAP_SH(1);

    }

    if (isOutIntPacked) {

        outFormatter ^= DOSWAP_SH(1);

    }

#endif

    if (DF_ICC_BUF_SIZE < size*2) {

        iccArray = (cmsHPROFILE*) malloc(

            size*2*sizeof(cmsHPROFILE));

        if (iccArray == NULL) {

            (*env)->ReleaseLongArrayElements(env, profileIDs, ids, 0);

            J2dRlsTraceLn(J2D_TRACE_ERROR, "getXForm: iccArray == NULL");

            return 0L;

        }

    }

    j = 0;

    for (i = 0; i < size; i++) {

        cmsHPROFILE icc;

        cmsColorSpaceSignature cs;

        sTrans.j = ids[i];

        icc = sTrans.lcmsPf->pf;

        iccArray[j++] = icc;

        /* Middle non-abstract profiles should be doubled before passing to

         * the cmsCreateMultiprofileTransform function

         */

        cs = cmsGetColorSpace(icc);

        if (size > 2 && i != 0 && i != size - 1 &&

            cs != cmsSigXYZData && cs != cmsSigLabData)

        {

            iccArray[j++] = icc;

        }

    }

    sTrans.xf = cmsCreateMultiprofileTransform(iccArray, j,

        inFormatter, outFormatter, renderType, 0);

    (*env)->ReleaseLongArrayElements(env, profileIDs, ids, 0);

    if (sTrans.xf == NULL) {

        J2dRlsTraceLn(J2D_TRACE_ERROR, "LCMS_createNativeTransform: "

                                       "sTrans.xf == NULL");

        if ((*env)->ExceptionOccurred(env) == NULL) {

            JNU_ThrowByName(env, "java/awt/color/CMMException",

                            "Cannot get color transform");

        }

    } else {

        Disposer_AddRecord(env, disposerRef, LCMS_freeTransform, sTrans.j);

    }

    if (iccArray != &_iccArray[0]) {

        free(iccArray);

    }

    return sTrans.j;

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    loadProfile

 * Signature: ([B,Lsun/java2d/cmm/lcms/LCMSProfile;)V

 */

JNIEXPORT jlong JNICALL Java_sun_java2d_cmm_lcms_LCMS_loadProfileNative

  (JNIEnv *env, jobject obj, jbyteArray data, jobject disposerRef)

{

    jbyte* dataArray;

    jint dataSize;

    storeID_t sProf;

    cmsHPROFILE pf;

    if (JNU_IsNull(env, data)) {

        JNU_ThrowIllegalArgumentException(env, "Invalid profile data");

        return 0L;

    }

    sProf.j = 0L;

    dataArray = (*env)->GetByteArrayElements (env, data, 0);

    dataSize = (*env)->GetArrayLength (env, data);

    if (dataArray == NULL) {

        JNU_ThrowIllegalArgumentException(env, "Invalid profile data");

        return 0L;

    }

    pf = cmsOpenProfileFromMem((const void *)dataArray,

                                     (cmsUInt32Number) dataSize);

    (*env)->ReleaseByteArrayElements (env, data, dataArray, 0);

    if (pf == NULL) {

        JNU_ThrowIllegalArgumentException(env, "Invalid profile data");

    } else {

        /* Sanity check: try to save the profile in order

         * to force basic validation.

         */

        cmsUInt32Number pfSize = 0;

        if (!cmsSaveProfileToMem(pf, NULL, &pfSize) ||

            pfSize < sizeof(cmsICCHeader))

        {

            JNU_ThrowIllegalArgumentException(env, "Invalid profile data");

            cmsCloseProfile(pf);

            pf = NULL;

        }

    }

    if (pf != NULL) {

        // create profile holder

        sProf.lcmsPf = (lcmsProfile_p)malloc(sizeof(lcmsProfile_t));

        if (sProf.lcmsPf != NULL) {

            // register the disposer record

            sProf.lcmsPf->pf = pf;

        } else {

            cmsCloseProfile(pf);

        }

    }

    return sProf.j;

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    getProfileSizeNative

 * Signature: (J)I

 */

JNIEXPORT jint JNICALL Java_sun_java2d_cmm_lcms_LCMS_getProfileSizeNative

  (JNIEnv *env, jobject obj, jlong id)

{

    storeID_t sProf;

    cmsUInt32Number pfSize = 0;

    sProf.j = id;

    if (cmsSaveProfileToMem(sProf.lcmsPf->pf, NULL, &pfSize) && ((jint)pfSize > 0)) {

        return (jint)pfSize;

    } else {

      JNU_ThrowByName(env, "java/awt/color/CMMException",

                      "Can not access specified profile.");

        return -1;

    }

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    getProfileDataNative

 * Signature: (J[B)V

 */

JNIEXPORT void JNICALL Java_sun_java2d_cmm_lcms_LCMS_getProfileDataNative

  (JNIEnv *env, jobject obj, jlong id, jbyteArray data)

{

    storeID_t sProf;

    jint size;

    jbyte* dataArray;

    cmsUInt32Number pfSize = 0;

    cmsBool status;

    sProf.j = id;

    // determine actual profile size

    if (!cmsSaveProfileToMem(sProf.lcmsPf->pf, NULL, &pfSize)) {

        JNU_ThrowByName(env, "java/awt/color/CMMException",

                        "Can not access specified profile.");

        return;

    }

    // verify java buffer capacity

    size = (*env)->GetArrayLength(env, data);

    if (0 >= size || pfSize > (cmsUInt32Number)size) {

        JNU_ThrowByName(env, "java/awt/color/CMMException",

                        "Insufficient buffer capacity.");

        return;

    }

    dataArray = (*env)->GetByteArrayElements (env, data, 0);

    status = cmsSaveProfileToMem(sProf.lcmsPf->pf, dataArray, &pfSize);

    (*env)->ReleaseByteArrayElements (env, data, dataArray, 0);

    if (!status) {

        JNU_ThrowByName(env, "java/awt/color/CMMException",

                        "Can not access specified profile.");

        return;

    }

}

/* Get profile header info */

static cmsBool _getHeaderInfo(cmsHPROFILE pf, jbyte* pBuffer, jint bufferSize);

static cmsBool _setHeaderInfo(cmsHPROFILE pf, jbyte* pBuffer, jint bufferSize);

static cmsHPROFILE _writeCookedTag(cmsHPROFILE pfTarget, cmsTagSignature sig, jbyte *pData, jint size);

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    getTagData

 * Signature: (JI[B)V

 */

JNIEXPORT jbyteArray JNICALL Java_sun_java2d_cmm_lcms_LCMS_getTagNative

  (JNIEnv *env, jobject obj, jlong id, jint tagSig)

{

    storeID_t sProf;

    TagSignature_t sig;

    cmsInt32Number tagSize;

    jbyte* dataArray = NULL;

    jbyteArray data = NULL;

    jint bufSize;

    sProf.j = id;

    sig.j = tagSig;

    if (tagSig == SigHead) {

        cmsBool status;

        // allocate java array

        bufSize = sizeof(cmsICCHeader);

        data = (*env)->NewByteArray(env, bufSize);

        if (data == NULL) {

            JNU_ThrowByName(env, "java/awt/color/CMMException",

                            "Unable to allocate buffer");

            return NULL;

        }

        dataArray = (*env)->GetByteArrayElements (env, data, 0);

        if (dataArray == NULL) {

           JNU_ThrowByName(env, "java/awt/color/CMMException",

                            "Unable to get buffer");

           return NULL;

        }

        status = _getHeaderInfo(sProf.lcmsPf->pf, dataArray, bufSize);

        (*env)->ReleaseByteArrayElements (env, data, dataArray, 0);

        if (!status) {

            JNU_ThrowByName(env, "java/awt/color/CMMException",

                            "ICC Profile header not found");

            return NULL;

        }

        return data;

    }

    if (cmsIsTag(sProf.lcmsPf->pf, sig.cms)) {

        tagSize = cmsReadRawTag(sProf.lcmsPf->pf, sig.cms, NULL, 0);

    } else {

        JNU_ThrowByName(env, "java/awt/color/CMMException",

                        "ICC profile tag not found");

        return NULL;

    }

    // allocate java array

    data = (*env)->NewByteArray(env, tagSize);

    if (data == NULL) {

        JNU_ThrowByName(env, "java/awt/color/CMMException",

                        "Unable to allocate buffer");

        return NULL;

    }

    dataArray = (*env)->GetByteArrayElements (env, data, 0);

    if (dataArray == NULL) {

        JNU_ThrowByName(env, "java/awt/color/CMMException",

                        "Unable to get buffer");

        return NULL;

    }

    bufSize = cmsReadRawTag(sProf.lcmsPf->pf, sig.cms, dataArray, tagSize);

    (*env)->ReleaseByteArrayElements (env, data, dataArray, 0);

    if (bufSize != tagSize) {

        JNU_ThrowByName(env, "java/awt/color/CMMException",

                        "Can not get tag data.");

        return NULL;

    }

    return data;

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    setTagData

 * Signature: (JI[B)V

 */

JNIEXPORT void JNICALL Java_sun_java2d_cmm_lcms_LCMS_setTagDataNative

  (JNIEnv *env, jobject obj, jlong id, jint tagSig, jbyteArray data)

{

    storeID_t sProf;

    cmsHPROFILE pfReplace = NULL;

    TagSignature_t sig;

    cmsBool status = FALSE;

    jbyte* dataArray;

    int tagSize;

    sProf.j = id;

    sig.j = tagSig;