

#include "tools/toolutils.h"
#include "tools/toolhandle.h"
#include "toonzqt/imageutils.h"
#include "trop.h"
#include "tools/tool.h"
#include "tstroke.h"
#include "timageinfo.h"
#include "timagecache.h"
#include "tgl.h"

#include "toonz/txsheethandle.h"
#include "toonz/tframehandle.h"
#include "toonz/txshlevelhandle.h"
#include "toonz/tscenehandle.h"
#include "toonz/txshleveltypes.h"
#include "toonz/tcolumnhandle.h"
#include "toonz/tpalettehandle.h"
#include "toonz/txshlevel.h"
#include "toonz/txshcell.h"
#include "toonz/txshsimplelevel.h"
#include "toonz/imagemanager.h"
#include "toonz/ttileset.h"
#include "toonz/toonzimageutils.h"
#include "toonz/levelproperties.h"
#include "toonz/tstageobjectspline.h"
#include "toonz/tobjecthandle.h"
#include "toonz/tstageobject.h"
#include "toonz/trasterimageutils.h"
#include "toonz/levelset.h"
#include "toonz/toonzscene.h"
#include "toonz/preferences.h"
#include "toonz/palettecontroller.h"
#include "toonz/txshchildlevel.h"

#include "toonzqt/tselectionhandle.h"
#include "toonzqt/icongenerator.h"
#include "toonzqt/selection.h"
#include "toonzqt/gutil.h"

#include "tools/strokeselection.h"

#include <QPainter>
#include <QPainterPath>
#include <QGLWidget>  // for QGLWidget::convertToGLFormat
#include <QFont>
#include <QFontMetrics>

//****************************************************************************************
//    Local namespace
//****************************************************************************************

namespace {

//! Riempie il vettore \b theVect con gli indici degli stroke contenuti nel
//! mapping \b theMap.
void mapToVector(const std::map<int, VIStroke *> &theMap,
                 std::vector<int> &theVect) {
  assert(theMap.size() == theVect.size());
  std::map<int, VIStroke *>::const_iterator it = theMap.begin();
  UINT i                                       = 0;
  for (; it != theMap.end(); ++it) {
    theVect[i++] = it->first;
  }
}

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

void updateSaveBox(const TToonzImageP &ti) {
  if (ti) {
    assert(ti->getRaster());            // Image should have a raster
    assert(ti->getSubsampling() == 1);  // Image should not be subsampled -
                                        // modified images must be the ORIGINAL
                                        // ones

    const TRect &savebox = ti->getSavebox();
    {
      TRect newSaveBox;
      TRop::computeBBox(ti->getRaster(), newSaveBox);  // This iterates the
                                                       // WHOLE raster to find
                                                       // its new savebox!

      if (!Preferences::instance()->isMinimizeSaveboxAfterEditing())
        newSaveBox +=
            savebox;  // If not minimizing the savebox, it cannot be shrunk.

      ti->setSavebox(newSaveBox);
    }
  }
}

}  // namespace

//****************************************************************************************
//    ToolUtils namespace
//****************************************************************************************

void ToolUtils::updateSaveBox(const TXshSimpleLevelP &sl, const TFrameId &fid) {
  // TODO: Savebox updates should not happen on mouse updates. This is,
  // unfortunately, what currently happens.
  sl->setDirtyFlag(true);

  TImageP img = sl->getFrame(fid, true);  // The image will be modified (it
                                          // should already have been, though)
  if (!img) return;
  // Observe that the returned image will forcedly have subsampling 1
  ::updateSaveBox(img);

  TImageInfo *info = sl->getFrameInfo(fid, true);
  ImageBuilder::setImageInfo(*info, img.getPointer());
}

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

void ToolUtils::updateSaveBox(const TXshSimpleLevelP &sl, const TFrameId &fid,
                              TImageP img) {
  sl->setFrame(fid, img);
  ToolUtils::updateSaveBox(sl, fid);
}

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

void ToolUtils::updateSaveBox() {
  TTool::Application *application = TTool::getApplication();
  if (!application) return;

  TXshLevel *xl = application->getCurrentLevel()->getLevel();
  if (!xl) return;

  TXshSimpleLevel *sl = xl->getSimpleLevel();
  if (!sl || sl->getType() != TZP_XSHLEVEL) return;

  TFrameId fid = getFrameId();
  ToolUtils::updateSaveBox(sl, fid);
}

//-----------------------------------------------------------------------------
//! Return the right value in both case: LevelFrame and SceneFrame.
TFrameId ToolUtils::getFrameId() {
  TTool::Application *app = TTool::getApplication();
  if (!app) return TFrameId();

  TFrameHandle *frameHandle = app->getCurrentFrame();
  if (frameHandle->isEditingScene()) {
    TXsheet *xsh = app->getCurrentXsheet()->getXsheet();
    if (!xsh) return 0;
    int row = frameHandle->getFrame();
    int col = app->getCurrentColumn()->getColumnIndex();
    if (col < 0) return 0;
    TXshCell cell = xsh->getCell(row, col);
    return cell.getFrameId();
  } else
    return frameHandle->getFid();
}

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

void ToolUtils::drawRect(const TRectD &rect, const TPixel32 &color,
                         unsigned short stipple, bool doContrast) {
  GLint src, dst;
  bool isEnabled;
  tglColor(color);
  if (doContrast) {
    if (color == TPixel32::Black) tglColor(TPixel32(90, 90, 90));
    isEnabled = glIsEnabled(GL_BLEND);
    glGetIntegerv(GL_BLEND_SRC, &src);
    glGetIntegerv(GL_BLEND_DST, &dst);
    glEnable(GL_BLEND);
    glBlendFunc(GL_ONE_MINUS_DST_COLOR, GL_ONE_MINUS_SRC_ALPHA);
  }

  if (stipple != 0xffff) {
    glLineStipple(1, stipple);
    glEnable(GL_LINE_STIPPLE);
  }

  glBegin(GL_LINE_STRIP);
  tglVertex(rect.getP00());
  tglVertex(rect.getP01());
  tglVertex(rect.getP11());
  tglVertex(rect.getP10());
  tglVertex(rect.getP00());
  glEnd();
  glDisable(GL_LINE_STIPPLE);
  if (doContrast) {
    if (!isEnabled) glDisable(GL_BLEND);
    glBlendFunc(src, dst);
  }
}

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

void ToolUtils::fillRect(const TRectD &rect, const TPixel32 &color) {
  tglColor(color);
  glBegin(GL_QUADS);
  tglVertex(rect.getP00());
  tglVertex(rect.getP01());
  tglVertex(rect.getP11());
  tglVertex(rect.getP10());
  tglVertex(rect.getP00());
  glEnd();
}

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

void ToolUtils::drawPoint(const TPointD &q, double pixelSize) {
  double size = pixelSize * 2.0;
  glBegin(GL_QUADS);
  glVertex2d(q.x - size, q.y - size);
  glVertex2d(q.x - size, q.y + size);
  glVertex2d(q.x + size, q.y + size);
  glVertex2d(q.x + size, q.y - size);
  glEnd();
}

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

void ToolUtils::drawCross(const TPointD &q, double pixelSize) {
  double size = pixelSize;
  glBegin(GL_LINES);
  glVertex2d(q.x - size, q.y);
  glVertex2d(q.x + size, q.y);
  glEnd();
  glBegin(GL_LINES);
  glVertex2d(q.x, q.y - size);
  glVertex2d(q.x, q.y + size);
  glEnd();
}

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

void ToolUtils::drawArrow(const TSegment &s, double pixelSize) {
  TPointD v, vn;

  double length = s.getLength() * pixelSize;

  if (length == 0) return;

  v  = normalize(s.getSpeed());
  vn = v;

  TPointD p1 = s.getP0() + v * length;
  glBegin(GL_LINES);
  tglVertex(s.getP0());
  tglVertex(p1);
  glEnd();

  v = v * length * 0.7;

  vn = vn * length * 0.2;

  TPointD p;

  glBegin(GL_TRIANGLES);
  p = s.getP0() + v + rotate90(vn);
  tglVertex(p);
  tglVertex(p1);
  p = s.getP0() + v + rotate270(vn);
  tglVertex(p);
  glEnd();
}

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

void ToolUtils::drawSquare(const TPointD &pos, double r,
                           const TPixel32 &color) {
  TRectD rect(pos - TPointD(r, r), pos + TPointD(r, r));
  tglColor(color);
  glBegin(GL_LINE_STRIP);
  tglVertex(rect.getP00());
  tglVertex(rect.getP01());
  tglVertex(rect.getP11());
  tglVertex(rect.getP10());
  tglVertex(rect.getP00());
  glEnd();
}

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

void ToolUtils::drawRectWhitArrow(const TPointD &pos, double r) {
  if (TTool::getApplication()->getCurrentObject()->isSpline()) return;
  TRectD rect(pos - TPointD(14 * r, 2 * r), pos + TPointD(14 * r, 2 * r));
  tglColor(TPixel32::Black);
  glBegin(GL_POLYGON);
  tglVertex(rect.getP00());
  tglVertex(rect.getP10());
  tglVertex(rect.getP11());
  tglVertex(rect.getP01());
  glEnd();

  double par = 5 * r;

  TPointD p01 = 0.5 * (rect.getP00() + rect.getP10());
  TPointD p02 = 0.5 * (rect.getP01() + rect.getP11());
  TPointD p11 = TPointD(p01.x, p01.y - par);
  TPointD p12 = TPointD(p02.x, p02.y + par);
  TPointD p;

  tglColor(TPixel32(130, 130, 130));

  glBegin(GL_TRIANGLES);
  p = p11 + rotate90(TPointD(0, par));
  tglVertex(p);
  tglVertex(p01);
  p = p11 + rotate270(TPointD(0, par));
  tglVertex(p);
  glEnd();

  glBegin(GL_TRIANGLES);
  p = p12 + rotate90(TPointD(0, -par));
  tglVertex(p);
  tglVertex(p02);
  p = p12 + rotate270(TPointD(0, -par));
  tglVertex(p);
  glEnd();
}

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

QRadialGradient ToolUtils::getBrushPad(int size, double hardness) {
  hardness        = tcrop(hardness, 0.0, 0.97);
  double halfSize = size * 0.5;
  double x        = halfSize * hardness;
  TQuadratic q(TPointD(x, 1.0), TPointD((halfSize + x) * 0.5, 0.0),
               TPointD(halfSize, 0.0));
  QRadialGradient rd(QPointF(halfSize, halfSize), halfSize,
                     QPointF(halfSize, halfSize));
  rd.setColorAt(0, QColor(0, 0, 0));

  double t;
  double offset = halfSize - x;
  assert(offset > 0);
  for (t = 0; t <= 1; t += 1.0 / offset) {
    TPointD p = q.getPoint(t);
    int value = 255 * p.y;
    rd.setColorAt(p.x / halfSize, QColor(0, 0, 0, value));
  }
  return rd;
}

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

QList<TRect> ToolUtils::splitRect(const TRect &first, const TRect &second) {
  TRect intersection = first * second;
  QList<TRect> rects;
  if (intersection.isEmpty()) {
    rects.append(first);
    return rects;
  }

  TRect rect;
  if (first.x0 < intersection.x0) {
    rect = TRect(first.getP00(), TPoint(intersection.x0 - 1, first.y1));
    rects.append(rect);
  }
  if (intersection.x1 < first.x1) {
    rect = TRect(TPoint(intersection.x1 + 1, first.y0), first.getP11());
    rects.append(rect);
  }
  if (intersection.y1 < first.y1) {
    rect =
        TRect(intersection.x0, intersection.y1 + 1, intersection.x1, first.y1);
    rects.append(rect);
  }
  if (first.y0 < intersection.y0) {
    rect =
        TRect(intersection.x0, first.y0, intersection.x1, intersection.y0 - 1);
    rects.append(rect);
  }
  return rects;
}

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

TRaster32P ToolUtils::convertStrokeToImage(TStroke *stroke,
                                           const TRect &imageBounds,
                                           TPoint &pos, bool pencilMode) {
  int count = stroke->getControlPointCount();
  if (count == 0) return TRaster32P();
  TPointD imgCenter = TPointD((imageBounds.x0 + imageBounds.x1) * 0.5,
                              (imageBounds.y0 + imageBounds.y1) * 0.5);

  TStroke s(*stroke);

  // check self looped stroke
  TThickPoint first = s.getControlPoint(0);
  TThickPoint back  = s.getControlPoint(count - 1);
  if (first != back) {
    TPointD mid = (first + back) * 0.5;
    s.setControlPoint(count, mid);
    s.setControlPoint(count + 1, first);
  }

  // check bounds intersection
  s.transform(TTranslation(imgCenter));
  TRectD bbox = s.getBBox();
  TRect rect(tfloor(bbox.x0), tfloor(bbox.y0), tfloor(bbox.x1),
             tfloor(bbox.y1));
  pos = rect.getP00();
  pos = TPoint(pos.x > 0 ? pos.x : 0, pos.y > 0 ? pos.y : 0);
  rect *= imageBounds;
  if (rect.isEmpty()) return TRaster32P();

  // creates the image
  QImage img(rect.getLx(), rect.getLy(), QImage::Format_ARGB32);
  img.fill(Qt::transparent);
  QColor color = Qt::black;
  QPainter p(&img);
  p.setPen(QPen(color, 1, Qt::SolidLine));
  p.setBrush(color);
  p.setRenderHint(QPainter::Antialiasing, !pencilMode);
  QPainterPath path = strokeToPainterPath(&s);
  QRectF pathRect   = path.boundingRect();
  p.translate(-toQPoint(pos));
  p.drawPath(path);
  return rasterFromQImage(img, true, false);
}

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

TStroke *ToolUtils::merge(const ArrayOfStroke &a) {
  std::vector<TThickPoint> v;

  TStroke *ref      = 0;
  int controlPoints = 0;

  for (UINT i = 0; i < a.size(); ++i) {
    ref = a[i];
    assert(ref);
    if (!ref) continue;

    controlPoints = ref->getControlPointCount() - 1;

    for (int j = 0; j < controlPoints; ++j)
      v.push_back(ref->getControlPoint(j));
  }

  if (controlPoints > 0) v.push_back(ref->getControlPoint(controlPoints));

  TStroke *out = new TStroke(v);

  return out;
}

//================================================================================================
//
// TToolUndo
//
//================================================================================================

ToolUtils::TToolUndo::TToolUndo(TXshSimpleLevel *level, const TFrameId &frameId,
                                bool createdFrame, bool createdLevel,
                                const TPaletteP &oldPalette,
                                bool renumberedLevel)
    : TUndo()
    , m_level(level)
    , m_frameId(frameId)
    , m_oldPalette(oldPalette)
    , m_col(-2)
    , m_row(-1)
    , m_isEditingLevel(false)
    , m_createdFrame(createdFrame)
    , m_createdLevel(createdLevel)
    , m_renumberedLevel(renumberedLevel)
    , m_imageId("") {
  TTool::Application *app = TTool::getApplication();
  m_isEditingLevel        = app->getCurrentFrame()->isEditingLevel();
  if (!m_isEditingLevel) {
    m_col       = app->getCurrentColumn()->getColumnIndex();
    m_row       = app->getCurrentFrame()->getFrameIndex();
    m_cellsData = TTool::m_cellsData;
  }
  if (m_renumberedLevel) {
    m_oldFids = TTool::m_oldFids;
    m_newFids = TTool::m_newFids;
  }
  if (createdFrame) {
    m_imageId = "TToolUndo" + std::to_string(m_idCount++);
    TImageCache::instance()->add(m_imageId, level->getFrame(frameId, false),
                                 false);
  }
}

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

ToolUtils::TToolUndo::~TToolUndo() {
  TImageCache::instance()->remove(m_imageId);
}

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

void ToolUtils::TToolUndo::insertLevelAndFrameIfNeeded() const {
  TTool::Application *app = TTool::getApplication();
  if (m_renumberedLevel) {
    TXsheet *xsh = app->getCurrentScene()->getScene()->getTopXsheet();
    std::vector<TXshChildLevel *> childLevels;
    ToolUtils::doUpdateXSheet(m_level.getPointer(), m_oldFids, m_newFids, xsh,
                              childLevels);
    m_level->renumber(m_newFids);
    app->getCurrentXsheet()->notifyXsheetChanged();
  }
  if (m_createdLevel) {
    TLevelSet *levelSet = app->getCurrentScene()->getScene()->getLevelSet();
    if (levelSet) {
      levelSet->insertLevel(m_level.getPointer());
      app->getCurrentScene()->notifyCastChange();
    }
  }
  if (m_createdFrame) {
    TXsheet *xsh = app->getCurrentXsheet()->getXsheet();
    TImageP img  = TImageCache::instance()->get(m_imageId, false);
    m_level->setFrame(m_frameId, img);
    if (!m_isEditingLevel) {
      for (const TTool::CellOps &cellOps : m_cellsData) {
        TXshCell cell;
        if (cellOps.type == TTool::CellOps::BlankToExisting)
          cell = xsh->getCell(cellOps.r0 - 1, m_col);
        else
          cell = TXshCell(m_level.getPointer(), m_frameId);
        for (int r = cellOps.r0; r <= cellOps.r1; r++)
          xsh->setCell(r, m_col, cell);
      }
    }
    app->getCurrentLevel()->notifyLevelChange();
  }
}

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

void ToolUtils::TToolUndo::removeLevelAndFrameIfNeeded() const {
  TTool::Application *app = TTool::getApplication();
  if (m_createdFrame) {
    m_level->eraseFrame(m_frameId);
    if (!m_isEditingLevel) {
      TXsheet *xsh = app->getCurrentXsheet()->getXsheet();
      TXshCell cell;
      for (const TTool::CellOps &cellOps : m_cellsData) {
        if (cellOps.type == TTool::CellOps::ExistingToNew)
          cell = xsh->getCell(cellOps.r0 - 1, m_col);
        for (int r = cellOps.r0; r <= cellOps.r1; r++)
          xsh->setCell(r, m_col, cell);
      }
      if (m_cellsData.size() < 1) {
        xsh->setCell(m_row, m_col, cell);
      }
    }
    if (m_createdLevel) {
      // butta il livello
      TLevelSet *levelSet = app->getCurrentScene()->getScene()->getLevelSet();
      if (levelSet) {
        levelSet->removeLevel(m_level.getPointer());
        app->getCurrentScene()->notifyCastChange();
      }
    }
  }
  if (m_oldPalette.getPointer()) {
    m_level->getPalette()->assign(m_oldPalette->clone());
    app->getPaletteController()
        ->getCurrentLevelPalette()
        ->notifyPaletteChanged();
  }
  if (m_renumberedLevel) {
    TXsheet *xsh = app->getCurrentScene()->getScene()->getTopXsheet();
    std::vector<TXshChildLevel *> childLevels;
    ToolUtils::doUpdateXSheet(m_level.getPointer(), m_newFids, m_oldFids, xsh,
                              childLevels);
    m_level->renumber(m_oldFids);
    app->getCurrentXsheet()->notifyXsheetChanged();
  }
  if (m_createdFrame || m_createdLevel)
    app->getCurrentLevel()->notifyLevelChange();
}

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

void ToolUtils::TToolUndo::notifyImageChanged() const {
  TTool::Application *app = TTool::getApplication();

  TXshSimpleLevel *currentSl = 0;
  TFrameId currentFrameId;
  if (app->getCurrentFrame()->isEditingLevel()) {
    TXshLevel *xl = app->getCurrentLevel()->getLevel();
    if (!xl) return;
    currentSl      = xl->getSimpleLevel();
    currentFrameId = app->getCurrentFrame()->getFid();
  } else {
    int row = app->getCurrentFrame()->getFrame();
    int col = app->getCurrentColumn()->getColumnIndex();
    if (col < 0) return;
    TXsheet *xsh = app->getCurrentXsheet()->getXsheet();
    if (!xsh) return;
    TXshCell cell  = xsh->getCell(row, col);
    currentSl      = cell.getSimpleLevel();
    currentFrameId = cell.getFrameId();
  }
  if (currentSl == m_level.getPointer() && currentFrameId == m_frameId) {
    TTool *tool = app->getCurrentTool()->getTool();
    if (tool) tool->onImageChanged();
  }

  IconGenerator::instance()->invalidate(m_level.getPointer(), m_frameId);
  IconGenerator::instance()->invalidateSceneIcon();

  if (m_level && m_level->getType() == PLI_XSHLEVEL) {
    std::string id = m_level->getImageId(m_frameId) + "_rasterized";
    ImageManager::instance()->invalidate(id);
  }
}

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

void ToolUtils::TToolUndo::onAdd() {
  // clean up the flags after registering undo
  TTool::m_isLevelCreated     = false;
  TTool::m_isFrameCreated     = false;
  TTool::m_isLevelRenumbererd = false;
}

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

int ToolUtils::TToolUndo::m_idCount = 0;

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

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

ToolUtils::TRasterUndo::TRasterUndo(TTileSetCM32 *tiles, TXshSimpleLevel *level,
                                    const TFrameId &frameId, bool createdFrame,
                                    bool createdLevel,
                                    const TPaletteP &oldPalette)
    : TToolUndo(level, frameId, createdFrame, createdLevel, oldPalette)
    , m_tiles(tiles) {}

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

ToolUtils::TRasterUndo::~TRasterUndo() {
  if (m_tiles) delete m_tiles;
}

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

TToonzImageP ToolUtils::TRasterUndo::getImage() const {
  if (m_level->isFid(m_frameId))
    return (TToonzImageP)m_level->getFrame(m_frameId, true);
  return 0;
}

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

int ToolUtils::TRasterUndo::getSize() const {
  int size = sizeof(*this);
  size += sizeof(*(m_level.getPointer()));
  size += sizeof(*(m_oldPalette.getPointer()));
  return m_tiles ? m_tiles->getMemorySize() + size : size;
}

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

void ToolUtils::TRasterUndo::undo() const {
  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  if (m_tiles && m_tiles->getTileCount() > 0) {
    TToonzImageP image = getImage();
    if (!image) return;

    ToonzImageUtils::paste(image, m_tiles);
    ToolUtils::updateSaveBox(m_level, m_frameId);
  }

  removeLevelAndFrameIfNeeded();

  if (m_level) {
    m_level->setDirtyFlag(true);
  }
  app->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

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

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

ToolUtils::TFullColorRasterUndo::TFullColorRasterUndo(
    TTileSetFullColor *tiles, TXshSimpleLevel *level, const TFrameId &frameId,
    bool createdFrame, bool createdLevel, const TPaletteP &oldPalette)
    : TToolUndo(level, frameId, createdFrame, createdLevel, oldPalette)
    , m_tiles(tiles) {}

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

ToolUtils::TFullColorRasterUndo::~TFullColorRasterUndo() { delete m_tiles; }

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

TRasterImageP ToolUtils::TFullColorRasterUndo::getImage() const {
  if (m_level->isFid(m_frameId))
    return (TRasterImageP)m_level->getFrame(m_frameId, true);
  return 0;
}

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

int ToolUtils::TFullColorRasterUndo::getSize() const {
  int size = sizeof(*this);
  size += sizeof(*(m_level.getPointer()));
  size += sizeof(*(m_oldPalette.getPointer()));
  return m_tiles ? m_tiles->getMemorySize() + size : size;
}

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

void ToolUtils::TFullColorRasterUndo::undo() const {
  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  if (m_tiles && m_tiles->getTileCount() > 0) {
    TRasterImageP image = getImage();
    if (!image) return;
    std::vector<TRect> rects = paste(image, m_tiles);
    int i;
    TRect resRect = rects[0];
    for (i = 1; i < (int)rects.size(); i++) resRect += rects[i];
  }

  removeLevelAndFrameIfNeeded();

  app->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

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

std::vector<TRect> ToolUtils::TFullColorRasterUndo::paste(
    const TRasterImageP &ti, const TTileSetFullColor *tileSet) const {
  std::vector<TRect> rects;
  TRasterP raster = ti->getRaster();
  for (int i = 0; i < tileSet->getTileCount(); i++) {
    const TTileSetFullColor::Tile *tile = tileSet->getTile(i);
    TRasterP ras;
    tile->getRaster(ras);
    assert(!!ras);
    raster->copy(ras, tile->m_rasterBounds.getP00());
    rects.push_back(tile->m_rasterBounds);
  }
  return rects;
}

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

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

ToolUtils::UndoModifyStroke::UndoModifyStroke(TXshSimpleLevel *level,
                                              const TFrameId &frameId,
                                              int strokeIndex)
    : TToolUndo(level, frameId), m_strokeIndex(strokeIndex) {
  TVectorImageP image = level->getFrame(frameId, true);
  assert(image);
  TStroke *stroke = image->getStroke(m_strokeIndex);
  int n           = stroke->getControlPointCount();
  for (int i = 0; i < n; i++) m_before.push_back(stroke->getControlPoint(i));

  m_selfLoopBefore = stroke->isSelfLoop();

  TTool::Application *app = TTool::getApplication();
  m_row                   = app->getCurrentFrame()->getFrame();
  m_column                = app->getCurrentColumn()->getColumnIndex();
}

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

ToolUtils::UndoModifyStroke::~UndoModifyStroke() {}

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

void ToolUtils::UndoModifyStroke::onAdd() {
  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(image);
  if (!image) return;

  TStroke *stroke = image->getStroke(m_strokeIndex);
  assert(stroke);
  int n = stroke->getControlPointCount();
  for (int i = 0; i < n; i++) m_after.push_back(stroke->getControlPoint(i));

  m_selfLoopAfter = stroke->isSelfLoop();
}

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

void ToolUtils::UndoModifyStroke::undo() const {
  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  if (app->getCurrentFrame()->isEditingScene()) {
    app->getCurrentColumn()->setColumnIndex(m_column);
    app->getCurrentFrame()->setFrame(m_row);
  } else
    app->getCurrentFrame()->setFid(m_frameId);

  TSelection *selection = app->getCurrentSelection()->getSelection();
  if (selection) selection->selectNone();

  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(image);
  if (!image) return;
  QMutexLocker lock(image->getMutex());
  TStroke *stroke = 0;
  if (image->getStrokeCount() ==
      1)  // && image->getStroke(0)->getStyle()==SplineStyle)
    stroke = image->getStroke(0);
  else
    stroke = image->getStroke(m_strokeIndex);
  if (!stroke) return;
  TStroke *oldStroke = new TStroke(*stroke);
  stroke->reshape(&m_before[0], m_before.size());
  stroke->setSelfLoop(m_selfLoopBefore);
  image->notifyChangedStrokes(m_strokeIndex, oldStroke);
  notifyImageChanged();
  delete oldStroke;
}

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

void ToolUtils::UndoModifyStroke::redo() const {
  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  if (app->getCurrentFrame()->isEditingScene()) {
    app->getCurrentColumn()->setColumnIndex(m_column);
    app->getCurrentFrame()->setFrame(m_row);
  } else
    app->getCurrentFrame()->setFid(m_frameId);

  TSelection *selection = app->getCurrentSelection()->getSelection();
  if (selection) selection->selectNone();

  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(image);
  if (!image) return;
  QMutexLocker lock(image->getMutex());
  TStroke *stroke = 0;
  if (image->getStrokeCount() ==
      1)  //&& image->getStroke(0)->getStyle()==SplineStyle)
    stroke = image->getStroke(0);
  else
    stroke = image->getStroke(m_strokeIndex);
  if (!stroke) return;
  TStroke *oldStroke = new TStroke(*stroke);
  stroke->reshape(&m_after[0], m_after.size());
  stroke->setSelfLoop(m_selfLoopAfter);
  image->notifyChangedStrokes(m_strokeIndex, oldStroke);
  delete oldStroke;
  app->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

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

int ToolUtils::UndoModifyStroke::getSize() const {
  return (m_before.capacity() + m_after.capacity()) * sizeof(TThickPoint) +
         sizeof(*this) + 500;
}

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

ToolUtils::UndoModifyStrokeAndPaint::UndoModifyStrokeAndPaint(
    TXshSimpleLevel *level, const TFrameId &frameId, int strokeIndex)
    : UndoModifyStroke(level, frameId, strokeIndex), m_fillInformation(0) {
  TVectorImageP image = level->getFrame(frameId, true);
  assert(image);
  TStroke *stroke = image->getStroke(strokeIndex);
  m_oldBBox       = stroke->getBBox();
}

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

void ToolUtils::UndoModifyStrokeAndPaint::onAdd() {
  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(!!image);
  if (!image) return;

  UndoModifyStroke::onAdd();
  TStroke *stroke   = image->getStroke(m_strokeIndex);
  m_fillInformation = new std::vector<TFilledRegionInf>;
  ImageUtils::getFillingInformationOverlappingArea(
      image, *m_fillInformation, m_oldBBox, stroke->getBBox());
}

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

void ToolUtils::UndoModifyStrokeAndPaint::undo() const {
  TTool::Application *application = TTool::getApplication();
  if (!application) return;

  UndoModifyStroke::undo();
  TRegion *reg;
  UINT size = m_fillInformation->size();
  if (!size) {
    application->getCurrentXsheet()->notifyXsheetChanged();
    notifyImageChanged();
    return;
  }

  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(!!image);
  if (!image) return;

  // image->validateRegions();
  image->findRegions();
  for (UINT i = 0; i < size; i++) {
    reg = image->getRegion((*m_fillInformation)[i].m_regionId);
    assert(reg);
    if (reg) reg->setStyle((*m_fillInformation)[i].m_styleId);
  }

  application->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}
//-----------------------------------------------------------------------------

ToolUtils::UndoModifyStrokeAndPaint::~UndoModifyStrokeAndPaint() {
  delete m_fillInformation;
}

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

int ToolUtils::UndoModifyStrokeAndPaint::getSize() const {
  int size = m_fillInformation
                 ? m_fillInformation->size() * sizeof(TFilledRegionInf)
                 : 0;
  return UndoModifyStroke::getSize() + size;
}

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

ToolUtils::UndoModifyListStroke::UndoModifyListStroke(
    TXshSimpleLevel *level, const TFrameId &frameId,
    const std::vector<TStroke *> &strokeVect)
    : TToolUndo(level, frameId), m_fillInformation(0) {
  UINT strokeNum      = strokeVect.size();
  TVectorImageP image = level->getFrame(frameId, true);
  assert(image);
  for (UINT i = 0; i < strokeNum; i++) {
    m_oldBBox += (strokeVect[i])->getBBox();
    int strokeIndex = image->getStrokeIndex(strokeVect[i]);
    m_strokeList.push_back(new UndoModifyStroke(level, frameId, strokeIndex));
  }

  m_beginIt = m_strokeList.begin();
  m_endIt   = m_strokeList.end();
}

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

ToolUtils::UndoModifyListStroke::~UndoModifyListStroke() {
  clearPointerContainer(m_strokeList);
  delete m_fillInformation;
}

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

void ToolUtils::UndoModifyListStroke::onAdd() {
  std::list<UndoModifyStroke *>::iterator it = m_beginIt;
  TRectD newBBox;

  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(!!image);
  if (!image) return;

  for (; it != m_endIt; ++it) {
    TStroke *s = image->getStroke((*it)->m_strokeIndex);
    (*it)->onAdd();
  }
  m_fillInformation = new std::vector<TFilledRegionInf>;

  if (m_beginIt != m_endIt)
    ImageUtils::getFillingInformationOverlappingArea(image, *m_fillInformation,
                                                     m_oldBBox, newBBox);
}

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

void ToolUtils::UndoModifyListStroke::undo() const {
  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  std::list<UndoModifyStroke *>::iterator stroke_it = m_beginIt;
  if (m_beginIt == m_endIt) return;

  for (; stroke_it != m_endIt; ++stroke_it) {
    (*stroke_it)->undo();
  }

  UINT size = m_fillInformation->size();
  if (!size) {
    app->getCurrentXsheet()->notifyXsheetChanged();
    app->getCurrentTool()->getTool()->notifyImageChanged();
    return;
  }

  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(!!image);
  if (!image) return;

  image->findRegions();

  TRegion *reg;
  for (UINT i = 0; i < size; i++) {
    reg = image->getRegion((*m_fillInformation)[i].m_regionId);
    assert(reg);
    if (reg) reg->setStyle((*m_fillInformation)[i].m_styleId);
  }
  app->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

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

void ToolUtils::UndoModifyListStroke::redo() const {
  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  std::list<UndoModifyStroke *>::iterator it = m_beginIt;

  for (; it != m_endIt; ++it) {
    (*it)->redo();
  }

  app->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

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

int ToolUtils::UndoModifyListStroke::getSize() const {
  int sum = 0;

  std::list<UndoModifyStroke *>::iterator it = m_beginIt;

  for (; it != m_endIt; ++it) {
    sum += (*it)->getSize();
  }

  if (m_fillInformation)
    sum += m_fillInformation->capacity() * sizeof(TFilledRegionInf);

  return sum;
}

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

ToolUtils::UndoPencil::UndoPencil(
    TStroke *stroke, std::vector<TFilledRegionInf> *fillInformation,
    TXshSimpleLevel *level, const TFrameId &frameId, bool createdFrame,
    bool createdLevel, bool autogroup, bool autofill)
    : TToolUndo(level, frameId, createdFrame, createdLevel, 0)
    , m_strokeId(stroke->getId())
    , m_fillInformation(fillInformation)
    , m_autogroup(autogroup)
    , m_autofill(autofill) {
  m_stroke = new TStroke(*stroke);
}

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

ToolUtils::UndoPencil::~UndoPencil() {
  delete m_fillInformation;
  delete m_stroke;
}

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

void ToolUtils::UndoPencil::undo() const {
  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  /*if(app->getCurrentFrame()->isEditingScene() && m_col!=-2 && m_row!=-1)
  {
          app->getCurrentColumn()->setColumnIndex(m_col);
          app->getCurrentFrame()->setFrame(m_row);
  }
  else
          app->getCurrentFrame()->setFid(m_frameId);*/

  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(image);
  if (!image) return;

  QMutexLocker sl(image->getMutex());
  VIStroke *stroke = image->getStrokeById(m_strokeId);
  if (!stroke) return;
  image->deleteStroke(stroke);
  TSelection *selection            = app->getCurrentSelection()->getSelection();
  StrokeSelection *strokeSelection = dynamic_cast<StrokeSelection *>(selection);
  if (strokeSelection) strokeSelection->selectNone();

  UINT size = m_fillInformation->size();
  TRegion *reg;
  for (UINT i = 0; i < size; i++) {
    reg = image->getRegion((*m_fillInformation)[i].m_regionId);
    assert(reg);
    if (reg) reg->setStyle((*m_fillInformation)[i].m_styleId);
  }

  removeLevelAndFrameIfNeeded();

  app->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

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

void ToolUtils::UndoPencil::redo() const {
  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  insertLevelAndFrameIfNeeded();

  /*if(app->getCurrentFrame()->isEditingScene())
  {
          app->getCurrentColumn()->setColumnIndex(m_col);
          app->getCurrentFrame()->setFrame(m_row);
  }
  else
          app->getCurrentFrame()->setFid(m_frameId);*/

  TVectorImageP image = m_level->getFrame(m_frameId, true);
  if (!image) return;

  QMutexLocker sl(image->getMutex());
  TStroke *stroke = new TStroke(*m_stroke);
  stroke->setId(m_strokeId);
  image->addStroke(stroke);
  if (image->isComputedRegionAlmostOnce()) image->findRegions();

  if (m_autogroup && stroke->isSelfLoop()) {
    int index = image->getStrokeCount() - 1;
    image->group(index, 1);
    if (m_autofill) {
      // to avoid filling other strokes, I enter into the new stroke group
      int currentGroup = image->exitGroup();
      image->enterGroup(index);
      image->selectFill(stroke->getBBox().enlarge(1, 1), 0, stroke->getStyle(),
                        false, true, false);
      if (currentGroup != -1)
        image->enterGroup(currentGroup);
      else
        image->exitGroup();
    }
  }
  app->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

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

int ToolUtils::UndoPencil::getSize() const {
  return sizeof(*this) +
         m_fillInformation->capacity() * sizeof(TFilledRegionInf) + 500;
}

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

ToolUtils::UndoRasterPencil::UndoRasterPencil(
    TXshSimpleLevel *level, const TFrameId &frameId, TStroke *stroke,
    bool selective, bool filled, bool doAntialias, bool createdFrame,
    bool createdLevel, std::string primitiveName)
    : TRasterUndo(0, level, frameId, createdFrame, createdLevel, 0)
    , m_selective(selective)
    , m_filled(filled)
    , m_doAntialias(doAntialias)
    , m_primitiveName(primitiveName) {
  TRasterCM32P raster = getImage()->getRaster();
  TDimension d        = raster->getSize();
  m_tiles             = new TTileSetCM32(d);
  TRect rect =
      convert(stroke->getBBox()) + TPoint((int)(d.lx * 0.5), (int)(d.ly * 0.5));
  m_tiles->add(raster, rect.enlarge(2));
  m_stroke = new TStroke(*stroke);
}

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

ToolUtils::UndoRasterPencil::~UndoRasterPencil() { delete m_stroke; }

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

void ToolUtils::UndoRasterPencil::redo() const {
  insertLevelAndFrameIfNeeded();
  TToonzImageP image = getImage();
  if (!image) return;

  ToonzImageUtils::addInkStroke(image, m_stroke, m_stroke->getStyle(),
                                m_selective, m_filled, TConsts::infiniteRectD,
                                m_doAntialias);
  ToolUtils::updateSaveBox();
  TTool::getApplication()->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

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

int ToolUtils::UndoRasterPencil::getSize() const {
  return TRasterUndo::getSize() +
         m_stroke->getControlPointCount() * sizeof(TThickPoint) + 100 +
         sizeof(this);
}

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

ToolUtils::UndoFullColorPencil::UndoFullColorPencil(
    TXshSimpleLevel *level, const TFrameId &frameId, TStroke *stroke,
    double opacity, bool doAntialias, bool createdFrame, bool createdLevel)
    : TFullColorRasterUndo(0, level, frameId, createdFrame, createdLevel, 0)
    , m_opacity(opacity)
    , m_doAntialias(doAntialias) {
  TRasterP raster = getImage()->getRaster();
  TDimension d    = raster->getSize();
  m_tiles         = new TTileSetFullColor(d);
  TRect rect =
      convert(stroke->getBBox()) + TPoint((int)(d.lx * 0.5), (int)(d.ly * 0.5));
  m_tiles->add(raster, rect.enlarge(2));
  m_stroke = new TStroke(*stroke);
}

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

ToolUtils::UndoFullColorPencil::~UndoFullColorPencil() { delete m_stroke; }

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

void ToolUtils::UndoFullColorPencil::redo() const {
  insertLevelAndFrameIfNeeded();
  TRasterImageP image = getImage();
  if (!image) return;
  TRasterImageUtils::addStroke(image, m_stroke, TRectD(), m_opacity,
                               m_doAntialias);
  TTool::getApplication()->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

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

int ToolUtils::UndoFullColorPencil::getSize() const {
  return TFullColorRasterUndo::getSize() +
         m_stroke->getControlPointCount() * sizeof(TThickPoint) + 100 +
         sizeof(this);
}

//=============================================================================================
//
// undo class (path strokes). call it BEFORE and register it AFTER path change
//
ToolUtils::UndoPath::UndoPath(TStageObjectSpline *spline) : m_spline(spline) {
  assert(!!m_spline);

  const TStroke *stroke = m_spline->getStroke();
  assert(stroke);
  int n = stroke->getControlPointCount();
  for (int i = 0; i < n; i++) m_before.push_back(stroke->getControlPoint(i));
  m_selfLoopBefore = stroke->isSelfLoop();
}

ToolUtils::UndoPath::~UndoPath() {}

void ToolUtils::UndoPath::onAdd() {
  assert(!!m_spline);

  const TStroke *stroke = m_spline->getStroke();
  assert(stroke);
  int n = stroke->getControlPointCount();
  for (int i = 0; i < n; i++) m_after.push_back(stroke->getControlPoint(i));
}

void ToolUtils::UndoPath::undo() const {
  assert(!!m_spline);

  TTool::Application *app = TTool::getApplication();
  TSelection *selection   = app->getCurrentSelection()->getSelection();
  if (selection) selection->selectNone();

  TStroke *stroke = new TStroke(*m_spline->getStroke());
  stroke->reshape(&m_before[0], m_before.size());
  stroke->setSelfLoop(m_selfLoopBefore);
  m_spline->setStroke(stroke);

  if (!app->getCurrentObject()->isSpline()) return;

  TStageObjectId currentObjectId = app->getCurrentObject()->getObjectId();
  TStageObject *stageObject =
      app->getCurrentXsheet()->getXsheet()->getStageObject(currentObjectId);
  if (stageObject->getSpline()->getId() == m_spline->getId())
    app->getCurrentObject()->setSplineObject(m_spline);

  app->getCurrentTool()->getTool()->notifyImageChanged();
}

void ToolUtils::UndoPath::redo() const {
  assert(!!m_spline);

  TTool::Application *app = TTool::getApplication();
  TSelection *selection   = app->getCurrentSelection()->getSelection();
  if (selection) selection->selectNone();

  TStroke *stroke = new TStroke(*m_spline->getStroke());
  stroke->reshape(&m_after[0], m_after.size());
  stroke->setSelfLoop(m_selfLoopBefore);
  m_spline->setStroke(stroke);

  if (!app->getCurrentObject()->isSpline()) return;

  TStageObjectId currentObjectId = app->getCurrentObject()->getObjectId();
  TStageObject *stageObject =
      app->getCurrentXsheet()->getXsheet()->getStageObject(currentObjectId);
  if (stageObject->getSpline()->getId() == m_spline->getId())
    app->getCurrentObject()->setSplineObject(m_spline);

  app->getCurrentTool()->getTool()->notifyImageChanged();
}

int ToolUtils::UndoPath::getSize() const { return sizeof(*this) + 500; }

//=============================================================================================
//
// UndoControlPointEditor
//

ToolUtils::UndoControlPointEditor::UndoControlPointEditor(
    TXshSimpleLevel *level, const TFrameId &frameId)
    : TToolUndo(level, frameId), m_isStrokeDelete(false) {
  TVectorImageP image = level->getFrame(frameId, true);
  assert(image);
  if (!image) return;

  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  m_row    = app->getCurrentFrame()->getFrame();
  m_column = app->getCurrentColumn()->getColumnIndex();
}

ToolUtils::UndoControlPointEditor::~UndoControlPointEditor() {
  deleteVIStroke(m_oldStroke.second);
  if (!m_isStrokeDelete) deleteVIStroke(m_newStroke.second);
}

void ToolUtils::UndoControlPointEditor::onAdd() {
  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(image);
  if (!image) return;
  QMutexLocker lock(image->getMutex());
  if (m_isStrokeDelete) return;
  addNewStroke(m_oldStroke.first, image->getVIStroke(m_oldStroke.first));
}

void ToolUtils::UndoControlPointEditor::addOldStroke(int index, VIStroke *vs) {
  VIStroke *s = cloneVIStroke(vs);
  m_oldStroke = std::make_pair(index, s);
}

void ToolUtils::UndoControlPointEditor::addNewStroke(int index, VIStroke *vs) {
  VIStroke *s = cloneVIStroke(vs);
  m_newStroke = std::make_pair(index, s);
}

void ToolUtils::UndoControlPointEditor::undo() const {
  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  if (app->getCurrentFrame()->isEditingScene()) {
    app->getCurrentColumn()->setColumnIndex(m_column);
    app->getCurrentFrame()->setFrame(m_row);
  } else
    app->getCurrentFrame()->setFid(m_frameId);

  TSelection *selection = app->getCurrentSelection()->getSelection();
  if (selection) selection->selectNone();
  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(image);
  if (!image) return;
  QMutexLocker lock(image->getMutex());

  if (!m_isStrokeDelete) image->removeStroke(m_newStroke.first, false);

  UINT i      = 0;
  VIStroke *s = cloneVIStroke(m_oldStroke.second);
  image->insertStrokeAt(s, m_oldStroke.first);

  if (image->isComputedRegionAlmostOnce())
    image->findRegions();  // in futuro togliere. Serve perche' la
                           // removeStrokes, se gli si dice
  // di non calcolare le regioni, e' piu' veloce ma poi chrash tutto

  app->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

void ToolUtils::UndoControlPointEditor::redo() const {
  TTool::Application *app = TTool::getApplication();
  if (!app) return;

  if (app->getCurrentFrame()->isEditingScene()) {
    app->getCurrentColumn()->setColumnIndex(m_column);
    app->getCurrentFrame()->setFrame(m_row);
  } else
    app->getCurrentFrame()->setFid(m_frameId);
  TSelection *selection = app->getCurrentSelection()->getSelection();
  if (selection) selection->selectNone();
  TVectorImageP image = m_level->getFrame(m_frameId, true);
  assert(image);
  if (!image) return;
  QMutexLocker lock(image->getMutex());

  image->removeStroke(m_oldStroke.first, false);

  if (!m_isStrokeDelete) {
    VIStroke *s = cloneVIStroke(m_newStroke.second);
    image->insertStrokeAt(s, m_newStroke.first);
  }

  if (image->isComputedRegionAlmostOnce())
    image->findRegions();  // in futuro togliere. Serve perche' la
                           // removeStrokes, se gli si dice
  // di non calcolare le regioni, e' piu' veloce ma poi chrash tutto

  app->getCurrentXsheet()->notifyXsheetChanged();
  notifyImageChanged();
}

//=============================================================================================
//
// Menu
//

ToolUtils::DragMenu::DragMenu() : QMenu() {}

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

QAction *ToolUtils::DragMenu::exec(const QPoint &p, QAction *action) {
  QAction *execAct = QMenu::exec(p, action);
  if (execAct) return execAct;

  return defaultAction();
}

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

void ToolUtils::DragMenu::mouseReleaseEvent(QMouseEvent *e) {
  QMenu::mouseReleaseEvent(e);
  hide();
}

//=============================================================================================
//
// ColumChooserMenu
//

ToolUtils::ColumChooserMenu::ColumChooserMenu(
    TXsheet *xsh, const std::vector<int> &columnIndexes)
    : DragMenu() {
  int size = columnIndexes.size();
  int i;
  for (i = size - 1; i >= 0; i--) {
    int index                 = columnIndexes[i];
    TStageObjectId id         = TStageObjectId::ColumnId(index);
    TStageObject *stageObject = xsh->getStageObject(id);
    QString cmdStr = "Column " + QString::fromStdString(stageObject->getName());
    QAction *act   = new QAction(cmdStr, this);
    act->setData(index);
    addAction(act);
    if (size - 1 == i) {
      setDefaultAction(act);
      setActiveAction(act);
    }
  }
}

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

int ToolUtils::ColumChooserMenu::execute() {
  QAction *executeAct = exec(QCursor::pos());
  return (!executeAct) ? -1 : executeAct->data().toInt();
}

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

double ToolUtils::ConeSubVolume::compute(double cover) {
  double x = (10 * tcrop(cover, -1.0, 1.0)) + 10;
  assert(0 <= x && x <= 20);
  int i = tfloor(x);
  if (i == 20)
    return m_values[i];
  else
    // Linear interpolation.
    return (-(x - (i + 1)) * m_values[i]) - (-(x - i) * m_values[i + 1]);
}

const double ToolUtils::ConeSubVolume::m_values[] = {
    1.0,      0.99778,  0.987779,  0.967282,  0.934874,  0.889929,   0.832457,
    0.763067, 0.683002, 0.594266,  0.5,       0.405734,  0.316998,   0.236933,
    0.167543, 0.110071, 0.0651259, 0.0327182, 0.0122208, 0.00221986, 0.0};

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

void ToolUtils::drawBalloon(const TPointD &pos, std::string text,
                            const TPixel32 &color, TPoint delta,
                            double pixelSize, bool isPicking,
                            std::vector<TRectD> *otherBalloons) {
  TToolViewer *viewer =
      TTool::getApplication()->getCurrentTool()->getTool()->getViewer();
  int devPixRatio = getDevicePixelRatio(viewer->viewerWidget());
  QString qText   = QString::fromStdString(text);
  QFont font("Arial");  // ,QFont::Bold);
  font.setPixelSize(13 * devPixRatio);
  QFontMetrics fm(font);
  QRect textRect = fm.boundingRect(qText);

  int baseLine = -textRect.top();
  int mrg      = 3 * devPixRatio;

  // avoid other balloons
  if (otherBalloons) {
    QRect tmpTextRect = textRect;
    int tmpMrg        = mrg;
    // for high dpi monitors
    if (devPixRatio != 1) {
      QFont tmpFont = font;
      tmpFont.setPixelSize(13);
      QFontMetrics tmpFm(tmpFont);
      tmpTextRect = tmpFm.boundingRect(qText);
      tmpMrg      = 3;
    }

    std::vector<TRectD> &balloons = *otherBalloons;
    int n                         = (int)balloons.size();
    TDimensionD balloonSize(pixelSize * (tmpTextRect.width() + tmpMrg * 2),
                            pixelSize * (tmpTextRect.height() + tmpMrg * 2));
    TRectD balloonRect;
    for (;;) {
      balloonRect = TRectD(
          pos + TPointD(delta.x * pixelSize, delta.y * pixelSize), balloonSize);
      int i = 0;
      while (i < n && !balloons[i].overlaps(balloonRect)) i++;
      if (i == n) break;
      double y = balloons[i].y0 - balloonSize.ly - 2 * tmpMrg * pixelSize;
      delta.y  = (y - pos.y) / pixelSize;
    }
    balloons.push_back(balloonRect);
  }

  delta.x *= devPixRatio;
  delta.y *= devPixRatio;

  textRect.moveTo(std::max(delta.x, 10 * devPixRatio + mrg),
                  std::max(mrg + 2 * devPixRatio, -delta.y - baseLine));

  int y  = textRect.top() + baseLine;
  int x0 = textRect.left() - mrg;
  int x1 = textRect.right() + mrg;
  int y0 = textRect.top() - mrg;
  int y1 = textRect.bottom() + mrg;

  if (isPicking) {
    if (viewer->is3DView()) {
      double x0 = pos.x + textRect.left() * pixelSize,
             y0 = pos.y + delta.y * pixelSize;
      double x1 = x0 + pixelSize * textRect.width();
      double y1 = y0 + pixelSize * textRect.height();
      double d  = pixelSize * 5;
      glRectd(x0 - d, y0 - d, x1 + d, y1 + d);
    } else {
      TPointD posBalloon = viewer->worldToPos(pos);

      double d  = 5;
      double x0 = posBalloon.x + textRect.left() - d;
      double y0 = posBalloon.y + delta.y - d;
      double x1 = x0 + textRect.width() + d;
      double y1 = y0 + textRect.height() + d;

      TPointD p1(x0, y0);
      TPointD p2(x1, y0);
      TPointD p3(x0, y1);
      TPointD p4(x1, y1);

      TPointD w1(viewer->winToWorld(p1));
      TPointD w2(viewer->winToWorld(p2));
      TPointD w3(viewer->winToWorld(p3));
      TPointD w4(viewer->winToWorld(p4));

      glBegin(GL_QUADS);
      glVertex2d(w1.x, w1.y);
      glVertex2d(w2.x, w2.y);
      glVertex2d(w4.x, w4.y);
      glVertex2d(w3.x, w3.y);
      glEnd();
    }

    return;
  }

  QSize size(textRect.width() + textRect.left() + mrg,
             std::max(textRect.bottom() + mrg, y + delta.y) + 3 * devPixRatio);

  QImage label(size.width(), size.height(), QImage::Format_ARGB32);
  label.fill(Qt::transparent);
  // label.fill(qRgba(200,200,0,200));
  QPainter p(&label);
  p.setRenderHints(QPainter::Antialiasing | QPainter::TextAntialiasing);
  p.setBrush(QColor(color.r, color.g, color.b, color.m));
  p.setPen(Qt::NoPen);

  QPainterPath pp;
  pp.moveTo(x0, y - 8 * devPixRatio);
  pp.lineTo(0, y + delta.y);
  pp.lineTo(x0, y);
  /* rounded edges
  int arcSize = 10;
  pp.arcTo(x0,y1-arcSize,arcSize,arcSize,180,90);
  pp.arcTo(x1-arcSize,y1-arcSize,arcSize,arcSize,270,90);
  pp.arcTo(x1-arcSize,y0,arcSize,arcSize,0,90);
  pp.arcTo(x0,y0,arcSize,arcSize,90,90);
  */
  // sharp edges
  pp.lineTo(x0, y1);
  pp.lineTo(x1, y1);
  pp.lineTo(x1, y0);
  pp.lineTo(x0, y0);

  pp.closeSubpath();

  p.drawPath(pp);

  p.setPen(Qt::black);
  p.setFont(font);
  p.drawText(textRect, Qt::AlignCenter | Qt::TextDontClip, qText);

  QImage texture = QGLWidget::convertToGLFormat(label);

  glRasterPos2f(pos.x, pos.y);
  glBitmap(0, 0, 0, 0, 0, -size.height() + (y + delta.y), NULL);  //
  glEnable(GL_BLEND);
  glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
  glDrawPixels(texture.width(), texture.height(), GL_RGBA, GL_UNSIGNED_BYTE,
               texture.bits());
  glDisable(GL_BLEND);
  glColor3d(0, 0, 0);
}

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

void ToolUtils::drawHook(const TPointD &pos, ToolUtils::HookType type,
                         bool highlighted, bool onionSkin) {
  TToolViewer *viewer =
      TTool::getApplication()->getCurrentTool()->getTool()->getViewer();
  int devPixRatio = getDevicePixelRatio(viewer->viewerWidget());
  int r = 10, d = r + r;
  QImage image(d * devPixRatio, d * devPixRatio, QImage::Format_ARGB32);
  image.fill(Qt::transparent);
  // image.fill(qRgba(200,200,0,200));
  QPainter painter(&image);
  painter.scale(devPixRatio, devPixRatio);
  // painter.setRenderHints(QPainter::Antialiasing|QPainter::TextAntialiasing);

  int matte = onionSkin ? 100 : 255;

  QColor color(0, 0, 0, matte);
  if (highlighted) color = QColor(0, 175, 175, matte);

  if (type == NormalHook || type == PassHookA) {
    painter.setPen(QPen(QColor(255, 255, 255, matte), 3));
    painter.drawEllipse(5, 5, d - 10, d - 10);
    painter.setPen(color);
    painter.drawEllipse(5, 5, d - 10, d - 10);
  } else if (type == OtherLevelHook) {
    QColor color(0, 200, 200, 200);
    // painter.setPen(QPen(Qt::white,3));
    // painter.drawEllipse(5,5,d-10,d-10);
    painter.setPen(Qt::white);
    painter.setBrush(color);
    painter.drawEllipse(6, 6, d - 12, d - 12);
  }
  if (type == NormalHook || type == PassHookB) {
    painter.setPen(QPen(QColor(255, 255, 255, matte), 3));
    painter.drawLine(0, r, d, r);
    painter.drawLine(r, 0, r, d);
    painter.setPen(color);
    painter.drawLine(0, r, d, r);
    painter.drawLine(r, 0, r, d);
  }

  QImage texture = QGLWidget::convertToGLFormat(image);
  glRasterPos2f(pos.x, pos.y);
  glBitmap(0, 0, 0, 0, -r * devPixRatio, -r * devPixRatio, NULL);
  glEnable(GL_BLEND);
  glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
  glDrawPixels(texture.width(), texture.height(), GL_RGBA, GL_UNSIGNED_BYTE,
               texture.bits());
  glDisable(GL_BLEND);
}

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

bool ToolUtils::isJustCreatedSpline(TImage *image) {
  TVectorImageP vi = image;
  if (!vi) return false;
  if (vi->getStrokeCount() != 1) return false;
  TStroke *stroke = vi->getStroke(0);
  if (stroke->getControlPointCount() != 3) return false;
  TPointD p0 = stroke->getControlPoint(0);
  TPointD p1 = stroke->getControlPoint(1);
  TPointD p2 = stroke->getControlPoint(2);
  double d   = 30.0;
  return p0 == TPointD(-d, 0) && p1 == TPointD(0, 0) && p2 == TPointD(d, 0);
}

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

TRectD ToolUtils::interpolateRect(const TRectD &rect1, const TRectD &rect2,
                                  double t) {
  assert(rect1.x0 <= rect1.x1);
  assert(rect1.y0 <= rect1.y1);
  assert(rect2.x0 <= rect2.x1);
  assert(rect2.y0 <= rect2.y1);

  return TRectD(rect1.x0 + (rect2.x0 - rect1.x0) * t,
                rect1.y0 + (rect2.y0 - rect1.y0) * t,
                rect1.x1 + (rect2.x1 - rect1.x1) * t,
                rect1.y1 + (rect2.y1 - rect1.y1) * t);
}

//-----------------------------------------------------------------------------
/*
bool ToolUtils::isASubRegion(int reg, const std::vector<TRegion*> &regions)
{
        TRegion *region=regions[reg];
        for (int i=0; i<(int)regions.size(); i++)
        {
                if(i!=reg)
                        if(region->isSubRegionOf(*regions[i]))
                                return true;
        }
        return false;
}
*/
//-----------------------------------------------------------------------------

TRectD ToolUtils::getBounds(const std::vector<TThickPoint> &points,
                            double maxThickness) {
  TThickPoint p = points[0];
  double radius = maxThickness == 0 ? p.thick * 0.5 : maxThickness * 0.5;
  TRectD rect(p - TPointD(radius, radius), p + TPointD(radius, radius));
  int i;
  for (i = 1; i < (int)points.size(); i++) {
    p      = points[i];
    radius = maxThickness == 0 ? p.thick * 0.5 : maxThickness * 0.5;
    rect =
        rect + TRectD(p - TPointD(radius, radius), p + TPointD(radius, radius));
  }
  return rect;
}

//-----------------------------------------------------------------------------
/*
template <typename PIXEL>
TRasterPT<PIXEL> ToolUtils::rotate90(const TRasterPT<PIXEL> &ras, bool toRight)
{
        if(!ras)
                return 0;
        int lx=ras->getLy();
        int ly=ras->getLx();
        TRasterPT<PIXEL> workRas(lx,ly);
        for (int i=0; i<ras->getLx(); i++)
        {
                for (int j=0; j<ras->getLy(); j++)
                {
                        if(toRight)
                                workRas->pixels(ly-1-i)[j]=ras->pixels(j)[i];
                        else
                                workRas->pixels(i)[lx-1-j]=ras->pixels(j)[i];
                }
        }
        return workRas;
}*/

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

bool ToolUtils::doUpdateXSheet(TXshSimpleLevel *sl,
                               std::vector<TFrameId> oldFids,
                               std::vector<TFrameId> newFids, TXsheet *xsh,
                               std::vector<TXshChildLevel *> &childLevels) {
  bool ret = false;
  for (int c = 0; c < xsh->getColumnCount(); ++c) {
    int r0, r1;
    int n = xsh->getCellRange(c, r0, r1);
    if (n > 0) {
      bool changed = false;
      std::vector<TXshCell> cells(n);
      xsh->getCells(r0, c, n, &cells[0]);
      for (int i = 0; i < n; i++) {
        TXshCell currCell = cells[i];
        // check the sub xsheets too
        if (!cells[i].isEmpty() &&
            cells[i].m_level->getType() == CHILD_XSHLEVEL) {
          TXshChildLevel *level = cells[i].m_level->getChildLevel();
          // make sure we haven't already checked the level
          if (level && std::find(childLevels.begin(), childLevels.end(),
                                 level) == childLevels.end()) {
            childLevels.push_back(level);
            TXsheet *subXsh = level->getXsheet();
            ret |= doUpdateXSheet(sl, oldFids, newFids, subXsh, childLevels);
          }
        }
        for (int j = 0; j < oldFids.size(); j++) {
          if (oldFids.at(j) == newFids.at(j)) continue;
          TXshCell tempCell(sl, oldFids.at(j));
          bool sameSl  = tempCell.getSimpleLevel() == currCell.getSimpleLevel();
          bool sameFid = tempCell.getFrameId() == currCell.getFrameId();
          if (sameSl && sameFid) {
            TXshCell newCell(sl, newFids.at(j));
            cells[i] = newCell;
            changed  = true;
            break;
          }
        }
      }
      if (changed) {
        xsh->setCells(r0, c, n, &cells[0]);
        ret = true;
        // TApp::instance()->getCurrentXsheet()->notifyXsheetChanged();
      }
    }
  }
  return ret;
}

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

bool ToolUtils::renumberForInsertFId(TXshSimpleLevel *sl, const TFrameId &fid,
                                     const TFrameId &maxFid, TXsheet *xsh) {
  auto getNextLetter = [](const QString &letter) {
    if (letter.isEmpty()) return QString('a');
    if (letter == 'z' || letter == 'Z') return QString();
    QByteArray byteArray = letter.toUtf8();
    // return incrementing the last letter
    byteArray.data()[byteArray.size() - 1]++;
    return QString::fromUtf8(byteArray);
  };

  std::vector<TFrameId> fids;
  std::vector<TFrameId> oldFrames;
  sl->getFids(oldFrames);
  sl->getFids(fids);
  std::vector<TFrameId>::iterator it = std::find(fids.begin(), fids.end(), fid);
  if (it == fids.end()) return false;

  std::set<TFrameId> fidsSet(fids.begin(), fids.end());
  QList<TFrameId> fIdsToBeShifted;
  TFrameId tmpFid = fid;
  for (auto itr = fidsSet.upper_bound(maxFid); itr != fidsSet.end(); ++itr) {
    if (*itr > tmpFid) break;
    fIdsToBeShifted.push_back(*itr);
    if (!fid.getLetter().isEmpty()) {
      QString nextLetter = getNextLetter((*itr).getLetter());
      if (!nextLetter.isEmpty())
        tmpFid = TFrameId((*itr).getNumber(), nextLetter);
      else
        tmpFid = TFrameId((*itr).getNumber() + 1);
    } else
      tmpFid = TFrameId((*itr).getNumber() + 1, (*itr).getLetter());
  }

  if (fIdsToBeShifted.isEmpty()) return false;

  for (TFrameId &tmpFid : fids) {
    if (fIdsToBeShifted.contains(tmpFid)) {
      if (!fid.getLetter().isEmpty()) {
        QString nextLetter = getNextLetter(tmpFid.getLetter());
        if (!nextLetter.isEmpty())
          tmpFid = TFrameId(tmpFid.getNumber(), nextLetter);
        else
          tmpFid = TFrameId(tmpFid.getNumber() + 1);
      } else
        tmpFid = TFrameId(tmpFid.getNumber() + 1, tmpFid.getLetter());
    }
  }

  std::vector<TXshChildLevel *> childLevels;
  doUpdateXSheet(sl, oldFrames, fids, xsh, childLevels);
  sl->renumber(fids);

  return true;
}
