khanat-opennel-code/code/nel/src/3d/water_pool_manager.cpp
2015-01-13 19:47:14 +01:00

253 lines
6.7 KiB
C++

// 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)
{
NLMISC::fromString(args[2], whmb.FilterWeight);
}
if (numArgs == 4)
{
NLMISC::fromString(args[3], whmb.UnitSize);
}
if (numArgs == 5)
{
NLMISC::fromString(args[4], whmb.WaveIntensity);
}
if (numArgs == 4)
{
NLMISC::fromString(args[5], whmb.WavePeriod);
}
// 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 (uint)_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 = (uint32)_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