#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_smooth, 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_smooth;

  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