355 lines
12 KiB
C++
355 lines
12 KiB
C++
|
// Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
|
||
|
// 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 "stdpch.h"
|
||
|
#include "water_map.h"
|
||
|
#include "continent_manager.h"
|
||
|
#include "user_entity.h"
|
||
|
#include "misc.h"
|
||
|
#include "global.h"
|
||
|
#include "client_cfg.h"
|
||
|
//
|
||
|
#include "r2/editor.h"
|
||
|
#include "r2/island_collision.h"
|
||
|
//
|
||
|
#include "nel/3d/u_scene.h"
|
||
|
#include "nel/3d/u_driver.h"
|
||
|
//
|
||
|
#include "nel/misc/path.h"
|
||
|
#include "nel/misc/file.h"
|
||
|
|
||
|
|
||
|
extern NL3D::UScene *Scene;
|
||
|
extern NL3D::UDriver *Driver;
|
||
|
extern CContinentManager ContinentMngr;
|
||
|
|
||
|
bool DisplayWaterMap = false;
|
||
|
|
||
|
|
||
|
using namespace NLMISC;
|
||
|
|
||
|
H_AUTO_DECL(RZ_WaterMap)
|
||
|
|
||
|
// *********************************************************************************************************
|
||
|
CWaterMap::CWaterMap()
|
||
|
{
|
||
|
H_AUTO_USE(RZ_WaterMap)
|
||
|
// Construct
|
||
|
_CellSize = 0.f;
|
||
|
_Width = 0;
|
||
|
_Height = 0;
|
||
|
|
||
|
}
|
||
|
|
||
|
// *********************************************************************************************************
|
||
|
void CWaterMap::init(const NLMISC::CVector2f &minCorner, const NLMISC::CVector2f &maxCorner, float cellSize /*=20.f*/)
|
||
|
{
|
||
|
H_AUTO_USE(RZ_WaterMap)
|
||
|
release();
|
||
|
if (cellSize <= 0.f)
|
||
|
{
|
||
|
nlwarning("Invalid cell size");
|
||
|
return;
|
||
|
}
|
||
|
_MinCorner = minCorner;
|
||
|
_MaxCorner = maxCorner;
|
||
|
if (_MinCorner.y > _MaxCorner.y) std::swap(_MinCorner.y, _MaxCorner.y);
|
||
|
if (_MinCorner.y > _MaxCorner.y) std::swap(_MinCorner.y, _MaxCorner.y);
|
||
|
_CellSize = cellSize;
|
||
|
_Width = (uint) ceilf((_MaxCorner.x - _MinCorner.x) / _CellSize);
|
||
|
_Height = (uint) ceilf((_MaxCorner.y - _MinCorner.y) / _CellSize);
|
||
|
_Grid.resize(_Width * _Height, 0);
|
||
|
// init index 0 as an empty list of water surface
|
||
|
_WaterInfoVect.push_back(CWaterInfo());
|
||
|
_WaterInfoVect.back().Height = 0.f;
|
||
|
_WaterInfoVect.back().SplashEnabled = false;
|
||
|
_WaterInfoIndexVectVect.push_back(TWaterInfoIndexVect(1, 0));
|
||
|
Scene->setWaterCallback(this);
|
||
|
}
|
||
|
|
||
|
// *********************************************************************************************************
|
||
|
void CWaterMap::release()
|
||
|
{
|
||
|
H_AUTO_USE(RZ_WaterMap)
|
||
|
NLMISC::contReset(_Grid);
|
||
|
NLMISC::contReset(_WaterInfoVect);
|
||
|
NLMISC::contReset(_WaterInfoIndexVectVect);
|
||
|
_CellSize = 0.f;
|
||
|
_Width = 0;
|
||
|
_Height = 0;
|
||
|
Scene->setWaterCallback(NULL);
|
||
|
}
|
||
|
|
||
|
// *********************************************************************************************************
|
||
|
bool CWaterMap::getWaterHeight(const NLMISC::CVector2f &pos, float &height, bool &splashEnabled)
|
||
|
{
|
||
|
H_AUTO_USE(RZ_WaterMap)
|
||
|
if (_Grid.empty()) return false;
|
||
|
float x = (pos.x - _MinCorner.x) / _CellSize;
|
||
|
float y = (pos.y - _MinCorner.y) / _CellSize;
|
||
|
sint ix = (sint) x;
|
||
|
sint iy = (sint) y;
|
||
|
if (ix < 0 || ix >= _Width) return false;
|
||
|
if (iy < 0 || iy >= _Height) return false;
|
||
|
const TWaterInfoIndexVect &wiiv = _WaterInfoIndexVectVect[_Grid[ix + iy * _Width]];
|
||
|
if (wiiv.empty()) return false;
|
||
|
// search if there's an intersection with one of the water surfaces
|
||
|
for(uint k = 0; k < wiiv.size(); ++k)
|
||
|
{
|
||
|
const CWaterInfo &wi = _WaterInfoVect[wiiv[k]];
|
||
|
if (wi.Shape.Vertices.empty()) continue;
|
||
|
// test intersection with each surface, and pick the first match
|
||
|
if (wi.Shape.contains(NLMISC::CVector2f(x, y)))
|
||
|
{
|
||
|
height = wi.Height;
|
||
|
splashEnabled = wi.SplashEnabled;
|
||
|
return true;
|
||
|
}
|
||
|
}
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
// *********************************************************************************************************
|
||
|
void CWaterMap::waterSurfaceAdded(const NLMISC::CPolygon2D &shape, const NLMISC::CMatrix &worldMatrix, bool splashEnabled, bool usesSceneWaterEnvMap)
|
||
|
{
|
||
|
if (ClientCfg.R2EDEnabled)
|
||
|
{
|
||
|
R2::getEditor().getIslandCollision().waterSurfaceAdded(shape, worldMatrix);
|
||
|
}
|
||
|
H_AUTO_USE(RZ_WaterMap)
|
||
|
if (usesSceneWaterEnvMap) ++WaterEnvMapRefCount;
|
||
|
if (_Grid.empty()) return;
|
||
|
if (_CellSize == 0.f) return;
|
||
|
float height = worldMatrix.getPos().z;
|
||
|
// transform the water shape in grid coordinates
|
||
|
CWaterInfo wi;
|
||
|
uint numVerts = (uint)shape.Vertices.size();
|
||
|
wi.Shape.Vertices.resize(numVerts);
|
||
|
wi.SplashEnabled = splashEnabled;
|
||
|
NLMISC::CMatrix toGridMatrix;
|
||
|
toGridMatrix.scale(NLMISC::CVector(1.f / _CellSize, 1.f / _CellSize, 1.f));
|
||
|
toGridMatrix.translate(- _MinCorner);
|
||
|
toGridMatrix = toGridMatrix * worldMatrix;
|
||
|
for(uint k = 0; k < numVerts; ++k)
|
||
|
{
|
||
|
wi.Shape.Vertices[k] = toGridMatrix * shape.Vertices[k];
|
||
|
}
|
||
|
// see if water surface already added
|
||
|
for(uint k = 0; k < _WaterInfoVect.size(); ++k)
|
||
|
{
|
||
|
if (_WaterInfoVect[k].Height == height && _WaterInfoVect[k].Shape.Vertices.size() == shape.Vertices.size())
|
||
|
{
|
||
|
if (std::equal(wi.Shape.Vertices.begin(), wi.Shape.Vertices.end(), _WaterInfoVect[k].Shape.Vertices.begin()))
|
||
|
{
|
||
|
// already inserted -> do nothing
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
// insert new CWaterInfo in the list
|
||
|
wi.Height = height;
|
||
|
_WaterInfoVect.push_back(wi);
|
||
|
// build rasters
|
||
|
NLMISC::CPolygon2D::TRasterVect rasters;
|
||
|
sint minY;
|
||
|
//wi.Shape.computeBordersLarge(rasters, minY);
|
||
|
wi.Shape.computeOuterBorders(rasters, minY);
|
||
|
if (!rasters.empty())
|
||
|
{
|
||
|
sint numRasters = (sint) rasters.size();
|
||
|
for(sint y = 0; y < numRasters; ++y)
|
||
|
{
|
||
|
if ((y + minY) < 0 || (y + minY) >= _Height) continue; // outisde of map
|
||
|
sint lastX = rasters[y].second + 1;
|
||
|
for (sint x = rasters[y].first; x < lastX; ++x)
|
||
|
{
|
||
|
if (x < 0 || x >= _Width) continue; // outside of map
|
||
|
const TWaterInfoIndexVect *currWI = &_WaterInfoIndexVectVect[_Grid[x + (y + minY) * _Width]];
|
||
|
// see if there's already a list of water surfaces that match the list for that grid cell
|
||
|
// such a list should contains the surfaces already found for that grid cell + the new one
|
||
|
sint goodList = -1;
|
||
|
for(uint k = 0; k < _WaterInfoIndexVectVect.size(); ++k)
|
||
|
{
|
||
|
if (_WaterInfoIndexVectVect[k].size() == currWI->size() + 1)
|
||
|
{
|
||
|
if (std::equal(_WaterInfoIndexVectVect[k].begin(), _WaterInfoIndexVectVect[k].begin() + currWI->size(), currWI->begin()) &&
|
||
|
_WaterInfoIndexVectVect[k].back() == _WaterInfoVect.size() - 1)
|
||
|
{
|
||
|
// this list match what we want
|
||
|
goodList = k;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
if (goodList == -1)
|
||
|
{
|
||
|
// must create a new list
|
||
|
_WaterInfoIndexVectVect.push_back(TWaterInfoIndexVect());
|
||
|
// vector has grown up, so currWI pointer becomes invalid -> rebuild it
|
||
|
currWI = &_WaterInfoIndexVectVect[_Grid[x + (y + minY) * _Width]];
|
||
|
_WaterInfoIndexVectVect.back().resize(currWI->size() + 1);
|
||
|
std::copy(currWI->begin(), currWI->end(), _WaterInfoIndexVectVect.back().begin());
|
||
|
_WaterInfoIndexVectVect.back().back() = (uint16)_WaterInfoVect.size() - 1;
|
||
|
goodList = (sint)_WaterInfoIndexVectVect.size() - 1;
|
||
|
}
|
||
|
_Grid[x + (y + minY) * _Width] = goodList; // reassign new list
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// *********************************************************************************************************
|
||
|
void CWaterMap::waterSurfaceRemoved(bool usesSceneWaterEnvMap)
|
||
|
{
|
||
|
H_AUTO_USE(RZ_WaterMap)
|
||
|
if (usesSceneWaterEnvMap) --WaterEnvMapRefCount;
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
static const NLMISC::CRGBA DebugCols[] =
|
||
|
{
|
||
|
NLMISC::CRGBA(255, 32, 32),
|
||
|
NLMISC::CRGBA(32, 255, 32),
|
||
|
NLMISC::CRGBA(255, 255, 32),
|
||
|
NLMISC::CRGBA(32, 255, 255),
|
||
|
NLMISC::CRGBA(255, 32, 255),
|
||
|
NLMISC::CRGBA(255, 127, 32),
|
||
|
NLMISC::CRGBA(255, 255, 255)
|
||
|
};
|
||
|
static const uint NumDebugCols = sizeof(DebugCols) / sizeof(DebugCols[0]);
|
||
|
|
||
|
|
||
|
// *********************************************************************************************************
|
||
|
void CWaterMap::dump(const std::string &filename)
|
||
|
{
|
||
|
H_AUTO_USE(RZ_WaterMap)
|
||
|
if (_Grid.empty())
|
||
|
{
|
||
|
nlwarning("Grid not built");
|
||
|
return;
|
||
|
}
|
||
|
NLMISC::CBitmap bm;
|
||
|
bm.resize(_Width, _Height, NLMISC::CBitmap::RGBA);
|
||
|
NLMISC::CRGBA *pix = (NLMISC::CRGBA *) bm.getPixels(0).getPtr();
|
||
|
for(uint x = 0; x < _Width; ++x)
|
||
|
{
|
||
|
for(uint y = 0; y < _Height; ++y)
|
||
|
{
|
||
|
if (_Grid[x + y *_Width] == 0)
|
||
|
{
|
||
|
pix[x + y * _Width] = CRGBA(127, 127, 127);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
pix[x + y * _Width] = DebugCols[_Grid[x + y *_Width] % NumDebugCols];
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
// merge with world map if present
|
||
|
if (ContinentMngr.cur())
|
||
|
{
|
||
|
if (!ContinentMngr.cur()->WorldMap.empty())
|
||
|
{
|
||
|
std::string path = CPath::lookup(ContinentMngr.cur()->WorldMap, false);
|
||
|
if (!path.empty())
|
||
|
{
|
||
|
CIFile stream;
|
||
|
if (stream.open(path))
|
||
|
{
|
||
|
CBitmap worldMap;
|
||
|
if (worldMap.load(stream))
|
||
|
{
|
||
|
worldMap.flipV();
|
||
|
worldMap.convertToType(CBitmap::RGBA);
|
||
|
worldMap.resample(bm.getWidth(), bm.getHeight());
|
||
|
bm.blend(bm, worldMap, 127, true);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
//
|
||
|
drawDisc(bm, ((float) UserEntity->pos().x - _MinCorner.x) / _CellSize, ((float) UserEntity->pos().y - _MinCorner.y) / _CellSize, 2.f, CRGBA::Magenta);
|
||
|
//
|
||
|
NLMISC::COFile f;
|
||
|
if (!f.open(filename))
|
||
|
{
|
||
|
nlwarning("Can't open %s for writing", filename.c_str());
|
||
|
return;
|
||
|
}
|
||
|
bm.writeTGA(f, 24, true);
|
||
|
f.close();
|
||
|
}
|
||
|
|
||
|
// ******************************************************************************************************************
|
||
|
void CWaterMap::render(const NLMISC::CVector2f &camPos, float maxDist /*=100.f*/)
|
||
|
{
|
||
|
H_AUTO_USE(RZ_WaterMap)
|
||
|
if (_Grid.empty()) return;
|
||
|
Driver->setViewMatrix(Scene->getCam().getMatrix().inverted());
|
||
|
NL3D::CFrustum fr;
|
||
|
Scene->getCam().getFrustum(fr.Left, fr.Right, fr.Bottom, fr.Top, fr.Near, fr.Far);
|
||
|
fr.Perspective = true;
|
||
|
Driver->setFrustum(fr);
|
||
|
Driver->setModelMatrix(NLMISC::CMatrix::Identity);
|
||
|
float userZ = UserEntity ? (float) UserEntity->pos().z : 0.f;
|
||
|
for (uint x = 0; x < _Width; ++x)
|
||
|
{
|
||
|
for (uint y = 0; y < _Height; ++y)
|
||
|
{
|
||
|
uint16 index = _Grid[x + _Width * y];
|
||
|
if (index == 0) continue;
|
||
|
const TWaterInfoIndexVect &wii = _WaterInfoIndexVectVect[index];
|
||
|
// see if cell not too far
|
||
|
NLMISC::CVector2f pos(x * _CellSize + _MinCorner.x, y * _CellSize + _MinCorner.y);
|
||
|
if ((camPos - pos).norm() > maxDist) continue; // too far, don't display
|
||
|
// display box for each primitive type
|
||
|
NLMISC::CVector cornerMin(pos.x, pos.y, userZ - 5.f);
|
||
|
NLMISC::CVector cornerMax(pos.x + _CellSize, pos.y + _CellSize, userZ + 5.f);
|
||
|
for(uint l = 0; l < wii.size(); ++l)
|
||
|
{
|
||
|
// add a bias each time to see when several primitives are overlapped
|
||
|
NLMISC::CVector bias = (float) wii[l] * NLMISC::CVector(0.01f, 0.f, 0.1f);
|
||
|
drawBox(cornerMin + bias, cornerMax + bias, DebugCols[wii[l] % NumDebugCols]);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#if !FINAL_VERSION
|
||
|
|
||
|
// ******************************************************************************************************************
|
||
|
// dump water map in a tga file
|
||
|
NLMISC_COMMAND(dumpWaterMap, "dump water map", "<filename>")
|
||
|
{
|
||
|
if (args.size() != 1) return false;
|
||
|
if (!ContinentMngr.cur()) return false;
|
||
|
ContinentMngr.cur()->WaterMap.dump(args[0]);
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
// ******************************************************************************************************************
|
||
|
// display the water map
|
||
|
NLMISC_COMMAND(displayWaterMap, "dump water map", "<0 = on / 1 = off>")
|
||
|
{
|
||
|
if (args.size() != 1) return false;
|
||
|
fromString(args[0], DisplayWaterMap);
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
#endif
|