#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)
{
	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)
{
	map<TStroke *, vector<int> *>::iterator it = corners.begin();
	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;
	vector<TPointD> m_accumulator;

	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;

	// contains information about stroke which have
	//  intersection with benderSegment
	map<TStroke *, ArrayOfStroke> m_metaStroke;
	map<TStroke *, vector<int> *> m_hitStrokeCorners;

	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;

	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());

	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;
		//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

		// 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>;
			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;
	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;}