#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></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;</skeletonsubtools::magiclink>

	std::set<int> m_temporaryPinnedColumns;</int>

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

	void drawDrawingBrowser(const TXshCell &cell, const TPointD ¢er);

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

	{

		m_temporaryPinnedColumns = tmp;

	}

	bool isGlobalKeyframesEnabled() const;

};

#endif