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

	std::vector<tstroke> m_oldStrokes;</tstroke>

	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 ¢er,

									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