#pragma once

#ifndef CONTROLPOINT_SELECTION_INCLUDED

#define CONTROLPOINT_SELECTION_INCLUDED

#include "toonzqt/selection.h"

#include "tools/tool.h"

#include "tstroke.h"

#include "tvectorimage.h"

#include "tcurves.h"

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

// ControlPointEditorStroke

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

class ControlPointEditorStroke

{

private:

	class ControlPoint

	{

	public:

		int m_pointIndex;

		TThickPoint m_speedIn;

		TThickPoint m_speedOut;

		bool m_isCusp;

		ControlPoint(int i, TThickPoint speedIn, TThickPoint speedOut, bool isCusp = true)

			: m_pointIndex(i), m_speedIn(speedIn), m_speedOut(speedOut), m_isCusp(isCusp)

		{

		}

		ControlPoint()

		{

		}

	};

	QList<controlpoint> m_controlPoints;</controlpoint>

	TVectorImageP m_vi;

	int m_strokeIndex;

	void adjustChunkParity();

	/*! Reset m_controlPoints. */

	void resetControlPoints();

	TThickPoint getPureDependentPoint(int index) const;

	void getDependentPoints(int index, std::vector<std::pair<int, tthickpoint="">> &points) const;</std::pair<int,>

	void updatePoints();

	void updateDependentPoint(int index);

	inline int nextIndex(int index) const;

	inline int prevIndex(int index) const;

	void moveSpeedIn(int index, const TPointD &delta, double minDistance);

	void moveSpeedOut(int index, const TPointD &delta, double minDistance);

	void moveSingleControlPoint(int index, const TPointD &delta);

public:

	enum PointType {

		CONTROL_POINT,

		SPEED_IN,

		SPEED_OUT,

		SEGMENT,

		NONE

	};

	ControlPointEditorStroke()

		: m_vi() {}

	~ControlPointEditorStroke()

	{

		m_controlPoints.clear();

	}

	ControlPointEditorStroke *clone() const;

	/*! Modify stroke: between two linear or cusp point must be a pair chunk number.

      PAY ATTENTION: Can add control point in the stroke. */

	void setStroke(const TVectorImageP &vi, int strokeIndex);

	TStroke *getStroke() const { return m_vi ? m_vi->getStroke(m_strokeIndex) : 0; }

	void setStrokeIndex(int strokeIndex) { m_strokeIndex = strokeIndex; }

	int getStrokeIndex() const { return m_strokeIndex; }

	int getControlPointCount() const { return m_controlPoints.size(); }

	TThickPoint getControlPoint(int index) const;

	/*! From index point in \b ControlPointEditorStroke to index point in \b TStroke. */

	int getIndexPointInStroke(int index) const;

	TThickPoint getSpeedInPoint(int index) const;

	TThickPoint getSpeedOutPoint(int index) const;

	//! Return true if index point is a cusp.

	bool isCusp(int index) const;

	//! Set \b m_isCusp of \b index point to \b isCusp.

	/*! If isCusp == false check \b setSpeedIn and modify speed. */

	void setCusp(int index, bool isCusp, bool setSpeedIn);

	bool isSpeedInLinear(int index) const;

	bool isSpeedOutLinear(int index) const;

	void setLinearSpeedIn(int index, bool linear = true, bool updatePoints = true);

	void setLinearSpeedOut(int index, bool linear = true, bool updatePoints = true);

	/*! If isLinear==true set to "0" the speedIn and speedOut value of index point; otherwise set to default value.

      Return true if speedIn or speedOut is modified.

      If updatePoints is true update dependent point after.*/

	bool setLinear(int index, bool isLinear, bool updatePoints = true);

	bool setControlPointsLinear(std::set<int> points, bool isLinear);</int>

	/*! Move \b index ControlPoint of \b delta. */

	void moveControlPoint(int index, const TPointD &delta);

	/*! Add a ControlPoint in \b point. Return added ControlPoint index. */

	int addControlPoint(const TPointD &pos);

	void deleteControlPoint(int index);

	/*! Move speed in \b index of \b delta. */

	void moveSpeed(int index, const TPointD &delta, bool isIn, double minDistance);

	/*! Move segment between \b beforeIndex and \b nextIndex of \b delta. */

	void moveSegment(int beforeIndex, int nextIndex, const TPointD &delta, const TPointD &pos);

	/*!Return NONE if \b pos is far from stroke.*/

	PointType getPointTypeAt(const TPointD &pos, double &distance2, int &index) const;

	bool isSelfLoop() const

	{

		TStroke *stroke = getStroke();

		return stroke && stroke->isSelfLoop();

	}

};

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

// ControlPointSelection

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

class ControlPointSelection : public QObject, public TSelection

{

	Q_OBJECT

private:

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

	int m_strokeIndex;

	ControlPointEditorStroke *m_controlPointEditorStroke;

public:

	ControlPointSelection() : m_controlPointEditorStroke(0), m_strokeIndex(-1)

	{

	}

	void setControlPointEditorStroke(ControlPointEditorStroke *controlPointEditorStroke)

	{

		m_controlPointEditorStroke = controlPointEditorStroke;

	}

	bool isEmpty() const { return m_selectedPoints.empty(); }

	void selectNone() { m_selectedPoints.clear(); }

	bool isSelected(int index) const;

	void select(int index);

	void unselect(int index);

	void deleteControlPoints();

	void addMenuItems(QMenu *menu);

	void enableCommands();

protected slots:

	void setLinear();

	void setUnlinear();

};

#endif //CONTROLPOINT_SELECTION_INCLUDED