#ifndef FXNODESELECTION_H
#define FXNODESELECTION_H

#include "toonzqt/selection.h"
#include "tfx.h"
#include "toonz/fxcommand.h"
#include <QList>
#include <QPair>
#include <QMap>
#include <QSet>
#include <set>

//forward declaration
class TXsheetHandle;
class TFxHandle;
class SchematicLink;
class SchematicPort;
class FxSchematicScene;

using namespace TFxCommand;

//=========================================================
//
// FxSelection
//
//---------------------------------------------------------

class FxSelection : public QObject, public TSelection
{

	Q_OBJECT

	QList<Link> m_selectedLinks;
	QList<TFxP> m_selectedFxs;
	QList<int> m_selectedColIndexes;
	TXsheetHandle *m_xshHandle;
	TFxHandle *m_fxHandle;
	TPointD m_pastePosition;

	FxSchematicScene *m_schematicScene;

public:
	FxSelection();
	FxSelection(const FxSelection &src);
	~FxSelection();

	TSelection *clone() const;

	void setPastePosition(const TPointD &pos) { m_pastePosition = pos; }

	void enableCommands();

	//! Return true if the selection is empty
	bool isEmpty() const { return m_selectedFxs.empty() && m_selectedLinks.empty() && m_selectedColIndexes.isEmpty(); }

	//! Empty the selection
	void selectNone()
	{
		m_selectedFxs.clear();
		m_selectedLinks.clear();
		m_selectedColIndexes.clear();
	}

	//! Adds the \b fx to the m_selectedFxs container;
	void select(TFxP fx);
	//! Adds the \b colIndex to the m_selectedColIndexes container;
	void select(int colIndex);
	//! Removes the \b fx from the m_selectedFxs container;
	void unselect(TFxP fx);
	//! Removes the \b colIndex from the m_selectedColIndexes container;
	void unselect(int colIndex);
	//! Adds a QPair<TFxP,TFxP> to the m_selectedLinks container.
	//! The first element of of the QPair is the fx that has the link in input.
	//! The second element of of the QPair is the fx that has the link in output
	void select(SchematicLink *link);
	//! Removes a QPair<TFxP,TFxP> from the m_selectedLinks container.
	void unselect(SchematicLink *link);

	//! Returns true if the given \b fx is selected.
	bool isSelected(TFxP fx) const;
	//! Returns true if the given \b columnIndex is selected.
	bool isSelected(int columnIndex) const;
	//! Returns true if the QPair<TFxP,TFxP> of the fxs that bounds the given \b link is contained in the
	//! m_selectedLinks container.
	bool isSelected(SchematicLink *link);

	const QList<TFxP> &getFxs() const { return m_selectedFxs; }
	const QList<Link> &getLinks() const { return m_selectedLinks; }
	const QList<int> &getColumnIndexes() const { return m_selectedColIndexes; }

	void setXsheetHandle(TXsheetHandle *xshHandle) { m_xshHandle = xshHandle; }
	void setFxHandle(TFxHandle *fxHandle) { m_fxHandle = fxHandle; }
	//! Returns the size of the selection.
	//! The size is the number of fxs and link selected.
	int size() { return m_selectedFxs.size() + m_selectedLinks.size(); }

	//Commands

	//! Calls the TFxCommand::deleteSelection() to remove selected links or fxs.
	void deleteSelection();
	//! Copy selected fxs in the clipboard.
	void copySelection();
	//! Copy selected fxs in the clipboard and alls the TFxCommand::deleteSelection()
	//! to remove selected links or fxs.
	void cutSelection();
	//! Get fxs from the clipboards and calls the TFxCommand::pasteFxs(const QList<TFxP> &fxs, TXsheetHandle*)
	//! to insert fxs in the scene.
	void pasteSelection();

	//! Get fxs from the clipboards and calls the TFxCommand::insertPasteFxs(const QList<TFxP> &fxs, TXsheetHandle*)
	//! to insert fxs in the scene.
	//! Fxs are pasted only if they are connected!
	bool insertPasteSelection();
	//! Get fxs from the clipboards and calls the TFxCommand::addPasteFxs(const QList<TFxP> &fxs, TXsheetHandle*)
	//! to insert fxs in the scene.
	//! Fxs are pasted only if they are connected!
	bool addPasteSelection();
	//! Get fxs from the clipboards and calls the TFxCommand::replacePasteFxs(const QList<TFxP> &fxs, TXsheetHandle*)
	//! to insert fxs in the scene.
	//! Fxs are pasted only if they are connected!
	bool replacePasteSelection();

	void groupSelection();
	void ungroupSelection();
	void collapseSelection();
	void explodeChild();
	Link getBoundingFxs(SchematicLink *link);

	//! Return true if the selection is connected;
	//! A selection is connected if nodes and links selected create a connected graph
	bool isConnected();

	void setFxSchematicScene(FxSchematicScene *schematicScene) { m_schematicScene = schematicScene; }

private:
	// not implemented
	FxSelection &operator=(const FxSelection &);

	Link getBoundingFxs(SchematicPort *inputPort, SchematicPort *outputPort);

	void visitFx(TFx *fx, QList<TFx *> &visitedFxs);
	bool areLinked(TFx *outFx, TFx *inFx);

signals:
	void doCollapse(const QList<TFxP> &);
	void doExplodeChild(const QList<TFxP> &);
};

#endif