#include "rulertool.h"

#include "tools/toolhandle.h"

#include "toonz/tscenehandle.h"

#include "toonz/txshlevelhandle.h"

#include "tools/toolutils.h"

#include "tvectorimage.h"

#include "ttoonzimage.h"

#include "tools/cursors.h"

#include "trasterimage.h"

#include "tgl.h"

#include "toonz/stage2.h"

#include "toonz/txshsimplelevel.h"

#include "toonzqt/icongenerator.h"

#include "tenv.h"

#include "tconst.h"

#include "toonz/tframehandle.h"

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

RulerTool::RulerTool()

	: TTool("T_Ruler"), m_firstPos(TConst::nowhere), m_secondPos(TConst::nowhere), m_mousePos(TConst::nowhere), m_dragMode(MakeNewRuler), m_justClicked(false)

{

	bind(TTool::AllTargets);

}

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

void RulerTool::setToolOptionsBox(RulerToolOptionsBox *toolOptionsBox)

{

	m_toolOptionsBox.push_back(toolOptionsBox);

}

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

void RulerTool::onImageChanged()

{

	/*--位置をリセット--*/

	m_firstPos = TConst::nowhere;

	m_secondPos = TConst::nowhere;

	for (int i = 0; i < (int)m_toolOptionsBox.size(); i++) {

		m_toolOptionsBox[i]->resetValues();

	}

}

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

void RulerTool::draw()

{

	/*--- 始点が設定されていたら、描画 ---*/

	if (m_firstPos != TConst::nowhere) {

		tglColor((m_dragMode == MoveFirstPos) ? TPixel32(51, 204, 26) : TPixel32::Red);

		tglDrawCircle(m_firstPos, 4);

		tglDrawCircle(m_firstPos, 2);

		/*--- 終点が設定されていたら、その区間を描画 ---*/

		if (m_secondPos != TConst::nowhere) {

			tglColor((m_dragMode == MoveRuler) ? TPixel32(51, 204, 26) : TPixel32::Red);

			glBegin(GL_LINE_STRIP);

			tglVertex(m_firstPos);

			tglVertex(m_secondPos);

			glEnd();

			tglColor((m_dragMode == MoveSecondPos) ? TPixel32(51, 204, 26) : TPixel32::Red);

			tglDrawCircle(m_secondPos, 4);

		}

	}

}

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

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

{

	if (m_dragMode == MakeNewRuler)

		m_justClicked = true;

}

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

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

{

	/*-- 最初のドラッグ --*/

	if (m_justClicked && m_dragMode == MakeNewRuler) {

		m_firstPos = m_mousePos;

		m_justClicked = false;

	}

	/*-- ドラッグモードがMakeNewRuler/MoveSecondPosのとき、secondPosを更新 --*/

	if (m_dragMode == MakeNewRuler || m_dragMode == MoveSecondPos) {

		/*-- Shiftキーが押されていたら：0,45,90度に角度固定の直線を引く --*/

		if (e.isShiftPressed())

			m_secondPos = getHVCoordinatedPos(pos, m_firstPos);

		else

			m_secondPos = pos;

	}

	/*-- ドラッグモードがMoveFirstPosのとき、firstPosを更新 --*/

	else if (m_dragMode == MoveFirstPos) {

		/*-- Shiftキーが押されていたら：0,45,90度に角度固定の直線を引く --*/

		if (e.isShiftPressed())

			m_firstPos = getHVCoordinatedPos(pos, m_secondPos);

		else

			m_firstPos = pos;

	}

	/*-- Ruler全体を移動するモード --*/

	else {

		TPointD d = pos - m_mousePos;

		m_firstPos += d;

		m_secondPos += d;

		/*-- マウス位置を更新 --*/

		m_mousePos = pos;

	}

	updateToolOption(); /*-- ToolOptionの表示を更新 --*/

	invalidate();

}

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

void RulerTool::leftButtonUp(const TPointD &pos, const TMouseEvent &e)

{

	m_justClicked = false;

	invalidate();

}

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

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

{

	/*-- マウス位置を更新 --*/

	m_mousePos = pos;

	/*-- マウスクリックしてたら無視 --*/

	if (e.isLeftButtonPressed())

		return;

	/*-- マウスがRulerの端点/Rulerそのものに近い場合はドラッグモードを変更する。--*/

	if (m_firstPos != TConst::nowhere && tdistance2(pos, m_firstPos) < 16)

		m_dragMode = MoveFirstPos;

	else if (m_secondPos != TConst::nowhere && tdistance2(pos, m_secondPos) < 16)

		m_dragMode = MoveSecondPos;

	else if (isNearRuler())

		m_dragMode = MoveRuler;

	else

		m_dragMode = MakeNewRuler;

	invalidate();

}

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

void RulerTool::onActivate()

{

	/*-- 位置をリセット --*/

	m_firstPos = TConst::nowhere;

	m_secondPos = TConst::nowhere;

	for (int i = 0; i < (int)m_toolOptionsBox.size(); i++) {

		m_toolOptionsBox[i]->resetValues();

	}

}

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

int RulerTool::getCursorId() const

{

	if (m_dragMode == MakeNewRuler)

		return ToolCursor::RulerNewCursor;

	else

		return ToolCursor::RulerModifyCursor;

}

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

/*-- ToolOptionの表示を更新 --*/

void RulerTool::updateToolOption()

{

	TTool::Application *app = TTool::getApplication();

	TFrameHandle *currentFrame = app->getCurrentFrame();

	double x, y, w, h, a, l;

	/*-- Level編集モードのとき、そのLevelのDPIに合わせてInchの値を得る。Pixelの値も出力する。 --*/

	if (currentFrame->isEditingLevel()) {

		TXshLevelHandle *currentLevel = app->getCurrentLevel();

		TXshLevel *xl = currentLevel->getLevel();

		if (xl) {

			TXshSimpleLevel *sl = xl->getSimpleLevel();

			if (sl) {

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

				TPointD dpiScale = getViewer()->getDpiScale();

				TPointD pp1 = TPointD(m_firstPos.x / dpiScale.x, m_firstPos.y / dpiScale.y);

				TPointD pp2 = TPointD(m_secondPos.x / dpiScale.x, m_secondPos.y / dpiScale.y);

				TPointD p1 = TScale(1.0 / subsampling) * pp1 + TPointD(-0.5, -0.5);

				TPointD p2 = TScale(1.0 / subsampling) * pp2 + TPointD(-0.5, -0.5);

				TImageP image = getImage(false);

				TPoint pix1, pix2;

				TToonzImageP ti = image;

				TRasterImageP ri = image;

				if (ti || ri) {

					if (ti) {

						TDimension size = ti->getSize();

						pix1 = TPoint(tround(0.5 * size.lx + p1.x), tround(0.5 * size.ly + p1.y));

						pix2 = TPoint(tround(0.5 * size.lx + p2.x), tround(0.5 * size.ly + p2.y));

					} else if (ri) {

						TDimension size = ri->getRaster()->getSize();

						pix1 = TPoint(tround(0.5 * size.lx + p1.x), tround(0.5 * size.ly + p1.y));

						pix2 = TPoint(tround(0.5 * size.lx + p2.x), tround(0.5 * size.ly + p2.y));

					}

					int xPix, yPix, wPix, hPix;

					TPointD dpi = sl->getDpi(getCurrentFid());

					xPix = pix1.x;

					yPix = pix1.y;

					wPix = pix2.x - pix1.x;

					hPix = pix2.y - pix1.y;

					x = (double)xPix / dpi.x;

					y = (double)yPix / dpi.y;

					w = (double)wPix / dpi.x;

					h = (double)hPix / dpi.y;

					a = atan2(h, w) * 180.0 / 3.14159264;

					l = sqrt(w * w + h * h);

					for (int i = 0; i < (int)m_toolOptionsBox.size(); i++) {

						m_toolOptionsBox[i]->updateValues(true, x, y, w, h, a, l,

														  xPix, yPix, wPix, hPix);

					}

					return;

				}

			}

		}

	}

	/*-- シーン編集モードのとき、Stage::inchで割った値がInch値となる --*/

	x = m_firstPos.x / Stage::inch;

	y = m_firstPos.y / Stage::inch;

	w = (m_secondPos.x - m_firstPos.x) / Stage::inch;

	h = (m_secondPos.y - m_firstPos.y) / Stage::inch;

	a = atan2(h, w) * 180.0 / 3.14159264;

	l = sqrt(w * w + h * h);

	for (int i = 0; i < (int)m_toolOptionsBox.size(); i++) {

		m_toolOptionsBox[i]->updateValues(false, x, y, w, h, a, l);

	}

}

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

/*! 現在のマウス位置がRulerに十分近ければTrueを返す

*/

bool RulerTool::isNearRuler()

{

	double a, b, c;

	TPointD vec = m_secondPos - m_firstPos;

	a = -vec.y;

	b = vec.x;

	c = -a * m_firstPos.x - b * m_firstPos.y;

	double k = a * m_mousePos.x + b * m_mousePos.y + c;

	double d2 = k * k / (a * a + b * b);

	/*-- 距離が4より遠ければfalse --*/

	if (d2 > 16)

		return false;

	/*-- 垂線の足がRuler上にあるかを判断。Rulerを対角線とするRectにマウス位置が収まるかどうか --*/

	TRectD rect = TRectD(m_firstPos, m_secondPos).enlarge(4);

	return rect.contains(m_mousePos);

}

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

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

/*! 基準点に対し、マウス位置を0,45,90度にフィットさせた位置を返す

	斜め方向では、X/Yの値のうち絶対値の小さいほうに合わせる。

*/

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

TPointD RulerTool::getHVCoordinatedPos(TPointD p, TPointD centerPos)

{

	TPointD vec = p - centerPos;

	double degree = (vec.x == 0.0) ? 90.0 : atan(vec.y / vec.x) * 180.0 / 3.1415926536;

	TPointD outPoint;

	if (degree <= -67.5) /*--垂直--*/

	{

		outPoint.x = centerPos.x;

		outPoint.y = p.y;

	} else if (degree < -22.5) /*--右斜め下--*/

	{

		if (abs(vec.x) > abs(vec.y))

			outPoint = centerPos + TPointD(-vec.y, vec.y);

		else

			outPoint = centerPos + TPointD(vec.x, -vec.x);

	} else if (degree <= 22.5) /*--水平--*/

	{

		outPoint.x = p.x;

		outPoint.y = centerPos.y;

	} else if (degree < 67.5) /*--右斜め上--*/

	{

		if (abs(vec.x) > abs(vec.y))

			outPoint = centerPos + TPointD(vec.y, vec.y);

		else

			outPoint = centerPos + TPointD(vec.x, vec.x);

	} else /*--再び垂直--*/

	{

		outPoint.x = centerPos.x;

		outPoint.y = p.y;

	}

	return outPoint;

}

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

RulerTool RulerTool;