#pragma once

#ifndef ICON_GENERATOR_H
#define ICON_GENERATOR_H

// TnzCore includes
#include "tthread.h"
#include "tgeometry.h"
#include "tfilepath.h"
#include "traster.h"
#include "timage.h"
#include "tpixel.h"

// Qt includes
#include <QPixmap>
#include <QThreadStorage>
#include <QEventLoop>

// STD includes
#include <map>
#include <vector>
#include <set>

#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 declarations

class TFilePath;
class TXshLevel;
class TStageObjectSpline;
class ToonzScene;
class TOfflineGL;

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

//**********************************************************************************
//    IconGenerator  definition
//**********************************************************************************

/*!
  \brief    The class responsible for icons management in Toonz.

  \details  It's a singleton class - in particular, rendered icons are stored in
            a shared map container for fast retrieval upon repeated icon requests.

            IconGenerator provides methods to submit icon requests, and to invalidate or
            remove icons from the internal database. In order to keep outer entities
            informed of the icon generation status, an iconGenerated() signal is emitted
            once an icon has been generated.
*/

class DVAPI IconGenerator : public QObject
{
	Q_OBJECT

public:
	class Settings
	{
	public:
		bool m_blackBgCheck;
		bool m_transparencyCheck;
		bool m_inksOnly;
		int m_inkIndex;
		int m_paintIndex;

		Settings()
			: m_transparencyCheck(false), m_blackBgCheck(false), m_inksOnly(false), m_inkIndex(-1), m_paintIndex(-1)
		{
		}
	};

public:
	IconGenerator();
	~IconGenerator();

	static IconGenerator *instance();

	void setSettings(const Settings &settings) { m_settings = settings; }
	const Settings getSettings() const { return m_settings; }
	static void setFilmstripIconSize(const TDimension &dim);

	TDimension getIconSize() const;

	TOfflineGL *getOfflineGLContext();

	// icons from splines
	QPixmap getIcon(TStageObjectSpline *spline);
	void invalidate(TStageObjectSpline *spline);
	void remove(TStageObjectSpline *spline);

	// icons from toonz levels
	QPixmap getIcon(TXshLevel *sl, const TFrameId &fid, bool filmStrip = true, bool onDemand = false);
	void invalidate(TXshLevel *sl, const TFrameId &fid, bool onlyFilmStrip = false);
	void remove(TXshLevel *sl, const TFrameId &fid, bool onlyFilmStrip = false);

	// icons from files
	QPixmap getIcon(const TFilePath &path, const TFrameId &fid = TFrameId::NO_FRAME);
	void invalidate(const TFilePath &path, const TFrameId &fid = TFrameId::NO_FRAME);
	void remove(const TFilePath &path, const TFrameId &fid = TFrameId::NO_FRAME);

	QPixmap getSceneIcon(ToonzScene *scene); //Unused in Toonz
	void invalidateSceneIcon();

	void remap(const std::string &newIconId, const std::string &oldIconId);

	void clearRequests();
	void clearSceneIcons();

	static TRaster32P generateVectorFileIcon(const TFilePath &path, const TDimension &iconSize, const TFrameId &fid);
	static TRaster32P generateRasterFileIcon(const TFilePath &path, const TDimension &iconSize, const TFrameId &fid);
	static TRaster32P generateSceneFileIcon(const TFilePath &path, const TDimension &iconSize, int row);
	static TRaster32P generateSplineFileIcon(const TFilePath &path, const TDimension &iconSize);
	static TRaster32P generateMeshFileIcon(const TFilePath &path, const TDimension &iconSize, const TFrameId &fid);

signals:

	void iconGenerated();

public slots:

	void onStarted(TThread::RunnableP iconRenderer);
	void onCanceled(TThread::RunnableP iconRenderer);
	void onFinished(TThread::RunnableP iconRenderer);
	void onException(TThread::RunnableP iconRenderer);
	void onTerminated(TThread::RunnableP iconRenderer);

private:
	TThread::Executor m_executor;
	QThreadStorage<TOfflineGL *> m_contexts;
	TDimension m_iconSize;

	QEventLoop m_iconsTerminationLoop; //!< Event loop used to wait for icons termination.

	Settings m_settings;

private:
	void addTask(const std::string &id, TThread::RunnableP iconRenderer);
};

//**********************************************************************************
//    Related non-member functions
//**********************************************************************************

template <class It>
inline void invalidateIcons(TXshLevel *sl, It fBegin, It fEnd, bool onlyFilmStrip = false)
{
	for (It ft = fBegin; ft != fEnd; ++ft)
		IconGenerator::instance()->invalidate(sl, *ft, onlyFilmStrip);
}

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

template <class C>
inline void invalidateIcons(TXshLevel *sl, const C &fids, bool onlyFilmStrip = false)
{
	invalidateIcons(sl, fids.begin(), fids.end(), onlyFilmStrip);
}

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

template <typename It>
inline void removeIcons(TXshLevel *sl, It fBegin, It fEnd, bool onlyFilmStrip = false)
{
	for (It ft = fBegin; ft != fEnd; ++ft)
		IconGenerator::instance()->remove(sl, *ft, onlyFilmStrip);
}

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

template <typename C>
inline void removeIcons(TXshLevel *sl, const C &fids, bool onlyFilmStrip = false)
{
	removeIcons(sl, fids.begin(), fids.end(), onlyFilmStrip);
}

#endif // ICON_GENERATOR_H