#include "tools/rasterselection.h"
#include "tools/tool.h"
#include "tools/toolutils.h"
#include "tools/toolhandle.h"
#include "tpaletteutil.h"
#include "trop.h"
#include "drawutil.h"
#include "tconvert.h"
#include "timagecache.h"
#include "tpixelutils.h"
#include "toonzqt/rasterimagedata.h"
#include "toonzqt/strokesdata.h"
#include "toonzqt/selectioncommandids.h"
#include "toonzqt/tselectionhandle.h"
#include "toonzqt/dvdialog.h"
#include "toonz/stage.h"
#include "toonz/toonzimageutils.h"
#include "toonz/txshlevelhandle.h"
#include "toonz/txshsimplelevel.h"
#include "toonz/tpalettehandle.h"
#include "toonz/palettecontroller.h"
#include "toonz/toonzscene.h"
#include "toonz/tcamera.h"
#include "toonz/trasterimageutils.h"
#include "toonz/toonzimageutils.h"

#include <QApplication>
#include <QClipboard>

#include "timage_io.h"
#include "tropcm.h"

//=============================================================================
namespace
{
//-----------------------------------------------------------------------------

TRasterP getRaster(const TImageP image)
{
	if (TToonzImageP ti = (TToonzImageP)(image))
		return ti->getRaster();
	if (TRasterImageP ri = (TRasterImageP)(image))
		return ri->getRaster();
	return (TRasterP)(0);
}

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

TRect convertWorldToRaster(const TRectD area, const TRasterP ras)
{
	if (area.isEmpty())
		return TRect();
	if (!ras)
		return TRect(tfloor(area.x0), tfloor(area.y0), tfloor(area.x1) - 1, tfloor(area.y1) - 1);
	TRectD rect(area + ras->getCenterD());
	return TRect(tfloor(rect.x0), tfloor(rect.y0), tceil(rect.x1) - 1, tceil(rect.y1) - 1);
}

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

TRect convertWorldToRaster(const TRectD area, const TImageP image)
{
	TRasterImageP ri(image);
	TToonzImageP ti(image);

	//Watch out! TToonzImage::getRaster() returns a TRasterCM32P, while
	//TRasterImage::getRaster() returns a TRasterP!
	TRasterP ras = (ri) ? ri->getRaster() : (TRasterP)ti->getRaster();
	return convertWorldToRaster(area, ras);
}

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

TRectD convertRasterToWorld(const TRect area, const TImageP image)
{
	TToonzImageP ti(image);
	if (ti)
		return ToonzImageUtils::convertRasterToWorld(area, image);
	return TRasterImageUtils::convertRasterToWorld(area, image);
}

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

TRectD intersection(const TRectD &area, const TImageP image)
{
	TToonzImageP ti(image);
	if (ti)
		return area * ToonzImageUtils::convertRasterToWorld(ti->getRaster()->getBounds(), image);

	TRasterImageP ri(image);
	if (ri)
		return area * TRasterImageUtils::convertRasterToWorld(ri->getRaster()->getBounds(), ri);

	return area;
}

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

//The stroke is in raster coordinates
template <class PIXEL1, class PIXEL2>
TRasterPT<PIXEL1> getImageFromStroke(TRasterPT<PIXEL2> ras, const TStroke &stroke)
{
	TRectD regionsBoxD = stroke.getBBox();
	//E' volutamente allargato di un pixel!
	TRect regionsBox(tfloor(regionsBoxD.x0), tfloor(regionsBoxD.y0), tceil(regionsBoxD.x1), tceil(regionsBoxD.y1));
	regionsBox *= ras->getBounds();
	if (regionsBox.isEmpty())
		return (TRasterPT<PIXEL1>)0;
	TRasterPT<PIXEL1> buffer(regionsBox.getSize());
	buffer->clear();

	//Compute regions created by the vector
	TVectorImage app;
	app.addStroke(new TStroke(stroke));
	app.findRegions();
	int reg, j, k, y;
	ras->lock();
	for (reg = 0; reg < (int)app.getRegionCount(); reg++) {
		//For each region, pixels inside the region are copied in buffer!
		TRectD bBoxD = stroke.getBBox();
		TRect bBox(tfloor(bBoxD.x0), tfloor(bBoxD.y0), tceil(bBoxD.x1) - 1, tceil(bBoxD.y1) - 1);
		bBox *= ras->getBounds();
		for (y = bBox.y0; y <= bBox.y1; y++) {
			PIXEL2 *selectedLine = ras->pixels(y);
			int startY = y - regionsBox.y0;
			PIXEL1 *bufferLine = buffer->pixels(startY >= 0 ? startY : 0);
			vector<double> intersections;
			app.getRegion(reg)->computeScanlineIntersections(y, intersections);
			if (intersections.empty())
				app.getRegion(reg)->computeScanlineIntersections(y + 0.9, intersections);
			for (j = 0; j < (int)intersections.size(); j += 2) {
				if (intersections[j] == intersections[j + 1])
					continue;
				int from = tmax(tfloor(intersections[j]), bBox.x0);
				int to = tmin(tceil(intersections[j + 1]), bBox.x1);
				for (k = from; k <= to; k++) {
					TRasterCM32P bufferCM(buffer);
					TRaster32P buffer32(buffer);
					TRasterCM32P rasCM(ras);
					TRaster32P ras32(ras);
					TRasterGR8P rasGR8(ras);
					if (bufferCM && rasCM) {
						TPixelCM32 *bottomPix = (TPixelCM32 *)bufferLine + k - regionsBox.x0;
						TPixelCM32 *topPix = (TPixelCM32 *)selectedLine + k;
						*bottomPix = *topPix;
					} else if (buffer32 && ras32) {
						TPixel32 *bottomPix = (TPixel32 *)bufferLine + k - regionsBox.x0;
						TPixel32 *topPix = (TPixel32 *)selectedLine + k;
						*bottomPix = *topPix;
					} else if (buffer32 && rasGR8) {
						TPixel32 *bottomPix = (TPixel32 *)bufferLine + k - regionsBox.x0;
						TPixelGR8 *topPix = (TPixelGR8 *)selectedLine + k;
						*bottomPix = TPixel32(topPix->value, topPix->value, topPix->value, 255);
					} else
						assert(0);
				}
			}
		}
	}
	ras->unlock();
	return buffer;
}

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

template <class PIXEL1, class PIXEL2>
TRasterPT<PIXEL1> getImageFromSelection(TRasterPT<PIXEL2> &ras, RasterSelection &selection)
{
	if (selection.isEmpty())
		return (TRasterPT<PIXEL1>)0;
	TRectD wSelectionBound = selection.getSelectionBbox();
	TRect rSelectionBound = convertWorldToRaster(wSelectionBound, ras);
	rSelectionBound *= ras->getBounds();
	TRasterPT<PIXEL1> selectedRaster(rSelectionBound.getSize());
	selectedRaster->clear();
	vector<TStroke> strokes = selection.getStrokes();
	TPoint startPosition = rSelectionBound.getP00();
	unsigned int i;
	for (i = 0; i < strokes.size(); i++) {
		TStroke stroke = strokes[i];
		stroke.transform(TTranslation(ras->getCenterD()));
		TRasterPT<PIXEL1> app = getImageFromStroke<PIXEL1, PIXEL2>(ras, stroke);
		if (!app)
			continue;
		TRectD strokeRectD = stroke.getBBox();
		TRect strokeRect(tfloor(strokeRectD.x0), tfloor(strokeRectD.y0), tceil(strokeRectD.x1) - 1, tceil(strokeRectD.y1) - 1);
		TPoint offset((strokeRect * rSelectionBound).getP00() - rSelectionBound.getP00());
		TPoint startP = rSelectionBound.getP00() + offset;
		startPosition = TPoint(tmin(startPosition.x, startP.x), tmin(startPosition.y, startP.y));
		TRop::over(selectedRaster, app, offset);
	}

	selection.setStartPosition(startPosition);
	return selectedRaster;
}

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

TRasterP getImageFromSelection(const TImageP &image, RasterSelection &selection)
{
	if (TToonzImageP toonzImage = (TToonzImageP)image) {
		TRasterPT<TPixelCM32> ras = toonzImage->getRaster();
		return getImageFromSelection<TPixelCM32, TPixelCM32>(ras, selection);
	}
	if (TRasterImageP rasterImage = (TRasterImageP)image) {
		TRasterP ras = rasterImage->getRaster();
		if (TRaster32P ras32 = (TRaster32P)ras)
			return getImageFromSelection<TPixel32, TPixel32>(ras32, selection);
		if (TRasterGR8P rasGR8 = (TRasterGR8P)ras)
			return getImageFromSelection<TPixel32, TPixelGR8>(rasGR8, selection);
	}
	return (TRasterP)0;
}

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

template <typename PIXEL>
void deleteSelectionWithoutUndo(TRasterPT<PIXEL> &ras, const vector<TStroke> &strokes, PIXEL emptyValue)
{
	if (!ras)
		return;
	unsigned int i;
	for (i = 0; i < strokes.size(); i++) {
		TStroke s = strokes[i];
		s.transform(TTranslation(ras->getCenterD()));
		TRectD strokeRectD = s.getBBox();
		//E' volutamente allargato di un pixel!
		TRect strokeRect(tfloor(strokeRectD.x0), tfloor(strokeRectD.y0), tceil(strokeRectD.x1), tceil(strokeRectD.y1));
		if (!strokeRect.overlaps(ras->getBounds()))
			continue;

		//Compute regions created by the vector
		TVectorImage app;
		app.addStroke(new TStroke(s));
		app.findRegions();
		int reg, j, k, y;
		ras->lock();
		TRect rasRect(ras->getBounds());
		for (reg = 0; reg < (int)app.getRegionCount(); reg++) {
			//For each region, pixels inside the region are erased!
			TRectD bBoxD = app.getRegion(reg)->getBBox();
			TRect bBox(tfloor(bBoxD.x0), tfloor(bBoxD.y0), tceil(bBoxD.x1) - 1, tceil(bBoxD.y1) - 1);
			bBox *= rasRect;
			for (y = bBox.y0; y <= bBox.y1; y++) {
				PIXEL *selectedLine = ras->pixels(y);
				int startY = y - strokeRect.y0;
				vector<double> intersections;
				app.getRegion(reg)->computeScanlineIntersections(y, intersections);
				if (intersections.empty())
					app.getRegion(reg)->computeScanlineIntersections(y + 0.9, intersections);
				for (j = 0; j < (int)intersections.size(); j += 2) {
					if (intersections[j] == intersections[j + 1])
						continue;
					int from = tmax(tfloor(intersections[j]), bBox.x0);
					int to = tmin(tceil(intersections[j + 1]), bBox.x1);
					for (k = from; k <= to; k++)
						*(selectedLine + k) = emptyValue;
				}
			}
		}
		ras->unlock();
	}
}

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

void deleteSelectionWithoutUndo(const TImageP &image, const vector<TStroke> &strokes)
{
	if (TToonzImageP toonzImage = (TToonzImageP)image) {
		TRasterPT<TPixelCM32> ras = toonzImage->getRaster();
		deleteSelectionWithoutUndo<TPixelCM32>(ras, strokes, TPixelCM32());
	}
	if (TRasterImageP rasterImage = (TRasterImageP)image) {
		TRasterP ras = rasterImage->getRaster();
		if (TRaster32P ras32 = (TRaster32P)ras)
			deleteSelectionWithoutUndo<TPixel32>(ras32, strokes, TPixel32::Transparent);
		if (TRasterGR8P rasGR8 = (TRasterGR8P)ras)
			deleteSelectionWithoutUndo<TPixelGR8>(rasGR8, strokes, TPixelGR8::White);
	}
}

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

void pasteFloatingSelectionWithoutUndo(const TImageP &image, const TRasterP &floatingSelection,
									   const TAffine &transformation, const TRectD &wSelectionBound,
									   bool noAntialiasing)
{
	TRasterImageP ri = (TRasterImageP)image;
	TToonzImageP ti = (TToonzImageP)image;

	TRasterP targetRaster = (ri) ? ri->getRaster() : (TRasterP)ti->getRaster();
	if (!targetRaster || !floatingSelection)
		return;

	TRect rSelectionBound = convertWorldToRaster(wSelectionBound, targetRaster);
	TRop::over(targetRaster, floatingSelection, rSelectionBound.getP00(), transformation,
			   noAntialiasing ? TRop::ClosestPixel : TRop::Triangle);
}

//=============================================================================
// UndoDeleteSelection
//-----------------------------------------------------------------------------

class UndoDeleteSelection : public TUndo
{
	static int m_id;
	TXshSimpleLevelP m_level;
	TFrameId m_frameId;
	string m_erasedImageId;
	TPoint m_erasePoint;
	vector<TStroke> m_strokes;
	TTool *m_tool;

public:
	UndoDeleteSelection(RasterSelection *selection, TXshSimpleLevel *level)
		: TUndo(), m_level(level), m_frameId(selection->getFrameId()), m_strokes(selection->getOriginalStrokes())
	{
		TImageP image = m_level->getFrame(m_frameId, true);
		m_erasedImageId = "UndoDeleteSelection" + toString(m_id++);
		TRasterP ras = getRaster(image);
		TRasterP erasedRas;
		if (!selection->isFloating())
			erasedRas = TRasterP(getImageFromSelection(image, *selection));
		else
			erasedRas = TRasterP(selection->getOriginalFloatingSelection());
		TImageP erasedImage;
		if (TRasterCM32P toonzRas = (TRasterCM32P)(erasedRas))
			erasedImage = TToonzImageP(toonzRas, toonzRas->getBounds());
		else if (TRaster32P fullColorRas = (TRaster32P)(erasedRas))
			erasedImage = TRasterImageP(fullColorRas);
		TImageCache::instance()->add(m_erasedImageId, erasedImage, false);
		m_erasePoint = selection->getStartPosition();
		m_tool = TTool::getApplication()->getCurrentTool()->getTool();
	}

	~UndoDeleteSelection()
	{
		if (TImageCache::instance()->isCached(m_erasedImageId))
			TImageCache::instance()->remove(m_erasedImageId);
	}

	void undo() const
	{
		TImageP image = m_level->getFrame(m_frameId, true);
		if (!image)
			return;
		TRasterP ras = getRaster(image);
		if (!ras)
			return;
		TImageP erasedImage = TImageCache::instance()->get(m_erasedImageId, false);
		if (!erasedImage)
			return;
		TRasterP erasedRaster = getRaster(erasedImage);
		TRop::over(ras, erasedRaster, m_erasePoint);

		ToolUtils::updateSaveBox(m_level, m_frameId);

		if (!m_tool)
			return;
		m_tool->notifyImageChanged(m_frameId);
		m_tool->invalidate();
	}

	void redo() const
	{
		TImageP image = m_level->getFrame(m_frameId, true);
		TImageP erasedImage = TImageCache::instance()->get(m_erasedImageId, false);
		if (!erasedImage)
			return;
		deleteSelectionWithoutUndo(image, m_strokes);

		ToolUtils::updateSaveBox(m_level, m_frameId);
		if (!m_tool)
			return;
		m_tool->notifyImageChanged(m_frameId);
		m_tool->invalidate();
	}

	int getSize() const
	{
		return sizeof(*this);
	}
};

int UndoDeleteSelection::m_id = 0;

//=============================================================================
// UndoPasteSelection
//-----------------------------------------------------------------------------

class UndoPasteSelection : public TUndo
{
	RasterSelection *m_currentSelection, m_newSelection;

public:
	UndoPasteSelection(RasterSelection *currentSelection)
		: TUndo(), m_currentSelection(currentSelection), m_newSelection(*currentSelection)
	{
	}

	~UndoPasteSelection()
	{
	}

	void undo() const
	{
		m_currentSelection->setFloatingSeletion(TRasterP());
		m_currentSelection->selectNone();
		m_currentSelection->notify();
	}

	void redo() const
	{
		*m_currentSelection = m_newSelection;
		m_currentSelection->notify();
	}

	int getSize() const
	{
		return sizeof(*this);
	}

	QString getHistoryString()
	{
		return QObject::tr("Paste");
	}
};

//=============================================================================
// UndoPasteFloatingSelection
//-----------------------------------------------------------------------------

class UndoPasteFloatingSelection : public TUndo
{
	static int m_id;

	TXshCell m_imageCell; //!< Level/frame pair to the pasted-to image
						  //!< (seemingly cached as m_imageId)
	TPaletteP m_oldPalette, m_newPalette;
	string m_imageId, m_floatingImageId, m_undoImageId, m_oldFloatingImageId;
	vector<TStroke> m_strokes;
	TRectD m_selectionRect;
	TAffine m_transformation;
	TPoint m_startPos;
	bool m_isPastedSelection;
	bool m_noAntialiasing;

	TTool *m_tool;
	TFrameId m_frameId;

public:
	UndoPasteFloatingSelection(RasterSelection *currentSelection, TPalette *oldPalette, bool noAntialiasing)
		: TUndo(), m_imageCell(currentSelection->getCurrentImageCell()), m_oldPalette(oldPalette ? oldPalette->clone() : 0), m_strokes(currentSelection->getOriginalStrokes()), m_selectionRect(currentSelection->getSelectionBbox()), m_transformation(currentSelection->getTransformation()), m_isPastedSelection(currentSelection->isPastedSelection()), m_noAntialiasing(noAntialiasing), m_undoImageId(""), m_frameId(currentSelection->getFrameId())
	{
		TImageP image(currentSelection->getCurrentImage());
		if (!image)
			return;

		m_imageId = "UndoPasteImage_" + toString(m_id);
		TImageCache::instance()->add(m_imageId, image, false);

		m_floatingImageId = "UndoPasteFloatingSelection_floating_" + toString(m_id);
		TRasterP floatingRas = currentSelection->getFloatingSelection();
		TImageP floatingImage;
		if (TRasterCM32P toonzRas = (TRasterCM32P)(floatingRas))
			floatingImage = TToonzImageP(toonzRas, toonzRas->getBounds());
		else if (TRaster32P fullColorRas = (TRaster32P)(floatingRas))
			floatingImage = TRasterImageP(fullColorRas);
		else if (TRasterGR8P grRas = (TRasterGR8P)(floatingRas))
			floatingImage = TRasterImageP(grRas);
		TImageCache::instance()->add(m_floatingImageId, floatingImage, false);

		m_oldFloatingImageId = "UndoPasteFloatingSelection_oldFloating_" + toString(m_id);
		TRasterP oldFloatingRas = currentSelection->getOriginalFloatingSelection();
		TImageP olfFloatingImage;
		if (TRasterCM32P toonzRas = (TRasterCM32P)(oldFloatingRas))
			olfFloatingImage = TToonzImageP(toonzRas, toonzRas->getBounds());
		else if (TRaster32P fullColorRas = (TRaster32P)(oldFloatingRas))
			olfFloatingImage = TRasterImageP(fullColorRas);
		else if (TRasterGR8P grRas = (TRasterGR8P)(oldFloatingRas))
			olfFloatingImage = TRasterImageP(grRas);
		TImageCache::instance()->add(m_oldFloatingImageId, olfFloatingImage, false);

		TPaletteP imgPalette = image->getPalette();
		m_newPalette = imgPalette ? imgPalette->clone() : 0;
		TRasterP rasImage = getRaster(image);
		TRectD wRect = m_selectionRect.enlarge(2);
		TRect rRect = convertWorldToRaster(wRect, image);
		rRect *= rasImage->getBounds();
		if (!rRect.isEmpty()) {
			m_undoImageId = "UndoPasteFloatingSelection_undo" + toString(m_id);
			TRasterP undoRas = rasImage->extract(rRect)->clone();
			TImageP undoImage;
			if (TRasterCM32P toonzRas = (TRasterCM32P)(undoRas))
				undoImage = TToonzImageP(toonzRas, toonzRas->getBounds());
			else if (TRaster32P fullColorRas = (TRaster32P)(undoRas))
				undoImage = TRasterImageP(fullColorRas);
			else if (TRasterGR8P grRas = (TRasterGR8P)(undoRas))
				undoImage = TRasterImageP(grRas);
			TImageCache::instance()->add(m_undoImageId, undoImage, false);
		}
		m_startPos = currentSelection->getStartPosition();
		m_id++;
		m_tool = TTool::getApplication()->getCurrentTool()->getTool();
	}

	~UndoPasteFloatingSelection()
	{
		if (TImageCache::instance()->isCached(m_imageId))
			TImageCache::instance()->remove(m_imageId);
		if (TImageCache::instance()->isCached(m_floatingImageId))
			TImageCache::instance()->remove(m_floatingImageId);
		if (TImageCache::instance()->isCached(m_undoImageId))
			TImageCache::instance()->remove(m_undoImageId);
	}

	void undo() const
	{
		TImageP image = TImageCache::instance()->get(m_imageId, false);
		if (!image)
			return;
		TRasterP rasImage = getRaster(image);
		TRectD wRect = m_selectionRect.enlarge(2);
		TRect rRect = convertWorldToRaster(wRect, image);
		rRect *= rasImage->getBounds();
		if (!m_undoImageId.empty()) {
			TImageP undoImage = TImageCache::instance()->get(m_undoImageId, false);
			if (!undoImage)
				return;
			rasImage->copy(getRaster(undoImage), rRect.getP00());
		}

		TXshSimpleLevelP sl(m_imageCell.getSimpleLevel());
		const TFrameId &fid = m_imageCell.m_frameId;

		if (!m_isPastedSelection) {
			TImageP floatingImage = TImageCache::instance()->get(m_oldFloatingImageId, false);
			if (!floatingImage)
				return;
			TRasterP floatingRaster = getRaster(floatingImage);
			TRop::over(rasImage, floatingRaster, m_startPos);
		}

		ToolUtils::updateSaveBox(sl, fid);

		if (m_oldPalette)
			image->getPalette()->assign(m_oldPalette->clone());
		TTool::Application *app = TTool::getApplication();
		app->getPaletteController()->getCurrentLevelPalette()->notifyPaletteChanged();
		if (!m_tool)
			return;
		m_tool->notifyImageChanged(m_frameId);
		m_tool->invalidate();
	}

	void redo() const
	{
		TImageP image = TImageCache::instance()->get(m_imageId, false);
		TImageP floatingImage = TImageCache::instance()->get(m_floatingImageId, false);
		if (!floatingImage || !image)
			return;
		TRasterP floatingRas = getRaster(floatingImage);

		TXshSimpleLevelP sl(m_imageCell.getSimpleLevel());
		const TFrameId &fid = m_imageCell.m_frameId;

		if (!m_isPastedSelection)
			deleteSelectionWithoutUndo(image, m_strokes);

		TRasterP ras = getRaster(image);
		pasteFloatingSelectionWithoutUndo(image, floatingRas, m_transformation, m_selectionRect, m_noAntialiasing);

		ToolUtils::updateSaveBox(sl, fid);

		if (m_newPalette)
			image->getPalette()->assign(m_newPalette->clone());
		TTool::Application *app = TTool::getApplication();
		app->getPaletteController()->getCurrentLevelPalette()->notifyPaletteChanged();
		if (!m_tool)
			return;
		m_tool->notifyImageChanged(m_frameId);
		m_tool->invalidate();
	}

	int getSize() const
	{
		return sizeof(*this);
	}

	QString getHistoryString()
	{
		return QObject::tr("Paste");
	}
};

int UndoPasteFloatingSelection::m_id = 0;

//=============================================================================
// Next methods are used to compute intersection between selected stroke and image box
//-----------------------------------------------------------------------------
/*! Help function.*/
TSegment getSegmentByIndex(TRectD rect, int index)
{
	if (index == 0)
		return TSegment(rect.getP00(), rect.getP01());
	if (index == 1)
		return TSegment(rect.getP01(), rect.getP11());
	if (index == 2)
		return TSegment(rect.getP11(), rect.getP10());
	if (index == 3)
		return TSegment(rect.getP10(), rect.getP00());
	return TSegment();
}

//-----------------------------------------------------------------------------
/*! Help function. precPoint -> point */
bool isClockwise(TRectD bbox, int segmentIndex, TThickPoint precPoint, TThickPoint point)
{
	if (segmentIndex == 0)
		return precPoint.y > point.y;
	if (segmentIndex == 1)
		return precPoint.x > point.x;
	if (segmentIndex == 2)
		return precPoint.y < point.y;
	if (segmentIndex == 3)
		return precPoint.x < point.x;
	return true;
}

//-----------------------------------------------------------------------------
/*! Help function.*/
void addPointToVector(TThickPoint point, std::vector<TThickPoint> &points, bool insertMiddlePoint)
{
	if (insertMiddlePoint)
		points.push_back((points[points.size() - 1] + point) * 0.5);
	points.push_back(point);
}

//-----------------------------------------------------------------------------
/*! Return intersection between \b bbox and \b chuck;
    set segmentIndex to index of segment that contains intersection. */
TThickPoint getIntersectionPoint(TRectD bbox, const TThickQuadratic *chunk,
								 int &segmentIndex, bool secondChunkIntersection)
{
	TStroke stroke;
	std::vector<TThickPoint> points;
	points.push_back(chunk->getThickP0());
	points.push_back(chunk->getThickP1());
	points.push_back(chunk->getThickP2());
	stroke.reshape(&points[0], points.size());

	std::vector<DoublePair> intersectionInfo;

	std::vector<DoublePair> intersections;
	TSegment segment0(bbox.getP00(), bbox.getP01());
	int count0 = intersect(stroke, segment0, intersections);
	if (count0 > 0) {
		DoublePair pair;
		pair.first = intersections[0].first;
		pair.second = 0;
		intersectionInfo.push_back(pair);
		intersections.clear();
	}
	TSegment segment1(bbox.getP01(), bbox.getP11());
	int count1 = intersect(stroke, segment1, intersections);
	if (count1 > 0) {
		DoublePair pair;
		pair.first = intersections[0].first;
		pair.second = 1;
		intersectionInfo.push_back(pair);
		intersections.clear();
	}
	TSegment segment2(bbox.getP11(), bbox.getP10());
	int count2 = intersect(stroke, segment2, intersections);
	if (count2 > 0) {
		DoublePair pair;
		pair.first = intersections[0].first;
		pair.second = 2;
		intersectionInfo.push_back(pair);
		intersections.clear();
	}
	TSegment segment3(bbox.getP10(), bbox.getP00());
	int count3 = intersect(stroke, segment3, intersections);
	if (count3 > 0) {
		DoublePair pair;
		pair.first = intersections[0].first;
		pair.second = 3;
		intersectionInfo.push_back(pair);
		intersections.clear();
	}
	int infoSize = intersectionInfo.size();
	assert(infoSize <= 2);
	if (infoSize == 1) {
		segmentIndex = intersectionInfo[0].second;
		return stroke.getPoint(intersectionInfo[0].first);
	} else if (infoSize == 2) {
		double firstT = intersectionInfo[0].first;
		double secondT = intersectionInfo[1].first;
		if (!secondChunkIntersection) {
			if (firstT < secondT) {
				segmentIndex = intersectionInfo[0].second;
				return stroke.getPoint(intersectionInfo[0].first);
			} else {
				segmentIndex = intersectionInfo[1].second;
				return stroke.getPoint(intersectionInfo[1].first);
			}
		} else {
			if (firstT > secondT) {
				segmentIndex = intersectionInfo[0].second;
				return stroke.getPoint(intersectionInfo[0].first);
			} else {
				segmentIndex = intersectionInfo[1].second;
				return stroke.getPoint(intersectionInfo[1].first);
			}
		}
	}
	segmentIndex = -1;
	return TThickPoint();
}

//-----------------------------------------------------------------------------
/*! Insert \b bbox corners in \b points if \b bbox corners are contained in \b outPoints stroke. */
void insertBoxCorners(TRectD bbox, std::vector<TThickPoint> &points, std::vector<TThickPoint> outPoints,
					  int currentSegmentIndex, int precSegmentIndex)
{
	if (outPoints[0] != outPoints[(int)outPoints.size() - 1])
		addPointToVector(outPoints[0], outPoints, true);
	assert((int)outPoints.size() % 2 == 1);
	TStroke *outPointsStroke = new TStroke();
	outPointsStroke->reshape(&(outPoints[0]), outPoints.size());
	TVectorImageP vi(new TVectorImage());
	vi->addStroke(outPointsStroke);
	vi->findRegions();
	assert((int)vi->getRegionCount() > 0);
	bool sameIndex = (precSegmentIndex == currentSegmentIndex);
	if (currentSegmentIndex == -1)
		return;
	int j;
	for (j = sameIndex ? 1 : 0; j < 2; j++) {
		bool clockwise = j;
		if (sameIndex)
			clockwise = isClockwise(bbox, currentSegmentIndex, outPoints[outPoints.size() - 2], outPoints[outPoints.size() - 1]);
		int segmentIndex = precSegmentIndex;
		if (sameIndex)
			segmentIndex = clockwise ? currentSegmentIndex - 1 : currentSegmentIndex + 1;
		if (segmentIndex < 0)
			segmentIndex = 3;
		if (segmentIndex > 3)
			segmentIndex = 0;
		while (segmentIndex != currentSegmentIndex) {
			if (sameIndex) //controllo anche il segmento di partenza.
			{
				segmentIndex = currentSegmentIndex;
				sameIndex = false;
			}
			TSegment s = getSegmentByIndex(bbox, segmentIndex);
			TThickPoint corner = clockwise ? s.getP0() : s.getP1();
			int i;
			for (i = 0; i < (int)vi->getRegionCount(); i++)
				if (vi->getRegion(i)->contains(corner)) {
					if ((int)points.size() % 2 == 1)
						points.push_back((points[points.size() - 1] + corner) * 0.5);
					points.push_back(corner);
				}
			segmentIndex = clockwise ? segmentIndex - 1 : segmentIndex + 1;
			if (segmentIndex < 0)
				segmentIndex = 3;
			if (segmentIndex > 3)
				segmentIndex = 0;
		}
	}
}

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

TStroke getStrokeByRect(TRectD r)
{
	TStroke stroke;
	if (r.isEmpty())
		return stroke;
	vector<TThickPoint> points;
	points.push_back(r.getP00());
	points.push_back((r.getP00() + r.getP01()) * 0.5);
	points.push_back(r.getP01());
	points.push_back((r.getP01() + r.getP11()) * 0.5);
	points.push_back(r.getP11());
	points.push_back((r.getP11() + r.getP10()) * 0.5);
	points.push_back(r.getP10());
	points.push_back((r.getP10() + r.getP00()) * 0.5);
	points.push_back(r.getP00());
	stroke.reshape(&(points[0]), points.size());
	stroke.setSelfLoop(true);
	return stroke;
}

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

TStroke getIntersectedStroke(TStroke &stroke, TRectD bbox)
{
	int cpCount = stroke.getControlPointCount();
	if (cpCount == 0)
		return stroke;
	//isFirstTime, startSegmentIndex e startOutPoints sono usati nel il caso in cui lo stroke inizia fuori dalla bbox.
	bool isFirstTime = true;
	std::vector<TThickPoint> points, outPoints, startOutPoints;
	TThickPoint precPoint = stroke.getControlPoint(0);
	bool isPrecPointInternal = bbox.contains(precPoint);
	if (isPrecPointInternal)
		points.push_back(precPoint);
	else
		outPoints.push_back(precPoint);
	int i;
	int precSegmentIndex, currentSegmentIndex, startSegmentIndex, precChunkIndex = -1;
	for (i = 1; i < stroke.getControlPointCount(); i++) {
		TThickPoint point = stroke.getControlPoint(i);
		bool isPointInternal = bbox.contains(point);
		if (isPointInternal && isPrecPointInternal)
			addPointToVector(point, points, (int)points.size() % 2 != i % 2);
		if (!isPointInternal && !isPrecPointInternal)
			addPointToVector(point, outPoints, (int)outPoints.size() > 0 && (int)outPoints.size() % 2 != i % 2);
		if (isPointInternal != isPrecPointInternal) {
			//Devo trovare l'intersezione
			int chunkIndex = (i % 2 == 0) ? (i * 0.5) - 1 : i * 0.5;
			TThickPoint p = getIntersectionPoint(bbox, stroke.getChunk(chunkIndex), currentSegmentIndex, chunkIndex == precChunkIndex);
			precChunkIndex = chunkIndex;
			addPointToVector(p, outPoints, (int)outPoints.size() % 2 == 1);
			if (!isPrecPointInternal && points.size() > 0 && outPoints.size() > 0) {
				insertBoxCorners(bbox, points, outPoints, currentSegmentIndex, precSegmentIndex);
				outPoints.clear();
			} else if (outPoints.size() > 0 && isFirstTime) {
				startSegmentIndex = currentSegmentIndex;
				startOutPoints = outPoints;
				outPoints.clear();
			}
			isFirstTime = false;
			precSegmentIndex = currentSegmentIndex;
			addPointToVector(p, points, (int)points.size() % 2 == 1);
			addPointToVector(p, outPoints, (int)outPoints.size() % 2 == 1);
		}
		if (isPointInternal && !isPrecPointInternal)
			addPointToVector(point, points, (int)points.size() % 2 != i % 2);
		if (!isPointInternal && isPrecPointInternal)
			addPointToVector(point, outPoints, (int)outPoints.size() > 0 && (int)outPoints.size() % 2 != i % 2);
		isPrecPointInternal = isPointInternal;
	}
	//Caso in cui lo stroke aveva il primo punto fuori dalla bbox
	if (!isPrecPointInternal && points.size() > 0 && outPoints.size() > 0) {
		int t;
		for (t = 0; t < (int)outPoints.size(); t++)
			addPointToVector(outPoints[t], startOutPoints, (int)startOutPoints.size() % 2 != 2);
		insertBoxCorners(bbox, points, startOutPoints, startSegmentIndex, currentSegmentIndex);
		outPoints.clear();
	}

	//Caso particolare in cui lo stroke non ha intersezione con la bbox
	if (points.size() == 0) { //Lo stroke e' completamente contenuto nella bbox
		if (bbox.contains(precPoint))
			return stroke;
		else
			return getStrokeByRect(bbox);
	}
	if (points[0] != points[(int)points.size() - 1])
		addPointToVector(points[0], points, true);

	assert((int)points.size() % 2 == 1);
	TStroke intersectedStroke;
	intersectedStroke.reshape(&(points[0]), points.size());
	intersectedStroke.setSelfLoop(true);
	return intersectedStroke;
}

} // namespace

//=============================================================================
// RasterSelection
//-----------------------------------------------------------------------------

RasterSelection::RasterSelection()
	: TSelection(), m_currentImage(), m_oldPalette(0), m_selectionBbox(), m_affine(), m_startPosition(), m_floatingSelection(), m_originalfloatingSelection(), m_fid(), m_transformationCount(0), m_isPastedSelection(false), m_noAntialiasing(false)
{
	m_strokes.clear();
	m_originalStrokes.clear();
}

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

RasterSelection::RasterSelection(const RasterSelection &src)
	: TSelection(), m_currentImage(src.m_currentImage), m_oldPalette(src.m_oldPalette), m_selectionBbox(src.m_selectionBbox), m_strokes(src.m_strokes), m_originalStrokes(src.m_originalStrokes), m_affine(src.m_affine), m_startPosition(src.m_startPosition), m_fid(src.m_fid), m_transformationCount(src.m_transformationCount), m_isPastedSelection(src.m_isPastedSelection), m_noAntialiasing(src.m_noAntialiasing)

{
	setView(src.getView());
	if (src.isFloating()) {
		m_floatingSelection = src.m_floatingSelection->clone();
		if (src.m_originalfloatingSelection)
			m_originalfloatingSelection = src.m_originalfloatingSelection->clone();
		assert(isFloating());
	}
}

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

//!Returns the clone of this selection
TSelection *RasterSelection::clone() const
{
	RasterSelection *rs = new RasterSelection(*this);
	return rs;
}

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

//! Notify to the viewer that the selection is changed.
void RasterSelection::notify()
{
	RasterSelection *selection = dynamic_cast<RasterSelection *>(TTool::getApplication()->getCurrentSelection()->getSelection());
	if (selection)
		selection->notifyView();
}

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

//!Empty the selection.
//! If the selection is floating, the floating image is pasted using the current tranformation.
void RasterSelection::selectNone()
{
	if (isFloating()) {
		pasteFloatingSelection();
		notify();
		return;
	}
	m_selectionBbox = TRectD();
	m_strokes.clear();
	m_originalStrokes.clear();
	m_affine = TAffine();
	m_startPosition = TPoint();
	m_floatingSelection = TRasterP();
	m_originalfloatingSelection = TRasterP();
	m_transformationCount = 0;
	m_isPastedSelection = false;
	m_oldPalette = 0;
	notify();
}

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

void RasterSelection::select(TStroke &stroke)
{
	TRect box = getRaster(m_currentImage)->getBounds();
	TRectD rasterBbox = convertRasterToWorld(box, m_currentImage);
	TStroke intersectedStroke = getIntersectedStroke(stroke, rasterBbox);
	if ((int)intersectedStroke.getControlPointCount() == 0)
		return;
	m_strokes.push_back(intersectedStroke);
	m_originalStrokes.push_back(intersectedStroke);
	notify();
}

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

void RasterSelection::select(const TRectD &rect)
{
	assert(!!m_currentImage);
	TRectD r = rect;
	TRect box = getRaster(m_currentImage)->getBounds();
	r *= convertRasterToWorld(box, m_currentImage);
	if (!r.isEmpty()) {
		TStroke stroke = getStrokeByRect(r);
		if ((int)stroke.getControlPointCount() == 0)
			return;
		m_strokes.push_back(stroke);
		m_originalStrokes.push_back(stroke);
	}
	notify();
}

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

void RasterSelection::selectAll()
{
	if (!m_currentImage)
		return;
	selectNone();
	TRectD wRect = convertRasterToWorld(getRaster(m_currentImage)->getBounds(), m_currentImage);
	select(wRect);
}

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

bool RasterSelection::isEmpty() const
{
	return getStrokesBound(m_strokes).isEmpty();
}

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

void RasterSelection::enableCommands()
{
	enableCommand(this, MI_Clear, &RasterSelection::deleteSelection);
	enableCommand(this, MI_Cut, &RasterSelection::cutSelection);
	enableCommand(this, MI_Copy, &RasterSelection::copySelection);
	enableCommand(this, MI_Paste, &RasterSelection::pasteSelection);
	enableCommand(this, MI_SelectAll, &RasterSelection::selectAll);
}

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

bool RasterSelection::isFloating() const
{
	return m_floatingSelection;
}

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

void RasterSelection::transform(const TAffine &affine)
{
	m_affine = affine * m_affine;
}

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

void RasterSelection::makeFloating()
{
	if (isEmpty())
		return;
	if (!m_currentImage)
		return;
	m_floatingSelection = getImageFromSelection(m_currentImage, *this);
	m_originalfloatingSelection = m_floatingSelection->clone();
	deleteSelectionWithoutUndo(m_currentImage, m_strokes);

	ToolUtils::updateSaveBox();
	TTool *tool = TTool::getApplication()->getCurrentTool()->getTool();
	tool->notifyImageChanged(m_fid);
}

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

void RasterSelection::pasteFloatingSelection()
{
	if (!isFloating())
		return;

	assert(m_transformationCount != -1 && m_transformationCount != -2);

	if (m_isPastedSelection)
		TUndoManager::manager()->popUndo(m_transformationCount + 1);
	else
		TUndoManager::manager()->popUndo(m_transformationCount);

	if (m_transformationCount > 0 || m_isPastedSelection)
		TUndoManager::manager()->add(new UndoPasteFloatingSelection(
			this, m_oldPalette.getPointer(), m_noAntialiasing));
	else if (m_transformationCount == 0)
		TUndoManager::manager()->popUndo(-1, true);

	TRectD wRect = getSelectionBbox();
	pasteFloatingSelectionWithoutUndo(m_currentImage, m_floatingSelection, m_affine, wRect, m_noAntialiasing);

	ToolUtils::updateSaveBox(m_currentImageCell.getSimpleLevel(),
							 m_currentImageCell.getFrameId());

	setFloatingSeletion(TRasterP());
	selectNone();

	TTool *tool = TTool::getApplication()->getCurrentTool()->getTool();
	tool->notifyImageChanged(m_fid);
}

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

void RasterSelection::deleteSelection()
{
	if (!m_currentImage)
		return;
	TTool::Application *app = TTool::getApplication();
	TXshSimpleLevel *level = app->getCurrentLevel()->getSimpleLevel();
	// we have to remove all undo transformation and the undo for the makeFloating operation!
	if (isFloating()) {
		assert(m_transformationCount != -1 && m_transformationCount != -2);
		if (m_isPastedSelection)
			TUndoManager::manager()->popUndo(m_transformationCount + 1);
		else
			TUndoManager::manager()->popUndo(m_transformationCount);
	}
	if (!isPastedSelection() && !isEmpty())
		TUndoManager::manager()->add(new UndoDeleteSelection(this, level));

	if (!isFloating())
		deleteSelectionWithoutUndo(m_currentImage, m_strokes);
	else if (m_oldPalette)
		m_currentImage->getPalette()->assign(m_oldPalette.getPointer());
	m_floatingSelection = TRasterP();
	m_originalfloatingSelection = TRasterP();

	ToolUtils::updateSaveBox();
	selectNone();
	app->getPaletteController()->getCurrentLevelPalette()->notifyPaletteChanged();
	TTool *tool = app->getCurrentTool()->getTool();
	tool->notifyImageChanged(m_fid);
}

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

void RasterSelection::copySelection()
{
	if (isEmpty() || !m_currentImage)
		return;
	TRasterP ras;
	if (!isFloating())
		ras = getImageFromSelection(m_currentImage, *this);
	else
		ras = m_floatingSelection;

	double dpix, dpiy;
	std::vector<TRectD> rect;
	if (TToonzImageP ti = (TToonzImageP)(m_currentImage)) {
		ToonzImageData *data = new ToonzImageData();
		ti->getDpi(dpix, dpiy);
		data->setData(ras, ti->getPalette(), dpix, dpiy, ti->getSize(), rect, m_strokes, m_originalStrokes, m_affine);
		QApplication::clipboard()->setMimeData(cloneData(data));
	} else if (TRasterImageP ri = (TRasterImageP)(m_currentImage)) {
		FullColorImageData *data = new FullColorImageData();
		ri->getDpi(dpix, dpiy);
		data->setData(ras, ri->getPalette(), dpix, dpiy, ri->getRaster()->getSize(), rect, m_strokes, m_originalStrokes, m_affine);
		QApplication::clipboard()->setMimeData(cloneData(data));
	}
}

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

void RasterSelection::cutSelection()
{
	copySelection();
	deleteSelection();
}

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

void RasterSelection::pasteSelection(const RasterImageData *riData)
{
	std::vector<TRectD> rect;
	double currentDpiX, currentDpiY;
	double dpiX, dpiY;
	const ToonzImageData *toonzImageData = dynamic_cast<const ToonzImageData *>(riData);
	const FullColorImageData *fullColorData = dynamic_cast<const FullColorImageData *>(riData);
	if (TToonzImageP ti = (TToonzImageP)m_currentImage) {
		ti->getDpi(currentDpiX, currentDpiY);
		TRasterP cmRas;
		if (fullColorData) {
			DVGui::error(QObject::tr("The copied selection cannot be pasted in the current drawing."));
			return;
		}
		riData->getData(cmRas, dpiX, dpiY, rect, m_strokes, m_originalStrokes, m_affine, m_currentImage->getPalette());
		if (!cmRas)
			return;
		m_floatingSelection = cmRas;
	} else if (TRasterImageP ri = (TRasterImageP)m_currentImage) {
		ri->getDpi(currentDpiX, currentDpiY);
		TRasterP ras;
		riData->getData(ras, dpiX, dpiY, rect, m_strokes, m_originalStrokes, m_affine, ri->getPalette());
		if (!ras)
			return;
		if (TRasterCM32P rasCM = ras) {
			TDimension dim = rasCM->getSize();
			TRaster32P app = TRaster32P(dim.lx, dim.ly);
			TRop::convert(app, rasCM, ri->getPalette());
			ras = app;
		}
		m_floatingSelection = ras;
	}
	if (m_floatingSelection)
		m_originalfloatingSelection = m_floatingSelection->clone();
	TScale sc;
	if (dpiX != 0 && dpiY != 0 && currentDpiX != 0 && currentDpiY != 0)
		sc = TScale(currentDpiX / dpiX, currentDpiY / dpiY);
	m_affine = m_affine * sc;
}

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

void RasterSelection::pasteSelection()
{
	TTool::Application *app = TTool::getApplication();
	TTool *tool = app->getCurrentTool()->getTool();
	TImageP image = tool->getImage(true);
	m_currentImage = image;
	m_fid = tool->getCurrentFid();

	QClipboard *clipboard = QApplication::clipboard();
	const RasterImageData *riData = dynamic_cast<const RasterImageData *>(clipboard->mimeData());
	const StrokesData *stData = dynamic_cast<const StrokesData *>(clipboard->mimeData());
	if (!riData && !stData)
		return;
	if (isFloating())
		pasteFloatingSelection();
	selectNone();
	m_isPastedSelection = true;
	m_oldPalette = m_currentImage->getPalette()->clone();
	if (stData) {
		if (TToonzImageP ti = m_currentImage)
			riData = stData->toToonzImageData(ti);
		else {
			TRasterImageP ri = m_currentImage;
			assert(ri);
			double dpix, dpiy;
			ri->getDpi(dpix, dpiy);
			if (dpix == 0 || dpiy == 0) {
				TPointD dpi = tool->getXsheet()->getScene()->getCurrentCamera()->getDpi();
				dpix = dpi.x;
				dpiy = dpi.y;
				ri->setDpi(dpix, dpiy);
			}
			riData = stData->toFullColorImageData(ri);
		}
	}

	if (!riData)
		return;
	pasteSelection(riData);

	app->getPaletteController()->getCurrentLevelPalette()->notifyPaletteChanged();
	notify();
	TUndoManager::manager()->add(new UndoPasteSelection(this));
}

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

bool RasterSelection::isTransformed()
{
	return !m_affine.isIdentity();
}

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

TRectD RasterSelection::getStrokesBound(std::vector<TStroke> strokes) const
{
	int i;
	TRectD box = TRectD();
	for (i = 0; i < (int)strokes.size(); i++)
		box += strokes[i].getBBox();
	return box;
}

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

TRectD RasterSelection::getSelectionBound() const
{
	if (m_strokes.size() == 0)
		return TRectD();
	TRectD selectionBox = getStrokesBound(m_strokes);
	if (isFloating())
		selectionBox = m_affine * selectionBox;
	return selectionBox;
}

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

TRectD RasterSelection::getOriginalSelectionBound() const
{
	if (m_originalStrokes.size() == 0)
		return TRectD();
	return getStrokesBound(m_originalStrokes);
}

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

TRectD RasterSelection::getSelectionBbox() const
{
	TRectD rect = m_selectionBbox;
	if (isFloating())
		rect = m_affine * m_selectionBbox;
	return rect;
}

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

void RasterSelection::setSelectionBbox(const TRectD &rect)
{
	m_selectionBbox = rect;
	if (m_currentImage)
		m_selectionBbox = intersection(m_selectionBbox, m_currentImage);
}