#include "iwa_adjustexposurefx.h"

#include "tfxparam.h"

/*------------------------------------------------------------

 ソース画像を０〜１に正規化してホストメモリに読み込む

------------------------------------------------------------*/

template <typename pixel="" raster,="" typename=""></typename>

void Iwa_AdjustExposureFx::setSourceRaster(const RASTER srcRas,

										   float4 *dstMem,

										   TDimensionI dim)

{

	float4 *chann_p = dstMem;

	for (int j = 0; j < dim.ly; j++) {

		PIXEL *pix = srcRas->pixels(j);

		for (int i = 0; i < dim.lx; i++) {

			(*chann_p).x = (float)pix->r / (float)PIXEL::maxChannelValue;

			(*chann_p).y = (float)pix->g / (float)PIXEL::maxChannelValue;

			(*chann_p).z = (float)pix->b / (float)PIXEL::maxChannelValue;

			(*chann_p).w = (float)pix->m / (float)PIXEL::maxChannelValue;

			pix++;

			chann_p++;

		}

	}

}

/*------------------------------------------------------------

 出力結果をChannel値に変換してタイルに格納

------------------------------------------------------------*/

template <typename pixel="" raster,="" typename=""></typename>

void Iwa_AdjustExposureFx::setOutputRaster(float4 *srcMem,

										   const RASTER dstRas,

										   TDimensionI dim)

{

	float4 *chan_p = srcMem;

	for (int j = 0; j < dim.ly; j++) {

		PIXEL *pix = dstRas->pixels(j);

		for (int i = 0; i < dim.lx; i++) {

			float val;

			val = (*chan_p).x * (float)PIXEL::maxChannelValue + 0.5f;

			pix->r = (typename PIXEL::Channel)((val > (float)PIXEL::maxChannelValue) ? (float)PIXEL::maxChannelValue : val);

			val = (*chan_p).y * (float)PIXEL::maxChannelValue + 0.5f;

			pix->g = (typename PIXEL::Channel)((val > (float)PIXEL::maxChannelValue) ? (float)PIXEL::maxChannelValue : val);

			val = (*chan_p).z * (float)PIXEL::maxChannelValue + 0.5f;

			pix->b = (typename PIXEL::Channel)((val > (float)PIXEL::maxChannelValue) ? (float)PIXEL::maxChannelValue : val);

			val = (*chan_p).w * (float)PIXEL::maxChannelValue + 0.5f;

			pix->m = (typename PIXEL::Channel)((val > (float)PIXEL::maxChannelValue) ? (float)PIXEL::maxChannelValue : val);

			pix++;

			chan_p++;

		}

	}

}

//------------------------------------------------

Iwa_AdjustExposureFx::Iwa_AdjustExposureFx()

	: m_hardness(3.3), m_scale(0.0), m_offset(0.0)

{

	addInputPort("Source", m_source);

	bindParam(this, "hardness", m_hardness, false);

	bindParam(this, "scale", m_scale, false);

	bindParam(this, "offset", m_offset, false);

	m_hardness->setValueRange(0.05, 20.0);

	m_scale->setValueRange(-10.0, 10.0);

	m_offset->setValueRange(-0.5, 0.5);

}

//------------------------------------------------

void Iwa_AdjustExposureFx::doCompute(TTile &tile,

									 double frame,

									 const TRenderSettings &settings)

{

	/*- Sourceが無ければreturn -*/

	if (!m_source.isConnected()) {

		tile.getRaster()->clear();

		return;

	}

	m_source->compute(tile, frame, settings);

	float4 *tile_host;

	TDimensionI dim(tile.getRaster()->getSize());

	/*- ホストメモリ確保 -*/

	TRasterGR8P tile_host_ras(sizeof(float4) * dim.lx, dim.ly);

	tile_host_ras->lock();

	tile_host = (float4 *)tile_host_ras->getRawData();

	TRaster32P ras32 = tile.getRaster();

	TRaster64P ras64 = tile.getRaster();

	if (ras32)

		setSourceRaster<traster32p, tpixel32="">(ras32, tile_host, dim);</traster32p,>

	else if (ras64)

		setSourceRaster<traster64p, tpixel64="">(ras64, tile_host, dim);</traster64p,>

	doCompute_CPU(tile, frame, settings, dim, tile_host);

	/*-  出力結果をチャンネル値に変換 -*/

	tile.getRaster()->clear();

	if (ras32)

		setOutputRaster<traster32p, tpixel32="">(tile_host, ras32, dim);</traster32p,>

	else if (ras64)

		setOutputRaster<traster64p, tpixel64="">(tile_host, ras64, dim);</traster64p,>

	tile_host_ras->unlock();

}

//------------------------------------------------

void Iwa_AdjustExposureFx::doCompute_CPU(TTile &tile,

										 double frame,

										 const TRenderSettings &settings,

										 TDimensionI &dim,

										 float4 *tile_host)

{

	float hardness = (float)m_hardness->getValue(frame);

	float scale = (float)m_scale->getValue(frame);

	float offset = (float)m_offset->getValue(frame);

	float exposureOffset = (powf(10.0f, (float)(abs(offset) / hardness)) - 1.0f) * ((offset < 0.0f) ? -1.0f : 1.0f);

	float4 *pix = tile_host;

	int size = dim.lx * dim.ly;

	for (int i = 0; i < size; i++, pix++) {

		/*- RGB->Exposure -*/

		pix->x = powf(10, (pix->x - 0.5f) * hardness);

		pix->y = powf(10, (pix->y - 0.5f) * hardness);

		pix->z = powf(10, (pix->z - 0.5f) * hardness);

		/*- スケール -*/

		pix->x *= powf(10, scale);

		pix->y *= powf(10, scale);

		pix->z *= powf(10, scale);

		/*- オフセット -*/

		pix->x += exposureOffset;

		pix->y += exposureOffset;

		pix->z += exposureOffset;

		/*- Exposure->RGB -*/

		pix->x = log10f(pix->x) / hardness + 0.5f;

		pix->y = log10f(pix->y) / hardness + 0.5f;

		pix->z = log10f(pix->z) / hardness + 0.5f;

		/*- クランプ -*/

		pix->x = (pix->x > 1.0f) ? 1.0f : ((pix->x < 0.0f) ? 0.0f : pix->x);

		pix->y = (pix->y > 1.0f) ? 1.0f : ((pix->y < 0.0f) ? 0.0f : pix->y);

		pix->z = (pix->z > 1.0f) ? 1.0f : ((pix->z < 0.0f) ? 0.0f : pix->z);

	}

}

//------------------------------------------------

bool Iwa_AdjustExposureFx::doGetBBox(double frame,

									 TRectD &bBox,

									 const TRenderSettings &info)

{

	if (m_source.isConnected()) {

		bool ret = m_source->doGetBBox(frame, bBox, info);

		return ret;

	} else {

		bBox = TRectD();

		return false;

	}

}

//------------------------------------------------

bool Iwa_AdjustExposureFx::canHandle(const TRenderSettings &info,

									 double frame)

{

	return true;

}

FX_PLUGIN_IDENTIFIER(Iwa_AdjustExposureFx, "iwa_AdjustExposureFx")