#include "toonz/rasterstrokegenerator.h"

#include "trastercm.h"

#include "toonz/rasterbrush.h"

#include "trop.h"

RasterStrokeGenerator::RasterStrokeGenerator(const TRasterCM32P &raster, Tasks task, ColorType colorType,

											 int styleId, const TThickPoint &p, bool selective, int selectedStyle, bool keepAntialias)

	: m_raster(raster), m_boxOfRaster(TRect(raster->getSize())), m_styleId(styleId), m_selective(selective), m_task(task), m_colorType(colorType), m_eraseStyle(4095), m_selectedStyle(selectedStyle), m_keepAntiAlias(keepAntialias), m_doAnArc(false)

{

	TThickPoint pp = p;

	m_points.push_back(pp);

	if (task == ERASE)

		m_styleId = m_eraseStyle;

}

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

RasterStrokeGenerator::~RasterStrokeGenerator()

{

}

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

bool RasterStrokeGenerator::add(const TThickPoint &p)

{

	TThickPoint pp = p;

	TThickPoint mid((m_points.back() + pp) * 0.5, (p.thick + m_points.back().thick) * 0.5);

	m_points.push_back(mid);

	m_points.push_back(pp);

	return true;

}

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

//Disegna il tratto interamente

void RasterStrokeGenerator::generateStroke(bool isPencil) const

{

	vector<tthickpoint> points(m_points);</tthickpoint>

	int size = points.size();

	//Prende un buffer trasparente di appoggio

	TRect box = getBBox(points);

	TPoint newOrigin = box.getP00();

	TRasterCM32P rasBuffer(box.getSize());

	rasBuffer->clear();

	//Trasla i punti secondo il nuovo sitema di riferimento

	translatePoints(points, newOrigin);

	vector<tthickpoint> partialPoints;</tthickpoint>

	if (size == 1) {

		rasterBrush(rasBuffer, points, m_styleId, !isPencil);

		placeOver(m_raster, rasBuffer, newOrigin);

	} else if (size <= 3) {

		vector<tthickpoint> partialPoints;</tthickpoint>

		partialPoints.push_back(points[0]);

		partialPoints.push_back(points[1]);

		rasterBrush(rasBuffer, partialPoints, m_styleId, !isPencil);

		placeOver(m_raster, rasBuffer, newOrigin);

	} else if (size % 2 == 1) /*-- 奇数の場合 --*/

	{

		int strokeCount = (size - 1) / 2 - 1;

		vector<tthickpoint> partialPoints;</tthickpoint>

		partialPoints.push_back(points[0]);

		partialPoints.push_back(points[1]);

		rasterBrush(rasBuffer, partialPoints, m_styleId, !isPencil);

		placeOver(m_raster, rasBuffer, newOrigin);

		for (int i = 0; i < strokeCount; i++) {

			partialPoints.clear();

			rasBuffer->clear();

			partialPoints.push_back(points[i * 2 + 1]);

			partialPoints.push_back(points[i * 2 + 2]);

			partialPoints.push_back(points[i * 2 + 3]);

			if (i == strokeCount - 1)

				partialPoints.push_back(points[i * 2 + 4]);

			rasterBrush(rasBuffer, partialPoints, m_styleId, !isPencil);

			placeOver(m_raster, rasBuffer, newOrigin);

		}

	} else {

		vector<tthickpoint> partialPoints;</tthickpoint>

		partialPoints.push_back(points[0]);

		partialPoints.push_back(points[1]);

		rasterBrush(rasBuffer, partialPoints, m_styleId, !isPencil);

		placeOver(m_raster, rasBuffer, newOrigin);

		if (size > 2) {

			partialPoints.clear();

			vector<tthickpoint>::iterator it = points.begin();</tthickpoint>

			it++;

			partialPoints.insert(partialPoints.begin(), it, points.end());

			rasterBrush(rasBuffer, partialPoints, m_styleId, !isPencil);

			placeOver(m_raster, rasBuffer, newOrigin);

		}

	}

}

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

TRect RasterStrokeGenerator::generateLastPieceOfStroke(bool isPencil, bool closeStroke)

{

	vector<tthickpoint> points;</tthickpoint>

	int size = m_points.size();

	if (size == 3) {

		points.push_back(m_points[0]);

		points.push_back(m_points[1]);

	} else if (size == 1)

		points.push_back(m_points[0]);

	else {

		points.push_back(m_points[size - 4]);

		points.push_back(m_points[size - 3]);

		points.push_back(m_points[size - 2]);

		if (closeStroke)

			points.push_back(m_points[size - 1]);

	}

	TRect box = getBBox(points);

	TPoint newOrigin = box.getP00();

	TRasterCM32P rasBuffer(box.getSize());

	rasBuffer->clear();

	//Trasla i punti secondo il nuovo sitema di riferimento

	translatePoints(points, newOrigin);

	rasterBrush(rasBuffer, points, m_styleId, !isPencil);

	placeOver(m_raster, rasBuffer, newOrigin);

	return box;

}

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

//Ritorna il rettangolo contenente i dischi generati con centri in "points" e diametro "points.thick" +3 pixel a bordo

TRect RasterStrokeGenerator::getBBox(const vector<tthickpoint> &points) const</tthickpoint>

{

	double x0 = (std::numeric_limits<double>::max)(), y0 = (std::numeric_limits<double>::max)(),</double></double>

		   x1 = -(std::numeric_limits<double>::max)(), y1 = -(std::numeric_limits<double>::max)();</double></double>

	for (int i = 0; i < (int)points.size(); i++) {

		double radius = points[i].thick * 0.5;

		if (points[i].x - radius < x0)

			x0 = points[i].x - radius;

		if (points[i].x + radius > x1)

			x1 = points[i].x + radius;

		if (points[i].y - radius < y0)

			y0 = points[i].y - radius;

		if (points[i].y + radius > y1)

			y1 = points[i].y + radius;

	}

	return TRect(TPoint((int)floor(x0 - 3), (int)floor(y0 - 3)), TPoint((int)ceil(x1 + 3), (int)ceil(y1 + 3)));

}

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

//Ricalcola i punti in un nuovo sistema di riferimento

void RasterStrokeGenerator::translatePoints(vector<tthickpoint> &points, const TPoint &newOrigin) const</tthickpoint>

{

	TPointD p(newOrigin.x, newOrigin.y);

	for (int i = 0; i < (int)points.size(); i++)

		points[i] -= p;

}

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

//Effettua la over.

void RasterStrokeGenerator::placeOver(const TRasterCM32P &out, const TRasterCM32P &in, const TPoint &p) const

{

	TRect inBox = in->getBounds() + p;

	TRect outBox = out->getBounds();

	TRect box = inBox * outBox;

	if (box.isEmpty())

		return;

	TRasterCM32P rOut = out->extract(box);

	TRect box2 = box - p;

	TRasterCM32P rIn = in->extract(box2);

	for (int y = 0; y < rOut->getLy(); y++) {

		/*-- Finger Toolの境界条件 --*/

		if (m_task == FINGER && (y == 0 || y == rOut->getLy() - 1))

			continue;

		TPixelCM32 *inPix = rIn->pixels(y);

		TPixelCM32 *outPix = rOut->pixels(y);

		TPixelCM32 *outEnd = outPix + rOut->getLx();

		for (; outPix < outEnd; ++inPix, ++outPix) {

			if (m_task == BRUSH) {

				int inTone = inPix->getTone();

				int outTone = outPix->getTone();

				if (inPix->isPureInk() && !m_selective) {

					*outPix = TPixelCM32(inPix->getInk(), outPix->getPaint(), inTone);

					continue;

				}

				if (outPix->isPureInk() && m_selective) {

					*outPix = TPixelCM32(outPix->getInk(), outPix->getPaint(), outTone);

					continue;

				}

				if (inTone <= outTone) {

					*outPix = TPixelCM32(inPix->getInk(), outPix->getPaint(), inTone);

				}

			}

			if (m_task == ERASE) {

				if (m_colorType == INK) {

					if (!m_keepAntiAlias) {

						if (inPix->getTone() == 0 && (!m_selective || (m_selective && outPix->getInk() == m_selectedStyle))) {

							outPix->setTone(255);

						}

					} else if (inPix->getTone() < 255 && (!m_selective || (m_selective && outPix->getInk() == m_selectedStyle))) {

						outPix->setTone(tmax(outPix->getTone(), 255 - inPix->getTone()));

					}

				}

				if (m_colorType == PAINT) {

					if (inPix->getTone() == 0 && (!m_selective || (m_selective && outPix->getPaint() == m_selectedStyle)))

						outPix->setPaint(0);

				}

				if (m_colorType == INKNPAINT) {

					if (inPix->getTone() < 255 && (!m_selective || (m_selective && outPix->getPaint() == m_selectedStyle)))

						outPix->setPaint(0);

					if (!m_keepAntiAlias) {

						if (inPix->getTone() == 0 && (!m_selective || (m_selective && outPix->getInk() == m_selectedStyle))) {

							outPix->setTone(255);

						}

					} else if (inPix->getTone() < 255 && (!m_selective || (m_selective && outPix->getInk() == m_selectedStyle))) {

						outPix->setTone(tmax(outPix->getTone(), 255 - inPix->getTone()));

					}

				}

			} else if (m_task == PAINTBRUSH) {

				if (!inPix->isPureInk())

					continue;

				bool changePaint = !m_selective || (m_selective && outPix->getPaint() == 0);

				if (m_colorType == INK)

					*outPix = TPixelCM32(inPix->getInk(), outPix->getPaint(), outPix->getTone());

				if (m_colorType == PAINT)

					if (changePaint)

						*outPix = TPixelCM32(outPix->getInk(), inPix->getInk(), outPix->getTone());

				if (m_colorType == INKNPAINT)

					*outPix = TPixelCM32(inPix->getInk(), changePaint ? inPix->getInk() : outPix->getPaint(), outPix->getTone());

			}

			/*-- Finger tool --*/

			else if (m_task == FINGER) {

				/*-- 境界条件 --*/

				if (outPix == rOut->pixels(y) || outPix == outEnd - 1)

					continue;

				int inkId = inPix->getInk();

				if (inkId == 0)

					continue;

				TPixelCM32 *neighbourPixels[4];

				neighbourPixels[0] = outPix - 1;			   /* 左 */

				neighbourPixels[1] = outPix + 1;			   /* 右 */

				neighbourPixels[2] = outPix - rOut->getWrap(); /* 上 */

				neighbourPixels[3] = outPix + rOut->getWrap(); /* 下 */

				int count = 0;

				int tone = outPix->getTone();

				/*--- Invertがオフのとき：穴を埋める操作 ---*/

				if (!m_selective) {

					/*-- 4近傍のピクセルについて --*/

					int minTone = tone;

					for (int p = 0; p < 4; p++) {

						/*-- 自分Pixelより線が濃い（Toneが低い）ものを集計する --*/

						if (neighbourPixels[p]->getTone() < tone) {

							count++;

							if (neighbourPixels[p]->getTone() < minTone)

								minTone = neighbourPixels[p]->getTone();

						}

					}

					/*--- 周りの３つ以上のピクセルが濃ければ、最小Toneと置き換える ---*/

					if (count <= 2)

						continue;

					*outPix = TPixelCM32(inkId, outPix->getPaint(), minTone);

				}

				/*--- InvertがONのとき：出っ張りを削る操作 ---*/

				else {

					if (outPix->isPurePaint() || outPix->getInk() != inkId)

						continue;

					/*-- 4近傍のピクセルについて --*/

					int maxTone = tone;

					for (int p = 0; p < 4; p++) {

						/*-- Ink＃がCurrentでない、または、自分Pixelより線が薄い（Toneが高い）ものを集計する --*/

						if (neighbourPixels[p]->getInk() != inkId) {

							count++;

							maxTone = 255;

						} else if (neighbourPixels[p]->getTone() > tone) {

							count++;

							if (neighbourPixels[p]->getTone() > maxTone)

								maxTone = neighbourPixels[p]->getTone();

						}

					}

					/*--- 周りの３つ以上のピクセルが薄ければ、最大Toneと置き換える ---*/

					if (count <= 2)

						continue;

					*outPix = TPixelCM32((maxTone == 255) ? 0 : inkId, outPix->getPaint(), maxTone);

				}

			}

		}

	}

}

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

TRect RasterStrokeGenerator::getLastRect() const

{

	vector<tthickpoint> points;</tthickpoint>

	int size = m_points.size();

	if (size == 3) {

		points.push_back(m_points[0]);

		points.push_back(m_points[1]);

	} else if (size == 1)

		points.push_back(m_points[0]);

	else {

		points.push_back(m_points[size - 4]);

		points.push_back(m_points[size - 3]);

		points.push_back(m_points[size - 2]);

		points.push_back(m_points[size - 1]);

	}

	return getBBox(points);

}