#include <iostream>
#include <cmath>

#include <utils/RgbTransform.h>

RgbTransform::RgbTransform()
{
	init(1.0, 1.0, 1.0, 0.0, 1.0);
}

RgbTransform::RgbTransform(double gammaR, double gammaG, double gammaB, double brightnessLow, double brightnessHigh)
{
	init(gammaR, gammaG, gammaB, brightnessLow, brightnessHigh);
}

void RgbTransform::init(double gammaR, double gammaG, double gammaB, double brightnessLow, double brightnessHigh)
{
	setGamma(gammaR,gammaG,gammaB);
	setBrightnessMin(brightnessLow);
	setBrightness(brightnessHigh);
	initializeMapping();
}

RgbTransform::~RgbTransform()
{
}

double RgbTransform::getGammaR() const
{
	return _gammaR;
}

double RgbTransform::getGammaG() const
{
	return _gammaG;
}

double RgbTransform::getGammaB() const
{
	return _gammaB;
}

void RgbTransform::setGamma(double gammaR, double gammaG, double gammaB)
{
	_gammaR = gammaR;
	_gammaG = (gammaG < 0.0) ? _gammaR : gammaG;
	_gammaB = (gammaB < 0.0) ? _gammaR : gammaB;
	initializeMapping();
}

void RgbTransform::initializeMapping()
{
	for (int i = 0; i < 256; ++i)
	{
		_mappingR[i] = std::min(std::max((int)(std::pow(i / 255.0, _gammaR) * 255), 0), 255);
		_mappingG[i] = std::min(std::max((int)(std::pow(i / 255.0, _gammaG) * 255), 0), 255);
		_mappingB[i] = std::min(std::max((int)(std::pow(i / 255.0, _gammaB) * 255), 0), 255);
	}
}


double RgbTransform::getBrightnessMin() const
{
	return _brightnessLow;
}

void RgbTransform::setBrightnessMin(double brightness)
{
	_brightnessLow    = brightness;
	_sumBrightnessLow = 765.0 * ((std::pow(2.0,brightness*2)-1) / 3.0);
}

double RgbTransform::getBrightness() const
{
	return _brightnessHigh;
}

void RgbTransform::setBrightness(double brightness)
{
	_brightnessHigh    = brightness;
	_sumBrightnessHigh = 765.0 * ((std::pow(2.0,brightness*2)-1) / 3.0);
}

void RgbTransform::transform(uint8_t & red, uint8_t & green, uint8_t & blue)
{
	// apply gamma
	red   = _mappingR[red];
	green = _mappingR[green];
	blue  = _mappingR[blue];

	//std::cout << (int)red << " " << (int)green << " " << (int)blue << " => ";
	// apply brightnesss
	if (red  ==0) red   = 1;
	if (green==0) green = 1;
	if (blue ==0) blue  = 1;

	int rgbSum = red+green+blue;

	if (rgbSum > _sumBrightnessHigh)
	{
		double cH = _sumBrightnessHigh / rgbSum;
		red   *= cH;
		green *= cH;
		blue  *= cH;
	}
	else if (rgbSum < _sumBrightnessLow)
	{
		double cL = _sumBrightnessLow / rgbSum;
		red   *= cL;
		green *= cL;
		blue  *= cL;
	}
	//std::cout << (int)red << " " << (int)green << " " << (int)blue << std::endl;
}