#include "toonzvectorbrushtool.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/tonionskinmaskhandle.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 V_VectorBrushMinSize("InknpaintVectorBrushMinSize", 1);

TEnv::DoubleVar V_VectorBrushMaxSize("InknpaintVectorBrushMaxSize", 5);

TEnv::IntVar V_VectorCapStyle("InknpaintVectorCapStyle", 1);

TEnv::IntVar V_VectorJoinStyle("InknpaintVectorJoinStyle", 1);

TEnv::IntVar V_VectorMiterValue("InknpaintVectorMiterValue", 4);

TEnv::DoubleVar V_BrushAccuracy("InknpaintBrushAccuracy", 20);

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

TEnv::IntVar V_BrushBreakSharpAngles("InknpaintBrushBreakSharpAngles", 0);

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

TEnv::IntVar V_VectorBrushFrameRange("VectorBrushFrameRange", 0);

TEnv::IntVar V_VectorBrushSnap("VectorBrushSnap", 0);

TEnv::IntVar V_VectorBrushSnapSensitivity("VectorBrushSnapSensitivity", 0);

TEnv::IntVar V_VectorBrushAssistants("VectorBrushAssistants", 1);

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

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

#define ROUNDC_WSTR L"round_cap"

#define BUTT_WSTR L"butt_cap"

#define PROJECTING_WSTR L"projecting_cap"

#define ROUNDJ_WSTR L"round_join"

#define BEVEL_WSTR L"bevel_join"

#define MITER_WSTR L"miter_join"

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

#define LINEAR_WSTR L"Linear"

#define EASEIN_WSTR L"In"

#define EASEOUT_WSTR L"Out"

#define EASEINOUT_WSTR L"In&Out"

#define LOW_WSTR L"Low"

#define MEDIUM_WSTR L"Med"

#define HIGH_WSTR L"High"

const double SNAPPING_LOW    = 5.0;

const double SNAPPING_MEDIUM = 25.0;

const double SNAPPING_HIGH   = 100.0;

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

//

// (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 autoGroup,

                      bool autoFill, 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,

                                                  autoGroup, autoFill));

    }

    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,

                                                autoGroup, autoFill));

  }

  if (autoGroup && stroke->isSelfLoop()) {

    int index = vi->getStrokeCount() - 1;

    vi->group(index, 1);

    if (autoFill) {

      // to avoid filling other strokes, I enter into the new stroke group

      int currentGroup = vi->exitGroup();

      vi->enterGroup(index);

      vi->selectFill(stroke->getBBox().enlarge(1, 1), 0, stroke->getStyle(),

                     false, true, false);

      if (currentGroup != -1)

        vi->enterGroup(currentGroup);

      else

        vi->exitGroup();

    }

  }

  // 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 autoGroup,

                      bool autoFill, bool frameCreated, bool levelCreated,

                      TXshSimpleLevel *sLevel = NULL,

                      TFrameId id             = TFrameId::NO_FRAME) {

  QMutexLocker lock(vi->getMutex());

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

            autoFill, 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);

    }

  }

}

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

double computeThickness(double pressure, const TDoublePairProperty &property,

                        bool enablePressure, bool isPath ) {

  if (isPath) return 0.0;

  if (!enablePressure) return property.getValue().second*0.5;

  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;

}

}  // namespace

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

//

// ToonzVectorBrushTool

//

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

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

    : TTool(name)

    , m_thickness("Size", 0, 1000, 0, 5)

    , m_accuracy("Accuracy:", 1, 100, 20)

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

    , m_preset("Preset:")

    , m_breakAngles("Break", true)

    , m_pressure("Pressure", true)

    , m_snap("Snap", false)

    , m_frameRange("Range:")

    , m_snapSensitivity("Sensitivity:")

    , m_capStyle("Cap")

    , m_joinStyle("Join")

    , m_miterJoinLimit("Miter:", 0, 100, 4)

    , m_assistants("Assistants", true)

    , m_firstStroke()

    , m_styleId()

    , m_minThick()

    , m_maxThick()

    , m_col()

    , m_firstFrame()

    , m_veryFirstFrame()

    , m_veryFirstCol()

    , m_targetType(targetType)

    , m_pixelSize()

    , m_minDistance2()

    , m_snapped()

    , m_snappedSelf()

    , m_active()

    , m_isPrompting()

    , m_firstTime(true)

    , m_isPath()

    , m_presetsLoaded()

    , m_firstFrameRange(true)

    , m_propertyUpdating()

{

  bind(targetType);

  m_thickness.setNonLinearSlider();

  m_prop[0].bind(m_thickness);

  m_prop[0].bind(m_accuracy);

  m_prop[0].bind(m_smooth);

  m_prop[0].bind(m_breakAngles);

  m_prop[0].bind(m_pressure);

  m_prop[0].bind(m_frameRange);

  m_frameRange.addValue(L"Off");

  m_frameRange.addValue(LINEAR_WSTR);

  m_frameRange.addValue(EASEIN_WSTR);

  m_frameRange.addValue(EASEOUT_WSTR);

  m_frameRange.addValue(EASEINOUT_WSTR);

  m_prop[0].bind(m_snap);

  m_prop[0].bind(m_snapSensitivity);

  m_snapSensitivity.addValue(LOW_WSTR);

  m_snapSensitivity.addValue(MEDIUM_WSTR);

  m_snapSensitivity.addValue(HIGH_WSTR);

  m_prop[0].bind(m_assistants);

  m_prop[0].bind(m_preset);

  m_preset.addValue(CUSTOM_WSTR);

  m_prop[1].bind(m_capStyle);

  m_capStyle.addValue(BUTT_WSTR, QString::fromStdWString(BUTT_WSTR));

  m_capStyle.addValue(ROUNDC_WSTR, QString::fromStdWString(ROUNDC_WSTR));

  m_capStyle.addValue(PROJECTING_WSTR,

                      QString::fromStdWString(PROJECTING_WSTR));

  m_prop[1].bind(m_joinStyle);

  m_joinStyle.addValue(MITER_WSTR, QString::fromStdWString(MITER_WSTR));

  m_joinStyle.addValue(ROUNDJ_WSTR, QString::fromStdWString(ROUNDJ_WSTR));

  m_joinStyle.addValue(BEVEL_WSTR, QString::fromStdWString(BEVEL_WSTR));

  m_prop[1].bind(m_miterJoinLimit);

  m_breakAngles.setId("BreakSharpAngles");

  m_frameRange.setId("FrameRange");

  m_snap.setId("Snap");

  m_snapSensitivity.setId("SnapSensitivity");

  m_preset.setId("BrushPreset");

  m_pressure.setId("PressureSensitivity");

  m_capStyle.setId("Cap");

  m_joinStyle.setId("Join");

  m_miterJoinLimit.setId("Miter");

  m_assistants.setId("Assistants");

  m_inputmanager.setHandler(this);

  m_modifierLine               = new TModifierLine();

  m_modifierTangents           = new TModifierTangents();

  m_modifierAssistants         = new TModifierAssistants();

  m_modifierSegmentation       = new TModifierSegmentation();

  m_modifierSmoothSegmentation = new TModifierSegmentation(TPointD(1, 1), 3);

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

    m_modifierSmooth[i]        = new TModifierSmooth();

  m_modifierSimplify           = new TModifierSimplify();

#ifndef NDEBUG

  m_modifierTest = new TModifierTest();

#endif

  m_inputmanager.addModifier(

      TInputModifierP(m_modifierAssistants.getPointer()));

}

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

ToolOptionsBox *ToonzVectorBrushTool::createOptionsBox() {

  TPaletteHandle *currPalette =

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

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

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

}

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

void ToonzVectorBrushTool::updateTranslation() {

  m_thickness.setQStringName(tr("Size"));

  m_accuracy.setQStringName(tr("Accuracy:"));

  m_smooth.setQStringName(tr("Smooth:"));

  m_preset.setQStringName(tr("Preset:"));

  m_preset.setItemUIName(CUSTOM_WSTR, tr("<custom>"));</custom>

  m_breakAngles.setQStringName(tr("Break"));

  m_pressure.setQStringName(tr("Pressure"));

  m_capStyle.setQStringName(tr("Cap"));

  m_joinStyle.setQStringName(tr("Join"));

  m_miterJoinLimit.setQStringName(tr("Miter:"));

  m_frameRange.setQStringName(tr("Range:"));

  m_snap.setQStringName(tr("Snap"));

  m_snapSensitivity.setQStringName("");

  m_assistants.setQStringName(tr("Assistants"));

  m_frameRange.setItemUIName(L"Off", tr("Off"));

  m_frameRange.setItemUIName(LINEAR_WSTR, tr("Linear"));

  m_frameRange.setItemUIName(EASEIN_WSTR, tr("In"));

  m_frameRange.setItemUIName(EASEOUT_WSTR, tr("Out"));

  m_frameRange.setItemUIName(EASEINOUT_WSTR, tr("In&Out"));

  m_snapSensitivity.setItemUIName(LOW_WSTR, tr("Low"));

  m_snapSensitivity.setItemUIName(MEDIUM_WSTR, tr("Med"));

  m_snapSensitivity.setItemUIName(HIGH_WSTR, tr("High"));

  m_capStyle.setItemUIName(BUTT_WSTR, tr("Butt cap"));

  m_capStyle.setItemUIName(ROUNDC_WSTR, tr("Round cap"));

  m_capStyle.setItemUIName(PROJECTING_WSTR, tr("Projecting cap"));

  m_joinStyle.setItemUIName(MITER_WSTR, tr("Miter join"));

  m_joinStyle.setItemUIName(ROUNDJ_WSTR, tr("Round join"));

  m_joinStyle.setItemUIName(BEVEL_WSTR, tr("Bevel join"));

}

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

void ToonzVectorBrushTool::onActivate() {

  if (m_firstTime) {

    m_firstTime = false;

    std::wstring wpreset =

        QString::fromStdString(V_VectorBrushPreset.getValue()).toStdWString();

    if (wpreset != CUSTOM_WSTR) {

      initPresets();

      if (!m_preset.isValue(wpreset)) wpreset = CUSTOM_WSTR;

      m_preset.setValue(wpreset);

      V_VectorBrushPreset = m_preset.getValueAsString();

      loadPreset();

    } else

      loadLastBrush();

  }

  resetFrameRange();

  // TODO:app->editImageOrSpline();

}

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

void ToonzVectorBrushTool::onDeactivate() {

  /*---

   * ドラッグ中にツールが切り替わった場合に備え、onDeactivateにもMouseReleaseと同じ処理を行う

   * ---*/

  // End current stroke.

  m_inputmanager.finishTracks();

  resetFrameRange();

}

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

void ToonzVectorBrushTool::inputMouseMove(

  const TPointD &position, const TInputState &state )

{

  struct Locals {

    ToonzVectorBrushTool *m_this;

    void setValue(TDoublePairProperty &prop,

                  const TDoublePairProperty::Value &value) {

      prop.setValue(value);

      m_this->onPropertyChanged(prop.getName());

      TTool::getApplication()->getCurrentTool()->notifyToolChanged();

    }

    void addMinMax(TDoublePairProperty &prop, double min, double max) {

      if (min == 0.0 && max == 0.0) return;

      const TDoublePairProperty::Range &range = prop.getRange();

      TDoublePairProperty::Value value = prop.getValue();

      value.first += min;

      value.second += max;

      if (value.first > value.second) value.first = value.second;

      value.first  = tcrop(value.first, range.first, range.second);

      value.second = tcrop(value.second, range.first, range.second);

      setValue(prop, value);

    }

  } locals = {this};

  TPointD halfThick(m_maxThick * 0.5, m_maxThick * 0.5);

  TRectD invalidateRect(m_brushPos - halfThick, m_brushPos + halfThick);

  bool alt     = state.isKeyPressed(TInputState::Key::alt);

  bool shift   = state.isKeyPressed(TInputState::Key::shift);

  bool control = state.isKeyPressed(TInputState::Key::control);

  if ( alt && control && !shift

    && Preferences::instance()->useCtrlAltToResizeBrushEnabled() )

  {

    // Resize the brush if CTRL+ALT is pressed and the preference is enabled.

    const TPointD &diff = position - m_mousePos;

    double max          = diff.x / 2;

    double min          = diff.y / 2;

    locals.addMinMax(m_thickness, min, max);

    double radius = m_thickness.getValue().second * 0.5;

    halfThick = TPointD(radius, radius);

  } else {

    m_brushPos = m_mousePos = position;

  }

  invalidateRect += TRectD(m_brushPos - halfThick, m_brushPos + halfThick);

  if (m_minThick == 0 && m_maxThick == 0) {

    m_minThick = m_thickness.getValue().first;

    m_maxThick = m_thickness.getValue().second;

  }

  invalidate(invalidateRect.enlarge(2));

}

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

void ToonzVectorBrushTool::inputSetBusy(bool busy) {

  if (m_active == busy) return;

  if (busy) {

    // begin painting //////////////////////////

    TTool::Application *app = TTool::getApplication();

    if (!app)

      return;

    m_isPath = app->getCurrentObject()->isSpline();

    if (m_isPath) {

      m_currentColor = TPixel32::Red;

      m_active = true;

      return;

    }

    // todo: gestire autoenable

    if ( app->getCurrentColumn()->getColumnIndex() < 0

      && !app->getCurrentFrame()->isEditingLevel() )

      return;

    if (!getImage(true) || !touchImage())

      return;

    if (!app->getCurrentLevel()->getLevel())

      return;

    // nel caso che il colore corrente sia un cleanup/studiopalette color

    // oppure il colore di un colorfield

    if (TColorStyle *cs = app->getCurrentLevelStyle()) {

      TRasterStyleFx *rfx = cs->getRasterStyleFx();

      if (!cs->isStrokeStyle() && (!rfx || !rfx->isInkStyle()))

        return;

      m_styleId = app->getCurrentLevelStyleIndex();

      m_currentColor = cs->getAverageColor();

      m_currentColor.m = 255;

    } else {

      m_styleId = 1;

      m_currentColor = TPixel32::Black;

    }

    m_active = true;

    return; // painting has begun

  }

  // end painting //////////////////////////

  m_active = false;

  // clear tracks automatically when return from this function

  struct Cleanup {

    ToonzVectorBrushTool &owner;

    inline ~Cleanup() { owner.m_track.clear(); owner.invalidate(); }

  } cleanup = {*this};

  // make motion path (if need)

  if (m_isPath) {

    double error = 20.0 * m_pixelSize;

    TStroke *stroke = m_track.makeStroke(error);

    TVectorImageP vi = getImage(true);

    if (!isJustCreatedSpline(vi.getPointer())) {

      m_isPrompting = true;

      invalidate();

      int ret = DVGui::MsgBox(

        QString("Are you sure you want to replace the motion path?"),

        QObject::tr("Yes"), QObject::tr("No"), 0 );

      m_isPrompting = false;

      if (ret != 1)

        return; // 1 here means "Yes" button

    }

    QMutexLocker lock(vi->getMutex());

    TUndo *undo = new UndoPath(

      getXsheet()->getStageObject(getObjectId())->getSpline() );

    while(vi->getStrokeCount() > 0) vi->deleteStroke(0);

    vi->addStroke(stroke, false);

    notifyImageChanged();

    TUndoManager::manager()->add(undo);

    return; // done with motion path

  }

  // paint regular strokes