#include "texception.h"
#include "stdfx.h"
#include "tpixelutils.h"
#include "tbasefx.h"
#include "tfxparam.h"
#include "trop.h"

#include "tparamset.h"

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

void backlit(TRaster32P lighted, TRaster32P light, TRaster32P out, double blur, const TPixel &color,
			 double fade, double scale)
{
	assert(light && lighted && out);

	assert(lighted->getSize() == light->getSize());
	assert(lighted->getSize() == out->getSize());

	int j;

	if (fade > 0.0) {
		assert(fade <= 1.0);
		light->lock();
		for (j = 0; j < light->getLy(); j++) {
			TPixel *pix = light->pixels(j);
			TPixel *endPix = pix + light->getLx();

			while (pix < endPix) {

				if (pix->m > 0) {
					if (pix->m == 255) {
						pix->r = troundp(pix->r + fade * (color.r - pix->r));
						pix->g = troundp(pix->g + fade * (color.g - pix->g));
						pix->b = troundp(pix->b + fade * (color.b - pix->b));
						pix->m = troundp(pix->m + fade * (color.m - pix->m));
					} else {
						int val;
						double factor = pix->m / 255.0;
						val = troundp(pix->r + fade * (color.r * factor - pix->r));
						pix->r = (val > 255) ? 255 : val;
						val = troundp(pix->g + fade * (color.g * factor - pix->g));
						pix->g = (val > 255) ? 255 : val;
						val = troundp(pix->b + fade * (color.b * factor - pix->b));
						pix->b = (val > 255) ? 255 : val;
						val = troundp(pix->m + fade * (color.m * factor - pix->m));
						pix->m = (val > 255) ? 255 : val;
					}
				}
				++pix;
			}
		}
		light->unlock();
	}

	TRop::ropout(light, lighted, light);

	if (blur > 0) {
		double _blur = blur * scale;
		int bi = tceil(_blur);
		TRaster32P ras(light->getLx() + bi, light->getLy() + bi);

		TRect rect(light->getBounds() + TPoint(bi, bi));
		TRop::copy(ras->extract(rect), light);

		TRop::blur(out, ras, _blur, -bi, -bi, true);

		double ampl = 1.0 + blur / 15.0;
		out->lock();
		for (j = 0; j < out->getLy(); j++) {
			TPixel *pix = out->pixels(j);
			TPixel *endPix = pix + out->getLx();
			while (pix < endPix) {
				if (pix->m != 0) {
					int val = troundp(pix->r * ampl);
					pix->r = (val > 204) ? 204 : val;
					val = troundp(pix->g * ampl);
					pix->g = (val > 204) ? 204 : val;
					val = troundp(pix->b * ampl);
					pix->b = (val > 204) ? 204 : val;
					val = troundp(pix->m * ampl);
					pix->m = (val > 204) ? 204 : val;
				}
				pix++;
			}
		}
		out->unlock();
		TRop::over(out, lighted, out);
	} else
		TRop::over(out, lighted, light);
}

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

void backlit(TRaster64P lighted, TRaster64P light, TRaster64P out, double blur, const TPixel &color32, double fade, double scale)
{
	assert(light && lighted && out);

	assert(lighted->getSize() == light->getSize());
	assert(lighted->getSize() == out->getSize());

	int j;
	TPixel64 color = toPixel64(color32);

	if (fade > 0.0) {
		assert(fade <= 1.0);
		light->lock();
		for (j = 0; j < light->getLy(); j++) {
			TPixel64 *pix = light->pixels(j);
			TPixel64 *endPix = pix + light->getLx();

			while (pix < endPix) {

				if (pix->m > 0) {
					if (pix->m == 65535) {
						pix->r = troundp(pix->r + fade * (color.r - pix->r));
						pix->g = troundp(pix->g + fade * (color.g - pix->g));
						pix->b = troundp(pix->b + fade * (color.b - pix->b));
						pix->m = troundp(pix->m + fade * (color.m - pix->m));
					} else {
						int val;
						double factor = pix->m / 65535.0;
						val = troundp(pix->r + fade * (color.r * factor - pix->r));
						pix->r = (val > 65535) ? 65535 : val;
						val = troundp(pix->g + fade * (color.g * factor - pix->g));
						pix->g = (val > 65535) ? 65535 : val;
						val = troundp(pix->b + fade * (color.b * factor - pix->b));
						pix->b = (val > 65535) ? 65535 : val;
						val = troundp(pix->m + fade * (color.m * factor - pix->m));
						pix->m = (val > 65535) ? 65535 : val;
					}
				}
				++pix;
			}
		}
		light->unlock();
	}

	TRop::ropout(light, lighted, light);

	if (blur > 0) {
		double _blur = blur * scale;
		int bi = tceil(_blur);
		TRaster64P ras(light->getLx() + bi, light->getLy() + bi);

		TRect rect(light->getBounds() + TPoint(bi, bi));
		TRop::copy(ras->extract(rect), light);

		TRop::blur(out, ras, _blur, -bi, -bi, false);

		double ampl = 1.0 + blur / 15.0;
		out->lock();
		int fac = (204 << 8) | 204;

		for (j = 0; j < out->getLy(); j++) {
			TPixel64 *pix = out->pixels(j);
			TPixel64 *endPix = pix + out->getLx();
			while (pix < endPix) {
				if (pix->m != 0) {
					int val = troundp(pix->r * ampl);
					pix->r = (val > fac) ? fac : val;
					val = troundp(pix->g * ampl);
					pix->g = (val > fac) ? fac : val;
					val = troundp(pix->b * ampl);
					pix->b = (val > fac) ? fac : val;
					val = troundp(pix->m * ampl);
					pix->m = (val > fac) ? fac : val;
				}
				pix++;
			}
		}
		out->unlock();
		TRop::over(out, lighted, out);
	} else
		TRop::over(out, lighted, light);
}

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

class BacklitFx : public TBaseRasterFx
{
	FX_DECLARATION(BacklitFx)
	TRasterFxPort m_lighted, m_light;
	TDoubleParamP m_value;
	TDoubleParamP m_fade;

	TPixelParamP m_color;

public:
	BacklitFx() : m_value(0.0), m_color(TPixel::White), m_fade(0.0)
	{
		m_color->enableMatte(true);
		m_value->setValueRange(0, (std::numeric_limits<double>::max)());
		m_fade->setValueRange(0.0, 100.0);
		bindParam(this, "value", m_value);
		bindParam(this, "color", m_color);
		bindParam(this, "fade", m_fade);

		addInputPort("Light", m_light);
		addInputPort("Source", m_lighted);
	}
	~BacklitFx() {}
	bool doGetBBox(double frame, TRectD &bbox, const TRenderSettings &info)
	{
		if (getActiveTimeRegion().contains(frame))
			if (m_light.isConnected()) {
				if (m_lighted.isConnected()) {
					TRectD b0, b1;
					bool ret = m_light->doGetBBox(frame, b0, info);
					ret = ret && m_lighted->doGetBBox(frame, b1, info);
					bbox = b0.enlarge(tceil(m_value->getValue(frame))) + b1;
					return ret;
				} else
					return m_light->doGetBBox(frame, bbox, info);
			} else if (m_lighted.isConnected())
				return m_lighted->doGetBBox(frame, bbox, info);
		return false;
	}

	void doDryCompute(TRectD &rect, double frame, const TRenderSettings &info)
	{
		if (!m_light.isConnected())
			return;
		if (!m_lighted.isConnected()) {
			m_light->dryCompute(rect, frame, info);
			return;
		}

		double value = m_value->getValue(frame);
		double scale = sqrt(fabs(info.m_affine.det()));
		int brad = tceil(value * scale);

		TRectD inRect = rect.enlarge(brad);
		inRect = TRectD(tfloor(inRect.x0), tfloor(inRect.y0), tceil(inRect.x1), tceil(inRect.y1));

		m_lighted->dryCompute(inRect, frame, info);
		m_light->dryCompute(inRect, frame, info);
	}

	void doCompute(TTile &tile, double frame, const TRenderSettings &ri)
	{
		if (!m_light.isConnected())
			return;

		if (!m_lighted.isConnected()) {
			m_light->compute(tile, frame, ri);
			return;
		}

		double value = m_value->getValue(frame);
		double scale = sqrt(fabs(ri.m_affine.det()));
		int brad = tceil(value * scale);

		TDimension tileSize(tile.getRaster()->getSize());
		TRectD rect(tile.m_pos, TDimensionD(tileSize.lx, tileSize.ly));

		//TRectD rect(tile.m_pos,convert(tile.getRaster()->getBounds().getP11())+tile.m_pos);
		rect = rect.enlarge(brad);
		TRect rectI(tfloor(rect.x0), tfloor(rect.y0), tceil(rect.x1) - 1, tceil(rect.y1) - 1);
		//m_light->compute(tile, frame, ri);
		TTile srcTile;
		m_lighted->allocateAndCompute(srcTile, rect.getP00(), rectI.getSize(), tile.getRaster(), frame, ri);
		TTile ctrTile;
		m_light->allocateAndCompute(ctrTile, rect.getP00(), rectI.getSize(), tile.getRaster(), frame, ri);
		if ((TRaster32P)srcTile.getRaster() && (TRaster32P)tile.getRaster())
			backlit((TRaster32P)srcTile.getRaster(), (TRaster32P)ctrTile.getRaster(), (TRaster32P)ctrTile.getRaster(),
					value, m_color->getValue(frame), m_fade->getValue(frame) / 100.0, scale);
		else if ((TRaster64P)srcTile.getRaster() && (TRaster64P)tile.getRaster())
			backlit((TRaster64P)srcTile.getRaster(), (TRaster64P)ctrTile.getRaster(), (TRaster64P)ctrTile.getRaster(),
					value, m_color->getValue(frame), m_fade->getValue(frame) / 100.0, scale);
		else
			throw TRopException("TRop::max invalid raster combination");
		tile.getRaster()->copy(ctrTile.getRaster(), TPoint(-brad, -brad));
	}

	bool canHandle(const TRenderSettings &info, double frame) { return true; }

	int getMemoryRequirement(const TRectD &rect, double frame, const TRenderSettings &info)
	{
		double value = m_value->getValue(frame);
		double scale = sqrt(fabs(info.m_affine.det()));
		int brad = tceil(value * scale);

		return TRasterFx::memorySize(rect.enlarge(brad), info.m_bpp);
	}
};

//==================================================================

FX_PLUGIN_IDENTIFIER(BacklitFx, "backlitFx")
//==================================================================