#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 : public TTool
{
	int m_oldStyleId, m_currentStyleId;

	TEnumProperty m_colorType;
	TPropertyGroup m_prop;

	TBoolProperty m_passivePick;

public:
	TPropertyGroup *getProperties(int targetType) { 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 { return TTool::LevelReadTool; }

	void draw()
	{
	}

	void leftButtonDown(const TPointD &pos, const TMouseEvent &e)
	{
		m_oldStyleId = m_currentStyleId = getApplication()->getCurrentLevelStyleIndex();
		pick(pos, e);
	}
	void leftButtonDrag(const TPointD &pos, const TMouseEvent &e)
	{
		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);
			if (styleSelection)
				styleSelection->selectNone();
		}

		getApplication()->setCurrentLevelStyleIndex(styleId);
	}

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

	void onActivate()
	{
	}

	int getCursorId() const
	{

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