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

/*! Finger Tool : 線のノイズを埋めるツール

*/

#include "tstroke.h"

#include "tools/toolutils.h"

#include "tools/tool.h"

#include "tmathutil.h"

#include "tools/cursors.h"

#include "drawutil.h"

#include "tcolorstyles.h"

#include "tundo.h"

#include "tvectorimage.h"

#include "ttoonzimage.h"

#include "tproperty.h"

#include "toonz/strokegenerator.h"

#include "toonz/ttilesaver.h"

#include "toonz/txshsimplelevel.h"

#include "toonz/observer.h"

#include "toonz/toonzimageutils.h"

#include "toonz/levelproperties.h"

#include "toonz/stage2.h"

#include "toonz/ttileset.h"

#include "toonz/rasterstrokegenerator.h"

#include "tgl.h"

#include "tenv.h"

#include "trop.h"

#include "tinbetween.h"

#include "ttile.h"

#include "toonz/tpalettehandle.h"

#include "toonz/txsheethandle.h"

#include "toonz/txshlevelhandle.h"

#include "toonz/tframehandle.h"

#include "tools/toolhandle.h"

#include "tools/toolutils.h"

// For Qt translation support

#include <qcoreapplication></qcoreapplication>

#include "tools/stylepicker.h"

#include "toonzqt/tselectionhandle.h"

#include "toonzqt/styleselection.h"

#include "historytypes.h"

using namespace ToolUtils;

TEnv::IntVar FingerInvert("InknpaintFingerInvert", 0);

TEnv::DoubleVar FingerSize("InknpaintFingerSize", 10);

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

const int BackgroundStyle = 0;

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

namespace

{

class FingerUndo : public TRasterUndo

{

	std::vector<tthickpoint> m_points;</tthickpoint>

	int m_styleId;

	bool m_invert;

public:

	FingerUndo(TTileSetCM32 *tileSet,

			   const std::vector<tthickpoint> &points,</tthickpoint>

			   int styleId, bool invert,

			   TXshSimpleLevel *level,

			   const TFrameId &frameId)

		: TRasterUndo(tileSet, level, frameId, false, false, 0), m_points(points), m_styleId(styleId), m_invert(invert)

	{

	}

	void redo() const

	{

		TToonzImageP image = m_level->getFrame(m_frameId, true);

		TRasterCM32P ras = image->getRaster();

		RasterStrokeGenerator m_rasterTrack(ras, FINGER, INK, m_styleId, m_points[0], m_invert, 0, false);

		m_rasterTrack.setPointsSequence(m_points);

		m_rasterTrack.generateStroke(true);

		image->setSavebox(image->getSavebox() + m_rasterTrack.getBBox(m_rasterTrack.getPointsSequence()));

		ToolUtils::updateSaveBox();

		TTool::getApplication()->getCurrentXsheet()->notifyXsheetChanged();

		notifyImageChanged();

	}

	int getSize() const

	{

		return sizeof(*this) + TRasterUndo::getSize();

	}

	virtual QString getToolName()

	{

		return QString("Finger Tool");

	}

	int getHistoryType()

	{

		return HistoryType::FingerTool;

	}

};

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

void drawLine(const TPointD &point, const TPointD ¢re, bool horizontal, bool isDecimal)

{

	if (!isDecimal) {

		if (horizontal) {

			tglDrawSegment(TPointD(point.x - 1.5, point.y + 0.5) + centre, TPointD(point.x - 0.5, point.y + 0.5) + centre);

			tglDrawSegment(TPointD(point.y - 0.5, -point.x + 1.5) + centre, TPointD(point.y - 0.5, -point.x + 0.5) + centre);

			tglDrawSegment(TPointD(-point.x + 0.5, -point.y + 0.5) + centre, TPointD(-point.x - 0.5, -point.y + 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 0.5, point.x - 0.5) + centre, TPointD(-point.y - 0.5, point.x + 0.5) + centre);

			tglDrawSegment(TPointD(point.y - 0.5, point.x + 0.5) + centre, TPointD(point.y - 0.5, point.x - 0.5) + centre);

			tglDrawSegment(TPointD(point.x - 0.5, -point.y + 0.5) + centre, TPointD(point.x - 1.5, -point.y + 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 0.5, -point.x + 0.5) + centre, TPointD(-point.y - 0.5, -point.x + 1.5) + centre);

			tglDrawSegment(TPointD(-point.x - 0.5, point.y + 0.5) + centre, TPointD(-point.x + 0.5, point.y + 0.5) + centre);

		} else {

			tglDrawSegment(TPointD(point.x - 1.5, point.y + 1.5) + centre, TPointD(point.x - 1.5, point.y + 0.5) + centre);

			tglDrawSegment(TPointD(point.x - 1.5, point.y + 0.5) + centre, TPointD(point.x - 0.5, point.y + 0.5) + centre);

			tglDrawSegment(TPointD(point.y + 0.5, -point.x + 1.5) + centre, TPointD(point.y - 0.5, -point.x + 1.5) + centre);

			tglDrawSegment(TPointD(point.y - 0.5, -point.x + 1.5) + centre, TPointD(point.y - 0.5, -point.x + 0.5) + centre);

			tglDrawSegment(TPointD(-point.x + 0.5, -point.y - 0.5) + centre, TPointD(-point.x + 0.5, -point.y + 0.5) + centre);

			tglDrawSegment(TPointD(-point.x + 0.5, -point.y + 0.5) + centre, TPointD(-point.x - 0.5, -point.y + 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 1.5, point.x - 0.5) + centre, TPointD(-point.y - 0.5, point.x - 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 0.5, point.x - 0.5) + centre, TPointD(-point.y - 0.5, point.x + 0.5) + centre);

			tglDrawSegment(TPointD(point.y + 0.5, point.x - 0.5) + centre, TPointD(point.y - 0.5, point.x - 0.5) + centre);

			tglDrawSegment(TPointD(point.y - 0.5, point.x - 0.5) + centre, TPointD(point.y - 0.5, point.x + 0.5) + centre);

			tglDrawSegment(TPointD(point.x - 1.5, -point.y - 0.5) + centre, TPointD(point.x - 1.5, -point.y + 0.5) + centre);

			tglDrawSegment(TPointD(point.x - 1.5, -point.y + 0.5) + centre, TPointD(point.x - 0.5, -point.y + 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 1.5, -point.x + 1.5) + centre, TPointD(-point.y - 0.5, -point.x + 1.5) + centre);

			tglDrawSegment(TPointD(-point.y - 0.5, -point.x + 1.5) + centre, TPointD(-point.y - 0.5, -point.x + 0.5) + centre);

			tglDrawSegment(TPointD(-point.x + 0.5, point.y + 1.5) + centre, TPointD(-point.x + 0.5, point.y + 0.5) + centre);

			tglDrawSegment(TPointD(-point.x + 0.5, point.y + 0.5) + centre, TPointD(-point.x - 0.5, point.y + 0.5) + centre);

		}

	} else {

		if (horizontal) {

			tglDrawSegment(TPointD(point.x - 0.5, point.y + 0.5) + centre, TPointD(point.x + 0.5, point.y + 0.5) + centre);

			tglDrawSegment(TPointD(point.y + 0.5, point.x - 0.5) + centre, TPointD(point.y + 0.5, point.x + 0.5) + centre);

			tglDrawSegment(TPointD(point.y + 0.5, -point.x + 0.5) + centre, TPointD(point.y + 0.5, -point.x - 0.5) + centre);

			tglDrawSegment(TPointD(point.x + 0.5, -point.y - 0.5) + centre, TPointD(point.x - 0.5, -point.y - 0.5) + centre);

			tglDrawSegment(TPointD(-point.x - 0.5, -point.y - 0.5) + centre, TPointD(-point.x + 0.5, -point.y - 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 0.5, -point.x + 0.5) + centre, TPointD(-point.y - 0.5, -point.x - 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 0.5, point.x - 0.5) + centre, TPointD(-point.y - 0.5, point.x + 0.5) + centre);

			tglDrawSegment(TPointD(-point.x + 0.5, point.y + 0.5) + centre, TPointD(-point.x - 0.5, point.y + 0.5) + centre);

		} else {

			tglDrawSegment(TPointD(point.x - 0.5, point.y + 1.5) + centre, TPointD(point.x - 0.5, point.y + 0.5) + centre);

			tglDrawSegment(TPointD(point.x - 0.5, point.y + 0.5) + centre, TPointD(point.x + 0.5, point.y + 0.5) + centre);

			tglDrawSegment(TPointD(point.y + 1.5, point.x - 0.5) + centre, TPointD(point.y + 0.5, point.x - 0.5) + centre);

			tglDrawSegment(TPointD(point.y + 0.5, point.x - 0.5) + centre, TPointD(point.y + 0.5, point.x + 0.5) + centre);

			tglDrawSegment(TPointD(point.y + 1.5, -point.x + 0.5) + centre, TPointD(point.y + 0.5, -point.x + 0.5) + centre);

			tglDrawSegment(TPointD(point.y + 0.5, -point.x + 0.5) + centre, TPointD(point.y + 0.5, -point.x - 0.5) + centre);

			tglDrawSegment(TPointD(point.x - 0.5, -point.y - 1.5) + centre, TPointD(point.x - 0.5, -point.y - 0.5) + centre);

			tglDrawSegment(TPointD(point.x - 0.5, -point.y - 0.5) + centre, TPointD(point.x + 0.5, -point.y - 0.5) + centre);

			tglDrawSegment(TPointD(-point.x + 0.5, -point.y - 1.5) + centre, TPointD(-point.x + 0.5, -point.y - 0.5) + centre);

			tglDrawSegment(TPointD(-point.x + 0.5, -point.y - 0.5) + centre, TPointD(-point.x - 0.5, -point.y - 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 1.5, -point.x + 0.5) + centre, TPointD(-point.y - 0.5, -point.x + 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 0.5, -point.x + 0.5) + centre, TPointD(-point.y - 0.5, -point.x - 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 1.5, point.x - 0.5) + centre, TPointD(-point.y - 0.5, point.x - 0.5) + centre);

			tglDrawSegment(TPointD(-point.y - 0.5, point.x - 0.5) + centre, TPointD(-point.y - 0.5, point.x + 0.5) + centre);

			tglDrawSegment(TPointD(-point.x + 0.5, point.y + 1.5) + centre, TPointD(-point.x + 0.5, point.y + 0.5) + centre);

			tglDrawSegment(TPointD(-point.x + 0.5, point.y + 0.5) + centre, TPointD(-point.x - 0.5, point.y + 0.5) + centre);

		}

	}

}

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

void drawEmptyCircle(int thick, const TPointD &mousePos, bool isPencil, bool isLxEven, bool isLyEven)

{

	TPointD pos = mousePos;

	if (isLxEven)

		pos.x += 0.5;

	if (isLyEven)

		pos.y += 0.5;

	if (!isPencil)

		tglDrawCircle(pos, (thick + 1) * 0.5);

	else {

		int x = 0, y = tround((thick * 0.5) - 0.5);

		int d = 3 - 2 * (int)(thick * 0.5);

		bool horizontal = true, isDecimal = thick % 2 != 0;

		drawLine(TPointD(x, y), pos, horizontal, isDecimal);

		while (y > x) {

			if (d < 0) {

				d = d + 4 * x + 6;

				horizontal = true;

			} else {

				d = d + 4 * (x - y) + 10;

				horizontal = false;

				y--;

			}

			x++;

			drawLine(TPointD(x, y), pos, horizontal, isDecimal);

		}

	}

}

} //namespace

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

class FingerTool : public TTool

{

	Q_DECLARE_TR_FUNCTIONS(FingerTool)

	RasterStrokeGenerator *m_rasterTrack;

	bool m_firstTime;

	double m_pointSize, m_distance2;

	bool m_selecting;

	TTileSaverCM32 *m_tileSaver;

	TPointD m_mousePos;

	TIntProperty m_toolSize;

	TBoolProperty m_invert;

	TPropertyGroup m_prop;

	int m_cursor;

	/*---	作業中のFrameIdをクリック時に保存し、マウスリリース時（Undoの登録時）

			に別のフレームに移動している場合があるため ---*/

	TFrameId m_workingFrameId;

	/*-- 最初のクリックでStyleを切り替える --*/

	void pick(const TPointD &pos);

public:

	FingerTool();

	void draw();

	void update(TToonzImageP ti, TRectD area);

	void updateTranslation();

	void leftButtonDown(const TPointD &pos, const TMouseEvent &e);

	void leftButtonDrag(const TPointD &pos, const TMouseEvent &e);

	void leftButtonUp(const TPointD &pos, const TMouseEvent &);

	void mouseMove(const TPointD &pos, const TMouseEvent &e);

	void onEnter();

	void onLeave();

	void onActivate();

	void onDeactivate();

	bool onPropertyChanged(std::string propertyName);

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

	ToolType getToolType() const { return TTool::LevelWriteTool; }

	int getCursorId() const { return m_cursor; }

	int getColorClass() const { return 2; }

	/*-- ドラッグ中にツールが切り替わった場合に備え、onDeactivateにもMouseReleaseと同じ処理を行う --*/

	void finishBrush();

};

FingerTool fingerTool;

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

//

//  InkPaintTool implemention

//

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

FingerTool::FingerTool()

	: TTool("T_Finger"), m_rasterTrack(0), m_pointSize(-1), m_selecting(false), m_tileSaver(0), m_cursor(ToolCursor::EraserCursor), m_toolSize("Size:", 1, 100, 10, false), m_invert("Invert", false), m_firstTime(true), m_workingFrameId(TFrameId())

{

	bind(TTool::ToonzImage);

	m_prop.bind(m_toolSize);

	m_prop.bind(m_invert);

	m_invert.setId("Invert");

}

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

void FingerTool::updateTranslation()

{

	m_toolSize.setQStringName(tr("Size:"));

	m_invert.setQStringName(tr("Invert", NULL));

}

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

void FingerTool::draw()

{

	if (m_pointSize == -1) {

		return;

	}

	TToonzImageP ti = (TToonzImageP)getImage(false);

	if (!ti)

		return;

	TRasterP ras = ti->getRaster();

	int lx = ras->getLx();

	int ly = ras->getLy();

	if ((ToonzCheck::instance()->getChecks() & ToonzCheck::eInk) || (ToonzCheck::instance()->getChecks() & ToonzCheck::ePaint))

		glColor3d(0.5, 0.8, 0.8);

	else

		glColor3d(1.0, 0.0, 0.0);

	drawEmptyCircle(m_toolSize.getValue(), m_mousePos, true, lx % 2 == 0, ly % 2 == 0);

}

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

const UINT pointCount = 20;

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

bool FingerTool::onPropertyChanged(std::string propertyName)

{

	/*-- サイズ --*/

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

		FingerSize = m_toolSize.getValue();

		double x = m_toolSize.getValue();

		double minRange = 1;

		double maxRange = 100;

		double minSize = 0.01;

		double maxSize = 100;

		m_pointSize = (x - minRange) / (maxRange - minRange) * (maxSize - minSize) + minSize;

		invalidate();

	}

	//Invert

	else if (propertyName == m_invert.getName()) {

		FingerInvert = (int)(m_invert.getValue());

	}

	return true;

}

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

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

{

	pick(pos);

	m_selecting = true;

	TImageP image(getImage(true));

	if (TToonzImageP ti = image) {

		TRasterCM32P ras = ti->getRaster();

		if (ras) {

			int thickness = m_toolSize.getValue();

			int styleId = TTool::getApplication()->getCurrentLevelStyleIndex();

			TTileSetCM32 *tileSet = new TTileSetCM32(ras->getSize());

			m_tileSaver = new TTileSaverCM32(ras, tileSet);

			m_rasterTrack = new RasterStrokeGenerator(ras,

													  FINGER,

													  INK,

													  styleId,

													  TThickPoint(pos + convert(ras->getCenter()), thickness),

													  m_invert.getValue(),

													  0,

													  false);

			/*-- 作業中Fidを現在のFIDにする --*/

			m_workingFrameId = getFrameId();

			m_tileSaver->save(m_rasterTrack->getLastRect());

			TRect modifiedBbox = m_rasterTrack->generateLastPieceOfStroke(true);

			invalidate();

		}

	}

}

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

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

{

	if (!m_selecting)

		return;

	m_mousePos = pos;

	if (TToonzImageP ri = TImageP(getImage(true))) {

		/*---	マウスを動かしながらショートカットで切り替わった場合、

				いきなりleftButtonDragから呼ばれることがあり、

				m_rasterTrackが無くて落ちることがある。 ---*/

		if (m_rasterTrack) {

			int thickness = m_toolSize.getValue();

			bool isAdded = m_rasterTrack->add(TThickPoint(pos + convert(ri->getRaster()->getCenter()), thickness));

			if (isAdded) {

				m_tileSaver->save(m_rasterTrack->getLastRect());

				TRect modifiedBbox = m_rasterTrack->generateLastPieceOfStroke(true);

				invalidate();

			}

		}

	}

}

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

void FingerTool::leftButtonUp(const TPointD &pos, const TMouseEvent &)

{

	if (!m_selecting)

		return;

	m_mousePos = pos;

	finishBrush();

}

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

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

{

	m_mousePos = pos;

	TPointD pp(tround(pos.x), tround(pos.y));

	m_mousePos = pp;

	invalidate();

}

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

void FingerTool::onEnter()

{

	if (m_firstTime) {

		m_invert.setValue(FingerInvert ? 1 : 0);

		m_toolSize.setValue(FingerSize);

		m_firstTime = false;

	}

	double x = m_toolSize.getValue();

	double minRange = 1;

	double maxRange = 100;

	double minSize = 0.01;

	double maxSize = 100;

	m_pointSize = (x - minRange) / (maxRange - minRange) * (maxSize - minSize) + minSize;

	if ((TToonzImageP)getImage(false))

		m_cursor = ToolCursor::PenCursor;

	else

		m_cursor = ToolCursor::CURSOR_NO;

}

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

void FingerTool::onLeave()

{

	m_pointSize = -1;

}

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

void FingerTool::onActivate()

{

	onEnter();

}

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

void FingerTool::onDeactivate()

{

	/*--- マウスドラッグ中(m_selecting=true)にツールが切り替わったときに線を終わらせる ---*/

	if (m_selecting)

		finishBrush();

}

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

/*! ドラッグ中にツールが切り替わった場合に備え、onDeactivateにもMouseReleaseと同じ処理を行う

*/

void FingerTool::finishBrush()

{

	if (TToonzImageP ti = (TToonzImageP)getImage(true)) {

		if (m_rasterTrack) {

			int thickness = m_toolSize.getValue();

			bool isAdded = m_rasterTrack->add(TThickPoint(m_mousePos + convert(ti->getRaster()->getCenter()), thickness));

			if (isAdded) {

				m_tileSaver->save(m_rasterTrack->getLastRect());

				TRect modifiedBbox = m_rasterTrack->generateLastPieceOfStroke(true, true);

			}

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

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

			TXshSimpleLevelP simLevel = level->getSimpleLevel();

			TFrameId frameId = m_workingFrameId.isEmptyFrame() ? getCurrentFid() : m_workingFrameId;

			TUndoManager::manager()->add(new FingerUndo(m_tileSaver->getTileSet(),

														m_rasterTrack->getPointsSequence(),

														m_rasterTrack->getStyleId(),

														m_rasterTrack->isSelective(),

														simLevel.getPointer(),

														frameId));

			ToolUtils::updateSaveBox();

			/*! FIdを指定して、作業中にフレームが動いても、

				クリック時のFidのサムネイルが更新されるようにする。

			*/

			notifyImageChanged(frameId);

			invalidate();

			delete m_rasterTrack;

			m_rasterTrack = 0;

			delete m_tileSaver;

			/*-- 作業中fIdをリセット --*/

			m_workingFrameId = TFrameId();

		}

	}

	m_selecting = false;

}

void FingerTool::pick(const TPointD &pos)

{

	int modeValue = 2; //LINES

	TImageP image = getImage(false);

	TToonzImageP ti = image;

	TVectorImageP vi = image;

	TXshSimpleLevel *level = getApplication()->getCurrentLevel()->getSimpleLevel();

	if (!ti || !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 == 2) //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);

}