// TnzTools includes

#include "tools/toolutils.h"

#include "tools/cursors.h"

#include "tools/tool.h"

#include "tools/toolhandle.h"

#include "bluredbrush.h"

// TnzQt includes

#include "toonzqt/icongenerator.h"

// TnzLib includes

#include "toonz/strokegenerator.h"

#include "toonz/ttilesaver.h"

#include "toonz/txshsimplelevel.h"

#include "toonz/observer.h"

#include "toonz/toonzimageutils.h"

#include "toonz/levelproperties.h"

#include "toonz/stage2.h"

#include "toonz/ttileset.h"

#include "toonz/rasterstrokegenerator.h"

#include "toonz/txsheethandle.h"

#include "toonz/tframehandle.h"

#include "toonz/txshlevelhandle.h"

#include "toonz/tpalettehandle.h"

#include "toonz/tobjecthandle.h"

#include "toonz/tcolumnhandle.h"

#include "toonz/preferences.h"

// TnzBase includes

#include "tenv.h"

// TnzCore includes

#include "tstroke.h"

#include "tmathutil.h"

#include "drawutil.h"

#include "tcolorstyles.h"

#include "tundo.h"

#include "tvectorimage.h"

#include "ttoonzimage.h"

#include "tproperty.h"

#include "tgl.h"

#include "trop.h"

#include "tinbetween.h"

#include "ttile.h"

#include "tropcm.h"

#include "timage_io.h"

// Qt includes

#include <qcoreapplication>  // For Qt translation support</qcoreapplication>

#include <qpainter></qpainter>

using namespace ToolUtils;

#define LINES L"Lines"

#define AREAS L"Areas"

#define ALL L"Lines & Areas"

#define NORMALERASE L"Normal"

#define RECTERASE L"Rectangular"

#define FREEHANDERASE L"Freehand"

#define POLYLINEERASE L"Polyline"

TEnv::DoubleVar EraseSize("InknpaintEraseSize", 10);

TEnv::StringVar EraseType("InknpaintEraseType", "Normal");

TEnv::IntVar EraseSelective("InknpaintEraseSelective", 0);

TEnv::IntVar EraseInvert("InknpaintEraseInvert", 0);

TEnv::IntVar EraseRange("InknpaintEraseRange", 0);

TEnv::StringVar EraseColorType("InknpaintEraseColorType", "Lines");

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

TEnv::IntVar ErasePencil("InknpaintErasePencil", 0);

namespace {

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

//   Undo dei Tools

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

/*!

        Viene realizzata la funzione di "UNDO" per il cancellino nell'opzione

   RECTERASE

*/

class RectRasterUndo final : public TRasterUndo {

  TRectD m_modifyArea;

  TStroke *m_stroke;

  int m_styleId;

  std::wstring m_colorType;

  std::wstring m_eraseType;

  bool m_selective;

  bool m_invert;

  bool m_pencilMode;

public:

  RectRasterUndo(TTileSetCM32 *tileSet, const TRectD &modifyArea,

                 TStroke stroke, int styleId, std::wstring eraseType,

                 std::wstring colorType, TXshSimpleLevel *level, bool selective,

                 bool invert, bool pencil, const TFrameId &frameId)

      : TRasterUndo(tileSet, level, frameId, false, false, 0)

      , m_modifyArea(modifyArea)

      , m_styleId(styleId)

      , m_eraseType(eraseType)

      , m_colorType(colorType)

      , m_selective(selective)

      , m_invert(invert)

      , m_pencilMode(pencil) {

    m_stroke = new TStroke(stroke);

  }

  void redo() const override {

    TToonzImageP ti = getImage();

    if (!ti) return;

    bool eraseInk   = m_colorType == LINES || m_colorType == ALL;

    bool erasePaint = m_colorType == AREAS || m_colorType == ALL;

    if (m_eraseType == RECTERASE) {

      TRect rect = ToonzImageUtils::eraseRect(ti, m_modifyArea, m_styleId,

                                              eraseInk, erasePaint);

      if (!rect.isEmpty()) ToolUtils::updateSaveBox(m_level, m_frameId);

    } else if (m_eraseType == FREEHANDERASE || m_eraseType == POLYLINEERASE) {

      if (m_level) {

        TPoint pos;

        TRaster32P ras = convertStrokeToImage(

            m_stroke, ti->getRaster()->getBounds(), pos, m_pencilMode);

        if (!ras) return;

        ToonzImageUtils::eraseImage(ti, ras, pos, m_invert, eraseInk,

                                    erasePaint, m_selective, m_styleId);

        ToolUtils::updateSaveBox(m_level, m_frameId);

      }

    }

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

    notifyImageChanged();

  }

  int getSize() const override {

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

           m_stroke->getControlPointCount() * sizeof(TThickPoint) + 100;

  }

  ~RectRasterUndo() {

    if (m_stroke) delete m_stroke;

  }

  QString getToolName() override { return QString("Eraser Tool (Rect)"); }

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

};

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

/*!

        Viene realizzata la funzione di "UNDO" per il cancellino nell'opzione

   NORMALERAS

*/

class RasterEraserUndo final : public TRasterUndo {

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

  int m_styleId;

  bool m_selective;

  bool m_isPencil;

  ColorType m_colorType;

  int m_colorSelected;

public:

  RasterEraserUndo(TTileSetCM32 *tileSet,

                   const std::vector<tthickpoint> &points, ColorType colorType,</tthickpoint>

                   int styleId, bool selective, int colorSelected,

                   TXshSimpleLevel *level, const TFrameId &frameId,

                   bool isPencil)

      : TRasterUndo(tileSet, level, frameId, false, false, 0)

      , m_points(points)

      , m_styleId(styleId)

      , m_selective(selective)

      , m_colorType(colorType)

      , m_colorSelected(colorSelected)

      , m_isPencil(isPencil) {}

  void redo() const override {

    TToonzImageP image = m_level->getFrame(m_frameId, true);

    TRasterCM32P ras   = image->getRaster();

    RasterStrokeGenerator m_rasterTrack(ras, ERASE, m_colorType, 0, m_points[0],

                                        m_selective, m_colorSelected, false,

                                        !m_isPencil);

    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("Eraser Tool"); }

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

};

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

class RasterBluredEraserUndo final : public TRasterUndo {

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

  int m_styleId;

  bool m_selective;

  int m_size;

  double m_hardness;

  std::wstring m_mode;

public:

  RasterBluredEraserUndo(TTileSetCM32 *tileSet,

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

                         bool selective, TXshSimpleLevel *level,

                         const TFrameId &frameId, int size, double hardness,

                         const std::wstring &mode)

      : TRasterUndo(tileSet, level, frameId, false, false, 0)

      , m_points(points)

      , m_styleId(styleId)

      , m_selective(selective)

      , m_size(size)

      , m_hardness(hardness)

      , m_mode(mode) {}

  void redo() const override {

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

    TToonzImageP image     = getImage();

    TRasterCM32P ras       = image->getRaster();

    TRasterCM32P backupRas = ras->clone();

    TRaster32P workRaster(ras->getSize());

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

    workRaster->clear();

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

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

    points.push_back(m_points[0]);

    TRect bbox = brush.getBoundFromPoints(points);

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

    brush.eraseDrawing(ras, ras, bbox, m_selective, m_styleId, m_mode);

    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[0] + m_points[1]) * 0.5, m_points[1],

                   1, 1);

      brush.eraseDrawing(ras, backupRas, bbox, m_selective, m_styleId, m_mode);

      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.eraseDrawing(ras, backupRas, bbox, m_selective, m_styleId,

                           m_mode);

      }

    }

    ToolUtils::updateSaveBox();

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

    notifyImageChanged();

  }

  int getSize() const override {

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

  }

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

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

};

void eraseStroke(const TToonzImageP &ti, TStroke *stroke,

                 std::wstring eraseType, std::wstring colorType, bool invert,

                 bool selective, bool pencil, int styleId,

                 const TXshSimpleLevelP &level, const TFrameId &frameId) {

  assert(stroke);

  TPoint pos;

  TRasterCM32P ras = ti->getRaster();

  TRaster32P image =

      convertStrokeToImage(stroke, ras->getBounds(), pos, pencil);

  if (!image) return;

  TRect rasterErasedArea = image->getBounds() + pos;

  TRect area;

  if (!invert)

    area = rasterErasedArea.enlarge(2);

  else

    area                = ras->getBounds();

  TTileSetCM32 *tileSet = new TTileSetCM32(ras->getSize());

  tileSet->add(ras, area);

  TUndoManager::manager()->add(new RectRasterUndo(

      tileSet, convert(area), *stroke, selective ? styleId : -1, eraseType,

      colorType, level.getPointer(), selective, invert, pencil, frameId));

  bool eraseInk   = colorType == LINES || colorType == ALL;

  bool erasePaint = colorType == AREAS || colorType == ALL;

  ToonzImageUtils::eraseImage(ti, image, pos, invert, eraseInk, erasePaint,

                              selective, styleId);

}

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

void drawLine(const TPointD &point, const TPointD ¢re, bool horizontal,

              bool isDecimal) {

  if (!isDecimal) {

    if (horizontal) {

      tglDrawSegment(TPointD(point.x - 1.5, point.y + 0.5) + centre,

                     TPointD(point.x - 0.5, point.y + 0.5) + centre);

      tglDrawSegment(TPointD(point.y - 0.5, -point.x + 1.5) + centre,

                     TPointD(point.y - 0.5, -point.x + 0.5) + centre);

      tglDrawSegment(TPointD(-point.x + 0.5, -point.y + 0.5) + centre,

                     TPointD(-point.x - 0.5, -point.y + 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 0.5, point.x - 0.5) + centre,

                     TPointD(-point.y - 0.5, point.x + 0.5) + centre);

      tglDrawSegment(TPointD(point.y - 0.5, point.x + 0.5) + centre,

                     TPointD(point.y - 0.5, point.x - 0.5) + centre);

      tglDrawSegment(TPointD(point.x - 0.5, -point.y + 0.5) + centre,

                     TPointD(point.x - 1.5, -point.y + 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 0.5, -point.x + 0.5) + centre,

                     TPointD(-point.y - 0.5, -point.x + 1.5) + centre);

      tglDrawSegment(TPointD(-point.x - 0.5, point.y + 0.5) + centre,

                     TPointD(-point.x + 0.5, point.y + 0.5) + centre);

    } else {

      tglDrawSegment(TPointD(point.x - 1.5, point.y + 1.5) + centre,

                     TPointD(point.x - 1.5, point.y + 0.5) + centre);

      tglDrawSegment(TPointD(point.x - 1.5, point.y + 0.5) + centre,

                     TPointD(point.x - 0.5, point.y + 0.5) + centre);

      tglDrawSegment(TPointD(point.y + 0.5, -point.x + 1.5) + centre,

                     TPointD(point.y - 0.5, -point.x + 1.5) + centre);

      tglDrawSegment(TPointD(point.y - 0.5, -point.x + 1.5) + centre,

                     TPointD(point.y - 0.5, -point.x + 0.5) + centre);

      tglDrawSegment(TPointD(-point.x + 0.5, -point.y - 0.5) + centre,

                     TPointD(-point.x + 0.5, -point.y + 0.5) + centre);

      tglDrawSegment(TPointD(-point.x + 0.5, -point.y + 0.5) + centre,

                     TPointD(-point.x - 0.5, -point.y + 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 1.5, point.x - 0.5) + centre,

                     TPointD(-point.y - 0.5, point.x - 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 0.5, point.x - 0.5) + centre,

                     TPointD(-point.y - 0.5, point.x + 0.5) + centre);

      tglDrawSegment(TPointD(point.y + 0.5, point.x - 0.5) + centre,

                     TPointD(point.y - 0.5, point.x - 0.5) + centre);

      tglDrawSegment(TPointD(point.y - 0.5, point.x - 0.5) + centre,

                     TPointD(point.y - 0.5, point.x + 0.5) + centre);

      tglDrawSegment(TPointD(point.x - 1.5, -point.y - 0.5) + centre,

                     TPointD(point.x - 1.5, -point.y + 0.5) + centre);

      tglDrawSegment(TPointD(point.x - 1.5, -point.y + 0.5) + centre,

                     TPointD(point.x - 0.5, -point.y + 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 1.5, -point.x + 1.5) + centre,

                     TPointD(-point.y - 0.5, -point.x + 1.5) + centre);

      tglDrawSegment(TPointD(-point.y - 0.5, -point.x + 1.5) + centre,

                     TPointD(-point.y - 0.5, -point.x + 0.5) + centre);

      tglDrawSegment(TPointD(-point.x + 0.5, point.y + 1.5) + centre,

                     TPointD(-point.x + 0.5, point.y + 0.5) + centre);

      tglDrawSegment(TPointD(-point.x + 0.5, point.y + 0.5) + centre,

                     TPointD(-point.x - 0.5, point.y + 0.5) + centre);

    }

  } else {

    if (horizontal) {

      tglDrawSegment(TPointD(point.x - 0.5, point.y + 0.5) + centre,

                     TPointD(point.x + 0.5, point.y + 0.5) + centre);

      tglDrawSegment(TPointD(point.y + 0.5, point.x - 0.5) + centre,

                     TPointD(point.y + 0.5, point.x + 0.5) + centre);

      tglDrawSegment(TPointD(point.y + 0.5, -point.x + 0.5) + centre,

                     TPointD(point.y + 0.5, -point.x - 0.5) + centre);

      tglDrawSegment(TPointD(point.x + 0.5, -point.y - 0.5) + centre,

                     TPointD(point.x - 0.5, -point.y - 0.5) + centre);

      tglDrawSegment(TPointD(-point.x - 0.5, -point.y - 0.5) + centre,

                     TPointD(-point.x + 0.5, -point.y - 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 0.5, -point.x + 0.5) + centre,

                     TPointD(-point.y - 0.5, -point.x - 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 0.5, point.x - 0.5) + centre,

                     TPointD(-point.y - 0.5, point.x + 0.5) + centre);

      tglDrawSegment(TPointD(-point.x + 0.5, point.y + 0.5) + centre,

                     TPointD(-point.x - 0.5, point.y + 0.5) + centre);

    } else {

      tglDrawSegment(TPointD(point.x - 0.5, point.y + 1.5) + centre,

                     TPointD(point.x - 0.5, point.y + 0.5) + centre);

      tglDrawSegment(TPointD(point.x - 0.5, point.y + 0.5) + centre,

                     TPointD(point.x + 0.5, point.y + 0.5) + centre);

      tglDrawSegment(TPointD(point.y + 1.5, point.x - 0.5) + centre,

                     TPointD(point.y + 0.5, point.x - 0.5) + centre);

      tglDrawSegment(TPointD(point.y + 0.5, point.x - 0.5) + centre,

                     TPointD(point.y + 0.5, point.x + 0.5) + centre);

      tglDrawSegment(TPointD(point.y + 1.5, -point.x + 0.5) + centre,

                     TPointD(point.y + 0.5, -point.x + 0.5) + centre);

      tglDrawSegment(TPointD(point.y + 0.5, -point.x + 0.5) + centre,

                     TPointD(point.y + 0.5, -point.x - 0.5) + centre);

      tglDrawSegment(TPointD(point.x - 0.5, -point.y - 1.5) + centre,

                     TPointD(point.x - 0.5, -point.y - 0.5) + centre);

      tglDrawSegment(TPointD(point.x - 0.5, -point.y - 0.5) + centre,

                     TPointD(point.x + 0.5, -point.y - 0.5) + centre);

      tglDrawSegment(TPointD(-point.x + 0.5, -point.y - 1.5) + centre,

                     TPointD(-point.x + 0.5, -point.y - 0.5) + centre);

      tglDrawSegment(TPointD(-point.x + 0.5, -point.y - 0.5) + centre,

                     TPointD(-point.x - 0.5, -point.y - 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 1.5, -point.x + 0.5) + centre,

                     TPointD(-point.y - 0.5, -point.x + 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 0.5, -point.x + 0.5) + centre,

                     TPointD(-point.y - 0.5, -point.x - 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 1.5, point.x - 0.5) + centre,

                     TPointD(-point.y - 0.5, point.x - 0.5) + centre);

      tglDrawSegment(TPointD(-point.y - 0.5, point.x - 0.5) + centre,

                     TPointD(-point.y - 0.5, point.x + 0.5) + centre);

      tglDrawSegment(TPointD(-point.x + 0.5, point.y + 1.5) + centre,

                     TPointD(-point.x + 0.5, point.y + 0.5) + centre);

      tglDrawSegment(TPointD(-point.x + 0.5, point.y + 0.5) + centre,

                     TPointD(-point.x - 0.5, point.y + 0.5) + centre);

    }

  }

}

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

void drawEmptyCircle(int thick, const TPointD &mousePos, bool isPencil,

                     bool isLxEven, bool isLyEven) {

  TPointD pos = mousePos;

  if (isLxEven) pos.x += 0.5;

  if (isLyEven) pos.y += 0.5;

  if (!isPencil)

    tglDrawCircle(pos, (thick)*0.5);

  else {

    int x = 0, y = tround((thick * 0.5) - 0.5);

    int d           = 3 - 2 * (int)(thick * 0.5);

    bool horizontal = true, isDecimal = thick % 2 != 0;

    drawLine(TPointD(x, y), pos, horizontal, isDecimal);

    while (y > x) {

      if (d < 0) {

        d          = d + 4 * x + 6;

        horizontal = true;

      } else {

        d          = d + 4 * (x - y) + 10;

        horizontal = false;

        y--;

      }

      x++;

      drawLine(TPointD(x, y), pos, horizontal, isDecimal);

    }

  }

}

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

//

//  InkPaintTool class declaration

//

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

class EraserTool final : public TTool {

  Q_DECLARE_TR_FUNCTIONS(EraserTool)

public:

  EraserTool(std::string name);

  ~EraserTool() {

    if (m_firstStroke) delete m_firstStroke;

  }

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

  void updateTranslation() override;

  void draw() override;

  void update(const TToonzImageP &ti, const TPointD &pos);

  void saveUndo();

  void update(const TToonzImageP &ti, TRectD selArea,

              const TXshSimpleLevelP &level, bool multi = false,

              const TFrameId &frameId = -1);

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

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

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

  void leftButtonDoubleClick(const TPointD &pos, const TMouseEvent &e) override;

  void multiAreaEraser(const TXshSimpleLevelP &sl, TFrameId &firstFid,

                       TFrameId &lastFid, TStroke *firstStroke,

                       TStroke *lastStroke);

  void doMultiEraser(const TImageP &img, double t, const TXshSimpleLevelP &sl,

                     const TFrameId &fid, const TVectorImageP &firstImage,

                     const TVectorImageP &lastImage);

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

  void onEnter() override;

  void onLeave() override;

  void onActivate() override;

  bool onPropertyChanged(std::string propertyName) override;

  void onImageChanged() override;

  void multiUpdate(const TXshSimpleLevelP &level, TFrameId firstFrameId,

                   TFrameId lastFrameId, TRectD firstRect, TRectD lastRect);

  TPropertyGroup *getProperties(int targetType) override { return &m_prop; }

  int getCursorId() const override;

  void resetMulti();

  /*-- ドラッグ中にツールが切り替わった場合、処理を終了させる--*/

  void onDeactivate() override;

  /*-- Brush、PaintBrush、EraserToolがPencilModeのときにTrueを返す --*/

  bool isPencilModeActive() override;

  TPointD getCenteredCursorPos(const TPointD &originalCursorPos);

  TPointD fixMousePos(TPointD pos, bool precise = false);

private:

  /*-- 終了処理 --*/

  void storeUndoAndRefresh();

  enum Type { NONE = 0, BRUSH, INKCHANGE, ERASER };

private:

  TPropertyGroup m_prop;

  TEnumProperty m_eraseType;

  TIntProperty m_toolSize;

  TDoubleProperty m_hardness;

  TBoolProperty m_invertOption;

  TBoolProperty m_currentStyle;

  TBoolProperty m_multi;

  TBoolProperty m_pencil;

  TEnumProperty m_colorType;

  Type m_type;

  TXshSimpleLevelP m_level;

  std::pair<int, int=""> m_currCell;</int,>

  TFrameId m_firstFrameId, m_veryFirstFrameId;

  StrokeGenerator m_track;

  TStroke *m_firstStroke;

  TTileSaverCM32 *m_tileSaver;

  TTileSetCM32 *m_tileSet;

  RasterStrokeGenerator *m_normalEraser;

  TRectD m_selectingRect, m_firstRect;

  ColorType m_colorTypeEraser;

  TPointD m_mousePos, m_brushPos, m_firstPos;

  std::vector<tpointd> m_polyline;</tpointd>

  // gestione cancellino blurato

  TRasterCM32P m_backupRas;

  TRaster32P m_workRas;

  QRadialGradient m_brushPad;

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

  BluredBrush *m_bluredBrush;

  double m_pointSize, m_distance2, m_cleanerSize, m_thick;

  bool m_isXsheetCell, m_active, m_enabled, m_selecting, m_firstFrameSelected,

      m_firstTime;

  /*---  消しゴム開始時のFrameIdを保存し、マウスリリース時（Undoの登録時）に

          別のフレームに移動していた場合に備える

  ---*/

  TFrameId m_workingFrameId;

  /*--- マウスボタンDown→プロパティ変更→マウスボタンUpのように、

          ボタンDownからUpまでストレートに来なかった場合は処理を行わないようにする。

  ---*/

  bool m_isLeftButtonPressed;

};

EraserTool inkPaintEraserTool("T_Eraser");

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

//

//  InkPaintTool implementation

//

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

EraserTool::EraserTool(std::string name)

    : TTool(name)

    , m_toolSize("Size:", 1, 1000, 10, false)  // W_ToolOptions_EraserToolSize

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

    , m_eraseType("Type:")                // W_ToolOptions_Erasetype

    , m_colorType("Mode:")                // W_ToolOptions_InkOrPaint

    , m_currentStyle("Selective", false)  // W_ToolOptions_Selective

    , m_invertOption("Invert", false)     // W_ToolOptions_Invert

    , m_multi("Frame Range", false)       // W_ToolOptions_FrameRange

    , m_pencil("Pencil Mode", false)

    , m_currCell(-1, -1)

    , m_tileSaver(0)

    , m_bluredBrush(0)

    , m_pointSize(-1)

    , m_thick(0.5)

    , m_firstFrameSelected(false)

    , m_selecting(false)

    , m_active(false)

    , m_enabled(false)

    , m_isXsheetCell(false)

    , m_firstTime(true)

    , m_workingFrameId(TFrameId())

    , m_isLeftButtonPressed(false) {

  bind(TTool::ToonzImage);

  m_toolSize.setNonLinearSlider();

  m_prop.bind(m_toolSize);

  m_prop.bind(m_hardness);

  m_prop.bind(m_eraseType);

  m_eraseType.addValue(NORMALERASE);

  m_eraseType.addValue(RECTERASE);

  m_eraseType.addValue(FREEHANDERASE);

  m_eraseType.addValue(POLYLINEERASE);

  m_colorType.addValue(LINES);

  m_colorType.addValue(AREAS);

  m_colorType.addValue(ALL);

  m_prop.bind(m_colorType);

  m_prop.bind(m_currentStyle);

  m_prop.bind(m_invertOption);

  m_prop.bind(m_multi);

  m_prop.bind(m_pencil);

  m_currentStyle.setId("Selective");

  m_invertOption.setId("Invert");

  m_multi.setId("FrameRange");

  m_pencil.setId("PencilMode");

  m_colorType.setId("Mode");

  m_eraseType.setId("Type");

}

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

void EraserTool::updateTranslation() {

  m_toolSize.setQStringName(tr("Size:"));

  m_hardness.setQStringName(tr("Hardness:"));

  m_eraseType.setQStringName(tr("Type:"));

  m_eraseType.setItemUIName(NORMALERASE, tr("Normal"));

  m_eraseType.setItemUIName(RECTERASE, tr("Rectangular"));

  m_eraseType.setItemUIName(FREEHANDERASE, tr("Freehand"));

  m_eraseType.setItemUIName(POLYLINEERASE, tr("Polyline"));

  m_colorType.setQStringName(tr("Mode:"));

  m_colorType.setItemUIName(LINES, tr("Lines"));

  m_colorType.setItemUIName(AREAS, tr("Areas"));

  m_colorType.setItemUIName(ALL, tr("Lines & Areas"));

  m_currentStyle.setQStringName(tr("Selective"));

  m_invertOption.setQStringName(tr("Invert"));

  m_multi.setQStringName(tr("Frame Range"));

  m_pencil.setQStringName(tr("Pencil Mode"));

}

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

TPointD EraserTool::getCenteredCursorPos(const TPointD &originalCursorPos) {

  TXshLevelHandle *levelHandle = m_application->getCurrentLevel();

  TXshSimpleLevel *level = levelHandle ? levelHandle->getSimpleLevel() : 0;

  TDimension resolution =

      level ? level->getProperties()->getImageRes() : TDimension(0, 0);

  bool xEven = (resolution.lx % 2 == 0);

  bool yEven = (resolution.ly % 2 == 0);

  TPointD centeredCursorPos = originalCursorPos;

  if (xEven) centeredCursorPos.x -= 0.5;

  if (yEven) centeredCursorPos.y -= 0.5;

  return centeredCursorPos;

}

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

TPointD EraserTool::fixMousePos(TPointD pos, bool precise) {

  TPointD fixedPos = getCenteredCursorPos(pos);

  if (precise) {

    TPointD pp(tround(fixedPos.x), tround(fixedPos.y));

    fixedPos = pp;

  }

  return fixedPos;

}

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

void EraserTool::draw() {

  /*-- MouseLeave時に赤点が描かれるのを防ぐ --*/

  if (m_pointSize == -1 && m_cleanerSize == 0) return;

  double pixelSize2 = getPixelSize() * getPixelSize();

  m_thick           = sqrt(pixelSize2) / 2.0;

  TImageP img = getImage(false);

  if (!img) return;

  if (m_eraseType.getValue() == RECTERASE) {

    TPixel color = TPixel32::Red;

    if (m_multi.getValue() && m_firstFrameSelected)

      drawRect(m_firstRect, color, 0x3F33, true);

    if (m_selecting || (m_multi.getValue() && !m_firstFrameSelected))

      drawRect(m_selectingRect, color, 0xFFFF, true);

  }

  if (m_eraseType.getValue() == NORMALERASE) {

    // If toggled off, don't draw brush outline

    if (!Preferences::instance()->isCursorOutlineEnabled()) return;

    TToonzImageP image(img);

    TRasterP ras = image->getRaster();

    int lx       = ras->getLx();

    int ly       = ras->getLy();

    /*-- InkCheck, PaintCheck, Ink#1CheckがONのときは、BrushTipの描画色を変える

     * --*/

    if ((ToonzCheck::instance()->getChecks() & ToonzCheck::eInk) ||

        (ToonzCheck::instance()->getChecks() & ToonzCheck::ePaint) ||

        (ToonzCheck::instance()->getChecks() & ToonzCheck::eInk1))

      glColor3d(0.5, 0.8, 0.8);

    else

      glColor3d(1.0, 0.0, 0.0);

    drawEmptyCircle(tround(m_cleanerSize), m_brushPos,

                    (m_pencil.getValue() || m_colorType.getValue() == AREAS),

                    lx % 2 == 0, ly % 2 == 0);

  }

  if ((m_eraseType.getValue() == FREEHANDERASE ||

       m_eraseType.getValue() == POLYLINEERASE) &&

      m_multi.getValue()) {

    TPixel color = TPixel32::Red;

    tglColor(color);

    if (m_firstStroke) drawStrokeCenterline(*m_firstStroke, 1);

  }

  if (m_eraseType.getValue() == POLYLINEERASE && !m_polyline.empty()) {

    TPixel color = TPixel32::Red;

    tglColor(color);

    tglDrawCircle(m_polyline[0], 2);

    glBegin(GL_LINE_STRIP);

    for (UINT i = 0; i < m_polyline.size(); i++) tglVertex(m_polyline[i]);

    tglVertex(m_mousePos);

    glEnd();

  } else if (m_eraseType.getValue() == FREEHANDERASE && !m_track.isEmpty()) {

    TPixel color = ToonzCheck::instance()->getChecks() & ToonzCheck::eBlackBg

                       ? TPixel32::White

                       : TPixel32::Black;

    tglColor(color);

    glPushMatrix();

    m_track.drawAllFragments();

    glPopMatrix();

  }

}

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

int EraserTool::getCursorId() const {

  int ret;

  if (m_eraseType.getValue() == NORMALERASE)

    ret = ToolCursor::NormalEraserCursor;

  else {

    ret = ToolCursor::RectEraserCursor;

    if (m_eraseType.getValue() == FREEHANDERASE)

      ret = ret | ToolCursor::Ex_FreeHand;

    else if (m_eraseType.getValue() == POLYLINEERASE)

      ret = ret | ToolCursor::Ex_PolyLine;

    else if (m_eraseType.getValue() == RECTERASE)

      ret = ret | ToolCursor::Ex_Rectangle;

  }

  if (m_colorType.getValue() == LINES)

    ret = ret | ToolCursor::Ex_Line;

  else if (m_colorType.getValue() == AREAS)

    ret = ret | ToolCursor::Ex_Area;

  if (ToonzCheck::instance()->getChecks() & ToonzCheck::eBlackBg)

    ret = ret | ToolCursor::Ex_Negate;

  return ret;

}

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

void EraserTool::resetMulti() {

  m_isXsheetCell       = false;

  m_firstFrameSelected = false;

  m_firstRect.empty();

  m_selectingRect.empty();

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

  m_level                 = app->getCurrentLevel()->getLevel()

                ? app->getCurrentLevel()->getSimpleLevel()

                : 0;

  m_firstFrameId = m_veryFirstFrameId = getFrameId();

  if (m_firstStroke) {

    delete m_firstStroke;

    m_firstStroke = 0;

  }

}

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

void EraserTool::multiUpdate(const TXshSimpleLevelP &level, TFrameId firstFid,

                             TFrameId lastFid, TRectD firstRect,

                             TRectD lastRect) {

  bool backward = false;

  if (firstFid > lastFid) {

    std::swap(firstFid, lastFid);

    backward = true;

  }

  assert(firstFid <= lastFid);

  std::vector<tframeid> allFids;</tframeid>

  level->getFids(allFids);

  /*-- フレーム範囲に対応するFIdを取得 --*/

  std::vector<tframeid>::iterator i0 = allFids.begin();</tframeid>

  while (i0 != allFids.end() && *i0 < firstFid) i0++;

  if (i0 == allFids.end()) return;

  std::vector<tframeid>::iterator i1 = i0;</tframeid>

  while (i1 != allFids.end() && *i1 <= lastFid) i1++;

  assert(i0 < i1);

  std::vector<tframeid> fids(i0, i1);</tframeid>

  int m = fids.size();

  assert(m > 0);

  std::wstring levelName = level->getName();

  /*-- FrameRangeの各フレームについて --*/

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

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

    TFrameId fid = fids[i];

    assert(firstFid <= fid && fid <= lastFid);

    TToonzImageP ti = level->getFrame(fid, true);

    if (!ti) continue;

    /*--補間の係数を取得 --*/

    double t = m > 1 ? (double)i / (double)(m - 1) : 0.5;

    /*--invertがONのとき、外側領域を4つのRectに分けてupdate--*/

    if (m_invertOption.getValue()) {

      TRect rect =

          convert(interpolateRect(firstRect, lastRect, backward ? 1 - t : t));

      TRectD rect01 = TRectD(TPointD(-100000., -100000.),

                             TPointD((double)rect.x0, 100000.));

      update(ti, rect01, level, true, fid);

      TRectD rect02 =

          TRectD(convert(rect.getP01()), TPointD((double)rect.x1, 100000.));

      update(ti, rect02, level, true, fid);

      TRectD rect03 =

          TRectD(TPointD((double)rect.x0, -100000.), convert(rect.getP10()));

      update(ti, rect03, level, true, fid);

      TRectD rect04 =

          TRectD(TPointD((double)rect.x1, -100000.), TPointD(100000., 100000.));

      update(ti, rect04, level, true, fid);

    } else

      update(ti, interpolateRect(firstRect, lastRect, backward ? 1 - t : t),

             level, true, fid);

    TRect savebox;

    TRop::computeBBox(ti->getRaster(), savebox);

    ti->setSavebox(savebox);

    level->getProperties()->setDirtyFlag(true);

    notifyImageChanged(fid);

  }

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

}

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

void EraserTool::update(const TToonzImageP &ti, TRectD selArea,

                        const TXshSimpleLevelP &level, bool multi,

                        const TFrameId &frameId) {

  if (m_selectingRect.x0 > m_selectingRect.x1) {

    selArea.x1 = m_selectingRect.x0;

    selArea.x0 = m_selectingRect.x1;

  }

  if (m_selectingRect.y0 > m_selectingRect.y1) {

    selArea.y1 = m_selectingRect.y0;

    selArea.y0 = m_selectingRect.y1;

  }

  if (selArea.getLx() < 1 || selArea.getLy() < 1) return;

  bool selective = m_currentStyle.getValue();

  int styleId    = TTool::getApplication()->getCurrentLevelStyleIndex();

  TRasterCM32P raster   = ti->getRaster();

  TTileSetCM32 *tileSet = new TTileSetCM32(raster->getSize());

  tileSet->add(raster, ToonzImageUtils::convertWorldToRaster(selArea, ti));