#ifndef FUNCTIONPANELTOOLS_H

#define FUNCTIONPANELTOOLS_H

#include "toonzqt/functionpanel.h"

#include "tdoublekeyframe.h"

#include "toonz/doubleparamcmd.h"

// class QMouseEvent;

class KeyframeSetter;

class TDoubleParam;

class FunctionPanel::DragTool

{

public:

	DragTool() {}

	virtual ~DragTool() {}

	virtual void click(QMouseEvent *e) {}

	virtual void drag(QMouseEvent *e) {}

	virtual void release(QMouseEvent *e) {}

	virtual void draw(QPainter &p) {}

};

class MoveFrameDragTool : public FunctionPanel::DragTool

{

	FunctionPanel *m_panel;

	TFrameHandle *m_frameHandle;

public:

	MoveFrameDragTool(FunctionPanel *panel, TFrameHandle *frameHandle);

	void drag(QMouseEvent *e);

};

class PanDragTool : public FunctionPanel::DragTool

{

	FunctionPanel *m_panel;

	QPoint m_oldPos;

	bool m_xLocked, m_yLocked;

public:

	PanDragTool(FunctionPanel *panel, bool xLocked, bool yLocked);

	void click(QMouseEvent *e);

	void drag(QMouseEvent *e);

};

class ZoomDragTool : public FunctionPanel::DragTool

{

	FunctionPanel *m_panel;

	QPoint m_startPos, m_oldPos;

	int m_zoomType;

public:

	enum ZoomType { FrameZoom = 1,

					ValueZoom = 2 };

	ZoomDragTool(FunctionPanel *panel, ZoomType zoomType) : m_panel(panel), m_zoomType((int)zoomType) {}

	void click(QMouseEvent *e);

	void drag(QMouseEvent *e);

	void release(QMouseEvent *e);

};

class RectSelectTool : public FunctionPanel::DragTool

{

	FunctionPanel *m_panel;

	TDoubleParam *m_curve;

	QPoint m_startPos;

	QRect m_rect;

public:

	RectSelectTool(FunctionPanel *panel, TDoubleParam *curve) : m_panel(panel), m_curve(curve) {}

	void click(QMouseEvent *e);

	void drag(QMouseEvent *e);

	void release(QMouseEvent *e);

	void draw(QPainter &painter);

};

class MovePointDragTool : public FunctionPanel::DragTool

{

	FunctionPanel *m_panel;

	QPoint m_startPos, m_oldPos;

	double m_deltaFrame;

	// length and kIndex of speedinout handles which can change because of point moving

	double m_speed0Length;

	int m_speed0Index;

	double m_speed1Length;

	int m_speed1Index;

	std::vector<KeyframeSetter *> m_setters;

	bool m_groupEnabled;

	FunctionSelection *m_selection;

public:

	MovePointDragTool(

		FunctionPanel *panel,

		TDoubleParam *curve);

	~MovePointDragTool();

	void addKeyframe2(int kIndex);

	// void addKeyframe(int kIndex) {m_setter->selectKeyframe(kIndex);}

	void createKeyframe(double frame);

	void selectKeyframes(double frame);

	void setSelection(FunctionSelection *selection);

	void click(QMouseEvent *e);

	void drag(QMouseEvent *e);

	void release(QMouseEvent *e);

};

class MoveHandleDragTool : public FunctionPanel::DragTool

{

public:

	typedef FunctionPanel::Handle Handle;

private:

	FunctionPanel *m_panel;

	TDoubleParam *m_curve;

	QPoint m_startPos; //, m_oldPos;

	double m_deltaFrame;

	Handle m_handle;

	int m_kIndex;

	TDoubleKeyframe m_keyframe;

	KeyframeSetter m_setter;

	double m_segmentWidth;

	QPointF m_nSpeed; // speedInOut constraint

public:

	FunctionTreeModel::ChannelGroup *m_channelGroup;

	MoveHandleDragTool(

		FunctionPanel *panel,

		TDoubleParam *curve,

		int kIndex,

		Handle handle);

	void click(QMouseEvent *e);

	void drag(QMouseEvent *e);

	void release(QMouseEvent *e);

};

class MoveGroupHandleDragTool : public FunctionPanel::DragTool

{

public:

	typedef FunctionPanel::Handle Handle;

private:

	FunctionPanel *m_panel;

	double m_keyframePosition;

	Handle m_handle;

	std::vector<std::pair<TDoubleKeyframe, KeyframeSetter *>> m_setters;

public:

	MoveGroupHandleDragTool(

		FunctionPanel *panel,

		double keyframePosition,

		Handle handle);

	~MoveGroupHandleDragTool();

	void click(QMouseEvent *e);

	void drag(QMouseEvent *e);

	void release(QMouseEvent *e);

};

#endif