#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"

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

}

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

  pick(pos, e);

}

void StylePickerTool::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 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;

  }

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

}

int StylePickerTool::getCursorId() const {

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

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

  if (m_organizePalette.getValue())

    return (isBlackBG) ? ToolCursor::PickerCursorWhiteOrganize

                        : ToolCursor::PickerCursorOrganize;

  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;

}

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;

  }

}

StylePickerTool stylePickerTool;