#include "trop.h"

/*

See Alexander Reshetov's "Morphological Antialiasing" paper on Intel Labs site.

Basically, this antialiasing algorithm is based on the following ideas:

 - Suppose that our image is just made up of flat colors. Then, a simple antialiasing

   approach is that of assuming that the 'actual' line separating two distinct colors

   is the polyline that passes through the midpoint of each edge of its original jaggy

   counterpart.

   As pixels around the border are cut through by the polyline, the area of the pixel

   that is filled of a certain color is its weight in the output filtered pixel.

 - The above method can be applied on each single uniform piece of a scanline, considering

   the lines originated by the vertical extensions of its left and right edges.

 - Of these lines, only those which lie completely on pixels adjacent to the edge are

   considered - so that the antialiasing effect is kept only around the contours.

This algorithm would yield a good result at what may be considered 50% softness. Implementing

a generalized softness simply requires that the line slopes used above are modified

accordingly (divide by 2 * softFactor).

*/

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

namespace

{

template <typename pix=""></typename>

class PixelSelector

{

public:

	typedef PIX pixel_type;

private:

	int m_thresh;

public:

	PixelSelector(int thresh) : m_thresh(thresh) {}

	bool areEqual(const PIX &a, const PIX &b) const

	{

		return std::max({

			abs((int)a.r - b.r),

			abs((int)a.g - b.g),

			abs((int)a.b - b.b),

			abs((int)a.m - b.m)}) < m_thresh;

	}

};

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

template <>

class PixelSelector<tpixelcm32></tpixelcm32>

{

public:

	typedef TPixelCM32 pixel_type;

private:

	int m_thresh;

public:

	PixelSelector(int thresh) : m_thresh(thresh) {}

	bool areEqual(const TPixelCM32 &a, const TPixelCM32 &b) const

	{

		return (a.getInk() == b.getInk()) && (abs(a.getTone() - b.getTone()) < m_thresh);

	}

};

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

template <typename pix=""></typename>

inline void weightPix(PIX *out, const PIX *a, const PIX *b, double weightA, double weightB)

{

	out->r = a->r * weightA + b->r * weightB;

	out->g = a->g * weightA + b->g * weightB;

	out->b = a->b * weightA + b->b * weightB;

	out->m = a->m * weightA + b->m * weightB;

}

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

template <>

inline void weightPix<tpixelcm32>(TPixelCM32 *out, const TPixelCM32 *a, const TPixelCM32 *b,</tpixelcm32>

								  double weightA, double weightB)

{

	*out = TPixelCM32(

		out->isPurePaint() ? b->getInk() : a->getInk(), a->getPaint(),

		a->getTone() * weightA + b->getTone() * weightB);

}

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

//Returns 0 if pixels to connect are on the 00-11 diagonal, 1 on the 01-10 one.

template <typename pix,="" selector="" typename=""></typename>

inline bool checkNeighbourHood(int x, int y, PIX *pix, int lx, int ly, int dx, int dy,

							   const SELECTOR &sel)

{

	int count1 = 0, count2 = 0;

	int dx2 = 2 * dx, dy2 = 2 * dy;

	if (y > 1) {

		//Lower edge

		count1 +=

			(int)sel.areEqual(*(pix - dx), *(pix - dy2)) +

			(int)sel.areEqual(*(pix - dx), *(pix - dy2 - dx));

		count2 +=

			(int)sel.areEqual(*pix, *(pix - dy2)) +

			(int)sel.areEqual(*pix, *(pix - dy2 - dx));

	}

	if (y < ly - 1) {

		//Upper edge

		count1 +=

			(int)sel.areEqual(*(pix - dx), *(pix + dy)) +

			(int)sel.areEqual(*(pix - dx), *(pix + dy - dx));

		count2 +=

			(int)sel.areEqual(*pix, *(pix + dy)) +

			(int)sel.areEqual(*pix, *(pix + dy - dx));

	}

	if (x > 1) {

		//Left edge

		count1 +=

			(int)sel.areEqual(*(pix - dx), *(pix - dx2)) +

			(int)sel.areEqual(*(pix - dx), *(pix - dx2 - dy));

		count2 +=

			(int)sel.areEqual(*pix, *(pix - dx2)) +

			(int)sel.areEqual(*pix, *(pix - dx2 - dy));

	}

	if (x < lx - 1) {

		//Left edge

		count1 +=

			(int)sel.areEqual(*(pix - dx), *(pix + dx)) +

			(int)sel.areEqual(*(pix - dx), *(pix + dx - dy));

		count2 +=

			(int)sel.areEqual(*pix, *(pix + dx)) +

			(int)sel.areEqual(*pix, *(pix + dx - dy));

	}

	//Connect by minority: if there are more pixels like those on the 00-11 diagonal, connect the other,

	//and viceversa.

	return count1 > count2;

}

}

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

template <typename pix=""></typename>

inline void filterLine(PIX *inLPix, PIX *inUPix, PIX *outLPix, PIX *outUPix,

					   int ll, int inDl, int outLDl, int outUDl,

					   double hStart, double slope, bool filterLower)

{

	assert(hStart >= 0.0 && slope > 0.0);

	double h0 = hStart, h1, area;

	double base = hStart / slope;

	int i, end = std::min(tfloor(base), ll);

	if (filterLower) {

		//Filter lower line

		for (i = 0; i < end; ++i, h0 = h1,

			inLPix += inDl, inUPix += inDl, outLPix += outLDl) {

			h1 = h0 - slope;

			area = 0.5 * (h0 + h1);

			weightPix(outLPix, outLPix, inUPix, 1.0 - area, area);

		}

		if (i < ll) {

			double remnant = base - end;

			area = 0.5 * remnant * h0;

			weightPix(outLPix, outLPix, inUPix, 1.0 - area, area);

		}

	} else {

		//Filter upper line

		for (i = 0; i < end; ++i, h0 = h1,

			inLPix += inDl, inUPix += inDl, outUPix += outUDl) {

			h1 = h0 - slope;

			area = 0.5 * (h0 + h1);

			weightPix(outUPix, outUPix, inLPix, 1.0 - area, area);

		}

		if (i < ll) {

			double remnant = base - end;

			area = 0.5 * remnant * h0;

			weightPix(outUPix, outUPix, inLPix, 1.0 - area, area);

		}

	}

}

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

template <typename pix,="" selector="" typename=""></typename>

inline bool checkLength(int lLine, int y, int ly, int dy,

						PIX *pixL1, PIX *pixU1, PIX *pixL2, PIX *pixU2,

						bool uniteU, bool do1Line,

						const SELECTOR &sel)

{

	//1-length edges must be processed (as primary edges) only if explicitly required,

	//and only when its associated secondary edge is of the same length.

	return (lLine > 1) ||

		   (do1Line &&

			((uniteU && (y > 1 && !(sel.areEqual(*pixL1, *(pixL1 - dy)) && sel.areEqual(*pixL2, *(pixL2 - dy))))) ||

			 (y < ly - 1 && !(sel.areEqual(*pixU1, *(pixU1 + dy)) && sel.areEqual(*pixU2, *(pixU2 + dy))))));

}

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

template <typename pix,="" selector="" typename=""></typename>

void processLine(int r, int lx, int ly,

				 PIX *inLRow, PIX *inURow, PIX *outLRow, PIX *outURow,

				 int inDx, int inDy, int outLDx, int outUDx,

				 bool do1Line, double hStart, double slope,

				 const SELECTOR &sel)

{

	//Using a 'horizontal' notation here - but the same applies in vertical too

	++r;

	//As long as we don't reach row end, process uninterrupted separation lines

	//between colors

	PIX *inLL = inLRow, *inLR, *inUL = inURow, *inUR;

	PIX *inLL_1, *inUL_1, *inLR_1, *inUR_1;

	PIX *inLEnd = inLRow + lx * inDx;

	int x, lLine;

	bool uniteLL, uniteUL, uniteLR, uniteUR;

	//Special case: a line at row start has different weights

	if (!sel.areEqual(*inLL, *inUL)) {

		//Look for line ends

		for (inLR = inLL + inDx, inUR = inUL + inDx;

			 inLR != inLEnd && sel.areEqual(*inLL, *inLR) && sel.areEqual(*inUL, *inUR);

			 inLR += inDx, inUR += inDx)

			;

		if (inLR != inLEnd) {

			//Found a line to process

			lLine = (inLR - inLL) / inDx;

			inLR_1 = inLR - inDx, inUR_1 = inUR - inDx;

			x = (inLR_1 - inLRow) / inDx;

			uniteUR = sel.areEqual(*inUR_1, *inLR);

			uniteLR = sel.areEqual(*inLR_1, *inUR);

			if (uniteUR || uniteLR) {

				if (uniteUR && uniteLR)

					//Ambiguous case. Check neighborhood to find out which one must be actually united.

					uniteUR = !checkNeighbourHood(x + 1, r, inUR, lx, ly, inDx, inDy, sel);

				if (checkLength(lLine, r, ly, inDy, inLR_1, inUR_1, inLR, inUR, uniteUR, do1Line, sel))

					filterLine(inLR_1, inUR_1, outLRow + x * outLDx, outURow + x * outUDx,

							   lLine, -inDx, -outLDx, -outUDx, hStart, slope / (lLine << 1), uniteUR);

			}

		}

		//Update lefts

		inLL = inLR, inUL = inUR;

	}

	//Search for a line start

	for (; inLL != inLEnd && sel.areEqual(*inLL, *inUL); inLL += inDx, inUL += inDx)

		;

	while (inLL != inLEnd) {

		//Look for line ends

		for (inLR = inLL + inDx, inUR = inUL + inDx;

			 inLR != inLEnd && sel.areEqual(*inLL, *inLR) && sel.areEqual(*inUL, *inUR);

			 inLR += inDx, inUR += inDx)

			;

		if (inLR == inLEnd)

			break; //Dealt with later

		//Found a line to process

		lLine = (inLR - inLL) / inDx;

		//First, filter left to right

		inLL_1 = inLL - inDx, inUL_1 = inUL - inDx;

		x = (inLL - inLRow) / inDx;

		uniteUL = sel.areEqual(*inUL, *inLL_1);

		uniteLL = sel.areEqual(*inLL, *inUL_1);

		if (uniteUL || uniteLL) {

			if (uniteUL && uniteLL)

				uniteUL = checkNeighbourHood(x, r, inUL, lx, ly, inDx, inDy, sel);

			if (checkLength(lLine, r, ly, inDy, inLL_1, inUL_1, inLL, inUL, uniteUL, do1Line, sel))

				filterLine(inLL, inUL, outLRow + x * outLDx, outURow + x * outUDx,

						   lLine, inDx, outLDx, outUDx, hStart, slope / lLine, uniteUL);

		}

		//Then, filter right to left

		inLR_1 = inLR - inDx, inUR_1 = inUR - inDx;

		x = (inLR_1 - inLRow) / inDx;

		uniteUR = sel.areEqual(*inUR_1, *inLR);

		uniteLR = sel.areEqual(*inLR_1, *inUR);

		if (uniteUR || uniteLR) {

			if (uniteUR && uniteLR)

				uniteUR = !checkNeighbourHood(x + 1, r, inUR, lx, ly, inDx, inDy, sel);

			if (checkLength(lLine, r, ly, inDy, inLR_1, inUR_1, inLR, inUR, uniteUR, do1Line, sel))

				filterLine(inLR_1, inUR_1, outLRow + x * outLDx, outURow + x * outUDx,

						   lLine, -inDx, -outLDx, -outUDx, hStart, slope / lLine, uniteUR);

		}

		//Update lefts - search for a new line start

		inLL = inLR, inUL = inUR;

		for (; inLL != inLEnd && sel.areEqual(*inLL, *inUL); inLL += inDx, inUL += inDx)

			;

	}

	//Special case: filter the last line in the row

	if (inLL != inLEnd) {

		//Found a line to process

		lLine = (inLR - inLL) / inDx;

		inLL_1 = inLL - inDx, inUL_1 = inUL - inDx;

		x = (inLL - inLRow) / inDx;

		uniteUL = sel.areEqual(*inUL, *inLL_1);

		uniteLL = sel.areEqual(*inLL, *inUL_1);

		if (uniteUL || uniteLL) {

			if (uniteUL && uniteLL)

				uniteUL = checkNeighbourHood(x, r, inUL, lx, ly, inDx, inDy, sel);

			if (checkLength(lLine, r, ly, inDy, inLL_1, inUL_1, inLL, inUL, uniteUL, do1Line, sel))

				filterLine(inLL, inUL, outLRow + x * outLDx, outURow + x * outUDx,

						   lLine, inDx, outLDx, outUDx, hStart, slope / (lLine << 1), uniteUL);

		}

	}

}

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

template <typename pix=""></typename>

void makeAntialias(const TRasterPT<pix> &src, TRasterPT<pix> &dst, int threshold, int softness)</pix></pix>

{

	dst->copy(src);

	if (softness == 0)

		return;

	double slope = (50.0 / softness);

	double hStart = 0.5; //fixed for now

	src->lock();

	dst->lock();

	PixelSelector<pix> sel(threshold);</pix>

	//First, filter by rows

	int x, y, lx = src->getLx(), ly = src->getLy(), lx_1 = lx - 1, ly_1 = ly - 1;

	for (y = 0; y < ly_1; ++y) {

		processLine(y, lx, ly,

					src->pixels(y), src->pixels(y + 1),

					dst->pixels(y), dst->pixels(y + 1),

					1, src->getWrap(), 1, 1,

					true, hStart, slope,

					sel);

	}

	//Then, go by columns

	for (x = 0; x < lx_1; ++x) {

		processLine(x, ly, lx,

					src->pixels(0) + x, src->pixels(0) + x + 1,

					dst->pixels(0) + x, dst->pixels(0) + x + 1,

					src->getWrap(), 1, dst->getWrap(), dst->getWrap(),

					false, hStart, slope,

					sel);

	}

	dst->unlock();

	src->unlock();

}

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

void TRop::antialias(const TRasterP &src, const TRasterP &dst, int threshold, int softness)

{

	assert(src->getSize() == dst->getSize());

	TRaster32P src32(src), dst32(dst);

	if (src32 && dst32) {

		makeAntialias<tpixel32>(src32, dst32, threshold, softness);</tpixel32>

		return;

	}

	TRaster64P src64(src), dst64(dst);

	if (src64 && dst64) {

		makeAntialias<tpixel64>(src64, dst64, threshold << 8, softness);</tpixel64>

		return;

	}

	TRasterCM32P srcCM(src), dstCM(dst);

	if (srcCM && dstCM) {

		makeAntialias<tpixelcm32>(srcCM, dstCM, threshold, softness);</tpixelcm32>

		return;

	}

	assert(!"Source and destination rasters must be of the same type!");

}