#include "stylepickertool.h"

// TnzTools includes

#include "tools/tool.h"

#include "tools/cursors.h"

#include "tools/stylepicker.h"

#include "tools/toolhandle.h"

// TnzQt includes

#include "toonzqt/tselectionhandle.h"

#include "toonzqt/styleselection.h"

#include "toonzqt/gutil.h"

// TnzLib includes

#include "toonz/txshsimplelevel.h"

#include "toonz/txshlevelhandle.h"

#include "toonz/tpalettehandle.h"

#include "toonz/stage2.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"

#include "toonz/txshleveltypes.h"

#include "toonz/txshpalettelevel.h"

// TnzCore includes

#include "drawutil.h"

#include "tvectorimage.h"

#include "ttoonzimage.h"

#include "tundo.h"

#include "tmsgcore.h"

#define LINES L"Lines"

#define AREAS L"Areas"

#define ALL L"Lines & Areas"

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

// Pick Style Tool

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

StylePickerTool::StylePickerTool()

    : TTool("T_StylePicker")

    , m_currentStyleId(0)

    , m_colorType("Mode:")

    , m_passivePick("Passive Pick", false)

    , m_organizePalette("Organize Palette", false)

    , m_paletteToBeOrganized(NULL) {

  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");

  m_prop.bind(m_organizePalette);

  m_organizePalette.setId("OrganizePalette");

}

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

  m_oldStyleId = m_currentStyleId =

      getApplication()->getCurrentLevelStyleIndex();

  pick(pos, e, false);

}

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

  pick(pos, e);

}

void StylePickerTool::pick(const TPointD &pos, const TMouseEvent &e,

                           bool isDragging) {

  // 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()) {

    double pickRange = 10.0;

    int superPickedColumnId =

        getViewer()->posToColumnIndex(e.m_pos, pickRange, 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;

        TAffine aff =

            getViewer()->getViewMatrix() * getColumnMatrix(superPickedColumnId);

        double scale2 = aff.det();

        StylePicker superPicker(getViewer()->viewerWidget(), 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),

            pickRange, scale2, 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,

                                                      !isDragging);

          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;

  TAffine aff     = getViewer()->getViewMatrix() * getCurrentColumnMatrix();

  double scale2   = aff.det();

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

  StylePicker picker(getViewer()->viewerWidget(), image);

  int styleId =

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

                         10.0, scale2, 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();

  }

  // When clicking and switching between studio palette and level palette, the

  // signal broadcastColorStyleSwitched is not emitted if the picked style is

  // previously selected one.

  // Therefore here I set the "forceEmit" flag to true in order to emit the

  // signal whenever the picking with mouse press.

  getApplication()->setCurrentLevelStyleIndex(styleId, !isDragging);

}

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

  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;

  }

  TAffine aff     = getViewer()->getViewMatrix() * getCurrentColumnMatrix();

  double scale2   = aff.det();

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

  StylePicker picker(getViewer()->viewerWidget(), image);

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

  int inkStyleId   = picker.pickStyleId(pickPos, 10.0, scale2, 1);

  int paintStyleId = picker.pickStyleId(pickPos, 10.0, scale2, 0);

  int tone         = picker.pickTone(pickPos);

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

}

int StylePickerTool::getCursorId() const {

  int ret;

  if (!Preferences::instance()->isMultiLayerStylePickerEnabled()) {

    TImageP img      = getImage(false);

    TVectorImageP vi = img;

    TToonzImageP ti  = img;

    if (!vi && !ti) return ToolCursor::CURSOR_NO;

  }

  /* in case the "organize palette" option is active */

  if (m_organizePalette.getValue())

    ret = ToolCursor::PickerCursorOrganize;

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

    ret = ToolCursor::PickerCursorLine;

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

    ret = ToolCursor::PickerCursorArea;

  else  // line&areas

    ret = ToolCursor::PickerCursor;

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

    ret = ret | ToolCursor::Ex_Negate;

  return ret;

}

bool StylePickerTool::onPropertyChanged(std::string propertyName) {

  if (propertyName == m_organizePalette.getName()) {

    if (m_organizePalette.getValue()) {

      if (!startOrganizePalette()) {

        m_organizePalette.setValue(false);

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

        return false;

      }

    } else {

      std::cout << "End Organize Palette" << std::endl;

      m_paletteToBeOrganized = NULL;

    }

  }

  return true;

}

bool StylePickerTool::startOrganizePalette() {

  /* Check if the organizing operation is available */

  TXshLevel *level = getApplication()->getCurrentLevel()->getLevel();

  if (!level) {

    DVGui::error(tr("No current level."));

    return false;

  }

  if (level->getType() != PLI_XSHLEVEL && level->getType() != TZP_XSHLEVEL &&

      level->getType() != PLT_XSHLEVEL) {

    DVGui::error(tr("Current level has no available palette."));

    return false;

  }

  /* palette should have more than one page to organize */

  TPalette *pal = NULL;

  if (level->getType() == PLT_XSHLEVEL)

    pal = level->getPaletteLevel()->getPalette();

  else

    pal = level->getSimpleLevel()->getPalette();

  if (!pal || pal->getPageCount() < 2) {

    DVGui::error(

        tr("Palette must have more than one palette to be organized."));

    return false;

  }

  m_paletteToBeOrganized = pal;

  std::cout << "Start Organize Palette" << std::endl;

  return true;

}

/*

  If the working palette is changed, then deactivate the "organize palette"

  toggle.

*/

void StylePickerTool::onImageChanged() {

  std::cout << "StylePickerTool::onImageChanged" << std::endl;

  if (!m_organizePalette.getValue() || !m_paletteToBeOrganized) return;

  TXshLevel *level = getApplication()->getCurrentLevel()->getLevel();

  if (!level) {

    m_organizePalette.setValue(false);

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

    return;

  }

  TPalette *pal = NULL;

  if (level->getType() == PLT_XSHLEVEL)

    pal = level->getPaletteLevel()->getPalette();

  else if (level->getSimpleLevel()) {

    pal = level->getSimpleLevel()->getPalette();

  }

  if (!pal || pal != m_paletteToBeOrganized) {

    m_organizePalette.setValue(false);

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

    return;

  }

}

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

void StylePickerTool::updateTranslation() {

  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_passivePick.setQStringName(tr("Passive Pick"));

  m_organizePalette.setQStringName(tr("Organize Palette"));

}

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

StylePickerTool stylePickerTool;