417 lines
12 KiB
C++
417 lines
12 KiB
C++
#include "precompiled.h"
|
|
|
|
#include "lib/types.h"
|
|
#include "lib/res/file/vfs.h"
|
|
|
|
#include "Camera.h"
|
|
#include "CinemaTrack.h"
|
|
#include "LightEnv.h"
|
|
#include "MapReader.h"
|
|
#include "MapWriter.h"
|
|
#include "Model.h"
|
|
#include "ObjectBase.h"
|
|
#include "ObjectManager.h"
|
|
#include "Patch.h"
|
|
#include "Terrain.h"
|
|
#include "TextureEntry.h"
|
|
#include "TextureManager.h"
|
|
#include "Unit.h"
|
|
#include "UnitManager.h"
|
|
|
|
#include "maths/MathUtil.h"
|
|
#include "maths/NUSpline.h"
|
|
#include "ps/Loader.h"
|
|
#include "ps/Player.h"
|
|
#include "ps/VFSUtil.h"
|
|
#include "ps/XML/XMLWriter.h"
|
|
#include "renderer/SkyManager.h"
|
|
#include "renderer/WaterManager.h"
|
|
#include "simulation/EntityTemplate.h"
|
|
#include "simulation/EntityTemplateCollection.h"
|
|
#include "simulation/Entity.h"
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// CMapWriter constructor: nothing to do at the minute
|
|
CMapWriter::CMapWriter()
|
|
{
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// SaveMap: try to save the current map to the given file
|
|
void CMapWriter::SaveMap(const char* filename, CTerrain* pTerrain,
|
|
CUnitManager* pUnitMan, WaterManager* pWaterMan, SkyManager* pSkyMan,
|
|
CLightEnv* pLightEnv, CCamera* pCamera, CCinemaManager* pCinema)
|
|
{
|
|
CFilePacker packer(FILE_VERSION, "PSMP");
|
|
|
|
// build necessary data
|
|
PackMap(packer, pTerrain);
|
|
|
|
// write it out
|
|
packer.Write(filename);
|
|
|
|
CStr filename_xml (filename);
|
|
filename_xml = filename_xml.Left(filename_xml.Length()-4) + ".xml";
|
|
WriteXML(filename_xml, pUnitMan, pWaterMan, pSkyMan, pLightEnv, pCamera, pCinema);
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// GetHandleIndex: return the index of the given handle in the given list; or 0xffff if
|
|
// handle isn't in list
|
|
static u16 GetHandleIndex(const Handle handle, const std::vector<Handle>& handles)
|
|
{
|
|
const uint limit = MIN((uint)handles.size(), 0xFFFE); // paranoia
|
|
for (uint i=0;i<limit;i++) {
|
|
if (handles[i]==handle) {
|
|
return (u16)i;
|
|
}
|
|
}
|
|
|
|
return 0xffff;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// EnumTerrainTextures: build lists of textures used by map, and tile descriptions for
|
|
// each tile on the terrain
|
|
void CMapWriter::EnumTerrainTextures(CTerrain *pTerrain,
|
|
std::vector<CStr>& textures,
|
|
std::vector<STileDesc>& tiles)
|
|
{
|
|
// the list of all handles in use
|
|
std::vector<Handle> handles;
|
|
|
|
// resize tile array to required size
|
|
tiles.resize(SQR(pTerrain->GetVerticesPerSide()-1));
|
|
STileDesc* tileptr=&tiles[0];
|
|
|
|
// now iterate through all the tiles
|
|
u32 mapsize=pTerrain->GetPatchesPerSide();
|
|
for (u32 j=0;j<mapsize;j++) {
|
|
for (u32 i=0;i<mapsize;i++) {
|
|
for (u32 m=0;m<(u32)PATCH_SIZE;m++) {
|
|
for (u32 k=0;k<(u32)PATCH_SIZE;k++) {
|
|
CMiniPatch& mp=pTerrain->GetPatch(i,j)->m_MiniPatches[m][k];
|
|
u16 index=u16(GetHandleIndex(mp.Tex1,handles));
|
|
if (index==0xffff) {
|
|
index=(u16)handles.size();
|
|
handles.push_back(mp.Tex1);
|
|
}
|
|
|
|
tileptr->m_Tex1Index=index;
|
|
tileptr->m_Tex2Index=0xffff;
|
|
tileptr->m_Priority=mp.Tex1Priority;
|
|
tileptr++;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// now find the texture names for each handle
|
|
for (uint i=0;i<(uint)handles.size();i++) {
|
|
CStr texturename;
|
|
CTextureEntry* texentry=g_TexMan.FindTexture(handles[i]);
|
|
if (!texentry) {
|
|
// uh-oh, this shouldn't happen; set texturename to empty string
|
|
texturename="";
|
|
} else {
|
|
texturename=texentry->GetTag();
|
|
}
|
|
textures.push_back(texturename);
|
|
}
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// PackMap: pack the current world into a raw data stream
|
|
void CMapWriter::PackMap(CFilePacker& packer, CTerrain* pTerrain)
|
|
{
|
|
// now pack everything up
|
|
PackTerrain(packer, pTerrain);
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// PackTerrain: pack the terrain onto the end of the output data stream
|
|
// - data: map size, heightmap, list of textures used by map, texture tile assignments
|
|
void CMapWriter::PackTerrain(CFilePacker& packer, CTerrain* pTerrain)
|
|
{
|
|
// pack map size
|
|
u32 mapsize=pTerrain->GetPatchesPerSide();
|
|
packer.PackRaw(&mapsize,sizeof(mapsize));
|
|
|
|
// pack heightmap
|
|
packer.PackRaw(pTerrain->GetHeightMap(),sizeof(u16)*SQR(pTerrain->GetVerticesPerSide()));
|
|
|
|
// the list of textures used by map
|
|
std::vector<CStr> terrainTextures;
|
|
// descriptions of each tile
|
|
std::vector<STileDesc> tiles;
|
|
|
|
// build lists by scanning through the terrain
|
|
EnumTerrainTextures(pTerrain, terrainTextures, tiles);
|
|
|
|
// pack texture names
|
|
u32 numTextures=(u32)terrainTextures.size();
|
|
packer.PackRaw(&numTextures,sizeof(numTextures));
|
|
for (uint i=0;i<numTextures;i++) {
|
|
packer.PackString(terrainTextures[i]);
|
|
}
|
|
|
|
// pack tile data
|
|
packer.PackRaw(&tiles[0],(u32)(sizeof(STileDesc)*tiles.size()));
|
|
}
|
|
void CMapWriter::WriteXML(const char* filename,
|
|
CUnitManager* pUnitMan, WaterManager* pWaterMan, SkyManager* pSkyMan,
|
|
CLightEnv* pLightEnv, CCamera* pCamera, CCinemaManager* pCinema)
|
|
{
|
|
Handle h = vfs_open(filename, FILE_WRITE_TO_TARGET|FILE_NO_AIO);
|
|
if (h <= 0)
|
|
{
|
|
debug_warn("Failed to open map XML file");
|
|
return;
|
|
}
|
|
|
|
XML_Start("utf-8");
|
|
|
|
{
|
|
XML_Element("Scenario");
|
|
|
|
{
|
|
XML_Element("Environment");
|
|
|
|
XML_Setting("SkySet", pSkyMan->GetSkySet());
|
|
{
|
|
XML_Element("SunColour");
|
|
XML_Attribute("r", pLightEnv->m_SunColor.X); // yes, it's X/Y/Z...
|
|
XML_Attribute("g", pLightEnv->m_SunColor.Y);
|
|
XML_Attribute("b", pLightEnv->m_SunColor.Z);
|
|
}
|
|
{
|
|
XML_Element("SunElevation");
|
|
XML_Attribute("angle", pLightEnv->m_Elevation);
|
|
}
|
|
{
|
|
XML_Element("SunRotation");
|
|
XML_Attribute("angle", pLightEnv->m_Rotation);
|
|
}
|
|
{
|
|
XML_Element("TerrainAmbientColour");
|
|
XML_Attribute("r", pLightEnv->m_TerrainAmbientColor.X);
|
|
XML_Attribute("g", pLightEnv->m_TerrainAmbientColor.Y);
|
|
XML_Attribute("b", pLightEnv->m_TerrainAmbientColor.Z);
|
|
}
|
|
{
|
|
XML_Element("UnitsAmbientColour");
|
|
XML_Attribute("r", pLightEnv->m_UnitsAmbientColor.X);
|
|
XML_Attribute("g", pLightEnv->m_UnitsAmbientColor.Y);
|
|
XML_Attribute("b", pLightEnv->m_UnitsAmbientColor.Z);
|
|
}
|
|
|
|
{
|
|
XML_Element("Water");
|
|
{
|
|
XML_Element("WaterBody");
|
|
XML_Setting("Type", "default");
|
|
{
|
|
XML_Element("Colour");
|
|
XML_Attribute("r", pWaterMan->m_WaterColor.r);
|
|
XML_Attribute("g", pWaterMan->m_WaterColor.g);
|
|
XML_Attribute("b", pWaterMan->m_WaterColor.b);
|
|
}
|
|
XML_Setting("Height", pWaterMan->m_WaterHeight);
|
|
XML_Setting("Shininess", pWaterMan->m_Shininess);
|
|
XML_Setting("Waviness", pWaterMan->m_Waviness);
|
|
}
|
|
}
|
|
}
|
|
|
|
{
|
|
XML_Element("Camera");
|
|
|
|
{
|
|
XML_Element("Position");
|
|
CVector3D pos = pCamera->m_Orientation.GetTranslation();
|
|
XML_Attribute("x", pos.X);
|
|
XML_Attribute("y", pos.Y);
|
|
XML_Attribute("z", pos.Z);
|
|
}
|
|
|
|
CVector3D in = pCamera->m_Orientation.GetIn();
|
|
// Convert to spherical coordinates
|
|
float rotation = atan2(in.X, in.Z);
|
|
float declination = atan2(sqrt(in.X*in.X + in.Z*in.Z), in.Y) - M_PI_2;
|
|
|
|
{
|
|
XML_Element("Rotation");
|
|
XML_Attribute("angle", rotation);
|
|
}
|
|
{
|
|
XML_Element("Declination");
|
|
XML_Attribute("angle", declination);
|
|
}
|
|
}
|
|
|
|
{
|
|
XML_Element("Entities");
|
|
|
|
const std::vector<CUnit*>& units = pUnitMan->GetUnits();
|
|
for (std::vector<CUnit*>::const_iterator unit = units.begin(); unit != units.end(); ++unit)
|
|
{
|
|
|
|
CEntity* entity = (*unit)->GetEntity();
|
|
|
|
// Ignore objects that aren't entities
|
|
if (! entity)
|
|
continue;
|
|
|
|
XML_Element("Entity");
|
|
|
|
XML_Setting("Template", entity->m_base->m_Tag);
|
|
|
|
XML_Setting("Player", entity->GetPlayer()->GetPlayerID());
|
|
|
|
{
|
|
CVector3D position = entity->m_position;
|
|
|
|
XML_Element("Position");
|
|
XML_Attribute("x", position.X);
|
|
XML_Attribute("y", position.Y);
|
|
XML_Attribute("z", position.Z);
|
|
}
|
|
|
|
{
|
|
float angle = entity->m_orientation.Y;
|
|
|
|
XML_Element("Orientation");
|
|
XML_Attribute("angle", angle);
|
|
}
|
|
}
|
|
}
|
|
{
|
|
XML_Element("Nonentities");
|
|
|
|
const std::vector<CUnit*>& units = pUnitMan->GetUnits();
|
|
for (std::vector<CUnit*>::const_iterator unit = units.begin(); unit != units.end(); ++unit) {
|
|
|
|
// Ignore objects that are entities
|
|
if ((*unit)->GetEntity())
|
|
continue;
|
|
|
|
XML_Element("Nonentity");
|
|
|
|
XML_Setting("Actor", (*unit)->GetObject()->m_Base->m_Name);
|
|
|
|
{
|
|
CVector3D position = (*unit)->GetModel()->GetTransform().GetTranslation();
|
|
|
|
XML_Element("Position");
|
|
XML_Attribute("x", position.X);
|
|
XML_Attribute("y", position.Y);
|
|
XML_Attribute("z", position.Z);
|
|
}
|
|
|
|
{
|
|
CVector3D orient = (*unit)->GetModel()->GetTransform().GetIn();
|
|
float angle = atan2(-orient.X, -orient.Z);
|
|
|
|
XML_Element("Orientation");
|
|
XML_Attribute("angle", angle);
|
|
}
|
|
}
|
|
}
|
|
const std::map<CStrW, CCinemaTrack>& tracks = pCinema->GetAllTracks();
|
|
std::map<CStrW, CCinemaTrack>::const_iterator it = tracks.begin();
|
|
|
|
{
|
|
XML_Element("Tracks");
|
|
|
|
for ( ; it != tracks.end(); it++ )
|
|
{
|
|
const std::vector<CCinemaPath>& paths = it->second.GetAllPaths();
|
|
CStrW name = it->first;
|
|
size_t numPaths = paths.size();
|
|
CVector3D startRotation = it->second.GetRotation();
|
|
float timescale = it->second.GetTimescale();
|
|
|
|
{
|
|
XML_Element("Track");
|
|
XML_Attribute("name", name);
|
|
XML_Attribute("timescale", timescale);
|
|
|
|
{
|
|
XML_Element("StartRotation");
|
|
XML_Attribute("x", startRotation.X);
|
|
XML_Attribute("y", startRotation.Y);
|
|
XML_Attribute("z", startRotation.Z);
|
|
}
|
|
|
|
for ( size_t i=0; i<numPaths; ++i )
|
|
{
|
|
const std::vector<SplineData>& nodes =
|
|
paths[i].GetAllNodes();
|
|
const CCinemaData* data = paths[i].GetData();
|
|
{
|
|
XML_Element("Path");
|
|
|
|
{
|
|
CVector3D rot = data->m_TotalRotation;
|
|
XML_Element("Rotation");
|
|
XML_Attribute("x", rot.X);
|
|
XML_Attribute("y", rot.Y);
|
|
XML_Attribute("z", rot.Z);
|
|
}
|
|
|
|
{
|
|
XML_Element("Distortion");
|
|
XML_Attribute("mode", data->m_Mode);
|
|
XML_Attribute("style", data->m_Style);
|
|
XML_Attribute("growth", data->m_Growth);
|
|
XML_Attribute("switch", data->m_Switch);
|
|
}
|
|
|
|
for ( int j=(int)nodes.size()-1; j >= 0; --j )
|
|
{
|
|
{
|
|
XML_Element("Node");
|
|
XML_Attribute("x", nodes[j].Position.X);
|
|
XML_Attribute("y", nodes[j].Position.Y);
|
|
XML_Attribute("z", nodes[j].Position.Z);
|
|
XML_Attribute("t", nodes[j].Distance);
|
|
}
|
|
}
|
|
} //path data
|
|
}
|
|
} //track data
|
|
}
|
|
} //(Did this really happen) - I blame XML
|
|
}
|
|
|
|
if (! XML_StoreVFS(h))
|
|
{
|
|
debug_warn("Failed to write map XML file");
|
|
}
|
|
(void)vfs_close(h);
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
|
// RewriteAllMaps
|
|
void CMapWriter::RewriteAllMaps(CTerrain* pTerrain, CUnitManager* pUnitMan,
|
|
WaterManager* pWaterMan, SkyManager* pSkyMan,
|
|
CLightEnv* pLightEnv, CCamera* pCamera, CCinemaManager* pCinema)
|
|
{
|
|
VFSUtil::FileList files;
|
|
VFSUtil::FindFiles("maps/scenarios", "*.pmp", files);
|
|
|
|
for (VFSUtil::FileList::iterator it = files.begin(); it != files.end(); ++it)
|
|
{
|
|
CMapReader* reader = new CMapReader;
|
|
LDR_BeginRegistering();
|
|
reader->LoadMap(*it, pTerrain, pUnitMan, pWaterMan, pSkyMan, pLightEnv, pCamera, pCinema);
|
|
LDR_EndRegistering();
|
|
LDR_NonprogressiveLoad();
|
|
|
|
CStr n (*it);
|
|
n.Replace("scenarios/", "scenarios/new/");
|
|
CMapWriter writer;
|
|
writer.SaveMap(n, pTerrain, pUnitMan, pWaterMan, pSkyMan, pLightEnv, pCamera, pCinema);
|
|
}
|
|
}
|