// NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
// Copyright (C) 2010  Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

#include "std3d.h"

#include "nel/3d/water_pool_manager.h"
#include "nel/3d/texture_blend.h"
#include "nel/3d/water_shape.h"
#include "nel/misc/command.h"
#include "nel/3d/water_height_map.h"



namespace NL3D {


/*
NLMISC_COMMAND(setWaterPool, "Setup a pool of water in the water pool manager",
			   "<ID>,<Size>,<Damping>,<FilterWeight>,<UnitSize>,<WaveIntensity>,<WavePeriod>")
{
	CWaterPoolManager::CWaterHeightMapBuild whmb;
	const uint numArgs = args.size();
	if (numArgs == 0) return false;
	if (numArgs == 1)
	{
		NLMISC::fromString(args[0], whmb.ID);
	}
	if (numArgs == 2)
	{
		NLMISC::fromString(args[1], whmb.Size);
	}
	if (numArgs == 3)
	{
		whmb.FilterWeight = ::atof(args[2].c_str());
	}
	if (numArgs == 4)
	{
		whmb.UnitSize = ::atof(args[3].c_str());
	}
	if (numArgs == 5)
	{
		whmb.WaveIntensity = ::atof(args[4].c_str());
	}
	if (numArgs == 4)
	{
		whmb.WavePeriod = ::atof(args[5].c_str());
	}
	// create the water pool
	GetWaterPoolManager().createWaterPool(whmb);
	return true;
}
*/

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

CWaterPoolManager &GetWaterPoolManager()
{
	static CWaterPoolManager singleton;
	return singleton;
}

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

bool	CWaterPoolManager::hasPool(uint32 ID) const
{
	return _PoolMap.count(ID) != 0;
}

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

CWaterHeightMap *CWaterPoolManager::createWaterPool(const CWaterHeightMapBuild &params)
{
	CWaterHeightMap *whm = _PoolMap.count(params.ID) == 0 ? new CWaterHeightMap : _PoolMap[params.ID];
	whm->setDamping(params.Damping);
	whm->setFilterWeight(params.FilterWeight);
	whm->setSize(params.Size);
	whm->setUnitSize(params.UnitSize);
	whm->setWaves(params.WaveIntensity, params.WavePeriod, params.WaveRadius, params.BorderWaves);
	whm->enableWaves(params.WavesEnabled);
	whm->setName(params.Name);
	_PoolMap[params.ID] = whm; // in case it was just created
	return whm;
}

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

CWaterHeightMap &CWaterPoolManager::getPoolByID(uint32 ID)
{
	if(_PoolMap.count(ID))
	{
		return *_PoolMap[ID];
	}
	else
	{
		return *createWaterPool();
	}
}

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

void CWaterPoolManager::reset()
{
	for (TPoolMap::iterator it = _PoolMap.begin(); it != _PoolMap.end(); ++it)
	{
		delete it->second;
	}
	_PoolMap.clear ();
}


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

void CWaterPoolManager::registerWaterShape(CWaterShape *shape)
{
	nlassert(std::find(_WaterShapes.begin(), _WaterShapes.end(), shape) == _WaterShapes.end()); // Shape registered twice!
	_WaterShapes.push_back(shape);
}

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

void CWaterPoolManager::unRegisterWaterShape(CWaterShape *shape)
{
	TWaterShapeVect::iterator it = std::find(_WaterShapes.begin(), _WaterShapes.end(), shape);
//	nlassert(it != _WaterShapes.end()); // shape not registered!
	if (it != _WaterShapes.end())
		_WaterShapes.erase(it);
}

//===============================================================================================
void CWaterPoolManager::setBlendFactor(IDriver *drv, float factor)
{
	nlassert(factor >= 0 && factor <= 1);
	for (TWaterShapeVect::iterator it = _WaterShapes.begin(); it != _WaterShapes.end(); ++it)
	{
		CTextureBlend *tb;
		for (uint k = 0; k < 2; ++k)
		{
			tb = dynamic_cast<CTextureBlend *>((*it)->getEnvMap(k));
			if (tb && tb->setBlendFactor((uint16) (256.f * factor)))
			{
				tb->setReleasable(false);
				drv->setupTexture(*tb);
			}
		}
	}
}

//===============================================================================================
void CWaterPoolManager::releaseBlendTextures()
{
	for (TWaterShapeVect::iterator it = _WaterShapes.begin(); it != _WaterShapes.end(); ++it)
	{
		CTextureBlend *tb;
		for (uint k = 0; k < 2; ++k)
		{
			tb = dynamic_cast<CTextureBlend *>((*it)->getEnvMap(k));
			if (tb)
			{
				tb->setReleasable(true);
				tb->release();
			}
		}
	}
}

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

bool CWaterPoolManager::isWaterShapeObserver(const CWaterShape *shape) const
{
	return std::find(_WaterShapes.begin(), _WaterShapes.end(), shape) != _WaterShapes.end();
}

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

uint		CWaterPoolManager::getNumPools() const
{
	return _PoolMap.size();
}

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

uint		CWaterPoolManager::getPoolID(uint i) const
{
	nlassert(i < getNumPools());
	TPoolMap::const_iterator it =  _PoolMap.begin();
	while (i--) ++it;
	return it->first;
}

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

void	CWaterPoolManager::removePool(uint32 ID)
{
	nlassert(hasPool(ID));
	TPoolMap::iterator it = _PoolMap.find(ID);
	delete it->second;
	_PoolMap.erase(_PoolMap.find(ID));
}

//===============================================================================================
void CWaterPoolManager::serial(NLMISC::IStream &f)  throw(NLMISC::EStream)
{
	f.xmlPush("WaterPoolManager");
	(void)f.serialVersion(0);
	uint32 size;
	TPoolMap::iterator it;
	if (!f.isReading())
	{
		size = _PoolMap.size();
		it  = _PoolMap.begin();
	}
	else
	{
		reset();
	}
	f.xmlSerial(size, "NUM_POOLS");
	while (size --)
	{
		f.xmlPush("PoolDesc");
		if (f.isReading())
		{
			CWaterHeightMap *whm = NULL;
			uint32 id;
			f.xmlSerial(id, "POOL_ID");
			f.serialPtr(whm);
			_PoolMap[id] = whm;
		}
		else
		{
			uint32 id = it->first;
			f.xmlSerial(id, "POOL_ID");
			f.serialPtr(it->second);
			++it;
		}
		f.xmlPop();
	}
	f.xmlPop();
}

} // NL3D