#include "tools/strokeselection.h"

// TnzTools includes

#include "tools/imagegrouping.h"

#include "tools/toolhandle.h"

#include "tools/tool.h"

#include "tools/toolutils.h"

// TnzQt includes

#include "toonzqt/selectioncommandids.h"

#include "toonzqt/imageutils.h"

#include "toonzqt/tselectionhandle.h"

#include "toonzqt/strokesdata.h"

#include "toonzqt/rasterimagedata.h"

#include "toonzqt/dvdialog.h"

// TnzLib includes

#include "toonz/tpalettehandle.h"

#include "toonz/palettecontroller.h"

#include "toonz/tobjecthandle.h"

#include "toonz/txshlevelhandle.h"

#include "toonz/tscenehandle.h"

#include "toonz/tcenterlinevectorizer.h"

#include "toonz/stage.h"

#include "toonz/tstageobject.h"

#include "toonz/toonzscene.h"

#include "toonz/sceneproperties.h"

// TnzCore includes

#include "tthreadmessage.h"

#include "tundo.h"

#include "tstroke.h"

#include "tvectorimage.h"

#include "tcolorstyles.h"

#include "tpalette.h"

// Qt includes

#include <qapplication></qapplication>

#include <qclipboard></qclipboard>

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

namespace

{

void vectorizeToonzImageData(const TVectorImageP &image, const ToonzImageData *tiData,

							 std::set<int> &indexes, TPalette *palette,</int>

							 const VectorizerConfiguration &config)

{

	if (!tiData)

		return;

	QApplication::setOverrideCursor(Qt::WaitCursor);

	TRasterP ras;

	std::vector<trectd> rects;</trectd>

	std::vector<tstroke> strokes;</tstroke>

	std::vector<tstroke> originalStrokes;</tstroke>

	TAffine affine;

	double dpiX, dpiY;

	tiData->getData(ras, dpiX, dpiY, rects, strokes, originalStrokes, affine, image->getPalette());

	TRasterCM32P rasCM = ras;

	TToonzImageP ti(rasCM, rasCM->getBounds());

	VectorizerCore vc;

	TVectorImageP vi = vc.vectorize(ti, config, palette);

	assert(vi);

	vi->setPalette(palette);

	TScale sc(dpiX / Stage::inch, dpiY / Stage::inch);

	int i;

	TRectD selectionBounds;

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

		selectionBounds += rects[i];

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

		selectionBounds += strokes[i].getBBox();

	TTranslation tr(selectionBounds.getP00());

	for (i = 0; i < (int)vi->getStrokeCount(); i++) {

		TStroke *stroke = vi->getStroke(i);

		stroke->transform(sc.inv() * affine * tr, true);

	}

	UINT oldImageSize = image->getStrokeCount();

	image->mergeImage(vi, TAffine());

	UINT newImageSize = image->getStrokeCount();

	indexes.clear();

	for (UINT sI = oldImageSize; sI < newImageSize; sI++)

		indexes.insert(sI);

	QApplication::restoreOverrideCursor();

}

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

void copyStrokesWithoutUndo(TVectorImageP image, std::set<int> &indexes)</int>

{

	QClipboard *clipboard = QApplication::clipboard();

	StrokesData *data = new StrokesData();

	data->setImage(image, indexes);

	clipboard->setMimeData(data, QClipboard::Clipboard);

}

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

bool pasteStrokesWithoutUndo(TVectorImageP image, std::set<int> &outIndexes,</int>

							 TSceneHandle *sceneHandle, bool insert = true)

{

	QMutexLocker lock(image->getMutex());

	QClipboard *clipboard = QApplication::clipboard();

	const StrokesData *stData = dynamic_cast<const *="" strokesdata="">(clipboard->mimeData());</const>

	const ToonzImageData *tiData = dynamic_cast<const *="" toonzimagedata="">(clipboard->mimeData());</const>

	const FullColorImageData *fciData = dynamic_cast<const *="" fullcolorimagedata="">(clipboard->mimeData());</const>

	std::set<int> indexes = outIndexes;</int>

	if (stData)

		stData->getImage(image, indexes, insert);

	else if (tiData) {

		ToonzScene *scene = sceneHandle->getScene();

		assert(scene);

		const VectorizerParameters *vParams = scene->getProperties()->getVectorizerParameters();

		assert(vParams);

		std::auto_ptr<vectorizerconfiguration> config(vParams->getCurrentConfiguration(0.0));</vectorizerconfiguration>

		vectorizeToonzImageData(image, tiData, indexes, image->getPalette(), *config);

	} else if (fciData) {

		DVGui::error(QObject::tr("The copied selection cannot be pasted in the current drawing."));

		return false;

	} else

		return false;

	StrokeSelection *selection = dynamic_cast<strokeselection *="">(TTool::getApplication()->getCurrentSelection()->getSelection());</strokeselection>

	if (selection)

		selection->notifyView();

	outIndexes = indexes; //outIndexes is a reference  to current  selection, so the notifyImageChanged could reset it!

	return true;

}

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

void deleteStrokesWithoutUndo(TVectorImageP image, std::set<int> &indexes)</int>

{

	QMutexLocker lock(image->getMutex());

	vector<int> indexesV(indexes.begin(), indexes.end());</int>

	TRectD bbox;

	UINT i = 0;

	for (; i < indexesV.size(); i++)

		bbox += image->getStroke(indexesV[i])->getBBox();

	vector<tfilledregioninf> regions;</tfilledregioninf>

	ImageUtils::getFillingInformationOverlappingArea(image, regions, bbox);

	TVectorImageP other = image->splitImage(indexesV, true);

	indexes.clear();

	TTool::getApplication()->getCurrentTool()->getTool()->notifyImageChanged();

	StrokeSelection *selection = dynamic_cast<strokeselection *="">(TTool::getApplication()->getCurrentSelection()->getSelection());</strokeselection>

	if (selection)

		selection->notifyView();

}

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

void cutStrokesWithoutUndo(TVectorImageP image, std::set<int> &indexes)</int>

{

	copyStrokesWithoutUndo(image, indexes);

	deleteStrokesWithoutUndo(image, indexes);

}

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

// CopyStrokesUndo

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

class CopyStrokesUndo : public TUndo

{

	QMimeData *m_oldData;

	QMimeData *m_newData;

public:

	CopyStrokesUndo(QMimeData *oldData,

					QMimeData *newData)

		: m_oldData(oldData), m_newData(newData)

	{

	}

	void undo() const

	{

		QClipboard *clipboard = QApplication::clipboard();

		clipboard->setMimeData(cloneData(m_oldData), QClipboard::Clipboard);

	}

	void redo() const

	{

		QClipboard *clipboard = QApplication::clipboard();

		clipboard->setMimeData(cloneData(m_newData), QClipboard::Clipboard);

	}

	int getSize() const

	{

		return sizeof(*this);

	}

};

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

// PasteStrokesUndo

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

class PasteStrokesUndo : public TUndo

{

	TXshSimpleLevelP m_level;

	TFrameId m_frameId;

	std::set<int> m_indexes;</int>

	TPaletteP m_oldPalette;

	QMimeData *m_oldData;

	TSceneHandle *m_sceneHandle;

public:

	PasteStrokesUndo(TXshSimpleLevel *level,

					 const TFrameId &frameId,

					 std::set<int> &indexes,</int>

					 TPaletteP oldPalette,

					 TSceneHandle *sceneHandle)

		: m_level(level), m_frameId(frameId), m_indexes(indexes), m_oldPalette(oldPalette), m_sceneHandle(sceneHandle)

	{

		QClipboard *clipboard = QApplication::clipboard();

		m_oldData = cloneData(clipboard->mimeData());

	}

	~PasteStrokesUndo()

	{

		delete m_oldData;

	}

	void undo() const

	{

		TVectorImageP image = m_level->getFrame(m_frameId, true);

		// Se la selezione corrente e' la stroke selection devo svuotarla,

		// altrimenti puo' rimanere selezionato uno stroke che non esiste piu'.

		StrokeSelection *selection = dynamic_cast<strokeselection *="">(TTool::getApplication()->getCurrentSelection()->getSelection());</strokeselection>

		if (selection)

			selection->selectNone();

		std::set<int> indexes = m_indexes;</int>

		deleteStrokesWithoutUndo(image, indexes);

	}

	void redo() const

	{

		TVectorImageP image = m_level->getFrame(m_frameId, true);

		std::set<int> indexes = m_indexes;</int>

		QClipboard *clipboard = QApplication::clipboard();

		QMimeData *data = cloneData(clipboard->mimeData());

		clipboard->setMimeData(cloneData(m_oldData), QClipboard::Clipboard);

		pasteStrokesWithoutUndo(image, indexes, m_sceneHandle);

		TTool::getApplication()->getCurrentTool()->getTool()->notifyImageChanged();

		clipboard->setMimeData(data, QClipboard::Clipboard);

	}

	int getSize() const

	{

		return sizeof(*this);

	}

};

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

class RemoveEndpointsUndo : public TUndo

{

	TXshSimpleLevelP m_level;

	TFrameId m_frameId;

	vector<pair<int, *="" tstroke="">> m_strokes;</pair<int,>

public:

	RemoveEndpointsUndo(TXshSimpleLevel *level,

						const TFrameId &frameId,

						vector<pair<int, *="" tstroke="">> strokes)</pair<int,>

		: m_level(level), m_frameId(frameId), m_strokes(strokes)

	{

	}

	~RemoveEndpointsUndo()

	{

		int i;

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

			delete m_strokes[i].second;

	}

	void undo() const

	{

		TVectorImageP vi = m_level->getFrame(m_frameId, true);

		int i;

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

			TStroke *newS = new TStroke(*(m_strokes[i].second));

			newS->setId(m_strokes[i].second->getId());

			vi->restoreEndpoints(m_strokes[i].first, newS);

		}

		StrokeSelection *selection = dynamic_cast<strokeselection *="">(TTool::getApplication()->getCurrentSelection()->getSelection());</strokeselection>

		if (selection)

			selection->selectNone();

		TTool::getApplication()->getCurrentTool()->getTool()->notifyImageChanged();

	}

	void redo() const

	{

		int i;

		TVectorImageP vi = m_level->getFrame(m_frameId, true);

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

			TStroke *s = vi->removeEndpoints(m_strokes[i].first);

			delete s;

			//assert(s==m_strokes[i].second);

		}

		TTool::getApplication()->getCurrentTool()->getTool()->notifyImageChanged();

	}

	int getSize() const

	{

		return sizeof(*this);

	}

};

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

// DeleteFramesUndo

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

class DeleteStrokesUndo : public TUndo

{

protected:

	TXshSimpleLevelP m_level;

	TFrameId m_frameId;

	std::set<int> m_indexes;</int>

	QMimeData *m_data;

	TSceneHandle *m_sceneHandle;

public:

	DeleteStrokesUndo(TXshSimpleLevel *level,

					  const TFrameId &frameId,

					  std::set<int> indexes, QMimeData *data,</int>

					  TSceneHandle *sceneHandle)

		: m_level(level), m_frameId(frameId), m_indexes(indexes), m_data(data), m_sceneHandle(sceneHandle)

	{

	}

	~DeleteStrokesUndo()

	{

		delete m_data;

	}

	void undo() const

	{

		QClipboard *clipboard = QApplication::clipboard();

		QMimeData *oldData = cloneData(clipboard->mimeData());

		clipboard->setMimeData(cloneData(m_data), QClipboard::Clipboard);

		std::set<int> indexes = m_indexes;</int>

		TVectorImageP image = m_level->getFrame(m_frameId, true);

		pasteStrokesWithoutUndo(image, indexes, m_sceneHandle, false);

		TTool::getApplication()->getCurrentTool()->getTool()->notifyImageChanged();

		clipboard->setMimeData(oldData, QClipboard::Clipboard);

	}

	void redo() const

	{

		TVectorImageP image = m_level->getFrame(m_frameId, true);

		std::set<int> indexes = m_indexes;</int>

		deleteStrokesWithoutUndo(image, indexes);

	}

	int getSize() const

	{

		return sizeof(*this);

	}

};

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

// CutStrokesUndo

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

class CutStrokesUndo : public DeleteStrokesUndo

{

public:

	CutStrokesUndo(TXshSimpleLevel *level,

				   const TFrameId &frameId,

				   std::set<int> indexes, QMimeData *data,</int>

				   TSceneHandle *sceneHandle)

		: DeleteStrokesUndo(level, frameId, indexes, data, sceneHandle)

	{

	}

	~CutStrokesUndo()

	{

	}

	void redo() const

	{

		TVectorImageP image = m_level->getFrame(m_frameId, true);

		std::set<int> indexes = m_indexes;</int>

		cutStrokesWithoutUndo(image, indexes);

	}

};

} // namespace

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

//

// StrokeSelection ctor/dtor

//

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

StrokeSelection::StrokeSelection()

	: m_groupCommand(new TGroupCommand()), m_sceneHandle(), m_updateSelectionBBox(false)

{

	m_groupCommand->setSelection(this);

}

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

StrokeSelection::~StrokeSelection()

{

}

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

StrokeSelection::StrokeSelection(const StrokeSelection &other)

	: m_vi(other.m_vi), m_indexes(other.m_indexes), m_groupCommand(new TGroupCommand()), m_sceneHandle(other.m_sceneHandle), m_updateSelectionBBox(other.m_updateSelectionBBox)

{

	m_groupCommand->setSelection(this);

}

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

StrokeSelection &StrokeSelection::operator=(const StrokeSelection &other)

{

	m_vi = other.m_vi;

	m_indexes = other.m_indexes;

	m_sceneHandle = other.m_sceneHandle;

	m_updateSelectionBBox = other.m_updateSelectionBBox;

	return *this;

}

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

void StrokeSelection::select(int index, bool on)

{

	if (on)

		m_indexes.insert(index);

	else

		m_indexes.erase(index);

}

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

void StrokeSelection::toggle(int index)

{

	std::set<int>::iterator it = m_indexes.find(index);</int>

	if (it == m_indexes.end())

		m_indexes.insert(index);

	else

		m_indexes.erase(it);

}

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

//

// removeEndpoints

//

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

void StrokeSelection::removeEndpoints()

{

	if (!m_vi)

		return;

	if (m_indexes.empty())

		return;

	vector<pair<int, *="" tstroke="">> undoData;</pair<int,>

	m_vi->findRegions();

	set<int>::iterator it = m_indexes.begin();</int>

	for (; it != m_indexes.end(); ++it) {

		TStroke *s = m_vi->removeEndpoints(*it);

		if (s)

			undoData.push_back(pair<int, *="" tstroke="">(*it, s));</int,>

	}

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

	TXshSimpleLevel *level = TTool::getApplication()->getCurrentLevel()->getSimpleLevel();

	if (!undoData.empty())

		TUndoManager::manager()->add(new RemoveEndpointsUndo(level, tool->getCurrentFid(), undoData));

	m_updateSelectionBBox = true;

	tool->notifyImageChanged();

	m_updateSelectionBBox = false;

}

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

//

// deleteStrokes

//

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

void StrokeSelection::deleteStrokes()

{

	if (!m_vi)

		return;

	if (m_indexes.empty())

		return;

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

	if (!tool)

		return;

	bool isSpline = tool->getApplication()->getCurrentObject()->isSpline();

	TUndo *undo;

	if (isSpline)

		undo = new ToolUtils::UndoPath(tool->getXsheet()->getStageObject(tool->getObjectId())->getSpline());

	StrokesData *data = new StrokesData();

	data->setImage(m_vi, m_indexes);

	std::set<int> oldIndexes = m_indexes;</int>

	deleteStrokesWithoutUndo(m_vi, m_indexes);

	if (!isSpline) {

		TXshSimpleLevel *level = TTool::getApplication()->getCurrentLevel()->getSimpleLevel();

		TUndoManager::manager()->add(new DeleteStrokesUndo(level, tool->getCurrentFid(), oldIndexes, data, m_sceneHandle));

	} else {

		assert(undo);

		if (undo)

			TUndoManager::manager()->add(undo);

	}

}

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

//

// StrokeSelection::copy()

//

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

void StrokeSelection::copy()

{

	if (m_indexes.empty())

		return;

	QClipboard *clipboard = QApplication::clipboard();

	QMimeData *oldData = cloneData(clipboard->mimeData());

	copyStrokesWithoutUndo(m_vi, m_indexes);

	QMimeData *newData = cloneData(clipboard->mimeData());

	//TUndoManager::manager()->add(new CopyStrokesUndo(oldData, newData));

}

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

//

// StrokeSelection::paste()

//

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

void StrokeSelection::paste()

{

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

	if (!tool)

		return;

	if (TTool::getApplication()->getCurrentObject()->isSpline()) {

		const StrokesData *stData = dynamic_cast<const *="" strokesdata="">(QApplication::clipboard()->mimeData());</const>

		if (!stData)

			return;

		TVectorImageP splineImg = tool->getImage(true);

		TVectorImageP img = stData->m_image;

		if (!splineImg || !img)

			return;

		QMutexLocker lock(splineImg->getMutex());

		TUndo *undo = new ToolUtils::UndoPath(tool->getXsheet()->getStageObject(tool->getObjectId())->getSpline());

		while (splineImg->getStrokeCount() > 0)

			splineImg->deleteStroke(0);

		TStroke *stroke = img->getStroke(0);

		splineImg->addStroke(new TStroke(*stroke), false);

		TUndoManager::manager()->add(undo);

		tool->notifyImageChanged();

		tool->invalidate();

		return;

	}

	TVectorImageP tarImg = tool->touchImage();

	if (!tarImg)

		return;

	TPaletteP palette = tarImg->getPalette();

	TPaletteP oldPalette = new TPalette();

	if (palette)

		oldPalette = palette->clone();

	bool isPaste = pasteStrokesWithoutUndo(tarImg, m_indexes, m_sceneHandle);

	if (isPaste) {

		TXshSimpleLevel *level = TTool::getApplication()->getCurrentLevel()->getSimpleLevel();

		TUndoManager::manager()->add(new PasteStrokesUndo(level, tool->getCurrentFid(), m_indexes, oldPalette, m_sceneHandle));

		m_updateSelectionBBox = isPaste;

	}

	tool->notifyImageChanged();

	tool->getApplication()->getPaletteController()->getCurrentLevelPalette()->notifyPaletteChanged();

	m_updateSelectionBBox = false;

	tool->invalidate();

}

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

//

// StrokeSelection::cut()

//

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

void StrokeSelection::cut()

{

	if (m_indexes.empty())

		return;

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

	if (!tool)

		return;

	bool isSpline = tool->getApplication()->getCurrentObject()->isSpline();

	TUndo *undo;

	if (isSpline)

		undo = new ToolUtils::UndoPath(tool->getXsheet()->getStageObject(tool->getObjectId())->getSpline());

	StrokesData *data = new StrokesData();

	data->setImage(m_vi, m_indexes);

	std::set<int> oldIndexes = m_indexes;</int>

	cutStrokesWithoutUndo(m_vi, m_indexes);

	if (!isSpline) {

		TXshSimpleLevel *level = tool->getApplication()->getCurrentLevel()->getSimpleLevel();

		TUndoManager::manager()->add(new CutStrokesUndo(level, tool->getCurrentFid(), oldIndexes, data, m_sceneHandle));

	} else {

		assert(undo);

		if (undo)

			TUndoManager::manager()->add(undo);

	}

}

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

//

// enableCommands

//

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

void StrokeSelection::enableCommands()

{

	enableCommand(this, MI_Clear, &StrokeSelection::deleteStrokes);

	enableCommand(this, MI_Cut, &StrokeSelection::cut);

	enableCommand(this, MI_Copy, &StrokeSelection::copy);

	enableCommand(this, MI_Paste, &StrokeSelection::paste);

	enableCommand(m_groupCommand.get(), MI_Group, &TGroupCommand::group);

	enableCommand(m_groupCommand.get(), MI_Ungroup, &TGroupCommand::ungroup);

	enableCommand(m_groupCommand.get(), MI_BringToFront, &TGroupCommand::front);

	enableCommand(m_groupCommand.get(), MI_BringForward, &TGroupCommand::forward);

	enableCommand(m_groupCommand.get(), MI_SendBack, &TGroupCommand::back);

	enableCommand(m_groupCommand.get(), MI_SendBackward, &TGroupCommand::backward);

	enableCommand(m_groupCommand.get(), MI_EnterGroup, &TGroupCommand::enterGroup);

	enableCommand(m_groupCommand.get(), MI_ExitGroup, &TGroupCommand::exitGroup);

	enableCommand(this, MI_RemoveEndpoints, &StrokeSelection::removeEndpoints);

}

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

namespace

{

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

class UndoSetStrokeStyle : public TUndo

{

	TVectorImageP m_image;

	vector<int> m_strokeIndexes;</int>

	vector<int> m_oldStyles;</int>

	int m_newStyle;

public:

	UndoSetStrokeStyle(TVectorImageP image, int newStyle)

		: m_image(image), m_newStyle(newStyle)

	{

	}

	void addStroke(TStroke *stroke)

	{

		m_strokeIndexes.push_back(m_image->getStrokeIndex(stroke));

		m_oldStyles.push_back(stroke->getStyle());

	}

	void undo() const

	{

		UINT size = m_strokeIndexes.size();

		assert(size == m_oldStyles.size());

		for (UINT i = 0; i != size; i++) {

			int index = m_strokeIndexes[i];

			if (index != -1 && index < (int)m_image->getStrokeCount())

				m_image->getStroke(index)->setStyle(m_oldStyles[i]);

		}

		TTool::getApplication()->getCurrentTool()->getTool()->notifyImageChanged();

	}

	void redo() const

	{

		UINT size = m_strokeIndexes.size();

		assert(size == m_oldStyles.size());

		for (UINT i = 0; i != size; i++) {

			int index = m_strokeIndexes[i];

			if (index != -1 && index < (int)m_image->getStrokeCount())

				m_image->getStroke(index)->setStyle(m_newStyle);

		}

		TTool::getApplication()->getCurrentTool()->getTool()->notifyImageChanged();

	}

	int getSize() const

	{

		return sizeof(*this) +

			   m_strokeIndexes.capacity() * sizeof(m_strokeIndexes[0]) +

			   m_oldStyles.capacity() * sizeof(m_oldStyles[0]);

	}

};

} // namespace

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

//

// changeColorStyle

//

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

void StrokeSelection::changeColorStyle(int styleIndex)

{

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

	if (!tool)

		return;

	TVectorImageP img(tool->getImage(true));

	if (!img)

		return;

	TPalette *palette = img->getPalette();

	TColorStyle *cs = palette->getStyle(styleIndex);

	if (!cs->isStrokeStyle())

		return;

	if (m_indexes.empty())

		return;

	UndoSetStrokeStyle *undo = new UndoSetStrokeStyle(img, styleIndex);

	std::set<int>::iterator it;</int>

	for (it = m_indexes.begin(); it != m_indexes.end(); ++it) {

		int index = *it;

		assert(0 <= index && index < (int)img->getStrokeCount());

		TStroke *stroke = img->getStroke(index);

		undo->addStroke(stroke);

		stroke->setStyle(styleIndex);

	}

	tool->notifyImageChanged();

	TUndoManager::manager()->add(undo);

}