#pragma once
 
#ifndef RASTERSELECTIONTOOL_INCLUDED
#define RASTERSELECTIONTOOL_INCLUDED
 
#include "selectiontool.h"
#include "setsaveboxtool.h"
#include "tools/rasterselection.h"
 
// forward declaration
class RasterSelectionTool;
class VectorFreeDeformer;
 
//=============================================================================
// RasterFreeDeformer
//-----------------------------------------------------------------------------
 
class RasterFreeDeformer : public FreeDeformer
{
	TRasterP m_ras;
	TRasterP m_newRas;
	bool m_noAntialiasing;
 
public:
	RasterFreeDeformer(TRasterP ras);
	~RasterFreeDeformer();
 
	/*! Set \b index point to \b p, with index from 0 to 3. */
	void setPoint(int index, const TPointD &p);
	/*! Helper function. */
	void setPoints(const TPointD &p0, const TPointD &p1, const TPointD &p2, const TPointD &p3);
	TRasterP getImage() const { return m_newRas; }
	void deformImage();
	void setNoAntialiasing(bool value) { m_noAntialiasing = value; }
};
 
//=============================================================================
// DragSelectionTool
//-----------------------------------------------------------------------------
 
namespace DragSelectionTool
{
 
//=============================================================================
// UndoRasterDeform
//-----------------------------------------------------------------------------
 
class UndoRasterDeform : public TUndo
{
	static int m_id;
	RasterSelectionTool *m_tool;
	std::string m_oldFloatingImageId, m_newFloatingImageId;
	std::vector<TStroke> m_newStrokes;
	std::vector<TStroke> m_oldStrokes;
	DragSelectionTool::DeformValues m_oldDeformValues, m_newDeformValues;
	FourPoints m_oldBBox, m_newBBox;
	TPointD m_oldCenter, m_newCenter;
 
	TDimension m_dim;
	int m_pixelSize;
 
public:
	UndoRasterDeform(RasterSelectionTool *tool);
	~UndoRasterDeform();
 
	void registerRasterDeformation();
 
	void undo() const;
	void redo() const;
 
	int getSize() const;
 
	QString getHistoryString()
	{
		return QObject::tr("Deform Raster");
	}
};
 
//=============================================================================
// UndoRasterTransform
//-----------------------------------------------------------------------------
 
class UndoRasterTransform : public TUndo
{
	RasterSelectionTool *m_tool;
	TAffine m_oldTransform, m_newTransform;
	TPointD m_oldCenter, m_newCenter;
	FourPoints m_oldBbox, m_newBbox;
	DragSelectionTool::DeformValues m_oldDeformValues, m_newDeformValues;
 
public:
	UndoRasterTransform(RasterSelectionTool *tool);
	void setChangedValues();
	void undo() const;
	void redo() const;
 
	int getSize() const { return sizeof(*this); }
 
	QString getHistoryString()
	{
		return QObject::tr("Transform Raster");
	}
};
 
//=============================================================================
// RasterDeformTool
//-----------------------------------------------------------------------------
 
class RasterDeformTool : public DeformTool
{
protected:
	TAffine m_transform;
	UndoRasterTransform *m_transformUndo;
	UndoRasterDeform *m_deformUndo;
 
	//!It's true when use RasterFreeDeformer
	bool m_isFreeDeformer;
 
	void applyTransform(FourPoints bbox);
	void applyTransform(TAffine aff, bool modifyCenter);
	void addTransformUndo();
 
	void leftButtonDrag(const TPointD &pos, const TMouseEvent &e){};
	void draw(){};
 
public:
	RasterDeformTool(RasterSelectionTool *tool, bool freeDeformer);
};
 
//=============================================================================
// RasterRotationTool
//-----------------------------------------------------------------------------
 
class RasterRotationTool : public RasterDeformTool
{
	Rotation *m_rotation;
 
public:
	RasterRotationTool(RasterSelectionTool *tool);
	void transform(TAffine aff, double angle);
	void leftButtonDrag(const TPointD &pos, const TMouseEvent &e);
	void draw();
};
 
//=============================================================================
// RasterFreeDeformTool
//-----------------------------------------------------------------------------
 
class RasterFreeDeformTool : public RasterDeformTool
{
	FreeDeform *m_freeDeform;
 
public:
	RasterFreeDeformTool(RasterSelectionTool *tool);
	void leftButtonDrag(const TPointD &pos, const TMouseEvent &e);
};
 
//=============================================================================
// RasterMoveSelectionTool
//-----------------------------------------------------------------------------
 
class RasterMoveSelectionTool : public RasterDeformTool
{
	MoveSelection *m_moveSelection;
 
public:
	RasterMoveSelectionTool(RasterSelectionTool *tool);
	void transform(TAffine aff);
	void leftButtonDown(const TPointD &pos, const TMouseEvent &e);
	void leftButtonDrag(const TPointD &pos, const TMouseEvent &e);
};
 
//=============================================================================
// RasterScaleTool
//-----------------------------------------------------------------------------
 
class RasterScaleTool : public RasterDeformTool
{
	Scale *m_scale;
 
public:
	RasterScaleTool(RasterSelectionTool *tool, int type);
	/*! Return scale value. */
	TPointD transform(int index, TPointD newPos);
	void leftButtonDown(const TPointD &pos, const TMouseEvent &e);
	void leftButtonDrag(const TPointD &pos, const TMouseEvent &e);
};
 
} //namespace DragSelectionTool
 
//=============================================================================
// RasterSelectionTool
//-----------------------------------------------------------------------------
 
class RasterSelectionTool : public SelectionTool
{
	Q_DECLARE_TR_FUNCTIONS(RasterSelectionTool)
 
	RasterSelection m_rasterSelection;
	int m_transformationCount;
 
	//!Used in ToonzRasterImage to switch from selection tool to modify savebox tool.
	TBoolProperty m_modifySavebox;
	SetSaveboxTool *m_setSaveboxTool;
	TBoolProperty m_noAntialiasing;
 
	//!Used to deform bbox strokes.
	VectorFreeDeformer *m_selectionFreeDeformer;
 
	void modifySelectionOnClick(TImageP image, const TPointD &pos, const TMouseEvent &e);
 
	void drawFloatingSelection();
 
public:
	RasterSelectionTool(int targetType);
 
	void setBBox(const DragSelectionTool::FourPoints &points, int index = 0);
 
	void setNewFreeDeformer();
	VectorFreeDeformer *getSelectionFreeDeformer() const;
 
	bool isFloating() const;
 
	void leftButtonDown(const TPointD &pos, const TMouseEvent &);
	void mouseMove(const TPointD &pos, const TMouseEvent &e);
	void leftButtonDrag(const TPointD &pos, const TMouseEvent &);
	void leftButtonUp(const TPointD &pos, const TMouseEvent &);
	void leftButtonDoubleClick(const TPointD &, const TMouseEvent &e);
 
	void draw();
 
	TSelection *getSelection();
	bool isSelectionEmpty();
 
	void computeBBox();
 
	void doOnActivate();
	void doOnDeactivate();
 
	void onImageChanged();
 
	void transformFloatingSelection(const TAffine &affine, const TPointD &center,
									const DragSelectionTool::FourPoints &points);
	void increaseTransformationCount();
	void decreaseTransformationCount();
 
	void onActivate();
	TBoolProperty *getModifySaveboxProperty()
	{
		if (m_targetType & ToonzImage)
			return &m_modifySavebox;
		return 0;
	}
	bool onPropertyChanged(std::string propertyName);
	bool getNoAntialiasingValue() { return m_noAntialiasing.getValue(); }
 
protected:
	void updateTranslation();
};
 
#endif //RASTERSELECTIONTOOL_INCLUDED