/*

 * Copyright (c) 2007, 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 "sun_java2d_cmm_lcms_LCMS.h"

#include "jni_util.h"

#include "Trace.h"

#include "Disposer.h"

#include "lcms.h"

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

#ifdef USE_BIG_ENDIAN

#define AdjustEndianess32(a)

#else

static

void AdjustEndianess32(LPBYTE pByte)

{

    BYTE temp1;

    BYTE 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

icInt32Number TransportValue32(icInt32Number Value)

{

    icInt32Number Temp = Value;

    AdjustEndianess32((LPBYTE) &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 20

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

    cmsHPROFILE pf;

    cmsHTRANSFORM xf;

    jobject jobj;

    jlong j;

} storeID_t, *storeID_p;

static jfieldID Trans_profileIDs_fID;

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 PF_ID_fID;

JavaVM *javaVM;

int errorHandler(int errorCode, const char *errorText) {

    JNIEnv *env;

    char errMsg[ERR_MSG_SIZE];

    /* We can safely use sprintf here because error message has limited size */

    sprintf(errMsg, "LCMS error %d", errorCode);

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

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

    return 1;

}

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

    javaVM = jvm;

    cmsSetErrorHandler(errorHandler);

    return JNI_VERSION_1_6;

}

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,

   jobject disposerRef)

{

    LPLCMSICCPROFILE _iccArray[DF_ICC_BUF_SIZE];

    LPLCMSICCPROFILE *iccArray = &_iccArray[0];

    cmsHTRANSFORM transform;

    storeID_t sTrans;

    int i, j, size;

    jlong* ids;

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

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

    if (DF_ICC_BUF_SIZE < size*2) {

        iccArray = (LPLCMSICCPROFILE*) malloc(

            size*2*sizeof(LPLCMSICCPROFILE));

        if (iccArray == NULL) {

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

            return NULL;

        }

    }

    j = 0;

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

        LPLCMSICCPROFILE icc;

        sTrans.j = ids[i];

        icc = sTrans.pf;

        iccArray[j++] = icc;

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

         * the cmsCreateMultiprofileTransform function

         */

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

            icc->ColorSpace != icSigXYZData &&

            icc->ColorSpace != icSigLabData)

        {

            iccArray[j++] = icc;

        }

    }

    sTrans.xf = cmsCreateMultiprofileTransform(iccArray, j,

        0, 0, renderType, 0);

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

    if (sTrans.xf == NULL) {

        J2dRlsTraceLn(J2D_TRACE_ERROR, "LCMS_createNativeTransform: "

                                       "sTrans.xf == NULL");

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

                        "Cannot get color transform");

    }

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

        free(iccArray);

    }

    return sTrans.j;

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    loadProfile

 * Signature: ([B)J

 */

JNIEXPORT jlong JNICALL Java_sun_java2d_cmm_lcms_LCMS_loadProfile

  (JNIEnv *env, jobject obj, jbyteArray data)

{

    jbyte* dataArray;

    jint dataSize;

    storeID_t sProf;

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

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

    sProf.pf = cmsOpenProfileFromMem((LPVOID)dataArray, (DWORD) dataSize);

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

    if (sProf.pf == NULL) {

        JNU_ThrowIllegalArgumentException(env, "Invalid profile data");

    }

    return sProf.j;

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    freeProfile

 * Signature: (J)V

 */

JNIEXPORT void JNICALL Java_sun_java2d_cmm_lcms_LCMS_freeProfile

  (JNIEnv *env, jobject obj, jlong id)

{

    storeID_t sProf;

    sProf.j = id;

    if (cmsCloseProfile(sProf.pf) == 0) {

        J2dRlsTraceLn1(J2D_TRACE_ERROR, "LCMS_freeProfile: cmsCloseProfile(%d)"

                       "== 0", id);

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

                        "Cannot close profile");

    }

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    getProfileSize

 * Signature: (J)I

 */

JNIEXPORT jint JNICALL Java_sun_java2d_cmm_lcms_LCMS_getProfileSize

  (JNIEnv *env, jobject obj, jlong id)

{

    LPLCMSICCPROFILE Icc;

    storeID_t sProf;

    unsigned char pfSize[4];

    sProf.j = id;

    Icc = (LPLCMSICCPROFILE) sProf.pf;

    Icc -> Seek(Icc, 0);

    Icc -> Read(pfSize, 4, 1, Icc);

    /* TODO: It's a correct but non-optimal for BE machines code, so should

     * be special cased for them

     */

    return (pfSize[0]<<24) | (pfSize[1]<<16) | (pfSize[2]<<8) |

        pfSize[3];

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    getProfileData

 * Signature: (J[B)V

 */

JNIEXPORT void JNICALL Java_sun_java2d_cmm_lcms_LCMS_getProfileData

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

{

    LPLCMSICCPROFILE Icc;

    storeID_t sProf;

    unsigned char pfSize[4];

    jint size;

    jbyte* dataArray;

    sProf.j = id;

    Icc = (LPLCMSICCPROFILE) sProf.pf;

    Icc -> Seek(Icc, 0);

    Icc -> Read(pfSize, 4, 1, Icc);

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

    Icc->Seek(Icc, 0);

    /* TODO: It's a correct but non-optimal for BE machines code, so should

     * be special cased for them

     */

    Icc->Read(dataArray, 1,

              (pfSize[0]<<24) | (pfSize[1]<<16) | (pfSize[2]<<8) | pfSize[3],

              Icc);

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

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    getTagSize

 * Signature: (JI)I

 */

JNIEXPORT jint JNICALL Java_sun_java2d_cmm_lcms_LCMS_getTagSize

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

{

    LPLCMSICCPROFILE Icc;

    storeID_t sProf;

    int i;

    jint result;

    sProf.j = id;

    Icc = (LPLCMSICCPROFILE) sProf.pf;

    if (tagSig == SigHead) {

        result = sizeof(icHeader);

    } else {

        i =  _cmsSearchTag(Icc, tagSig, FALSE);

        if (i >= 0) {

            result = Icc->TagSizes[i];

        } else {

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

                            "ICC profile tag not found");

            result = -1;

        }

    }

    return result;

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    getTagData

 * Signature: (JI[B)V

 */

JNIEXPORT void JNICALL Java_sun_java2d_cmm_lcms_LCMS_getTagData

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

{

    LPLCMSICCPROFILE Icc;

    storeID_t sProf;

    jbyte* dataArray;

    int i, tagSize;

    sProf.j = id;

    Icc = (LPLCMSICCPROFILE) sProf.pf;

    if (tagSig == SigHead) {

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

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

        Icc -> Seek(Icc, 0);

        Icc -> Read(dataArray, sizeof(icHeader), 1, Icc);

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

        return;

    }

    i =  _cmsSearchTag(Icc, tagSig, FALSE);

    if (i >=0) {

        tagSize = Icc->TagSizes[i];

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

        Icc->Seek(Icc, Icc->TagOffsets[i]);

        Icc->Read(dataArray, 1, tagSize, Icc);

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

        return;

    }

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

                    "ICC profile tag not found");

    return;

}

// Modify data for a tag in a profile

LCMSBOOL LCMSEXPORT _cmsModifyTagData(cmsHPROFILE hProfile,

                                 icTagSignature sig, void *data, size_t size);

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    setTagData

 * Signature: (JI[B)V

 */

JNIEXPORT void JNICALL Java_sun_java2d_cmm_lcms_LCMS_setTagData

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

{

    cmsHPROFILE profile;

    storeID_t sProf;

    jbyte* dataArray;

    int tagSize;

    if (tagSig == SigHead) {

        J2dRlsTraceLn(J2D_TRACE_ERROR, "LCMS_setTagData on icSigHead not "

                      "permitted");

        return;

    }

    sProf.j = id;

    profile = (cmsHPROFILE) sProf.pf;

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

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

    _cmsModifyTagData(profile, (icTagSignature) tagSig, dataArray, tagSize);

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

}

void* getILData (JNIEnv *env, jobject img, jint* pDataType,

                 jobject* pDataObject) {

    void* result = NULL;

    *pDataType = (*env)->GetIntField (env, img, IL_dataType_fID);

    *pDataObject = (*env)->GetObjectField(env, img, IL_dataArray_fID);

    switch (*pDataType) {

        case DT_BYTE:

            result = (*env)->GetByteArrayElements (env, *pDataObject, 0);

            break;

        case DT_SHORT:

            result = (*env)->GetShortArrayElements (env, *pDataObject, 0);

            break;

        case DT_INT:

            result = (*env)->GetIntArrayElements (env, *pDataObject, 0);

            break;

        case DT_DOUBLE:

            result = (*env)->GetDoubleArrayElements (env, *pDataObject, 0);

            break;

    }

    return result;

}

void releaseILData (JNIEnv *env, void* pData, jint dataType,

                    jobject dataObject) {

    switch (dataType) {

        case DT_BYTE:

            (*env)->ReleaseByteArrayElements(env,dataObject,(jbyte*)pData,0);

            break;

        case DT_SHORT:

            (*env)->ReleaseShortArrayElements(env,dataObject,(jshort*)pData, 0);

            break;

        case DT_INT:

            (*env)->ReleaseIntArrayElements(env,dataObject,(jint*)pData,0);

            break;

        case DT_DOUBLE:

            (*env)->ReleaseDoubleArrayElements(env,dataObject,(jdouble*)pData,

                                               0);

            break;

    }

}

/*

 * Class:     sun_java2d_cmm_lcms_LCMS

 * Method:    colorConvert

 * Signature: (Lsun/java2d/cmm/lcms/LCMSTransform;Lsun/java2d/cmm/lcms/LCMSImageLayout;Lsun/java2d/cmm/lcms/LCMSImageLayout;)V

 */

JNIEXPORT void JNICALL Java_sun_java2d_cmm_lcms_LCMS_colorConvert

  (JNIEnv *env, jclass obj, jobject trans, jobject src, jobject dst)

{

    storeID_t sTrans;

    int size, inFmt, outFmt, srcDType, dstDType, outSize, renderType;

    int srcOffset, srcNextRowOffset, dstOffset, dstNextRowOffset;

    int width, height, i;

    void* inputBuffer;

    void* outputBuffer;

    char* inputRow;

    char* outputRow;

    jobject srcData, dstData;

    inFmt = (*env)->GetIntField (env, src, IL_pixelType_fID);

    outFmt = (*env)->GetIntField (env, dst, IL_pixelType_fID);

    srcOffset = (*env)->GetIntField (env, src, IL_offset_fID);

    srcNextRowOffset = (*env)->GetIntField (env, src, IL_nextRowOffset_fID);

    dstOffset = (*env)->GetIntField (env, dst, IL_offset_fID);

    dstNextRowOffset = (*env)->GetIntField (env, dst, IL_nextRowOffset_fID);

    width = (*env)->GetIntField (env, src, IL_width_fID);

    height = (*env)->GetIntField (env, src, IL_height_fID);

#ifdef _LITTLE_ENDIAN

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

    if ((*env)->GetBooleanField (env, src, IL_isIntPacked_fID) == JNI_TRUE) {

        inFmt ^= DOSWAP_SH(1);

    }

    if ((*env)->GetBooleanField (env, dst, IL_isIntPacked_fID) == JNI_TRUE) {

        outFmt ^= DOSWAP_SH(1);

    }

#endif

    sTrans.j = (*env)->GetLongField (env, trans, Trans_ID_fID);

    cmsChangeBuffersFormat(sTrans.xf, inFmt, outFmt);

    if (sTrans.xf == NULL) {

        J2dRlsTraceLn(J2D_TRACE_ERROR, "LCMS_colorConvert: transform == NULL");

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

                        "Cannot get color transform");

        return;

    }

    inputBuffer = getILData (env, src, &srcDType, &srcData);

    if (inputBuffer == NULL) {

        J2dRlsTraceLn(J2D_TRACE_ERROR, "");

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

                        "Cannot get input data");

        return;

    }

    outputBuffer = getILData (env, dst, &dstDType, &dstData);

    if (outputBuffer == NULL) {

        releaseILData(env, inputBuffer, srcDType, srcData);

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

                        "Cannot get output data");

        return;

    }

    inputRow = (char*)inputBuffer + srcOffset;

    outputRow = (char*)outputBuffer + dstOffset;

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

        cmsDoTransform(sTrans.xf, inputRow, outputRow, width);

        inputRow += srcNextRowOffset;

        outputRow += dstNextRowOffset;

    }