#include "tools/tool.h"

#include "tools/cursors.h"

#include "tools/stylepicker.h"

#include "toonz/txshsimplelevel.h"

#include "toonz/txshlevelhandle.h"

#include "drawutil.h"

#include "tvectorimage.h"

#include "ttoonzimage.h"

#include "tundo.h"

#include "toonz/tpalettehandle.h"

#include "toonz/stage2.h"

#include "tproperty.h"

#include "toonzqt/tselectionhandle.h"

#include "toonzqt/styleselection.h"

#include "tools/toolhandle.h"

#include "toonz/tframehandle.h"

#include "toonz/txsheethandle.h"

#include "toonz/preferences.h"

#include "toonz/tcolumnhandle.h"

#include "toonz/dpiscale.h"

#include "toonz/palettecontroller.h"

#define LINES L"Lines"

#define AREAS L"Areas"

#define ALL L"Lines & Areas"

namespace {

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

// Pick Style Tool

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

class StylePickerTool final : public TTool {

  int m_oldStyleId, m_currentStyleId;

  TEnumProperty m_colorType;

  TPropertyGroup m_prop;

  TBoolProperty m_passivePick;

public:

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

  StylePickerTool()

      : TTool("T_StylePicker")

      , m_currentStyleId(0)

      , m_colorType("Mode:")

      , m_passivePick("Passive Pick", false) {

    m_prop.bind(m_colorType);

    m_colorType.addValue(AREAS);

    m_colorType.addValue(LINES);

    m_colorType.addValue(ALL);

    m_colorType.setId("Mode");

    bind(TTool::CommonLevels);

    m_prop.bind(m_passivePick);

    m_passivePick.setId("PassivePick");

  }

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

  void draw() override {}

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

    m_oldStyleId = m_currentStyleId =

        getApplication()->getCurrentLevelStyleIndex();

    pick(pos, e);

  }

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

    pick(pos, e);

  }

  void pick(const TPointD &pos, const TMouseEvent &e) {

    // Area = 0, Line = 1, All = 2

    int modeValue = m_colorType.getIndex();

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

    // MultiLayerStylePicker

    /*---

                    PickしたStyleId = 0、かつ

                    Preference で MultiLayerStylePickerが有効、かつ

                    Scene編集モード、かつ

                    下のカラムから拾った色がTransparentでない場合、

                    → カレントLevelを移動する。

    ---*/

    if (Preferences::instance()->isMultiLayerStylePickerEnabled() &&

        getApplication()->getCurrentFrame()->isEditingScene()) {

      int superPickedColumnId = getViewer()->posToColumnIndex(

          e.m_pos, getPixelSize() * getPixelSize(), false);

      if (superPickedColumnId >= 0 /*-- 何かColumnに当たった場合 --*/

          &&

          getApplication()->getCurrentColumn()->getColumnIndex() !=

              superPickedColumnId) /*-- かつ、Current Columnでない場合 --*/

      {

        /*-- そのColumnからPickを試みる --*/

        int currentFrame = getApplication()->getCurrentFrame()->getFrame();

        TXshCell pickedCell =

            getApplication()->getCurrentXsheet()->getXsheet()->getCell(

                currentFrame, superPickedColumnId);

        TImageP pickedImage           = pickedCell.getImage(false).getPointer();

        TToonzImageP picked_ti        = pickedImage;

        TVectorImageP picked_vi       = pickedImage;

        TXshSimpleLevel *picked_level = pickedCell.getSimpleLevel();

        if ((picked_ti || picked_vi) && picked_level) {

          TPointD tmpMousePosition =

              getColumnMatrix(superPickedColumnId).inv() *

              getViewer()->winToWorld(e.m_pos);

          TPointD tmpDpiScale =

              getCurrentDpiScale(picked_level, getCurrentFid());

          tmpMousePosition.x /= tmpDpiScale.x;

          tmpMousePosition.y /= tmpDpiScale.y;

          StylePicker superPicker(pickedImage);

          int picked_subsampling =

              picked_level->getImageSubsampling(pickedCell.getFrameId());

          int superPicked_StyleId = superPicker.pickStyleId(

              TScale(1.0 / picked_subsampling) * tmpMousePosition +

                  TPointD(-0.5, -0.5),

              getPixelSize() * getPixelSize(), modeValue);

          /*-- 何かStyleが拾えて、Transparentでない場合 --*/

          if (superPicked_StyleId > 0) {

            /*-- Levelの移動 --*/

            getApplication()->getCurrentLevel()->setLevel(picked_level);

            /*-- Columnの移動 --*/

            getApplication()->getCurrentColumn()->setColumnIndex(

                superPickedColumnId);

            /*-- 選択の解除 --*/

            if (getApplication()->getCurrentSelection()->getSelection())

              getApplication()

                  ->getCurrentSelection()

                  ->getSelection()

                  ->selectNone();

            /*-- StyleIdの移動 --*/

            getApplication()->setCurrentLevelStyleIndex(superPicked_StyleId);

            return;

          }

        }

      }

    }

    /*-- MultiLayerStylePicker ここまで --*/

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

    TImageP image    = getImage(false);

    TToonzImageP ti  = image;

    TVectorImageP vi = image;

    TXshSimpleLevel *level =

        getApplication()->getCurrentLevel()->getSimpleLevel();

    if ((!ti && !vi) || !level) return;

    /*-- 画面外をpickしても拾えないようにする --*/

    if (!m_viewer->getGeometry().contains(pos)) return;

    int subsampling = level->getImageSubsampling(getCurrentFid());

    StylePicker picker(image);

    int styleId = picker.pickStyleId(

        TScale(1.0 / subsampling) * pos + TPointD(-0.5, -0.5),

        getPixelSize() * getPixelSize(), modeValue);

    if (styleId < 0) return;

    if (modeValue == 1)  // LINES

    {

      /*-- pickLineモードのとき、取得Styleが0の場合はカレントStyleを変えない。

       * --*/

      if (styleId == 0) return;

      /*--

       * pickLineモードのとき、PurePaintの部分をクリックしてもカレントStyleを変えない

       * --*/

      if (ti &&

          picker.pickTone(TScale(1.0 / subsampling) * pos +

                          TPointD(-0.5, -0.5)) == 255)

        return;

    }

    /*--- Styleを選択している場合は選択を解除する ---*/

    TSelection *selection =

        TTool::getApplication()->getCurrentSelection()->getSelection();

    if (selection) {

      TStyleSelection *styleSelection =

          dynamic_cast<tstyleselection *="">(selection);</tstyleselection>

      if (styleSelection) styleSelection->selectNone();

    }

    getApplication()->setCurrentLevelStyleIndex(styleId);

  }

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

    if (!m_passivePick.getValue()) return;

    /*--- PassiveにStyleを拾う機能 ---*/

    PaletteController *controller =

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

    TImageP image    = getImage(false);

    TToonzImageP ti  = image;

    TVectorImageP vi = image;

    TXshSimpleLevel *level =

        getApplication()->getCurrentLevel()->getSimpleLevel();

    if ((!ti && !vi) || !level || !m_viewer->getGeometry().contains(pos)) {

      controller->notifyStylePassivePicked(-1, -1, -1);

      return;

    }

    int subsampling = level->getImageSubsampling(getCurrentFid());

    StylePicker picker(image);

    TPointD pickPos(TScale(1.0 / subsampling) * pos + TPointD(-0.5, -0.5));

    int inkStyleId =

        picker.pickStyleId(pickPos, getPixelSize() * getPixelSize(), 1);

    int paintStyleId =

        picker.pickStyleId(pickPos, getPixelSize() * getPixelSize(), 0);

    int tone = picker.pickTone(pickPos);

    controller->notifyStylePassivePicked(inkStyleId, paintStyleId, tone);

  }

  void onActivate() override {}

  int getCursorId() const override {

    bool isBlackBG = ToonzCheck::instance()->getChecks() & ToonzCheck::eBlackBg;

    if (m_colorType.getValue() == LINES)

      return (isBlackBG) ? ToolCursor::PickerCursorWhiteLine

                         : ToolCursor::PickerCursorLine;

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

      return (isBlackBG) ? ToolCursor::PickerCursorWhiteArea

                         : ToolCursor::PickerCursorArea;

    else  // line&areas

      return (isBlackBG) ? ToolCursor::PickerCursorWhite

                         : ToolCursor::PickerCursor;

  }

} stylePickerTool;

}