#ifndef KEYFRAMENAVIGATOR_INCLUDED
#define KEYFRAMENAVIGATOR_INCLUDED

#include "tpalette.h"
#include "tfxutil.h"

#include "toonz/tstageobject.h"
#include "toonz/tframehandle.h"
#include "toonz/tobjecthandle.h"
#include "toonz/txsheethandle.h"
#include "toonz/tpalettehandle.h"
#include "toonz/tfxhandle.h"
#include "toonz/tcolumnfx.h"

#include <QToolBar>

#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

class TStyleSelection;

//=============================================================================
// KeyframeNavigator
//-----------------------------------------------------------------------------

class DVAPI KeyframeNavigator : public QToolBar
{
	Q_OBJECT

	QAction *m_actPreviewKey;
	QAction *m_actKeyNo;
	QAction *m_actKeyPartial;
	QAction *m_actKeyTotal;
	QAction *m_actNextKey;

	TFrameHandle *m_frameHandle;

public:
	KeyframeNavigator(QWidget *parent = 0, TFrameHandle *frameHandle = 0);

	int getCurrentFrame() const
	{
		if (!m_frameHandle)
			return -1;
		return m_frameHandle->getFrameIndex();
	}

public slots:
	void setCurrentFrame(int frame)
	{
		m_frameHandle->setFrameIndex(frame);
		update();
	}

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

protected slots:
	void toggleKeyAct()
	{
		toggle();
		update();
	}
	void toggleNextKeyAct()
	{
		goNext();
		update();
	}
	void togglePrevKeyAct()
	{
		goPrev();
		update();
	}

protected:
	virtual bool hasNext() const = 0;
	virtual bool hasPrev() const = 0;
	virtual bool hasKeyframes() const = 0;
	virtual bool isKeyframe() const = 0;
	virtual bool isFullKeyframe() const = 0;
	virtual void toggle() = 0;
	virtual void goNext() = 0;
	virtual void goPrev() = 0;

	void showEvent(QShowEvent *);
	void hideEvent(QHideEvent *);
};

//=============================================================================
// ViewerKeyframeNavigator
//-----------------------------------------------------------------------------

class DVAPI ViewerKeyframeNavigator : public KeyframeNavigator
{
	Q_OBJECT

	TObjectHandle *m_objectHandle;
	TXsheetHandle *m_xsheetHandle;

public:
	ViewerKeyframeNavigator(QWidget *parent, TFrameHandle *frameHandle = 0)
		: KeyframeNavigator(parent, frameHandle), m_objectHandle(0), m_xsheetHandle(0)
	{
		update();
	}

	TObjectHandle *getObjectHandle() const { return m_objectHandle; }
	TXsheetHandle *getXsheetHandle() const { return m_xsheetHandle; }

public slots:
	void setObjectHandle(TObjectHandle *objectHandle)
	{
		m_objectHandle = objectHandle;
		update();
	}
	void setXsheetHandle(TXsheetHandle *xsheetHandle)
	{
		m_xsheetHandle = xsheetHandle;
		update();
	}

protected:
	TStageObject *getStageObject() const;
	bool hasNext() const;
	bool hasPrev() const;
	bool hasKeyframes() const;
	bool isKeyframe() const;
	bool isFullKeyframe() const;
	void toggle();
	void goNext();
	void goPrev();

	void showEvent(QShowEvent *);
	void hideEvent(QHideEvent *);
};

//=============================================================================
// PaletteKeyframeNavigator
//-----------------------------------------------------------------------------

class DVAPI PaletteKeyframeNavigator : public KeyframeNavigator
{
	Q_OBJECT

	TPaletteHandle *m_paletteHandle;
	TStyleSelection *m_selection;

public:
	PaletteKeyframeNavigator(QWidget *parent, TFrameHandle *frameHandle = 0)
		: KeyframeNavigator(parent, frameHandle), m_paletteHandle(0), m_selection(0)
	{
		update();
	}

	void setSelection(TStyleSelection *selection) { m_selection = selection; }

	TPalette *getPalette() const
	{
		if (!m_paletteHandle)
			return 0;
		return m_paletteHandle->getPalette();
	}
	int getStyleIndex() const
	{
		if (!m_paletteHandle)
			return 0;
		return m_paletteHandle->getStyleIndex();
	}

	TPaletteHandle *getPaletteHandle() const { return m_paletteHandle; }

public slots:
	void setPaletteHandle(TPaletteHandle *paletteHandle)
	{
		m_paletteHandle = paletteHandle;
		update();
	}

protected:
	bool hasNext() const;
	bool hasPrev() const;
	bool hasKeyframes() const;
	bool isKeyframe() const;
	bool isFullKeyframe() const { return isKeyframe(); }
	void toggle();
	void goNext();
	void goPrev();

	void showEvent(QShowEvent *);
	void hideEvent(QHideEvent *);
};

//=============================================================================
// FxKeyframeNavigator
//-----------------------------------------------------------------------------

class DVAPI FxKeyframeNavigator : public KeyframeNavigator
{
	Q_OBJECT

	TFxHandle *m_fxHandle;

public:
	FxKeyframeNavigator(QWidget *parent, TFrameHandle *frameHandle = 0)
		: KeyframeNavigator(parent, frameHandle), m_fxHandle(0)
	{
		update();
	}

	TFx *getFx() const
	{
		if (!m_fxHandle)
			return 0;
		TFx *fx = m_fxHandle->getFx();
		if (TZeraryColumnFx *zfx = dynamic_cast<TZeraryColumnFx *>(fx))
			fx = zfx->getZeraryFx();
		return fx;
	}

	TFxHandle *getFxHandle() const { return m_fxHandle; }

public slots:
	void setFxHandle(TFxHandle *fxHandle)
	{
		m_fxHandle = fxHandle;
		update();
	}

protected:
	bool hasNext() const;
	bool hasPrev() const;
	bool hasKeyframes() const;
	bool isKeyframe() const;
	bool isFullKeyframe() const;
	void toggle();
	void goNext();
	void goPrev();

	void showEvent(QShowEvent *);
	void hideEvent(QHideEvent *);
};

#endif