#pragma once

#ifndef SKELETON_TOOL_INCLUDED
#define SKELETON_TOOL_INCLUDED

#include "tools/tool.h"
#include "tgeometry.h"
#include "tproperty.h"
#include "toonz/hook.h"
#include "toonz/skeleton.h"

#include <QCoreApplication>

namespace SkeletonSubtools
{

class DragTool;
class CommandHandler;

class HookData
{
public:
	int m_columnIndex;
	int m_hookId; // 0=center (e.g. B,C,...), 1,2... = hook
	TPointD m_pos;
	std::string m_name; // name visualized in the balloons
	bool m_isPivot;		// true if this is the level "pivot", i.e. the current handle
	HookData(TXsheet *xsh, int columnIndex, int hookId, const TPointD &pos);
	std::string getHandle() const { return m_hookId > 0 ? "H" + m_name : m_name; }
};

class MagicLink
{
public:
	HookData m_h0, m_h1;
	double m_dist2;
	MagicLink(const HookData &h0, const HookData &h1, double dist2) : m_h0(h0), m_h1(h1), m_dist2(dist2) {}
	bool operator<(const MagicLink &link) const { return m_dist2 < link.m_dist2; }
};
}

class SkeletonTool : public TTool
{

	Q_DECLARE_TR_FUNCTIONS(SkeletonTool)

	SkeletonSubtools::DragTool *m_dragTool;

	TPointD m_lastPos;
	TPointD m_curPos;
	TPointD m_firstPos;
	TPointD m_curCenter;
	TPointD m_parentProbe;
	bool m_parentProbeEnabled;

	bool m_active;
	bool m_firstTime;
	int m_device;

	TEnumProperty m_mode;
	TBoolProperty m_showOnlyActiveSkeleton;
	TBoolProperty m_globalKeyframes;

	TPropertyGroup m_prop;

	std::vector<SkeletonSubtools::MagicLink> m_magicLinks;

	std::set<int> m_temporaryPinnedColumns;

	int m_currentFrame;

	SkeletonSubtools::CommandHandler *m_commandHandler;

	// column currently selected, during a click&drag operation
	int m_otherColumn;
	TRectD m_otherColumnBBox;
	TAffine m_otherColumnBBoxAff;

	TPointD m_labelPos;
	std::string m_label;

public:
	SkeletonTool();
	~SkeletonTool();

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

	void updateTranslation(); // QString localization stuff

	bool doesApply() const; // ritorna vero se posso deformare l'oggetto corrente

	void onEnter() {}
	void leftButtonDown(const TPointD &pos, const TMouseEvent &);
	void leftButtonDrag(const TPointD &pos, const TMouseEvent &);
	void leftButtonUp(const TPointD &pos, const TMouseEvent &);
	void mouseMove(const TPointD &, const TMouseEvent &e);

	void onImageChanged() { invalidate(); }

	void reset()
	{
		m_temporaryPinnedColumns.clear();
	}

	bool onPropertyChanged(std::string propertyName);

	void draw();

	void drawSkeleton(const Skeleton &skeleton, int row);
	void drawLevelBoundingBox(int frame, int columnIndex);
	void getImageBoundingBox(TRectD &bbox, TAffine &aff, int frame, int columnIndex);
	void drawIKJoint(const Skeleton::Bone *bone);
	void drawJoint(const TPointD &p, bool current);
	void drawBone(const TPointD &a, const TPointD &b, bool selected);
	void drawIKBone(const TPointD &a, const TPointD &b);
	void drawMainGadget(const TPointD &center);
	void drawDrawingBrowser(const TXshCell &cell, const TPointD &center);

	void setParentProbe(const TPointD &parentProbe)
	{
		m_parentProbe = parentProbe;
		m_parentProbeEnabled = true;
	}
	void resetParentProbe()
	{
		m_parentProbeEnabled = false;
	}

	void drawHooks();

	void computeMagicLinks();

	bool keyDown(int key, TUINT32 flags, const TPoint &pos);

	void onActivate();
	void onDeactivate();

	int getMagicLinkCount() const { return (int)m_magicLinks.size(); }
	SkeletonSubtools::MagicLink getMagicLink(int index) const;

	void magicLink(int index);

	int getCursorId() const;

	//TRaster32P getToolIcon() const {return Pixmaps::arrow;}
	TPropertyGroup *getProperties(int targetType) { return &m_prop; }

	void updateMatrix()
	{
		setMatrix(getCurrentObjectParentMatrix());
	}
	void addContextMenuItems(QMenu *menu);
	bool select(const TSelection *) { return false; }

	void togglePinnedStatus(int columnIndex, int frame, bool shiftPressed);

	void buildSkeleton(Skeleton &skeleton, int columnIndex);

	void setTemporaryPinnedColumns(const std::set<int> &tmp)
	{
		m_temporaryPinnedColumns = tmp;
	}
	bool isGlobalKeyframesEnabled() const;
};

#endif