khanat-code-old/code/ryzom/client/src/water_map.cpp

354 lines
11 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