#include "tools/cursormanager.h"

#include "tools/tool.h"

#include "tools/cursors.h"

#ifdef TOONZQT_EXPORTS

#include "toonz/preferences.h"

#else

#include "../toonz/preferences.h"

#endif

#include <qwidget></qwidget>

#include <qpixmap></qpixmap>

#include <assert.h></assert.h>

#include <map></map>

#include <qdebug></qdebug>

#include <qpainter></qpainter>

namespace {

const struct {

  int cursorType;

  const char *pixmapFilename;

  int x, y;

  bool flippable;

} cursorInfo[] = {

    {ToolCursor::PenCursor, "brush", 16, 15, false},

    {ToolCursor::PenLargeCursor, "brush_large", 16, 15, false},

    {ToolCursor::PenCrosshairCursor, "brush_crosshair", 16, 15, false},

    {ToolCursor::PenTriangleTopLeftCursor, "brush_triangle_top_left",

     16, 15, false},

    {ToolCursor::PenTriangleTopRightCursor, "brush_triangle_top_right",

     16, 15, false},

    {ToolCursor::PenTriangleBottomLeftCursor, "brush_triangle_bottom_left",

     16, 15, false},

    {ToolCursor::PenTriangleBottomRightCursor, "brush_triangle_bottom_right",

     16, 15, false},

    {ToolCursor::PenTriangleUpCursor, "brush_triangle_up",

     16, 15, false},

    {ToolCursor::PenTriangleDownCursor, "brush_triangle_down",

     16, 15, false},

    {ToolCursor::PenTriangleLeftCursor, "brush_triangle_left",

     16, 15, false},

    {ToolCursor::PenTriangleRightCursor, "brush_triangle_right",

     16, 15, false},

    {ToolCursor::BenderCursor, "bender", 9, 7, true},

    {ToolCursor::CutterCursor, "cutter", 6, 24, true},  // 12,20, ???},

    {ToolCursor::EraserCursor, "eraser", 7, 21, true},  // 15,16, ???},

    {ToolCursor::DistortCursor, "selection_distort", 11, 6, true},

    {ToolCursor::FillCursor, "fill", 3, 26, true},

    {ToolCursor::MoveCursor, "move", 15, 15, false},

    {ToolCursor::MoveEWCursor, "move_ew", 15, 15, false},

    {ToolCursor::MoveNSCursor, "move_ns", 15, 15, false},

    {ToolCursor::DisableCursor, "disable", 15, 15, false},

    {ToolCursor::MoveZCursor, "", 0, 0, false},

    {ToolCursor::MoveZCursorBase, "move_z_notext", 15, 15, false},

    {ToolCursor::FxGadgetCursor, "", 0, 0, false},

    {ToolCursor::FxGadgetCursorBase, "edit_FX_notext", 11, 6, true},

    {ToolCursor::FlipHCursor, "flip_h", 15, 15, false},

    {ToolCursor::FlipVCursor, "flip_v", 15, 15, false},

    {ToolCursor::IronCursor, "iron", 15, 15, true},

    {ToolCursor::LevelSelectCursor, "level_select", 7, 3, true},

    {ToolCursor::MagnetCursor, "magnet", 18, 18, true},

    {ToolCursor::PanCursor, "pan", 17, 17, true},

    {ToolCursor::PickerCursorLine, "", 0, 0, false},

    {ToolCursor::PickerCursorLineBase, "picker_style", 7, 22, true},

    {ToolCursor::PickerCursorArea, "", 0, 0, false},

    {ToolCursor::PickerCursorAreaBase, "picker_style", 7, 22, true},

    {ToolCursor::PickerCursor, "picker_style", 7, 22, true},

    {ToolCursor::PointingHandCursor, "pointing_hand", 13, 4, true},

    {ToolCursor::PumpCursor, "pump", 16, 23, false},

    {ToolCursor::RotCursor, "rot", 15, 15, false},

    {ToolCursor::RotTopLeft, "rot_top_left", 15, 15, false},

    {ToolCursor::RotBottomRight, "rot_bottom_right", 15, 15, false},

    {ToolCursor::RotBottomLeft, "rot_bottom_left", 15, 15, false},

    {ToolCursor::RotateCursor, "rotate", 15, 19, true},

    {ToolCursor::ScaleCursor, "scale", 15, 15, false},

    {ToolCursor::ScaleInvCursor, "scale_inv", 15, 15, false},

    {ToolCursor::ScaleHCursor, "scale_h", 15, 15, false},

    {ToolCursor::ScaleVCursor, "scale_v", 15, 15, false},

    {ToolCursor::EditFxCursor, "", 0, 0, false},

    {ToolCursor::EditFxCursorBase, "edit_FX_notext", 11, 6, true},

    {ToolCursor::ScaleGlobalCursor, "scale_global", 15, 15, false},

    {ToolCursor::ScaleHVCursor, "", 0, 0, false},

    {ToolCursor::ScaleHVCursorBase, "scale_hv_notext", 15, 15, false},

    {ToolCursor::StrokeSelectCursor, "stroke_select", 11, 6, true},

    {ToolCursor::TapeCursor, "tape", 4, 23, true},

    {ToolCursor::TrackerCursor, "tracker", 16, 15, false},

    {ToolCursor::TypeInCursor, "type_in", 16, 19, false},

    {ToolCursor::TypeOutCursor, "type_out", 16, 19, false},

    {ToolCursor::ZoomCursor, "zoom", 14, 14, true},

    {ToolCursor::PinchCursor, "pinch_curve", 6, 16, true},

    {ToolCursor::PinchAngleCursor, "pinch_angle", 6, 16, true},

    {ToolCursor::PinchWaveCursor, "pinch_wave", 6, 16, true},

    {ToolCursor::SplineEditorCursor, "stroke_select", 11, 6, true},

    {ToolCursor::SplineEditorCursorAdd, "selection_add", 11, 6, true},

    {ToolCursor::SplineEditorCursorSelect, "selection_convert", 11, 6, true},

    {ToolCursor::NormalEraserCursor, "normaleraser", 7, 19, true},

    {ToolCursor::RectEraserCursor, "recteraser", 3, 26, true},

    {ToolCursor::PickerCursorOrganize, "picker_style_organize", 7, 22, true},

    {ToolCursor::PickerRGB, "", 0, 0, false},

    {ToolCursor::PickerRGBBase, "picker_style", 7, 22, true},

    {ToolCursor::PickerRGBWhite, "picker_rgb_white", 7, 22, true},

    {ToolCursor::FillCursorL, "karasu", 7, 25, true},

    {ToolCursor::RulerModifyCursor, "ruler_modify", 7, 7, true},

    {ToolCursor::RulerNewCursor, "ruler_new", 7, 7, true},

    {ToolCursor::PickPrevCursor, "", 13, 4, false},

    {ToolCursor::PickNextCursor, "", 13, 4, false},

    {0, 0, 0, 0, false}};

struct CursorData {

  QPixmap pixmap;

  int x, y;

};

const struct {

  int decorateType;

  const char *pixmapFilename;

} decorateInfo[] = {{ToolCursor::Ex_FreeHand, "ex_freehand"},

                    {ToolCursor::Ex_PolyLine, "ex_polyline"},

                    {ToolCursor::Ex_Rectangle, "ex_rectangle"},

                    {ToolCursor::Ex_Line, "ex_line"},

                    {ToolCursor::Ex_Area, "ex_area"},

                    {ToolCursor::Ex_Fill_NoAutopaint, "ex_fill_no_autopaint"},

                    {ToolCursor::Ex_FX, "ex_FX"},

                    {ToolCursor::Ex_Z, "ex_z"},

                    {ToolCursor::Ex_StyleLine, "ex_style_line"},

                    {ToolCursor::Ex_StyleArea, "ex_style_area"},

                    {ToolCursor::Ex_RGB, "ex_rgb"},

                    {ToolCursor::Ex_HV, "ex_hv"},

                    {ToolCursor::Ex_Precise, "ex_precise"},

                    {ToolCursor::Ex_Prev, "ex_prev"},

                    {ToolCursor::Ex_Next, "ex_next"},

                    {ToolCursor::Ex_FreePick, "ex_freepick"},

                    {0, 0}};

};  // namespace

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

// CursorManager

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

class CursorManager {  // singleton

  std::map<int, cursordata=""> m_cursors;</int,>

  std::map<int, cursordata=""> m_cursorsLeft;</int,>

  CursorManager() {}

public:

  static CursorManager *instance() {

    static CursorManager _instance;

    return &_instance;

  }

  void doDecoration(QPixmap &pixmap, int decorationFlag, bool useLeft) {

    if (decorationFlag == 0) return;

    if (decorationFlag > ToolCursor::Ex_Negate) {

      QPainter p(&pixmap);

      p.setCompositionMode(QPainter::CompositionMode_SourceOver);

      for (int i = 0; decorateInfo[i].pixmapFilename; i++)

        if (decorationFlag & decorateInfo[i].decorateType) {

          QString leftStr = "";

          if (useLeft) leftStr = "_left";

          QString path = QString(":Resources/") +

                         decorateInfo[i].pixmapFilename + leftStr + ".png";

          p.drawPixmap(0, 0, QPixmap(path));

        }

    }

    // negate

    if (decorationFlag & ToolCursor::Ex_Negate) {

      QImage img = pixmap.toImage();

      img.invertPixels(QImage::InvertRgb);  // leave the alpha channel unchanged

      pixmap = QPixmap::fromImage(img);

    }

  }

  const CursorData &getCursorData(int cursorType) {

    // se e' gia' in tabella lo restituisco

    std::map<int, cursordata="">::iterator it;</int,>

    if (Preferences::instance()->getCursorBrushStyle() == "Simple")

      cursorType = ToolCursor::PenCursor;

    if (cursorType == ToolCursor::PenCursor) {

      QString brushType = Preferences::instance()->getCursorBrushType();

      if (brushType == "Large")

        cursorType = ToolCursor::PenLargeCursor;

      else if (brushType == "Crosshair")

        cursorType = ToolCursor::PenCrosshairCursor;

      else if (brushType == "Triangle Top Left")

        cursorType = ToolCursor::PenTriangleTopLeftCursor;

      else if (brushType == "Triangle Top Right")

        cursorType = ToolCursor::PenTriangleTopRightCursor;

      else if (brushType == "Triangle Bottom Left")

        cursorType = ToolCursor::PenTriangleBottomLeftCursor;

      else if (brushType == "Triangle Bottom Right")

        cursorType = ToolCursor::PenTriangleBottomRightCursor;

      else if (brushType == "Triangle Up")

        cursorType = ToolCursor::PenTriangleUpCursor;

      else if (brushType == "Triangle Down")

        cursorType = ToolCursor::PenTriangleDownCursor;

      else if (brushType == "Triangle Left")

        cursorType = ToolCursor::PenTriangleLeftCursor;

      else if (brushType == "Triangle Right")

        cursorType = ToolCursor::PenTriangleRightCursor;

    }

    bool useLeft =

        (Preferences::instance()->getCursorBrushStyle() == "Left-Handed");

    if (useLeft) {

      it = m_cursorsLeft.find(cursorType);

      if (it != m_cursorsLeft.end()) return it->second;

    } else {

      it = m_cursors.find(cursorType);

      if (it != m_cursors.end()) return it->second;

    }

    int decorationsFlag = cursorType & ~(0xFF);

    int baseCursorType  = cursorType & 0xFF;

    if (baseCursorType == ToolCursor::CURSOR_ARROW) {

      CursorData data;

      QCursor leftArrow(Qt::ArrowCursor);

      data.pixmap = leftArrow.pixmap();

      data.x      = leftArrow.hotSpot().x();

      data.y      = leftArrow.hotSpot().y();

      if (useLeft) {

        QImage target = (&data.pixmap)->toImage();

        (&data.pixmap)->convertFromImage(target.mirrored(true, false));

        data.x = data.pixmap.width() - data.x - 1;

        it     = m_cursorsLeft.insert(std::make_pair(cursorType, data)).first;

      } else

        it = m_cursors.insert(std::make_pair(cursorType, data)).first;

      return it->second;

    }

    // provo a cercarlo in cursorInfo[]

    int i;

    for (i = 0; cursorInfo[i].pixmapFilename; i++)

      if (baseCursorType == cursorInfo[i].cursorType) {

        QString path =

            QString(":Resources/") + cursorInfo[i].pixmapFilename + ".png";

        CursorData data;

        data.pixmap = QPixmap(path);

        if (data.pixmap.isNull())

          data = getCursorData(ToolCursor::CURSOR_ARROW);

        else {

          data.x = cursorInfo[i].x;

          data.y = cursorInfo[i].y;

          if (useLeft && cursorInfo[i].flippable) {

            QImage target = (&data.pixmap)->toImage();

            (&data.pixmap)->convertFromImage(target.mirrored(true, false));

            data.x = data.pixmap.width() - cursorInfo[i].x - 1;

          }

          if (decorationsFlag != 0)

            doDecoration(data.pixmap, decorationsFlag, useLeft);

        }

        if (useLeft)

          it = m_cursorsLeft.insert(std::make_pair(cursorType, data)).first;

        else

          it = m_cursors.insert(std::make_pair(cursorType, data)).first;

        return it->second;

      }

    // niente da fare. uso un default

    CursorData data;

    static const QPixmap standardCursorPixmap("cursors/hook.png");

    data.pixmap = standardCursorPixmap;

    data.x = data.y = 0;

    if (useLeft)

      it = m_cursorsLeft.insert(std::make_pair(cursorType, data)).first;

    else

      it = m_cursors.insert(std::make_pair(cursorType, data)).first;

    return it->second;

  }

  QCursor getCursor(int cursorType) {

    // assert(cursorType!=0);

    QCursor cursor;

    /*

if (cursorType == ToolCursor::CURSOR_ARROW)

  cursor = Qt::ArrowCursor;

else */

    if (cursorType == ToolCursor::ForbiddenCursor)

      cursor = Qt::ForbiddenCursor;

    else {

      const CursorData &data = getCursorData(cursorType);

      cursor                 = QCursor(data.pixmap, data.x, data.y);

    }

    return cursor;

  }

};

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

void setToolCursor(QWidget *viewer, int cursorType) {

  viewer->setCursor(CursorManager::instance()->getCursor(cursorType));

}

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

QCursor getToolCursor(int cursorType) {

  return CursorManager::instance()->getCursor(cursorType);

}