#include "tools/tool.h"

#include "tools/toolutils.h"

#include "tstroke.h"

#include "tstrokeutil.h"

#include "tstrokedeformations.h"

#include "tmathutil.h"

#include "tools/cursors.h"

#include "drawutil.h"

#include "toonz/tobjecthandle.h"

#include "toonz/txshlevelhandle.h"

#include "toonz/tstageobject.h"

using namespace ToolUtils;

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

namespace

{

const UINT MAX_SAMPLE = 2;

const int MY_ERROR = -1;

const double c_LenghtOfBenderRegion = 10.0;

const int IS_BEGIN = 0;

const int IS_END = 1;

const int IS_ALL = 2;

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

double _extractFirst(DoublePair val)

{

	return val.first;

}

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

double _extractSecond(DoublePair val)

{

	return val.second;

}

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

bool strokeIsConnected(const TStroke &s, double toll = TConsts::epsilon)

{

	bool out = true;

	int count = s.getChunkCount();

	long double toll2 = sq(TConsts::epsilon);

	if (count > 0) {

		const TThickQuadratic *refTQ = s.getChunk(0);

		for (int i = 1; i < count; ++i) {

			if (out &&

				tdistance2(refTQ->getP2(), s.getChunk(i)->getP0()) > toll2)

				out = false;

		}

	}

	return out;

}

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

inline bool strokeIsConnected(const TStroke *s, double toll = TConsts::epsilon)

{

	assert(s);

	return strokeIsConnected(*s, toll);

}

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

/*

Extract an value of a pair.

*/

void extract(const vector<doublepair> &s, ArrayOfDouble &d, bool isFirstOrSecond = true)</doublepair>

{

	if (isFirstOrSecond)

		std::transform(s.begin(), s.end(), back_inserter(d), _extractFirst);

	else

		std::transform(s.begin(), s.end(), back_inserter(d), _extractSecond);

}

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

double retrieveInitLength(double strokeLength, int beginEndOrAll)

{

	double initLength = MY_ERROR;

	switch (beginEndOrAll) {

	case IS_BEGIN:

		initLength = 0.0;

		break;

	case IS_END:

		initLength = strokeLength;

		break;

	case IS_ALL:

		initLength = strokeLength * 0.5;

		break;

	}

	return initLength;

}

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

inline bool isOdd(UINT val)

{

	return val & 1 ? true : false;

}

inline bool isEven(UINT val)

{

	return !isOdd(val);

}

void clearPointerMap(map<tstroke *,="" vector<int=""> *> &corners)</tstroke>

{

	map<tstroke *,="" vector<int=""> *>::iterator it = corners.begin();</tstroke>

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

		delete it->second;

	}

}

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

// Bender Tool

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

class BenderTool : public TTool

{

private:

	TUndo *m_undo;

	bool m_atLeastOneIsChanged;

	vector<bool> m_directionIsChanged;</bool>

	vector<tpointd> m_accumulator;</tpointd>

	void increaseCP(TStroke *, int);

	bool m_active;

	int m_cursor;

	enum PNT_SELECTED {

		BNDR_NULL = 0,

		BNDR_P0 = 1,

		BNDR_P1 = 2

	};

	struct benderStrokeInfo {

		TStroke *m_stroke;

		DoublePair m_extremes;

		int m_isBeginEndOrAll;

		benderStrokeInfo(TStroke *stroke,

						 DoublePair &info,

						 int isBeginEndOrAll)

		{

			m_stroke = stroke;

			m_extremes = info;

			m_isBeginEndOrAll = isBeginEndOrAll;

		}

	};

	std::vector<benderstrokeinfo> m_info;</benderstrokeinfo>

	// contains information about stroke which have

	//  intersection with benderSegment

	map<tstroke *,="" arrayofstroke=""> m_metaStroke;</tstroke>

	map<tstroke *,="" vector<int=""> *> m_hitStrokeCorners;</tstroke>

	bool m_showTangents;

	// value selected

	int m_buttonDownCounter;

	TSegment m_benderSegment;

	TPointD m_prevPoint;

	double m_rotationVersus;

	//!

	ArrayOfStroke m_strokesToRotate;

	//!

	ArrayOfStroke m_strokesToBend;

	vector<int> m_changedStrokes;</int>

	bool m_enableDragSelection;

	void findCurves(TVectorImageP &);

	void findVersus(const TPointD &p);

	double computeRotationVersus(const TPointD &,

								 const TPointD &);

public:

	void initBenderAction(TVectorImageP &, const TPointD &);

	// BenderTool(TVectorImageP  vimage, GLTestWidget* ref );

	BenderTool();

	virtual ~BenderTool();

	ToolType getToolType() const { return TTool::LevelWriteTool; }

	void draw();

	void leftButtonDown(const TPointD &, const TMouseEvent &);

	void leftButtonDrag(const TPointD &, const TMouseEvent &);

	void leftButtonUp(const TPointD &, const TMouseEvent &);

	void onEnter();

	void onActivate()

	{

		m_buttonDownCounter = 1;

		m_prevPoint = TConsts::napd;

		m_benderSegment.setP0(TConsts::napd);

		m_benderSegment.setP1(TConsts::napd);

		m_metaStroke.clear();

		clearPointerMap(m_hitStrokeCorners);

		m_hitStrokeCorners.clear();

	}

	int getCursorId() const { return m_cursor; }

} BenderTool;

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

BenderTool::BenderTool()

	: TTool("T_Bender"), m_showTangents(false), m_buttonDownCounter(1), m_rotationVersus(0.0), m_enableDragSelection(false), m_undo(0), m_cursor(ToolCursor::BenderCursor)

{

	bind(TTool::Vectors);

	m_prevPoint = TConsts::napd;

	m_benderSegment.setP0(TConsts::napd);

	m_benderSegment.setP1(TConsts::napd);

}

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

BenderTool::~BenderTool()

{

}

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

void BenderTool::onEnter()

{

	if ((TVectorImageP)getImage(false))

		m_cursor = ToolCursor::BenderCursor;

	else

		m_cursor = ToolCursor::CURSOR_NO;

}

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

void BenderTool::leftButtonUp(const TPointD &pos, const TMouseEvent &)

{

	m_active = false;

	TVectorImageP vi = TImageP(getImage(true));

	if (!vi)

		return;

	QMutexLocker lock(vi->getMutex());

	m_active = true;

	vector<tstroke *=""> oldStrokesArray(m_changedStrokes.size());</tstroke>

	int i;

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

		oldStrokesArray[i] = new TStroke(*(vi->getStroke(m_changedStrokes[i])));

	if (3 == m_buttonDownCounter) {

		m_rotationVersus = 0.0;

		// hide bender tool

		m_prevPoint = TConsts::napd;

		m_benderSegment.setP0(TConsts::napd);

		m_benderSegment.setP1(TConsts::napd);

		typedef std::map<tstroke *,="" arrayofstroke="">::iterator itAATS;</tstroke>

		//UINT count=0;

		for (itAATS it1 = m_metaStroke.begin(); it1 != m_metaStroke.end(); ++it1) {

			UINT i;

			// copy vector and...

			ArrayOfStroke &refA = it1->second;

			TStroke *s = it1->first;

			TStroke *out = merge(refA);

			out->reduceControlPoints(getPixelSize(), *(m_hitStrokeCorners[s]));

			if (it1->first->isSelfLoop()) {

				int cpCount = out->getControlPointCount();

				TThickPoint p1 = out->getControlPoint(0);

				TThickPoint p2 = out->getControlPoint(cpCount - 1);

				TThickPoint midP = (p1 + p2) * 0.5;

				out->setControlPoint(0, midP);

				out->setControlPoint(cpCount - 1, midP);

				out->setSelfLoop(true);

			}

			it1->first->swap(*out);

			it1->first->setStyle(out->getStyle());

			it1->first->outlineOptions() = out->outlineOptions();

			it1->first->invalidate();

			for (i = 0; i < refA.size(); ++i)

				delete refA[i];

			delete out;

		}

		m_metaStroke.clear();

		clearPointerMap(m_hitStrokeCorners);

		m_hitStrokeCorners.clear();

		m_buttonDownCounter = 1;

		UINT lastCpIndex, i, size = m_changedStrokes.size();

		TStroke *stroke;

		TThickPoint p1, p2, middleP;

		double loopError = 0.5 * getPixelSize();

		for (i = 0; i < size; i++) {

			stroke = vi->getStroke(m_changedStrokes[i]);

			if (m_directionIsChanged[i])

				stroke->changeDirection();

			lastCpIndex = stroke->getControlPointCount() - 1;

			p1 = stroke->getControlPoint(0);

			p2 = stroke->getControlPoint(lastCpIndex);

			if (isAlmostZero(p1.x - p2.x, loopError) && isAlmostZero(p1.y - p2.y, loopError)) {

				middleP = (p1 + p2) * 0.5;

				stroke->setControlPoint(0, middleP);

				stroke->setControlPoint(lastCpIndex, middleP);

			} else {

				stroke->setSelfLoop(false);

			}

		}

		vi->notifyChangedStrokes(m_changedStrokes, oldStrokesArray);

		notifyImageChanged();

		if (m_undo)

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

		m_undo = 0;

	}

	clearPointerContainer(oldStrokesArray);

	m_active = false;

	invalidate();

}

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

void BenderTool::leftButtonDrag(const TPointD &pos, const TMouseEvent &)

{

	if (!m_active)

		return;

	TVectorImageP vi = TImageP(getImage(true));

	if (!vi)

		return;

	QMutexLocker lock(vi->getMutex());

	double pixelSize = getPixelSize();

	if (tdistance2(pos, m_prevPoint) < 9.0 * pixelSize * pixelSize)

		return;

	if (m_buttonDownCounter < 3)

		return;

	if (m_enableDragSelection) {

		m_accumulator.push_back(pos);

		if (m_accumulator.size() < 3)

			return;

		TPointD middlePnt;

		accumulate(m_accumulator.begin(), m_accumulator.end(), middlePnt);

		static const double inv_of_3 = 1.0 / 2.0;

		middlePnt = inv_of_3 * middlePnt;

		m_accumulator.clear();

		initBenderAction(vi, pos + middlePnt);

		m_enableDragSelection = false;

		m_buttonDownCounter = 3;

	}

	TPointD p = pos;

	TPointD

		vc(p - m_benderSegment.getP0()),		   // current vector

		vp(m_prevPoint - m_benderSegment.getP0()); // previous vector

	TPointD s2v = m_benderSegment.getSpeed();

	// some check to jump invalid case

	double norm2BenderSeg = norm2(s2v);

	double norm2CurrentVect = norm2(vc);

	double norm2PreviousVect = norm2(vp);

	// invalid segments

	if (0.0 == norm2BenderSeg ||

		0.0 == norm2PreviousVect ||

		0.0 == norm2CurrentVect)

		return;

	// invalid rotation versus

	if (tsign(cross(s2v, vc)) != m_rotationVersus)

		return;

	// compute delta of rotation angle

	double diff = asin(cross(normalize(vp), normalize(vc)));

	// make a rototranslation matrix

	TRotation rot(m_benderSegment.getP0(), rad2degree(diff));

	// rotate references

	for (itAS its = m_strokesToRotate.begin(); its != m_strokesToRotate.end(); ++its)

		(*its)->transform(rot);

	// deform strokes

	for (UINT i = 0; i < m_info.size(); ++i) {

		TStroke &ref = *m_info[i].m_stroke;

		// DoublePair  &extr   = m_info[i].m_extremes;

		double strokeLength = ref.getLength();

		double initLength = retrieveInitLength(strokeLength, m_info[i].m_isBeginEndOrAll); // ? 0.0 : strokeLength;

		if (MY_ERROR == initLength)

			return;

		int innerOrOuter = m_info[i].m_isBeginEndOrAll == IS_ALL ? TStrokeBenderDeformation::OUTER : TStrokeBenderDeformation::INNER;

		TStrokeBenderDeformation

			def(&ref,

				m_benderSegment.getP0(),

				diff,

				initLength,

				innerOrOuter,

				strokeLength);

		modifyControlPoints(ref, def);

	}

	// fix previous point

	m_prevPoint = p;

	invalidate();

}

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

void BenderTool::leftButtonDown(const TPointD &p, const TMouseEvent &)

{

	m_active = false;

	TVectorImageP vi = TImageP(getImage(true));

	if (!vi)

		return;

	QMutexLocker lock(vi->getMutex());

	m_active = true;

	switch (m_buttonDownCounter) {

	case 1:

		findCurves(vi);

		m_strokesToRotate.clear();

		// le curve puntate sono state eliminate dalla clear precedente

		m_info.clear();

		m_benderSegment.setP0(p);

		m_benderSegment.setP1(p);

		break;

	case 2: // second buttonDown

		m_prevPoint = p;

		m_benderSegment.setP1(p);

		m_enableDragSelection = true;

		break;

		/*

    case 3: // third buttonDown      

    // initBenderAction( vi, p );

    break;

    */

	}

	++m_buttonDownCounter;

	invalidate();

	//vi->validateRegionEdges(vi->getStroke( m_strokeIndex ), true);

}

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

void BenderTool::findVersus(const TPointD &p)

{

	TPointD

		v1 = m_benderSegment.getSpeed(),

		v2 = p - m_benderSegment.getP0();

	// fix rotation versus

	m_rotationVersus = tsign(cross(v1, v2));

	if (isAlmostZero(m_rotationVersus))

		m_rotationVersus = 1.0;

}

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

inline void one_minus_x(double &x)

{

	x = 1.0 - x;

}

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

void BenderTool::findCurves(TVectorImageP &vi)

{

	ArrayOfStroke strokeToModify;

	m_changedStrokes.clear();

	m_directionIsChanged.clear();

	UINT j;

	for (UINT i = 0; i < vi->getStrokeCount(); ++i) // for all stroke in image

	{

		if (!vi->inCurrentGroup(i))

			continue;

		TStroke *s = vi->getStroke(i); // a useful reference

		vector<doublepair> pair_intersection; // informations about extremes</doublepair>

		// if there is interesection between stroke and bender tool

		if (intersect(*s, m_benderSegment, pair_intersection)) {

			if (s->isSelfLoop()) {

				//make the semgnet longer

				// such as the points are just a littleBit

				// outside the stroke bbox

				const double littleBit = 0.1;

				TRectD bbox = s->getBBox();

				TPointD bboxP0 = bbox.getP00();

				TPointD bboxP1 = bbox.getP11();

				double bboxX0 = tmin(bboxP0.x, bboxP1.x);

				double bboxX1 = tmax(bboxP0.x, bboxP1.x);

				double bboxY0 = tmin(bboxP0.y, bboxP1.y);

				double bboxY1 = tmax(bboxP0.y, bboxP1.y);

				TSegment segment;

				TPointD pp0 = m_benderSegment.getP0();

				TPointD pp1 = m_benderSegment.getP1();

				TPointD newP0;

				TPointD newP1;

				if (bbox.contains(pp1)) {

					double x = (pp0.x < pp1.x) ? bboxX1 : bboxX0;

					double y = (pp0.y < pp1.y) ? bboxY1 : bboxY0;

					double t;

					if (pp1.x == pp0.x && pp1.y == pp0.y) {

						assert(!"segmento del bender puntiforme");

						return;

					}

					if (pp1.x == pp0.x)

						t = (y - pp0.y) / (pp1.y - pp0.y);

					else if (pp1.y == pp0.y)

						t = (x - pp0.x) / (pp1.x - pp0.x);

					else

						t = tmin((x - pp0.x) / (pp1.x - pp0.x), (y - pp0.y) / (pp1.y - pp0.y));

					newP1 = (t > 1) ? m_benderSegment.getPoint(t + littleBit) : pp1;

				} else

					newP1 = pp1;

				if (bbox.contains(pp0)) {

					double x = (pp0.x > pp1.x) ? bboxX1 : bboxX0;

					double y = (pp0.y > pp1.y) ? bboxY1 : bboxY0;

					double t;

					if (pp1.x == pp0.x && pp1.y == pp0.y) {

						assert(!"segmento del bender puntiforme");

						return;

					}

					if (pp1.x == pp0.x)

						t = (y - pp0.y) / (pp1.y - pp0.y);

					else if (pp1.y == pp0.y)

						t = (x - pp0.x) / (pp1.x - pp0.x);

					else

						t = tmax((x - pp0.x) / (pp1.x - pp0.x), (y - pp0.y) / (pp1.y - pp0.y));

					newP0 = (t < 0) ? m_benderSegment.getPoint(t - littleBit) : pp0;

				} else

					newP0 = pp0;

				segment = TSegment(newP0, newP1);

				pair_intersection.clear();

				intersect(*s, segment, pair_intersection);

				assert(isEven(pair_intersection.size()));

				assert(!pair_intersection.empty());

				if (pair_intersection.empty()) //non dovrebbe accadere

					continue;

			}

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

			strokeToModify.push_back(s);

			m_changedStrokes.push_back(i);

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

			m_metaStroke[s] = ArrayOfStroke();

			ArrayOfStroke &tempAS = m_metaStroke[s];

			// extract information about intersection parameter

			ArrayOfDouble intersection;

			extract(pair_intersection, intersection);

			// now add stroke to rotate in m_info struct

			TPointD v = s->getSpeed(intersection[0]);

			TPointD normalToBenderSeg;

			normalToBenderSeg = m_rotationVersus * rotate90(m_benderSegment.getSpeed());

			m_atLeastOneIsChanged = false;

			//m_directionIsChanged = false;

			if (tsign(v * normalToBenderSeg) < 0) {

				m_atLeastOneIsChanged = true;

				m_directionIsChanged.push_back(true);

				std::for_each(intersection.begin(), intersection.end(), one_minus_x);

				std::reverse(intersection.begin(), intersection.end());

				s->changeDirection();

			} else

				m_directionIsChanged.push_back(false);

			splitStroke(*s, intersection, tempAS);

			// number of curves is number of intersection plus one

			UINT numberOfCurves = intersection.size() + 1;

			// and begin increase of control point

			if (isEven(intersection.size()) && m_atLeastOneIsChanged) // if solution are even

			{

				for (j = 0; j < numberOfCurves; ++j) {

					if (isOdd(j))

						increaseCP(tempAS[j], IS_ALL);

					else

						m_strokesToRotate.push_back(tempAS[j]);

				}

			} else {

				increaseCP(tempAS[0], IS_END);

				for (j = 1; j < numberOfCurves - 1; ++j) {

					if (isEven(j))

						increaseCP(tempAS[j], IS_ALL);

					else

						m_strokesToRotate.push_back(tempAS[j]);

				}

				if (isOdd(numberOfCurves))

					increaseCP(tempAS.back(), IS_BEGIN);

				else

					m_strokesToRotate.push_back(tempAS.back());

			}

			TStroke *tempForCorners = merge(tempAS);

			vector<int> *corners = new vector<int>;</int></int>

			corners->push_back(0);

			detectCorners(tempForCorners, 20, *corners);

			corners->push_back(tempForCorners->getChunkCount());

			m_hitStrokeCorners[s] = corners;

			delete tempForCorners;

		}

	}

	if (!strokeToModify.empty()) {

		UINT i, size = strokeToModify.size();

		for (i = 0; i < size; i++)

			if (m_directionIsChanged[i])

				strokeToModify[i]->changeDirection();

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

			m_undo = new UndoPath(getXsheet()->getStageObject(getObjectId())->getSpline());

		else {

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

			assert(sl);

			TFrameId id = getCurrentFid();

			m_undo = new UndoModifyListStroke(sl, id, strokeToModify);

		}

		for (i = 0; i < size; i++)

			if (m_directionIsChanged[i])

				strokeToModify[i]->changeDirection();

	}

}

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

void BenderTool::increaseCP(TStroke *tmpStroke, int beginEndOrAll)

{

	DoublePair extremes;

	double strokeLength = tmpStroke->getLength();

	double initLength = retrieveInitLength(strokeLength, beginEndOrAll);

	if (MY_ERROR == initLength)

		return;

	TStrokeBenderDeformation deformer(tmpStroke, initLength, strokeLength * 0.5);

	increaseControlPoints(*tmpStroke, deformer, getPixelSize());

	tmpStroke->disableComputeOfCaches();

	m_info.push_back(benderStrokeInfo(tmpStroke, extremes, beginEndOrAll));

}

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

void BenderTool::draw()

{

	//TAffine viewMatrix = getViewer()->getViewMatrix();

	//glPushMatrix();

	//tglMultMatrix(viewMatrix);

	double pixelSize = getPixelSize();

	typedef std::map<tstroke *,="" arrayofstroke="">::const_iterator mapTACit;</tstroke>

	for (mapTACit cit1 = m_metaStroke.begin(); cit1 != m_metaStroke.end(); ++cit1) {

		const ArrayOfStroke &tmp = (*cit1).second;

		tglColor(TPixel32::Red);

		for (citAS cit2 = tmp.begin(); cit2 != tmp.end(); ++cit2)

			drawStrokeCenterline(**cit2, pixelSize);

		//drawStroke(**cit2, TRect(0,0,1000,1000), TAffine() );

	}

	double length = m_benderSegment.getLength();

	// pnt to draw rotation vector

	TPointD pnt;

	if (length != 0.0) {

		TPointD v = m_prevPoint - m_benderSegment.getP0();

		double tmp = norm2(v);

		if (tmp != 0.0) {

			pnt = v * (length / sqrt(tmp));

			pnt += m_benderSegment.getP0();

		}

	} else

		pnt = m_prevPoint;

	// rotation vector

	if (m_buttonDownCounter == 3) {

		tglColor(TPixel::Black);

		tglDrawSegment(m_benderSegment.getP0(), pnt);

		drawPoint(pnt, pixelSize);

	}

	// bender vector

	tglColor(TPixel::Red);

	tglDrawSegment(m_benderSegment.getP0(), m_benderSegment.getP1());

	drawPoint(m_benderSegment.getP0(), pixelSize);

	drawPoint(m_benderSegment.getP1(), pixelSize);

	// point where is mouse pointer

	drawPoint(m_prevPoint, pixelSize);

	// arrow in up direction

	TPointD vDir = m_benderSegment.getSpeed();

	double length2 = norm2(vDir);

	if (length2 == 0.0) {

		//glPopMatrix();

		return;

	}

	TPointD vUp = 15.0 * normalize(rotate90(vDir));

	tglColor(TPixel::Magenta);

	TPointD middlePnt = 0.5 * (m_benderSegment.getP0() + m_benderSegment.getP1());

	drawArrow(TSegment(middlePnt, m_rotationVersus * vUp + middlePnt), pixelSize);

	//glPopMatrix();

}

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

void BenderTool::initBenderAction(TVectorImageP &vi, const TPointD &p)

{

	// reset counter

	// m_buttonDownCounter = 0;

	// Versus of bender segment depends from

	//  point selected to do rotation

	// P0 is always center of rotation

	if (tdistance2(p, m_benderSegment.getP0()) <

		tdistance2(p, m_benderSegment.getP1())) {

		TPointD tmpPnt = m_benderSegment.getP1();

		m_benderSegment.setP1(m_benderSegment.getP0());

		m_benderSegment.setP0(tmpPnt);

	}

	// fix information about versus

	findVersus(p);

	// find curves to bender and init data structures

	findCurves(vi);

	// now it's possible set first position of rotation

	// in the same position of extremes

	m_benderSegment.setP1(p);

	m_prevPoint = p;

}

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

} //namespace

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

//TTool *getBenderTool() {return &BenderTool;}