// TnzCore includes
#include "tgl.h"

// TnzExt includes
#include "ext/meshtexturizer.h"

// tcg includes
#include "tcg/tcg_list.h"

// Boost includes
#include <boost/make_shared.hpp>

// Qt includes
#include <QString>
#include <QCache>
#include <QMutex>
#include <QMutexLocker>

#include "ext/ttexturesstorage.h"

//***************************************************************************************
//    Local namespace - structures
//***************************************************************************************

struct TexturesContainer {
	MeshTexturizer m_texturizer; //!< The mesh texturizer - actual textures container
	tcg::list<QString> m_keys;   //!< Keys in the storage

public:
	TexturesContainer() {}

private:
	TexturesContainer(const TexturesContainer &);
	TexturesContainer &operator=(const TexturesContainer &);
};

//***************************************************************************************
//    Local namespace - variables
//***************************************************************************************

namespace
{

QMutex l_mutex(QMutex::Recursive); // A mutex is needed to synchronize access to the following objects

std::map<int, TexturesContainer *> l_texturesContainers;	 // Texture Containers by display lists space id
QCache<QString, DrawableTextureDataP> l_objects(500 * 1024); // 500 MB cache for now - NOTE: MUST be allocated before the following

} // namespace

//***************************************************************************************
//    Local namespace - global functions
//***************************************************************************************

namespace
{

inline QString textureString(int dlSpaceId, const std::string &texId)
{
	return QString::number(dlSpaceId) + "_" + QString::fromStdString(texId);
}

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

inline void deleteTexturesContainer(const std::pair<int, TexturesContainer *> &pair)
{
	delete pair.second;
}
}

//***************************************************************************************
//    DrawableTextureData implementation
//***************************************************************************************

DrawableTextureData::~DrawableTextureData()
{
	QMutexLocker locker(&l_mutex);

	TexturesContainer *texContainer = l_texturesContainers[m_dlSpaceId];

	if (m_dlSpaceId >= 0) {
		// Load the container's display lists space (remember current OpenGL context, too)
		TGLDisplayListsProxy *proxy = TGLDisplayListsManager::instance()->dlProxy(m_dlSpaceId);

		TGlContext currentContext = tglGetCurrentContext();

		// Unbind the textures
		{
			QMutexLocker locker(proxy->mutex());

			proxy->makeCurrent();
			texContainer->m_texturizer.unbindTexture(m_texId);
		}

		// Restore OpenGL context - equivalent to tglDoneCurrent if currentContext == TGlContext()
		tglMakeCurrent(currentContext);
	} else
		// Temporary - use current OpenGL context directly
		texContainer->m_texturizer.unbindTexture(m_texId);

	texContainer->m_keys.erase(m_objIdx);
}

//***************************************************************************************
//    TTexturesStorage implementation
//***************************************************************************************

TTexturesStorage::TTexturesStorage()
{
	// This singleton is dependent on TGLDisplayListsManager
	TGLDisplayListsManager::instance()->addObserver(this);
}

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

TTexturesStorage::~TTexturesStorage()
{
	l_objects.clear();
	std::for_each(l_texturesContainers.begin(), l_texturesContainers.end(), deleteTexturesContainer);
}

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

TTexturesStorage *TTexturesStorage::instance()
{
	static TTexturesStorage theInstance;
	return &theInstance;
}

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

DrawableTextureDataP TTexturesStorage::loadTexture(
	const std::string &textureId, const TRaster32P &ras, const TRectD &geometry)
{
	// Try to retrieve the proxy associated to current OpenGL context
	TGlContext currentContext = tglGetCurrentContext();
	int dlSpaceId = TGLDisplayListsManager::instance()->displayListsSpaceId(currentContext);

	QString texString(::textureString(dlSpaceId, textureId));

	// Deal with containers
	QMutexLocker locker(&l_mutex);

	// If necessary, allocate a textures container
	std::map<int, TexturesContainer *>::iterator it = l_texturesContainers.find(dlSpaceId);
	if (it == l_texturesContainers.end())
		it = l_texturesContainers.insert(std::make_pair(dlSpaceId, new TexturesContainer)).first;

	MeshTexturizer &texturizer = it->second->m_texturizer;

	DrawableTextureDataP dataPtr = boost::make_shared<DrawableTextureData>();
	DrawableTextureData *data = dataPtr.get();

	data->m_dlSpaceId = dlSpaceId;
	data->m_texId = texturizer.bindTexture(ras, geometry);
	data->m_objIdx = it->second->m_keys.push_back(texString);
	data->m_textureData = texturizer.getTextureData(data->m_texId);

	l_objects.insert(texString, new DrawableTextureDataP(dataPtr),
					 (ras->getLx() * ras->getLy() * ras->getPixelSize()) >> 10);

	if (dlSpaceId < 0) {
		// obj is a temporary. It was pushed in the cache to make space for it - however, it must not be
		// stored. Remove it now.
		l_objects.remove(texString);
	}

	return dataPtr;
}

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

void TTexturesStorage::unloadTexture(const std::string &textureId)
{
	QMutexLocker locker(&l_mutex);

	// Remove the specified texture from ALL the display lists spaces
	std::map<int, TexturesContainer *>::iterator it, iEnd(l_texturesContainers.end());
	for (it = l_texturesContainers.begin(); it != iEnd; ++it)
		l_objects.remove(::textureString(it->first, textureId));
}

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

void TTexturesStorage::onDisplayListDestroyed(int dlSpaceId)
{
	QMutexLocker locker(&l_mutex);

	// Remove the textures container associated with dlSpaceId
	std::map<int, TexturesContainer *>::iterator it = l_texturesContainers.find(dlSpaceId);
	if (it == l_texturesContainers.end())
		return;

	tcg::list<QString>::iterator st, sEnd(it->second->m_keys.end());

	for (st = it->second->m_keys.begin(); st != sEnd;) // Note that the increment is performed BEFORE the texture is removed.
		l_objects.remove(*st++);					   // This is because texture removal may destroy the key being addressed,
													   // whose iterator would then be invalidated.
	delete it->second;
	l_texturesContainers.erase(it);
}

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

DrawableTextureDataP TTexturesStorage::getTextureData(const std::string &textureId)
{
	// Get current display lists space
	TGlContext currentContext = tglGetCurrentContext();
	int dlSpaceId = TGLDisplayListsManager::instance()->displayListsSpaceId(currentContext);

	// If there is no known associated display lists space, the texture cannot be stored.
	if (dlSpaceId < 0)
		return DrawableTextureDataP();

	QMutexLocker locker(&l_mutex);

	// Search the texture object
	QString texString(::textureString(dlSpaceId, textureId));
	if (!l_objects.contains(texString))
		return DrawableTextureDataP();

	return *l_objects.object(texString);
}