#include "toonzrasterbrushtool.h"

// TnzTools includes

#include "tools/toolhandle.h"

#include "tools/toolutils.h"

#include "tools/tooloptions.h"

#include "bluredbrush.h"

// TnzQt includes

#include "toonzqt/dvdialog.h"

#include "toonzqt/imageutils.h"

// TnzLib includes

#include "toonz/tobjecthandle.h"

#include "toonz/txsheethandle.h"

#include "toonz/txshlevelhandle.h"

#include "toonz/tframehandle.h"

#include "toonz/tcolumnhandle.h"

#include "toonz/txsheet.h"

#include "toonz/tstageobject.h"

#include "toonz/tstageobjectspline.h"

#include "toonz/rasterstrokegenerator.h"

#include "toonz/ttileset.h"

#include "toonz/txshsimplelevel.h"

#include "toonz/toonzimageutils.h"

#include "toonz/palettecontroller.h"

#include "toonz/stage2.h"

#include "toonz/preferences.h"

#include "toonz/tpalettehandle.h"

#include "toonz/mypaintbrushstyle.h"

// TnzCore includes

#include "tstream.h"

#include "tcolorstyles.h"

#include "tvectorimage.h"

#include "tenv.h"

#include "tregion.h"

#include "tinbetween.h"

#include "tgl.h"

#include "trop.h"

// Qt includes

#include <qpainter></qpainter>

using namespace ToolUtils;

TEnv::DoubleVar RasterBrushMinSize("InknpaintRasterBrushMinSize", 1);

TEnv::DoubleVar RasterBrushMaxSize("InknpaintRasterBrushMaxSize", 5);

TEnv::DoubleVar BrushSmooth("InknpaintBrushSmooth", 0);

TEnv::IntVar BrushDrawOrder("InknpaintBrushDrawOrder", 0);

TEnv::IntVar RasterBrushPencilMode("InknpaintRasterBrushPencilMode", 0);

TEnv::IntVar BrushPressureSensitivity("InknpaintBrushPressureSensitivity", 1);

TEnv::DoubleVar RasterBrushHardness("RasterBrushHardness", 100);

TEnv::DoubleVar RasterBrushModifierSize("RasterBrushModifierSize", 0);

TEnv::StringVar RasterBrushPreset("RasterBrushPreset", "<custom>");</custom>

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

#define CUSTOM_WSTR L"<custom>"</custom>

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

//

// (Da mettere in libreria) : funzioni che spezzano una stroke

// nei suoi punti angolosi. Lo facciamo specialmente per limitare

// i problemi di fill.

//

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

//

// Split a stroke in n+1 parts, according to n parameter values

// Input:

//      stroke            = stroke to split

//      parameterValues[] = vector of parameters where I want to split the

//      stroke

//                          assert: 0

// Output:

//      strokes[]         = the split strokes

//

// note: stroke is unchanged

//

static void split(TStroke *stroke, const std::vector<double> ¶meterValues,</double>

                  std::vector<tstroke *=""> &strokes) {</tstroke>

  TThickPoint p2;

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

  TThickPoint lastPoint = stroke->getControlPoint(0);

  int n                 = parameterValues.size();

  int chunk;

  double t;

  int last_chunk = -1, startPoint = 0;

  double lastLocT = 0;

  for (int i = 0; i < n; i++) {

    points.push_back(lastPoint);  // Add first point of the stroke

    double w =

        parameterValues[i];  // Global parameter. along the stroke 0<=w<=1

    stroke->getChunkAndT(w, chunk,

                         t);  // t: local parameter in the chunk-th quadratic

    if (i == 0)

      startPoint = 1;

    else {

      int indexAfterLastT =

          stroke->getControlPointIndexAfterParameter(parameterValues[i - 1]);

      startPoint = indexAfterLastT;

      if ((indexAfterLastT & 1) && lastLocT != 1) startPoint++;

    }

    int endPoint = 2 * chunk + 1;

    if (lastLocT != 1 && i > 0) {

      if (last_chunk != chunk || t == 1)

        points.push_back(p2);  // If the last local t is not an extreme

                               // add the point p2

    }

    for (int j = startPoint; j < endPoint; j++)

      points.push_back(stroke->getControlPoint(j));

    TThickPoint p, A, B, C;

    p       = stroke->getPoint(w);

    C       = stroke->getControlPoint(2 * chunk + 2);

    B       = stroke->getControlPoint(2 * chunk + 1);

    A       = stroke->getControlPoint(2 * chunk);

    p.thick = A.thick;

    if (last_chunk != chunk) {

      TThickPoint p1 = (1 - t) * A + t * B;

      points.push_back(p1);

      p.thick = p1.thick;

    } else {

      if (t != 1) {

        // If the i-th cut point belong to the same chunk of the (i-1)-th cut

        // point.

        double tInters  = lastLocT / t;

        TThickPoint p11 = (1 - t) * A + t * B;

        TThickPoint p1  = (1 - tInters) * p11 + tInters * p;

        points.push_back(p1);

        p.thick = p1.thick;

      }

    }

    points.push_back(p);

    if (t != 1) p2 = (1 - t) * B + t * C;

    assert(points.size() & 1);

    // Add new stroke

    TStroke *strokeAdd = new TStroke(points);

    strokeAdd->setStyle(stroke->getStyle());

    strokeAdd->outlineOptions() = stroke->outlineOptions();

    strokes.push_back(strokeAdd);

    lastPoint  = p;

    last_chunk = chunk;

    lastLocT   = t;

    points.clear();

  }

  // Add end stroke

  points.push_back(lastPoint);

  if (lastLocT != 1) points.push_back(p2);

  startPoint =

      stroke->getControlPointIndexAfterParameter(parameterValues[n - 1]);

  if ((stroke->getControlPointIndexAfterParameter(parameterValues[n - 1]) &

       1) &&

      lastLocT != 1)

    startPoint++;

  for (int j = startPoint; j < stroke->getControlPointCount(); j++)

    points.push_back(stroke->getControlPoint(j));

  assert(points.size() & 1);

  TStroke *strokeAdd = new TStroke(points);

  strokeAdd->setStyle(stroke->getStyle());

  strokeAdd->outlineOptions() = stroke->outlineOptions();

  strokes.push_back(strokeAdd);

  points.clear();

}

// Compute Parametric Curve Curvature

// By Formula:

// k(t)=(|p'(t) x p''(t)|)/Norm2(p')^3

// p(t) is parametric curve

// Input:

//      dp  = First Derivate.

//      ddp = Second Derivate

// Output:

//      return curvature value.

//      Note: if the curve is a single point (that's dp=0) or it is a straight

//      line (that's ddp=0) return 0

static double curvature(TPointD dp, TPointD ddp) {

  if (dp == TPointD(0, 0))

    return 0;

  else

    return fabs(cross(dp, ddp) / pow(norm2(dp), 1.5));

}

// Find the max curvature points of a stroke.

// Input:

//      stroke.

//      angoloLim =  Value (radians) of the Corner between two tangent vector.

//                   Up this value the two corner can be considered angular.

//      curvMaxLim = Value of the max curvature.

//                   Up this value the point can be considered a max curvature

//                   point.

// Output:

//      parameterValues = vector of max curvature parameter points

static void findMaxCurvPoints(TStroke *stroke, const float &angoloLim,

                              const float &curvMaxLim,

                              std::vector<double> ¶meterValues) {</double>

  TPointD tg1, tg2;  // Tangent vectors

  TPointD dp, ddp;  // First and Second derivate.

  parameterValues.clear();

  int cpn = stroke ? stroke->getControlPointCount() : 0;

  for (int j = 2; j < cpn; j += 2) {

    TPointD p0 = stroke->getControlPoint(j - 2);

    TPointD p1 = stroke->getControlPoint(j - 1);

    TPointD p2 = stroke->getControlPoint(j);

    TPointD q = p1 - (p0 + p2) * 0.5;

    // Search corner point

    if (j > 2) {

      tg2 = -p0 + p2 + 2 * q;  // Tangent vector to this chunk at t=0

      double prod_scal =

          tg2 * tg1;  // Inner product between tangent vectors at t=0.

      assert(tg1 != TPointD(0, 0) || tg2 != TPointD(0, 0));

      // Compute corner between two tangent vectors

      double angolo =

          acos(prod_scal / (pow(norm2(tg2), 0.5) * pow(norm2(tg1), 0.5)));

      // Add corner point

      if (angolo > angoloLim) {

        double w = getWfromChunkAndT(stroke, (UINT)(0.5 * (j - 2)),

                                     0);  //  transform lacal t to global t

        parameterValues.push_back(w);

      }

    }

    tg1 = -p0 + p2 - 2 * q;  // Tangent vector to this chunk at t=1

    // End search corner point

    // Search max curvature point

    // Value of t where the curvature function has got an extreme.

    // (Point where first derivate is null)

    double estremo_int = 0;

    double t           = -1;

    if (q != TPointD(0, 0)) {

      t = 0.25 *

          (2 * q.x * q.x + 2 * q.y * q.y - q.x * p0.x + q.x * p2.x -

           q.y * p0.y + q.y * p2.y) /

          (q.x * q.x + q.y * q.y);

      dp  = -p0 + p2 + 2 * q - 4 * t * q;  // First derivate of the curve

      ddp = -4 * q;                        // Second derivate of the curve

      estremo_int = curvature(dp, ddp);

      double h    = 0.01;

      dp          = -p0 + p2 + 2 * q - 4 * (t + h) * q;

      double c_dx = curvature(dp, ddp);

      dp          = -p0 + p2 + 2 * q - 4 * (t - h) * q;

      double c_sx = curvature(dp, ddp);

      // Check the point is a max and not a minimum

      if (estremo_int < c_dx && estremo_int < c_sx) {

        estremo_int = 0;

      }

    }

    double curv_max = estremo_int;

    // Compute curvature at the extreme of interval [0,1]

    // Compute curvature at t=0 (Left extreme)

    dp                = -p0 + p2 + 2 * q;

    double estremo_sx = curvature(dp, ddp);

    // Compute curvature at t=1 (Right extreme)

    dp                = -p0 + p2 - 2 * q;

    double estremo_dx = curvature(dp, ddp);

    // Compare curvature at the extreme of interval [0,1] with the internal

    // value

    double t_ext;

    if (estremo_sx >= estremo_dx)

      t_ext = 0;

    else

      t_ext = 1;

    double maxEstremi = std::max(estremo_dx, estremo_sx);

    if (maxEstremi > estremo_int) {

      t        = t_ext;

      curv_max = maxEstremi;

    }

    // Add max curvature point

    if (t >= 0 && t <= 1 && curv_max > curvMaxLim) {

      double w = getWfromChunkAndT(stroke, (UINT)(0.5 * (j - 2)),

                                   t);  // transform local t to global t

      parameterValues.push_back(w);

    }

    // End search max curvature point

  }

  // Delete duplicate of parameterValues

  // Because some max cuvature point can coincide with the corner point

  if ((int)parameterValues.size() > 1) {

    std::sort(parameterValues.begin(), parameterValues.end());

    parameterValues.erase(

        std::unique(parameterValues.begin(), parameterValues.end()),

        parameterValues.end());

  }

}

static void addStroke(TTool::Application *application, const TVectorImageP &vi,

                      TStroke *stroke, bool breakAngles, bool frameCreated,

                      bool levelCreated, TXshSimpleLevel *sLevel = NULL,

                      TFrameId fid = TFrameId::NO_FRAME) {

  QMutexLocker lock(vi->getMutex());

  if (application->getCurrentObject()->isSpline()) {

    application->getCurrentXsheet()->notifyXsheetChanged();

    return;

  }

  std::vector<double> corners;</double>

  std::vector<tstroke *=""> strokes;</tstroke>

  const float angoloLim =

      1;  // Value (radians) of the Corner between two tangent vector.

          // Up this value the two corner can be considered angular.

  const float curvMaxLim = 0.8;  // Value of the max curvature.

  // Up this value the point can be considered a max curvature point.

  findMaxCurvPoints(stroke, angoloLim, curvMaxLim, corners);

  TXshSimpleLevel *sl;

  if (!sLevel) {

    sl = application->getCurrentLevel()->getSimpleLevel();

  } else {

    sl = sLevel;

  }

  TFrameId id = application->getCurrentTool()->getTool()->getCurrentFid();

  if (id == TFrameId::NO_FRAME && fid != TFrameId::NO_FRAME) id = fid;

  if (!corners.empty()) {

    if (breakAngles)

      split(stroke, corners, strokes);

    else

      strokes.push_back(new TStroke(*stroke));

    int n = strokes.size();

    TUndoManager::manager()->beginBlock();

    for (int i = 0; i < n; i++) {

      std::vector<tfilledregioninf> *fillInformation =</tfilledregioninf>

          new std::vector<tfilledregioninf>;</tfilledregioninf>

      ImageUtils::getFillingInformationOverlappingArea(vi, *fillInformation,

                                                       stroke->getBBox());

      TStroke *str = new TStroke(*strokes[i]);

      vi->addStroke(str);

      TUndoManager::manager()->add(new UndoPencil(str, fillInformation, sl, id,

                                                  frameCreated, levelCreated));

    }

    TUndoManager::manager()->endBlock();

  } else {

    std::vector<tfilledregioninf> *fillInformation =</tfilledregioninf>

        new std::vector<tfilledregioninf>;</tfilledregioninf>

    ImageUtils::getFillingInformationOverlappingArea(vi, *fillInformation,

                                                     stroke->getBBox());

    TStroke *str = new TStroke(*stroke);

    vi->addStroke(str);

    TUndoManager::manager()->add(new UndoPencil(str, fillInformation, sl, id,

                                                frameCreated, levelCreated));

  }

  // Update regions. It will call roundStroke() in

  // TVectorImage::Imp::findIntersections().

  // roundStroke() will slightly modify all the stroke positions.

  // It is needed to update information for Fill Check.

  vi->findRegions();

  for (int k = 0; k < (int)strokes.size(); k++) delete strokes[k];

  strokes.clear();

  application->getCurrentTool()->getTool()->notifyImageChanged();

}

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

//

// Gennaro: end

//

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

//===================================================================

//

// Helper functions and classes

//

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

namespace {

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

void addStrokeToImage(TTool::Application *application, const TVectorImageP &vi,

                      TStroke *stroke, bool breakAngles, bool frameCreated,

                      bool levelCreated, TXshSimpleLevel *sLevel = NULL,

                      TFrameId id = TFrameId::NO_FRAME) {

  QMutexLocker lock(vi->getMutex());

  addStroke(application, vi.getPointer(), stroke, breakAngles, frameCreated,

            levelCreated, sLevel, id);

  // la notifica viene gia fatta da addStroke!

  // getApplication()->getCurrentTool()->getTool()->notifyImageChanged();

}

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

enum DrawOrder { OverAll = 0, UnderAll, PaletteOrder };

void getAboveStyleIdSet(int styleId, TPaletteP palette,

                        QSet<int> &aboveStyles) {</int>

  if (!palette) return;

  for (int p = 0; p < palette->getPageCount(); p++) {

    TPalette::Page *page = palette->getPage(p);

    for (int s = 0; s < page->getStyleCount(); s++) {

      int tmpId = page->getStyleId(s);

      if (tmpId == styleId) return;

      if (tmpId != 0) aboveStyles.insert(tmpId);

    }

  }

}

//=========================================================================================================

class RasterBrushUndo final : public TRasterUndo {

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

  int m_styleId;

  bool m_selective;

  bool m_isPaletteOrder;

  bool m_isPencil;

public:

  RasterBrushUndo(TTileSetCM32 *tileSet, const std::vector<tthickpoint> &points,</tthickpoint>

                  int styleId, bool selective, TXshSimpleLevel *level,

                  const TFrameId &frameId, bool isPencil, bool isFrameCreated,

                  bool isLevelCreated, bool isPaletteOrder)

      : TRasterUndo(tileSet, level, frameId, isFrameCreated, isLevelCreated, 0)

      , m_points(points)

      , m_styleId(styleId)

      , m_selective(selective)

      , m_isPencil(isPencil)

      , m_isPaletteOrder(isPaletteOrder) {}

  void redo() const override {

    insertLevelAndFrameIfNeeded();

    TToonzImageP image = getImage();

    TRasterCM32P ras   = image->getRaster();

    RasterStrokeGenerator m_rasterTrack(ras, BRUSH, NONE, m_styleId,

                                        m_points[0], m_selective, 0,

                                        !m_isPencil, m_isPaletteOrder);

    if (m_isPaletteOrder) {

      QSet<int> aboveStyleIds;</int>

      getAboveStyleIdSet(m_styleId, image->getPalette(), aboveStyleIds);

      m_rasterTrack.setAboveStyleIds(aboveStyleIds);

    }

    m_rasterTrack.setPointsSequence(m_points);

    m_rasterTrack.generateStroke(m_isPencil);

    image->setSavebox(image->getSavebox() +

                      m_rasterTrack.getBBox(m_rasterTrack.getPointsSequence()));

    ToolUtils::updateSaveBox();

    TTool::getApplication()->getCurrentXsheet()->notifyXsheetChanged();

    notifyImageChanged();

  }

  int getSize() const override {

    return sizeof(*this) + TRasterUndo::getSize();

  }

  QString getToolName() override { return QString("Brush Tool"); }

  int getHistoryType() override { return HistoryType::BrushTool; }

};

//=========================================================================================================

class RasterBluredBrushUndo final : public TRasterUndo {

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

  int m_styleId;

  DrawOrder m_drawOrder;

  int m_maxThick;

  double m_hardness;

public:

  RasterBluredBrushUndo(TTileSetCM32 *tileSet,

                        const std::vector<tthickpoint> &points, int styleId,</tthickpoint>

                        DrawOrder drawOrder, TXshSimpleLevel *level,

                        const TFrameId &frameId, int maxThick, double hardness,

                        bool isFrameCreated, bool isLevelCreated)

      : TRasterUndo(tileSet, level, frameId, isFrameCreated, isLevelCreated, 0)

      , m_points(points)

      , m_styleId(styleId)

      , m_drawOrder(drawOrder)

      , m_maxThick(maxThick)

      , m_hardness(hardness) {}

  void redo() const override {

    if (m_points.size() == 0) return;

    insertLevelAndFrameIfNeeded();

    TToonzImageP image     = getImage();

    TRasterCM32P ras       = image->getRaster();

    TRasterCM32P backupRas = ras->clone();

    TRaster32P workRaster(ras->getSize());

    QRadialGradient brushPad = ToolUtils::getBrushPad(m_maxThick, m_hardness);

    workRaster->clear();

    BluredBrush brush(workRaster, m_maxThick, brushPad, false);

    if (m_drawOrder == PaletteOrder) {

      QSet<int> aboveStyleIds;</int>

      getAboveStyleIdSet(m_styleId, image->getPalette(), aboveStyleIds);

      brush.setAboveStyleIds(aboveStyleIds);

    }

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

    points.push_back(m_points[0]);

    TRect bbox = brush.getBoundFromPoints(points);

    brush.addPoint(m_points[0], 1);

    brush.updateDrawing(ras, ras, bbox, m_styleId, (int)m_drawOrder);

    if (m_points.size() > 1) {

      points.clear();

      points.push_back(m_points[0]);

      points.push_back(m_points[1]);

      bbox = brush.getBoundFromPoints(points);

      brush.addArc(m_points[0], (m_points[1] + m_points[0]) * 0.5, m_points[1],

                   1, 1);

      brush.updateDrawing(ras, backupRas, bbox, m_styleId, (int)m_drawOrder);

      int i;

      for (i = 1; i + 2 < (int)m_points.size(); i = i + 2) {

        points.clear();

        points.push_back(m_points[i]);

        points.push_back(m_points[i + 1]);

        points.push_back(m_points[i + 2]);

        bbox = brush.getBoundFromPoints(points);

        brush.addArc(m_points[i], m_points[i + 1], m_points[i + 2], 1, 1);

        brush.updateDrawing(ras, backupRas, bbox, m_styleId, (int)m_drawOrder);

      }

    }

    ToolUtils::updateSaveBox();

    TTool::getApplication()->getCurrentXsheet()->notifyXsheetChanged();

    notifyImageChanged();

  }

  int getSize() const override {

    return sizeof(*this) + TRasterUndo::getSize();

  }

  QString getToolName() override { return QString("Brush Tool"); }

  int getHistoryType() override { return HistoryType::BrushTool; }

};

//=========================================================================================================

class MyPaintBrushUndo final : public TRasterUndo {

  TPoint m_offset;

  QString m_id;

public:

  MyPaintBrushUndo(TTileSetCM32 *tileSet, TXshSimpleLevel *level,

                   const TFrameId &frameId, bool isFrameCreated,

                   bool isLevelCreated, const TRasterCM32P &ras,

                   const TPoint &offset)

      : TRasterUndo(tileSet, level, frameId, isFrameCreated, isLevelCreated, 0)

      , m_offset(offset) {

    static int counter = 0;

    m_id = QString("MyPaintBrushUndo") + QString::number(counter++);

    TImageCache::instance()->add(m_id.toStdString(),

                                 TToonzImageP(ras, TRect(ras->getSize())));

  }

  ~MyPaintBrushUndo() { TImageCache::instance()->remove(m_id); }

  void redo() const override {

    insertLevelAndFrameIfNeeded();

    TToonzImageP image = getImage();

    TRasterCM32P ras   = image->getRaster();

    TImageP srcImg =

        TImageCache::instance()->get(m_id.toStdString(), false)->cloneImage();

    TToonzImageP tSrcImg = srcImg;

    assert(tSrcImg);

    ras->copy(tSrcImg->getRaster(), m_offset);

    ToolUtils::updateSaveBox();

    TTool::getApplication()->getCurrentXsheet()->notifyXsheetChanged();

    notifyImageChanged();

  }

  int getSize() const override {

    return sizeof(*this) + TRasterUndo::getSize();

  }

  QString getToolName() override { return QString("Brush Tool"); }

  int getHistoryType() override { return HistoryType::BrushTool; }

};

//=========================================================================================================

double computeThickness(double pressure, const TDoublePairProperty &property) {

  double t      = pressure * pressure * pressure;

  double thick0 = property.getValue().first;

  double thick1 = property.getValue().second;

  if (thick1 < 0.0001) thick0 = thick1 = 0.0;

  return (thick0 + (thick1 - thick0) * t) * 0.5;

}

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

int computeThickness(double pressure, const TIntPairProperty &property) {

  double t   = pressure * pressure * pressure;

  int thick0 = property.getValue().first;

  int thick1 = property.getValue().second;

  return tround(thick0 + (thick1 - thick0) * t);

}

}  // namespace

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

static void CatmullRomInterpolate(const TThickPoint &P0, const TThickPoint &P1,

                                  const TThickPoint &P2, const TThickPoint &P3,

                                  int samples,

                                  std::vector<tthickpoint> &points) {</tthickpoint>

  double x0 = P1.x;

  double x1 = (-P0.x + P2.x) * 0.5f;

  double x2 = P0.x - 2.5f * P1.x + 2.0f * P2.x - 0.5f * P3.x;

  double x3 = -0.5f * P0.x + 1.5f * P1.x - 1.5f * P2.x + 0.5f * P3.x;

  double y0 = P1.y;

  double y1 = (-P0.y + P2.y) * 0.5f;

  double y2 = P0.y - 2.5f * P1.y + 2.0f * P2.y - 0.5f * P3.y;

  double y3 = -0.5f * P0.y + 1.5f * P1.y - 1.5f * P2.y + 0.5f * P3.y;

  double z0 = P1.thick;

  double z1 = (-P0.thick + P2.thick) * 0.5f;

  double z2 = P0.thick - 2.5f * P1.thick + 2.0f * P2.thick - 0.5f * P3.thick;

  double z3 =

      -0.5f * P0.thick + 1.5f * P1.thick - 1.5f * P2.thick + 0.5f * P3.thick;

  for (int i = 1; i <= samples; ++i) {

    double t  = i / (double)(samples + 1);

    double t2 = t * t;

    double t3 = t2 * t;

    TThickPoint p;

    p.x     = x0 + x1 * t + x2 * t2 + x3 * t3;

    p.y     = y0 + y1 * t + y2 * t2 + y3 * t3;

    p.thick = z0 + z1 * t + z2 * t2 + z3 * t3;

    points.push_back(p);

  }

}

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

static void Smooth(std::vector<tthickpoint> &points, int radius) {</tthickpoint>

  int n = (int)points.size();

  if (radius < 1 || n < 3) {

    return;

  }

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

  float d = 1.0f / (radius * 2 + 1);

  for (int i = 1; i < n - 1; ++i) {

    int lower = i - radius;

    int upper = i + radius;

    TThickPoint total;

    total.x     = 0;

    total.y     = 0;

    total.thick = 0;

    for (int j = lower; j <= upper; ++j) {

      int idx = j;

      if (idx < 0) {

        idx = 0;

      } else if (idx >= n) {

        idx = n - 1;

      }

      total.x += points[idx].x;

      total.y += points[idx].y;

      total.thick += points[idx].thick;

    }

    total.x *= d;

    total.y *= d;

    total.thick *= d;

    result.push_back(total);

  }

  for (int i = 1; i < n - 1; ++i) {

    points[i].x     = result[i - 1].x;

    points[i].y     = result[i - 1].y;

    points[i].thick = result[i - 1].thick;

  }

  if (points.size() >= 3) {

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

    CatmullRomInterpolate(points[0], points[0], points[1], points[2], 10, pts);

    std::vector<tthickpoint>::iterator it = points.begin() + 1;</tthickpoint>

    points.insert(it, pts.begin(), pts.end());

    pts.clear();

    CatmullRomInterpolate(points[n - 3], points[n - 2], points[n - 1],

                          points[n - 1], 10, pts);

    it = points.begin();

    it += n - 1;

    points.insert(it, pts.begin(), pts.end());

  }

}

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

void SmoothStroke::beginStroke(int smooth) {

  m_smooth      = smooth;

  m_outputIndex = 0;

  m_readIndex   = -1;

  m_rawPoints.clear();

  m_outputPoints.clear();

}

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

void SmoothStroke::addPoint(const TThickPoint &point) {

  if (m_rawPoints.size() > 0 && m_rawPoints.back().x == point.x &&

      m_rawPoints.back().y == point.y) {

    return;

  }

  m_rawPoints.push_back(point);

  generatePoints();

}

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

void SmoothStroke::endStroke() {

  generatePoints();

  // force enable the output all segments

  m_outputIndex = m_outputPoints.size() - 1;

}

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

void SmoothStroke::clearPoints() {

  m_outputIndex = 0;

  m_readIndex   = -1;

  m_outputPoints.clear();

  m_rawPoints.clear();

}

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

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

  int n = m_outputPoints.size();

  for (int i = m_readIndex + 1; i <= m_outputIndex && i < n; ++i) {

    smoothPoints.push_back(m_outputPoints[i]);

  }

  m_readIndex = m_outputIndex;

}

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

void SmoothStroke::generatePoints() {

  int n = (int)m_rawPoints.size();

  if (n == 0) {

    return;

  }

  // if m_smooth = 0, then skip whole smoothing process

  if (m_smooth == 0) {

    for (int i = m_outputIndex; i < (int)m_outputPoints.size(); ++i) {

      if (m_outputPoints[i] != m_rawPoints[i]) {

        break;

      }

      ++m_outputIndex;

    }

    m_outputPoints = m_rawPoints;

    return;

  }

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

  // Add more stroke samples before applying the smoothing

  // This is because the raw inputs points are too few to support smooth result,

  // especially on stroke ends

  smoothedPoints.push_back(m_rawPoints.front());

  for (int i = 1; i < n; ++i) {

    const TThickPoint &p1 = m_rawPoints[i - 1];

    const TThickPoint &p2 = m_rawPoints[i];

    const TThickPoint &p0 = i - 2 >= 0 ? m_rawPoints[i - 2] : p1;

    const TThickPoint &p3 = i + 1 < n ? m_rawPoints[i + 1] : p2;

    int samples = 8;

    CatmullRomInterpolate(p0, p1, p2, p3, samples, smoothedPoints);

    smoothedPoints.push_back(p2);

  }

  // Apply the 1D box filter

  // Multiple passes result in better quality and fix the stroke ends break

  // issue

  for (int i = 0; i < 3; ++i) {

    Smooth(smoothedPoints, m_smooth);

  }

  // Compare the new smoothed stroke with old one

  // Enable the output for unchanged parts

  int outputNum = (int)m_outputPoints.size();

  for (int i = m_outputIndex; i < outputNum; ++i) {

    if (m_outputPoints[i] != smoothedPoints[i]) {

      break;

    }

    ++m_outputIndex;

  }

  m_outputPoints = smoothedPoints;

}

//===================================================================

//

// ToonzRasterBrushTool

//

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

ToonzRasterBrushTool::ToonzRasterBrushTool(std::string name, int targetType)

    : TTool(name)

    , m_rasThickness("Size", 1, 1000, 1, 5)

    , m_smooth("Smooth:", 0, 50, 0)

    , m_hardness("Hardness:", 0, 100, 100)

    , m_preset("Preset:")

    , m_drawOrder("Draw Order:")

    , m_pencil("Pencil", false)

    , m_pressure("Pressure", true)

    , m_modifierSize("ModifierSize", -3, 3, 0, true)

    , m_rasterTrack(0)

    , m_styleId(0)

    , m_bluredBrush(0)

    , m_active(false)

    , m_enabled(false)

    , m_isPrompting(false)

    , m_firstTime(true)

    , m_presetsLoaded(false)

    , m_targetType(targetType)

    , m_workingFrameId(TFrameId())

    , m_notifier(0) {

  bind(targetType);

  m_rasThickness.setNonLinearSlider();

  m_prop[0].bind(m_rasThickness);

  m_prop[0].bind(m_hardness);

  m_prop[0].bind(m_smooth);

  m_prop[0].bind(m_drawOrder);

  m_prop[0].bind(m_modifierSize);

  m_prop[0].bind(m_pencil);

  m_pencil.setId("PencilMode");

  m_drawOrder.addValue(L"Over All");

  m_drawOrder.addValue(L"Under All");

  m_drawOrder.addValue(L"Palette Order");

  m_drawOrder.setId("DrawOrder");

  m_prop[0].bind(m_pressure);

  m_prop[0].bind(m_preset);

  m_preset.setId("BrushPreset");

  m_preset.addValue(CUSTOM_WSTR);

  m_pressure.setId("PressureSensitivity");

}

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

ToolOptionsBox *ToonzRasterBrushTool::createOptionsBox() {

  TPaletteHandle *currPalette =

      TTool::getApplication()->getPaletteController()->getCurrentLevelPalette();

  ToolHandle *currTool = TTool::getApplication()->getCurrentTool();

  return new BrushToolOptionsBox(0, this, currPalette, currTool);

}

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