#include "trop.h"

#include "traster.h"

#include "tspectrumparam.h"

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

namespace

{

template <class t=""></class>

void doComputeRadialT(TRasterPT<t> ras, TPointD posTrasf, const TSpectrumT<t> &spectrum,</t></t>

					  double period, double count, double cycle, const TAffine &aff)

{

	int j;

	double maxRadius = period * count;

	double freq = 1.0 / period;

	ras->lock();

	for (j = 0; j < ras->getLy(); j++) {

		TPointD posAux = posTrasf;

		T *pix = ras->pixels(j);

		T *endPix = pix + ras->getLx();

		while (pix < endPix) {

			double radius = norm(posAux);

			double t = 1;

			if (radius < maxRadius) {

				t = (radius + cycle) * freq;

				t -= floor(t);

			}

			//double polinomfactor=(-2*t+3)*(t*t);

			*pix++ = spectrum.getPremultipliedValue(t);

			posAux.x += aff.a11;

			posAux.y += aff.a21;

		}

		posTrasf.x += aff.a12;

		posTrasf.y += aff.a22;

	}

	ras->unlock();

}

}

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

void multiRadial(

	const TRasterP &ras,

	TPointD posTrasf,

	const TSpectrumParamP colors,

	double period,

	double count,

	double cycle,

	const TAffine &aff,

	double frame)

{

	if ((TRaster32P)ras)

		doComputeRadialT<tpixel32>(ras, posTrasf, colors->getValue(frame), period, count, cycle, aff);</tpixel32>

	else if ((TRaster64P)ras)

		doComputeRadialT<tpixel64>(ras, posTrasf, colors->getValue64(frame), period, count, cycle, aff);</tpixel64>

	else

		throw TException("MultiRadialGradientFx: unsupported Pixel Type");

}

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

namespace

{

template <class t=""></class>

void doComputeLinearT(TRasterPT<t> ras, TPointD posTrasf, const TSpectrumT<t> &spectrum,</t></t>

					  double period, double count, double w_amplitude, double w_freq, double w_phase, double cycle, const TAffine &aff)

{

	double shift = 0;

	double maxRadius = period * count / 2.;

	double freq = 1.0 / period;

	int j;

	ras->lock();

	for (j = 0; j < ras->getLy(); j++) {

		TPointD posAux = posTrasf;

		//TPointD pos = tile.m_pos;

		//pos.y += j;

		T *pix = ras->pixels(j);

		T *endPix = pix + ras->getLx();

		while (pix < endPix) {

			if (w_amplitude)

				shift = w_amplitude * sin(w_freq * posAux.y + w_phase);

			double radius = posAux.x + shift;

			double t = 1;

			if (fabs(radius) < maxRadius) {

				t = (radius + maxRadius + cycle) * freq;

				t -= floor(t);

			} else if (radius < 0)

				t = 0;

			double polinomfactor = (-2 * t + 3) * (t * t);

			//pos.x += 1.0;

			*pix++ = spectrum.getPremultipliedValue(polinomfactor);

			posAux.x += aff.a11;

			posAux.y += aff.a21;

		}

		posTrasf.x += aff.a12;

		posTrasf.y += aff.a22;

	}

	ras->unlock();

}

}

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

void multiLinear(

	const TRasterP &ras,

	TPointD posTrasf,

	const TSpectrumParamP colors,

	double period,

	double count,

	double amplitude,

	double freq,

	double phase,

	double cycle,

	const TAffine &aff,

	double frame)

{

	if ((TRaster32P)ras)

		doComputeLinearT<tpixel32>(</tpixel32>

			ras, posTrasf,

			colors->getValue(frame),

			period, count, amplitude, freq, phase, cycle, aff);

	else if ((TRaster64P)ras)

		doComputeLinearT<tpixel64>(</tpixel64>

			ras, posTrasf,

			colors->getValue64(frame),

			period, count, amplitude, freq, phase, cycle, aff);

	else

		throw TException("MultiLinearGradientFx: unsupported Pixel Type");

}