#ifndef STROKE_SELECTION_H
#define STROKE_SELECTION_H

#include <memory>

// TnzQt includes
#include "toonzqt/selection.h"

// TnzCore includes
#include "tcommon.h"
#include "tvectorimage.h"

// STD includes
#include <set>

#undef DVAPI
#undef DVVAR
#ifdef TNZTOOLS_EXPORTS
#define DVAPI DV_EXPORT_API
#define DVVAR DV_EXPORT_VAR
#else
#define DVAPI DV_IMPORT_API
#define DVVAR DV_IMPORT_VAR
#endif

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

//    Forward declarations

class TGroupCommand;
class TSceneHandle;

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

//****************************************************************************
//    StrokeSelection  declaration
//****************************************************************************

class DVAPI StrokeSelection : public TSelection
{
public:
	typedef std::set<int> IndexesContainer;

public:
	StrokeSelection();
	~StrokeSelection();

	StrokeSelection(const StrokeSelection &other);
	StrokeSelection &operator=(const StrokeSelection &other);

	TGroupCommand *getGroupCommand() { return m_groupCommand.get(); }

	void setImage(const TVectorImageP &image) { m_vi = image; }
	const TVectorImageP &getImage() const { return m_vi; }

	const IndexesContainer &getSelection() const { return m_indexes; }
	IndexesContainer &getSelection() { return m_indexes; }

	bool isEmpty() const { return m_indexes.empty(); }
	bool updateSelectionBBox() const { return m_updateSelectionBBox; }
	bool isSelected(int index) const { return (m_indexes.count(index) > 0); }
	void select(int index, bool on);
	void toggle(int index);
	void selectNone() { m_indexes.clear(); }

	void setSceneHandle(TSceneHandle *tsh)
	{
		m_sceneHandle = tsh;
	}

public:
	// Commands

	void changeColorStyle(int styleIndex);
	void deleteStrokes();
	void copy();
	void cut();
	void paste();

	void removeEndpoints();

	void enableCommands();

private:
	TVectorImageP m_vi;			//!< Selected vector image.
	IndexesContainer m_indexes; //!< Selected stroke indexes in m_vi.

	std::unique_ptr<TGroupCommand> m_groupCommand; //!< Groups commands wrapper.
	TSceneHandle *m_sceneHandle;					 //!< Global scene handle. \deprecated  Use TApplication instead.

	/*!Set this boolean to true before call tool->notifyImageChanged() when you want to reset strokes bbox.
     N.B. after call tool->notifyImageChanged() set m_updateSelectionBBox to false.*/
	bool m_updateSelectionBBox;
};

#endif // STROKE_SELECTION_H