#pragma once

#ifndef FUNCTIONSELECTION_H
#define FUNCTIONSELECTION_H

#include "tcommon.h"
#include "functiontreeviewer.h"
#include "toonzqt/selection.h"
#include "toonzqt/dvmimedata.h"

#include "tdoublekeyframe.h"

#include <QWidget>
#include <QList>
#include <QSet>

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

// forward declaration
class TDoubleParam;
class TFrameHandle;

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

class ColumnToCurveMapper
{
public:
	virtual TDoubleParam *getCurve(int columnIndex) const = 0;
};

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

class FunctionSelection : public QObject, public TSelection
{
	Q_OBJECT
	QRect m_selectedCells; // yrange = rowrange of the selected keyframes;
						   // xrange = columnrange (functionsheet only)
	QList<QPair<TDoubleParam *, QSet<int>>> m_selectedKeyframes;
	// first = curve, second = set of selected keyframes index

	int m_selectedSegment; // index of the first keyframe of the segment; -1 if no segment selected
						   // (functionpanel only)
						   // assert(m_selectedSegment<0 || m_selectedKeyframes.size()==1)

	TFrameHandle *m_frameHandle;
	ColumnToCurveMapper *m_columnToCurveMapper;

	int getCurveIndex(TDoubleParam *curve) const;
	// finds i : m_selectedKeyframes[i].first == curve
	//-1 if curve not found
	int touchCurveIndex(TDoubleParam *curve);
	// as getCurve(); if curve not found then add it

public:
	FunctionSelection();
	~FunctionSelection();

	void setFrameHandle(TFrameHandle *frameHandle) { m_frameHandle = frameHandle; }

	// function graph
	void selectCurve(TDoubleParam *curve);
	void deselectAllKeyframes();

	// function sheet
	QRect getSelectedCells() const { return m_selectedCells; }
	void selectCells(const QRect &selectedCells, const QList<TDoubleParam *> &curves);
	void selectCells(const QRect &selectedCells);
	void deselectAllCells();

	bool isEmpty() const { return m_selectedKeyframes.empty(); }
	void selectNone();
	void select(TDoubleParam *curve, int k);
	bool isSelected(TDoubleParam *curve, int k) const;
	void selectSegment(TDoubleParam *, int k, QRect selectedCells = QRect()); // note: if a segment is selected then also the segment ends are selected
	int getSelectedKeyframeCount() const;
	QPair<TDoubleParam *, int> getSelectedKeyframe(int index) const; // if index<0 || index>=getSelectedKeyframeCount() returns (0,-1)

	QPair<TDoubleParam *, int> getSelectedSegment() const; // if no segment is selected returns (0,-1)
	bool isSegmentSelected(TDoubleParam *, int k) const;

	void setColumnToCurveMapper(ColumnToCurveMapper *mapper); // gets ownership

	TDoubleParam *getCurveFromColumn(int columnIndex) const
	{
		return m_columnToCurveMapper ? m_columnToCurveMapper->getCurve(columnIndex) : 0;
	}

	void enableCommands();

	void doCopy();
	void doPaste();
	void doCut();
	void doDelete();
	void insertCells();

signals:
	void selectionChanged();
};

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

class FunctionKeyframesData : public DvMimeData
{
public:
	FunctionKeyframesData();
	~FunctionKeyframesData();

	typedef std::vector<TDoubleKeyframe> Keyframes;

	void setColumnCount(int columnCount);
	int getColumnCount() const { return (int)m_keyframes.size(); }

	int getRowCount() const;

	void getData(int columnIndex, TDoubleParam *curve, double frame, const QSet<int> &kIndices);
	void setData(int columnIndex, TDoubleParam *curve, double frame) const;

	const Keyframes &getKeyframes(int columnIndex) const;

	DvMimeData *clone() const;

	bool isCircularReferenceFree(int columnIndex, TDoubleParam *curve) const;

private:
	std::vector<Keyframes> m_keyframes;
};

#endif