#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