/*
 *  This file is part of RawTherapee.
 *
 *  Copyright (c) 2004-2010 Gabor Horvath <hgabor@rawtherapee.com>
 *
 *  RawTherapee is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  RawTherapee 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 for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with RawTherapee.  If not, see <http://www.gnu.org/licenses/>.
 */
#include "rtengine.h"
#include "improcfun.h"
#include <glibmm.h>
#include "iccstore.h"
#include "iccmatrices.h"
#include "mytime.h"
#include "../rtgui/icmpanel.h"
#include "../rtgui/options.h"
#include "settings.h"
#include "curves.h"
#include "alignedbuffer.h"
#include "color.h"


#ifdef _OPENMP
#include <omp.h>
#endif

namespace rtengine {

#define CLIP01(a) ((a)>0?((a)<1?(a):1):0)
	
extern const Settings* settings;
	
const double (*wprof[])[3]  = {xyz_sRGB, xyz_adobe, xyz_prophoto, xyz_widegamut, xyz_bruce, xyz_beta, xyz_best};
const double (*iwprof[])[3] = {sRGB_xyz, adobe_xyz, prophoto_xyz, widegamut_xyz, bruce_xyz, beta_xyz, best_xyz};
const char* wprofnames[] = {"sRGB", "Adobe RGB", "ProPhoto", "WideGamut", "BruceRGB", "Beta RGB", "BestRGB"};
const int numprof = 7;

void ImProcFunctions::lab2monitorRgb (LabImage* lab, Image8* image) {
	//MyTime tBeg,tEnd;
 //   tBeg.set();
	//gamutmap(lab);

	if (monitorTransform) {

        int W = lab->W;
        int H = lab->H;
        unsigned char * data = image->data;

        // cmsDoTransform is relatively expensive
#ifdef _OPENMP
#pragma omp parallel firstprivate(lab, data, W, H)
#endif
{
        AlignedBuffer<unsigned short> pBuf(3*lab->W);
        unsigned short *buffer=pBuf.data;

#ifdef _OPENMP
#pragma omp for schedule(static)
#endif
		for (int i=0; i<H; i++) {
            // pre-conversion to integer, since the output is 8 bit anyway, but LCMS is MUCH faster not converting from float

            const int ix = i * 3 * W;
            int iy = 0;

			float* rL = lab->L[i];
			float* ra = lab->a[i];
			float* rb = lab->b[i];

            float fy,fx,fz,x_,y_,z_,LL;

			for (int j=0; j<W; j++) {
								
				fy = (0.00862069 * rL[j]) / 327.68 + 0.137932; // (L+16)/116
				fx = (0.002 * ra[j]) / 327.68 + fy;
				fz = fy - (0.005 * rb[j]) / 327.68;
				LL=rL[j]/327.68;
				x_ = Color::f2xyz(fx)*Color::D50x;
				//y_ = Color::f2xyz(fy);
				y_= (LL>Color::epskap) ? fy*fy*fy : LL/Color::kappa;
				
				z_ = Color::f2xyz(fz)*Color::D50z;

                buffer[iy++] = (unsigned short)CLIP(x_* MAXVALF+0.5);
                buffer[iy++] = (unsigned short)CLIP(y_* MAXVALF+0.5);
                buffer[iy++] = (unsigned short)CLIP(z_* MAXVALF+0.5);
			}

            cmsDoTransform (monitorTransform, buffer, data + ix, W);

} // End of parallelization

		}
	} else {

        int W = lab->W;
        int H = lab->H;
        unsigned char * data = image->data;

//#ifdef _OPENMP
//#pragma omp parallel for schedule(static) firstprivate(lab, data, W, H) if (multiThread)
//#endif
		for (int i=0; i<H; i++) {
			float* rL = lab->L[i];
			float* ra = lab->a[i];
			float* rb = lab->b[i];
			int ix = i * 3 * W;

			float R,G,B;
            float fy,fx,fz,x_,y_,z_,LL;

			for (int j=0; j<W; j++) {
			
				//float L1=rL[j],a1=ra[j],b1=rb[j];//for testing
				
				fy = (0.00862069 * rL[j]) / 327.68 + 0.137932; // (L+16)/116
				fx = (0.002 * ra[j]) / 327.68 + fy;
				fz = fy - (0.005 * rb[j]) / 327.68;
				LL=rL[j]/327.68;
				
				x_ = 65535.0 * Color::f2xyz(fx)*Color::D50x;
			//	y_ = 65535.0 * Color::f2xyz(fy);
				z_ = 65535.0 * Color::f2xyz(fz)*Color::D50z;
				y_= (LL>Color::epskap) ? 65535.0*fy*fy*fy : 65535.0*LL/Color::kappa;

				Color::xyz2srgb(x_,y_,z_,R,G,B);
				
				/* copy RGB */
				//int R1=((int)gamma2curve[(R)]) 
				data[ix++] = ((int)gamma2curve[CLIP(R)]) >> 8;
				data[ix++] = ((int)gamma2curve[CLIP(G)]) >> 8;
				data[ix++] = ((int)gamma2curve[CLIP(B)]) >> 8;
			}
		}
	}

    //tEnd.set();
    //printf("lab2rgb %i %d\n", lab->W, tEnd.etime(tBeg));
}

Image8* ImProcFunctions::lab2rgb (LabImage* lab, int cx, int cy, int cw, int ch, Glib::ustring profile, bool standard_gamma) {

	//gamutmap(lab);
	
    if (cx<0) cx = 0;
    if (cy<0) cy = 0;
    if (cx+cw>lab->W) cw = lab->W-cx;
    if (cy+ch>lab->H) ch = lab->H-cy;

    Image8* image = new Image8 (cw, ch);

    cmsHPROFILE oprof = iccStore->getProfile (profile);

    if (oprof) {
        cmsHPROFILE oprofG = oprof;
        if (standard_gamma) {
            oprofG = ICCStore::makeStdGammaProfile(oprof);
        }
        cmsHPROFILE iprof = iccStore->getXYZProfile ();
        lcmsMutex->lock ();
        cmsHTRANSFORM hTransform = cmsCreateTransform (iprof, TYPE_RGB_16, oprofG, TYPE_RGB_8, settings->colorimetricIntent,
            cmsFLAGS_NOOPTIMIZE | cmsFLAGS_NOCACHE );  // NOCACHE is important for thread safety
        lcmsMutex->unlock ();

        unsigned char *data = image->data;

        // cmsDoTransform is relatively expensive
#ifdef _OPENMP
#pragma omp parallel
#endif
{
        AlignedBuffer<unsigned short> pBuf(3*cw);
        unsigned short *buffer=pBuf.data;
        int condition = cy+ch;

#ifdef _OPENMP
#pragma omp for firstprivate(lab) schedule(static)
#endif
        for (int i=cy; i<condition; i++) {

            const int ix = i * 3 * cw;
            int iy = 0;

            float* rL = lab->L[i];
            float* ra = lab->a[i];
            float* rb = lab->b[i];

            for (int j=cx; j<cx+cw; j++) {
				
				float fy = (0.00862069 * rL[j])/327.68 + 0.137932; // (L+16)/116
				float fx = (0.002 * ra[j])/327.68 + fy;
				float fz = fy - (0.005 * rb[j])/327.68;
				float LL=rL[j]/327.68;
				
				float x_ = 65535.0 * Color::f2xyz(fx)*Color::D50x;
				//float y_ = 65535.0 * Color::f2xyz(fy);
				float z_ = 65535.0 * Color::f2xyz(fz)*Color::D50z;
				float y_= (LL>Color::epskap) ? 65535.0*fy*fy*fy : 65535.0*LL/Color::kappa;

                buffer[iy++] = CLIP((int)(x_+0.5));
                buffer[iy++] = CLIP((int)(y_+0.5));
                buffer[iy++] = CLIP((int)(z_+0.5));
            }

            cmsDoTransform (hTransform, buffer, data + ix, cw);
        }
} // End of parallelization

        cmsDeleteTransform(hTransform);
        if (oprofG != oprof)
            cmsCloseProfile(oprofG);
    } else {
		
		double rgb_xyz[3][3];
		
		for (int i=0; i<numprof; i++) {
			if (profile==wprofnames[i]) {
				for (int m=0; m<3; m++) 
					for (int n=0; n<3; n++) {
						rgb_xyz[m][n] = iwprof[i][m][n];
					}
				break;
			}
		}
		
		#pragma omp parallel for if (multiThread)
        for (int i=cy; i<cy+ch; i++) {
			float R,G,B;
            float* rL = lab->L[i];
            float* ra = lab->a[i];
            float* rb = lab->b[i];
			int ix = 3*i*cw;
            for (int j=cx; j<cx+cw; j++) {
				
				float fy = (0.00862069 * rL[j])/327.68 + 0.137932; // (L+16)/116
				float fx = (0.002 * ra[j])/327.68 + fy;
				float fz = fy - (0.005 * rb[j])/327.68;
				float LL=rL[j]/327.68;
				
				float x_ = 65535.0 * Color::f2xyz(fx)*Color::D50x;
				//float y_ = 65535.0 * Color::f2xyz(fy);
				float z_ = 65535.0 * Color::f2xyz(fz)*Color::D50z;
				float y_= (LL>Color::epskap) ? 65535.0*fy*fy*fy : 65535.0*LL/Color::kappa;

				Color::xyz2rgb(x_,y_,z_,R,G,B,rgb_xyz);

                image->data[ix++] = (int)gamma2curve[CLIP(R)] >> 8;
                image->data[ix++] = (int)gamma2curve[CLIP(G)] >> 8;
                image->data[ix++] = (int)gamma2curve[CLIP(B)] >> 8;
            }
        }
    }
    return image;
}
// for default (not gamma)
Image16* ImProcFunctions::lab2rgb16 (LabImage* lab, int cx, int cy, int cw, int ch, Glib::ustring profile, bool bw) {
	
	//gamutmap(lab);

    if (cx<0) cx = 0;
    if (cy<0) cy = 0;
    if (cx+cw>lab->W) cw = lab->W-cx;
    if (cy+ch>lab->H) ch = lab->H-cy;

    Image16* image = new Image16 (cw, ch);
   cmsHPROFILE oprof = iccStore->getProfile (profile);
 
 

    if (oprof) {
		#pragma omp parallel for if (multiThread)
		for (int i=cy; i<cy+ch; i++) {
			float* rL = lab->L[i];
			float* ra = lab->a[i];
			float* rb = lab->b[i];
			short* xa = (short*)image->r(i-cy);
			short* ya = (short*)image->g(i-cy);
			short* za = (short*)image->b(i-cy);
			for (int j=cx; j<cx+cw; j++) {
				
				float fy = (0.0086206897f * rL[j])/327.68f + 0.1379310345f; // (L+16)/116
				float fx = (0.002 * ra[j])/327.68f + fy;
				float fz = fy - (0.005f* rb[j])/327.68f;
				float LL=rL[j]/327.68f;
				
				float x_ = 65535.0f * (float) Color::f2xyz(fx)*Color::D50x;
				//float y_ = 65535.0 * Color::f2xyz(fy);
				float z_ = 65535.0f * (float) Color::f2xyz(fz)*Color::D50z;
				float y_= (LL>Color::epskap) ? 65535.0f*fy*fy*fy : 65535.0f*LL/Color::kappa;
				
				xa[j-cx] =	CLIP((int) 	round(x_));
				ya[j-cx] = 	CLIP((int)	round(y_));
				za[j-cx] = CLIP((int) 	round(z_));
				if(bw && y_ < 65535.f ){//force Bw value and take highlight into account
					xa[j-cx] =(int) round(y_* Color::D50x );
					za[j-cx] =(int) round(y_* Color::D50z);
				}
				
			}
		}

        cmsHPROFILE iprof = iccStore->getXYZProfile ();
        lcmsMutex->lock ();
		cmsHTRANSFORM hTransform = cmsCreateTransform (iprof, TYPE_RGB_16, oprof, TYPE_RGB_16, settings->colorimetricIntent, cmsFLAGS_NOOPTIMIZE | cmsFLAGS_NOCACHE);
        lcmsMutex->unlock ();

        image->ExecCMSTransform(hTransform);

		cmsDeleteTransform(hTransform);
	} else {
		#pragma omp parallel for if (multiThread)
		for (int i=cy; i<cy+ch; i++) {
			float R,G,B;
			float* rL = lab->L[i];
			float* ra = lab->a[i];
			float* rb = lab->b[i];
			for (int j=cx; j<cx+cw; j++) {
				
				float fy = (0.0086206897f * rL[j])/327.68f + 0.1379310345f; // (L+16)/116
				float fx = (0.002f * ra[j])/327.68f + fy;
				float fz = fy - (0.005f * rb[j])/327.68f;
				float LL=rL[j]/327.68f;
				
				float x_ = 65535.0f * (float) Color::f2xyz(fx)*Color::D50x;
				//float y_ = 65535.0 * Color::f2xyz(fy);
				float z_ = 65535.0f * (float) Color::f2xyz(fz)*Color::D50z;
				float y_= (LL>Color::epskap) ? (float) 65535.0f*fy*fy*fy : 65535.0f*LL/Color::kappa;

				Color::xyz2srgb(x_,y_,z_,R,G,B);

				image->r(i-cy,j-cx) = (int)gamma2curve[CLIP(R)];
				image->g(i-cy,j-cx) = (int)gamma2curve[CLIP(G)];
				image->b(i-cy,j-cx) = (int)gamma2curve[CLIP(B)];
			}
		}
	}
    return image;
}


// for gamma options (BT709...sRGB linear...)
Image16* ImProcFunctions::lab2rgb16b (LabImage* lab, int cx, int cy, int cw, int ch, Glib::ustring profile, Glib::ustring profi, Glib::ustring gam,  bool freegamma, double gampos, double slpos, double &ga0, double &ga1, double &ga2, double &ga3, double &ga4, double &ga5, double &ga6, bool bw) {
	
	//gamutmap(lab);

    if (cx<0) cx = 0;
    if (cy<0) cy = 0;
    if (cx+cw>lab->W) cw = lab->W-cx;
    if (cy+ch>lab->H) ch = lab->H-cy;

    Image16* image = new Image16 (cw, ch);
	float p1,p2,p3,p4,p5,p6;//primaries
	//double ga0,ga1,ga2,ga3,ga4,ga5=0.0,ga6=0.0;//gamma parameters
	double g_a0,g_a1,g_a2,g_a3,g_a4,g_a5;//gamma parameters
	double pwr;
	double ts;
	ga6=0.0;
	pwr=1.0/gampos;
	ts=slpos;
	int mode=0, imax=0;
	
	int t50;
	int select_temp =1;//5003K
	const double eps=0.000000001;// not divide by zero
	//primaries for 7 working profiles ==> output profiles
	// eventually to adapt primaries  if RT used special profiles !
	if(profi=="ProPhoto") 	  {p1=0.7347; p2=0.2653; p3=0.1596; p4=0.8404; p5=0.0366; p6=0.0001;select_temp=1;}//Prophoto primaries
	else if (profi=="WideGamut") {p1=0.7350; p2=0.2650; p3=0.1150; p4=0.8260; p5=0.1570; p6=0.0180;select_temp=1;}//Widegamut primaries
	else if (profi=="Adobe RGB") {p1=0.6400; p2=0.3300; p3=0.2100; p4=0.7100; p5=0.1500; p6=0.0600;select_temp=2;}//Adobe primaries
	else if (profi=="sRGB") {p1=0.6400; p2=0.3300; p3=0.3000; p4=0.6000; p5=0.1500; p6=0.0600;select_temp=2;} // sRGB primaries
	else if (profi=="BruceRGB") {p1=0.6400; p2=0.3300; p3=0.2800; p4=0.6500; p5=0.1500; p6=0.0600;select_temp=2;} // Bruce primaries
	else if (profi=="Beta RGB") {p1=0.6888; p2=0.3112; p3=0.1986; p4=0.7551; p5=0.1265; p6=0.0352;select_temp=1;} // Beta primaries
	else if (profi=="BestRGB") {p1=0.7347; p2=0.2653; p3=0.2150; p4=0.7750; p5=0.1300; p6=0.0350;select_temp=1;} // Best primaries
	if (!freegamma) {//if Free gamma not selected	
	// gamma : ga0,ga1,ga2,ga3,ga4,ga5 by calcul
    if(gam=="BT709_g2.2_s4.5") 		{ga0=2.22;ga1=0.909995;ga2=0.090005;ga3=0.222222; ga4=0.081071;ga5=0.0;}//BT709  2.2  4.5  - my prefered as D.Coffin	
	else if (gam=="sRGB_g2.4_s12.92")	{ga0=2.40; ga1=0.947858; ga2=0.052142;ga3=0.077399;ga4=0.039293;ga5=0.0;}//sRGB 2.4 12.92  - RT default as Lightroom	
	else if (gam=="High_g1.3_s3.35")	{ga0=1.3 ; ga1=0.998279; ga2=0.001721;ga3=0.298507;ga4=0.005746;ga5=0.0;}//for high dynamic images
	else if (gam== "Low_g2.6_s6.9")   {ga0=2.6 ; ga1=0.891161; ga2=0.108839;ga3=0.144928;ga4=0.076332;ga5=0.0;} //gamma 2.6 variable : for low contrast images
	else if (gam=="linear_g1.0")   {ga0=1.0; ga1=1.;ga2=0.;ga3=1./eps;ga4=0.;ga5=0.0;}//gamma=1 linear : for high dynamic images (cf : D.Coffin...)
	else if (gam=="standard_g2.2")   {ga0=2.2; ga1=1.;ga2=0.;ga3=1./eps;ga4=0.;ga5=0.0;}//gamma=2.2 (as gamma of Adobe, Widegamut...)
	else if (gam=="standard_g1.8")   {ga0=1.8; ga1=1.;ga2=0.;ga3=1./eps;ga4=0.;ga5=0.0;}//gamma=1.8  (as gamma of Prophoto)
	}
	else //free gamma selected
	{
	if(slpos==0) slpos=eps;
	Color::calcGamma(pwr, ts, mode, imax,g_a0,g_a1,g_a2,g_a3,g_a4,g_a5);// call to calcGamma with selected gamma and slope : return parameters for LCMS2
	ga4=g_a3*ts;
	//printf("g_a0=%f g_a1=%f g_a2=%f g_a3=%f g_a4=%f\n", g_a0,g_a1,g_a2,g_a3,g_a4);
	ga0=gampos;ga1=1./(1.0+g_a4);ga2=g_a4/(1.0 + g_a4);ga3=1./slpos;ga5=0.0;
	//printf("ga0=%f ga1=%f ga2=%f ga3=%f ga4=%f\n", ga0,ga1,ga2,ga3,ga4);

	}
	if(select_temp==1) t50=5003;// for Widegamut, Prophoto Best, Beta   D50
	else if (select_temp==2) t50=6504;// for sRGB, AdobeRGB, Bruce  D65

	cmsCIExyY       xyD;
	cmsCIExyYTRIPLE Primaries = {{p1, p2, 1.0},//red primaries
								{p3, p4, 1.0}, // green
								{p5, p6, 1.0} //blue
								};							   
    cmsToneCurve* GammaTRC[3];
	cmsFloat64Number Parameters[7];
    Parameters[0] = ga0;
    Parameters[1] = ga1;
    Parameters[2] = ga2;
    Parameters[3] = ga3;
    Parameters[4] = ga4;   
	Parameters[5] = ga5;   
	Parameters[6] = ga6;   
// 7 parameters for smoother curves
    cmsWhitePointFromTemp(&xyD, t50);
    GammaTRC[0] = GammaTRC[1] = GammaTRC[2] =   cmsBuildParametricToneCurve(NULL, 5, Parameters);//5 = more smoother than 4
    cmsHPROFILE oprofdef = cmsCreateRGBProfileTHR(NULL, &xyD, &Primaries, GammaTRC); //oprofdef  become Outputprofile

    cmsFreeToneCurve(GammaTRC[0]);

	
    if (oprofdef) {
		#pragma omp parallel for if (multiThread)
		for (int i=cy; i<cy+ch; i++) {
			float* rL = lab->L[i];
			float* ra = lab->a[i];
			float* rb = lab->b[i];
			short* xa = (short*)image->r(i-cy);
			short* ya = (short*)image->g(i-cy);
			short* za = (short*)image->b(i-cy);
			for (int j=cx; j<cx+cw; j++) {
				
				float fy = (0.0086206897f* rL[j])/327.68f + 0.1379310345f; // (L+16)/116
				float fx = (0.002f * ra[j])/327.68f + fy;
				float fz = fy - (0.005f * rb[j])/327.68f;
				float LL=rL[j]/327.68f;
				
				float x_ = 65535.0f * (float)Color::f2xyz(fx)*Color::D50x;
			//	float y_ = 65535.0 * Color::f2xyz(fy);
				float z_ = 65535.0f * (float)Color::f2xyz(fz)*Color::D50z;
				float y_= (LL>Color::epskap) ? (float) 65535.0*fy*fy*fy : 65535.0f*LL/Color::kappa;

				xa[j-cx] = CLIP((int) round(x_)) ;
				ya[j-cx] = CLIP((int) round(y_));
				za[j-cx] = CLIP((int) round(z_));
				if(bw && y_ < 65535.f){//force Bw value and take highlight into account				
					xa[j-cx] =(int) round(y_ * Color::D50x);
					za[j-cx] =(int) round(y_ * Color::D50z);
				}
				
			}
		}

        cmsHPROFILE iprof = iccStore->getXYZProfile ();
        lcmsMutex->lock ();
		cmsHTRANSFORM hTransform = cmsCreateTransform (iprof, TYPE_RGB_16, oprofdef, TYPE_RGB_16, settings->colorimetricIntent,  cmsFLAGS_NOOPTIMIZE | cmsFLAGS_NOCACHE);
        lcmsMutex->unlock ();

        image->ExecCMSTransform(hTransform);
		cmsDeleteTransform(hTransform);
	} else {
	// 
		#pragma omp parallel for if (multiThread)
		for (int i=cy; i<cy+ch; i++) {
			float R,G,B;
			float* rL = lab->L[i];
			float* ra = lab->a[i];
			float* rb = lab->b[i];
			for (int j=cx; j<cx+cw; j++) {
				
				float fy = (0.0086206897f * rL[j])/327.68f + 0.1379310345f; // (L+16)/116
				float fx = (0.002f * ra[j])/327.68f + fy;
				float fz = fy - (0.005f * rb[j])/327.68f;
				float LL=rL[j]/327.68f;
				
				float x_ = 65535.0f * (float) Color::f2xyz(fx)*Color::D50x;
				//float y_ = 65535.0 * Color::f2xyz(fy);
				float z_ = 65535.0f * (float) Color::f2xyz(fz)*Color::D50z;
				float y_= (LL>Color::epskap) ? (float) 65535.0*fy*fy*fy : 65535.0f*LL/Color::kappa;

				Color::xyz2srgb(x_,y_,z_,R,G,B);

				image->r(i-cy,j-cx) = (int)gamma2curve[CLIP(R)];
				image->g(i-cy,j-cx) = (int)gamma2curve[CLIP(G)];
				image->b(i-cy,j-cx) = (int)gamma2curve[CLIP(B)];
			}
		}
	}
    return image;
}
	
//#include "sRGBgamutbdy.cc"

}