#pragma once

#ifndef BRUSHTOOL_H

#define BRUSHTOOL_H

#include "tgeometry.h"

#include "tproperty.h"

#include "trasterimage.h"

#include "ttoonzimage.h"

#include "toonz/strokegenerator.h"

#include "tools/tool.h"

#include "tools/cursors.h"

#include <qcoreapplication></qcoreapplication>

#include <qradialgradient></qradialgradient>

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

//  Forward declarations

class TTileSetCM32;

class TTileSaverCM32;

class RasterStrokeGenerator;

class BluredBrush;

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

//************************************************************************

//    Brush Data declaration

//************************************************************************

struct BrushData : public TPersist {

	PERSIST_DECLARATION(BrushData)

	std::wstring m_name;

	double m_min, m_max, m_acc, m_hardness, m_opacityMin, m_opacityMax;

	bool m_selective, m_pencil, m_breakAngles, m_pressure;

	int m_cap, m_join, m_miter;

	BrushData();

	BrushData(const std::wstring &name);

	bool operator<(const BrushData &other) const { return m_name < other.m_name; }

	void saveData(TOStream &os);

	void loadData(TIStream &is);

};

//************************************************************************

//    Brush Preset Manager declaration

//************************************************************************

class BrushPresetManager

{

	TFilePath m_fp;				   //!< Presets file path

	std::set<brushdata> m_presets; //!< Current presets container</brushdata>

public:

	BrushPresetManager() {}

	void load(const TFilePath &fp);

	void save();

	const TFilePath &path() { return m_fp; };

	const std::set<brushdata> &presets() const { return m_presets; }</brushdata>

	void addPreset(const BrushData &data);

	void removePreset(const std::wstring &name);

};

//************************************************************************

//    Smooth Stroke declaration

//    Brush stroke smoothing buffer.

//************************************************************************

class SmoothStroke

{

public:

    SmoothStroke() {}

    ~SmoothStroke() {}

    // begin stroke

    // smooth is smooth strength, from 0 to 100

    void beginStroke(int smooth);

    // add stroke point

    void addPoint(const TThickPoint& point);

    // end stroke

    void endStroke();

    // Get generated stroke points which has been smoothed.

    // Both addPoint() and endStroke() generate new smoothed points.

    // This method will removed generated points 

    void getSmoothPoints(std::vector<tthickpoint>& smoothPoints);</tthickpoint>

private:

    void generatePoints();

private:

    int m_smooth;

    int m_outputIndex;

    int m_readIndex;

    std::vector<tthickpoint> m_rawPoints;</tthickpoint>

    std::vector<tthickpoint> m_outputPoints;</tthickpoint>

};

//************************************************************************

//    Brush Tool declaration

//************************************************************************

class BrushTool : public TTool

{

	Q_DECLARE_TR_FUNCTIONS(BrushTool)

public:

	BrushTool(std::string name, int targetType);

	ToolType getToolType() const { return TTool::LevelWriteTool; }

	ToolOptionsBox *createOptionsBox();

	void updateTranslation();

	void onActivate();

	void onDeactivate();

	bool preLeftButtonDown();

	void leftButtonDown(const TPointD &pos, const TMouseEvent &e);

	void leftButtonDrag(const TPointD &pos, const TMouseEvent &e);

	void leftButtonUp(const TPointD &pos, const TMouseEvent &e);

	void mouseMove(const TPointD &pos, const TMouseEvent &e);

	void draw();

	void onEnter();

	void onLeave();

	int getCursorId() const { return ToolCursor::PenCursor; }

	TPropertyGroup *getProperties(int targetType);

	bool onPropertyChanged(std::string propertyName);

	void onImageChanged();

	void setWorkAndBackupImages();

	void updateWorkAndBackupRasters(const TRect &rect);

	void initPresets();

	void loadPreset();

	void addPreset(QString name);

	void removePreset();

	void finishRasterBrush(const TPointD &pos, int pressureVal);

	//return true if the pencil mode is active in the Brush / PaintBrush / Eraser Tools.

	bool isPencilModeActive();

    void addTrackPoint(const TThickPoint& point, double pixelSize2);

    void flushTrackPoint();

protected:

	TPropertyGroup m_prop[2];

	TDoublePairProperty m_thickness;

	TDoublePairProperty m_rasThickness;

	TDoubleProperty m_accuracy;

	TDoubleProperty m_hardness;

	TEnumProperty m_preset;

	TBoolProperty m_selective;

	TBoolProperty m_breakAngles;

	TBoolProperty m_pencil;

	TBoolProperty m_pressure;

	TEnumProperty m_capStyle;

	TEnumProperty m_joinStyle;

	TIntProperty m_miterJoinLimit;

	StrokeGenerator m_track;

	RasterStrokeGenerator *m_rasterTrack;

	TTileSetCM32 *m_tileSet;

	TTileSaverCM32 *m_tileSaver;

	TPixel32 m_currentColor;

	int m_styleId;

	double m_minThick, m_maxThick;

	TRectD m_modifiedRegion;

	TPointD m_dpiScale,

		m_mousePos, //!< Current mouse position, in world coordinates.

		m_brushPos; //!< World position the brush will be painted at.

	BluredBrush *m_bluredBrush;

	QRadialGradient m_brushPad;

	TRasterCM32P m_backupRas;

	TRaster32P m_workRas;

	std::vector<tthickpoint> m_points;</tthickpoint>

	TRect m_strokeRect,

		m_lastRect;

    SmoothStroke m_smoothStroke;

	BrushPresetManager m_presetsManager; //!< Manager for presets of this tool instance

	bool m_active,

		m_enabled,

		m_isPrompting, //!< Whether the tool is prompting for spline substitution.

		m_firstTime,

		m_isPath,

		m_presetsLoaded;

	/*--- 作業中のFrameIdをクリック時に保存し、マウスリリース時（Undoの登録時）に別のフレームに

  移動していたときの不具合を修正する。---*/

	TFrameId m_workingFrameId;

protected:

	static void drawLine(const TPointD &point, const TPointD ¢re, bool horizontal, bool isDecimal);

	static void drawEmptyCircle(TPointD point, int thick, bool isLxEven, bool isLyEven, bool isPencil);

};

#endif //BRUSHTOOL_H