Something went wrong. Try again.
OpenMW is an open implementation of the Morrowind engine
Something went wrong. Try again.
6.4 kB · 208 lines
C++
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209#include "loadpgrd.hpp"
#include "esmreader.hpp"#include "esmwriter.hpp"
#include <components/misc/concepts.hpp>
namespace ESM{ template <Misc::SameAsWithoutCvref<Pathgrid::DATAstruct> T> void decompose(T&& v, const auto& f) { f(v.mX, v.mY, v.mGranularity, v.mPoints); }
template <Misc::SameAsWithoutCvref<Pathgrid::Point> T> void decompose(T&& v, const auto& f) { char padding[2] = { 0, 0 }; f(v.mX, v.mY, v.mZ, v.mAutogenerated, v.mConnectionNum, padding); }
Pathgrid::Point& Pathgrid::Point::operator=(const float rhs[3]) { mX = static_cast<int32_t>(rhs[0]); mY = static_cast<int32_t>(rhs[1]); mZ = static_cast<int32_t>(rhs[2]); mAutogenerated = 0; mConnectionNum = 0; return *this; } Pathgrid::Point::Point(const float rhs[3]) : mX(static_cast<int32_t>(rhs[0])) , mY(static_cast<int32_t>(rhs[1])) , mZ(static_cast<int32_t>(rhs[2])) , mAutogenerated(0) , mConnectionNum(0) { } Pathgrid::Point::Point() : mX(0) , mY(0) , mZ(0) , mAutogenerated(0) , mConnectionNum(0) { }
void Pathgrid::load(ESMReader& esm, bool& isDeleted) { isDeleted = false;
mPoints.clear(); mEdges.clear();
// keep track of total connections so we can reserve edge vector size size_t edgeCount = 0;
bool hasData = false; while (esm.hasMoreSubs()) { esm.getSubName(); switch (esm.retSubName().toInt()) { case SREC_NAME: mCell = esm.getRefId(); break; case fourCC("DATA"): esm.getSubComposite(mData); hasData = true; break; case fourCC("PGRP"): { esm.getSubHeader(); uint32_t size = esm.getSubSize(); if (size != getCompositeSize(Point{}) * mData.mPoints) esm.fail("Path point subrecord size mismatch"); else { mPoints.reserve(mData.mPoints); for (uint16_t i = 0; i < mData.mPoints; ++i) { Point p; esm.getComposite(p); mPoints.push_back(p); edgeCount += p.mConnectionNum; } } break; } case fourCC("PGRC"): { esm.getSubHeader(); uint32_t size = esm.getSubSize(); if (size % sizeof(uint32_t) != 0) esm.fail("PGRC size not a multiple of 4"); else { size_t rawConnNum = size / sizeof(uint32_t); std::vector<size_t> rawConnections; rawConnections.reserve(rawConnNum); for (size_t i = 0; i < rawConnNum; ++i) { uint32_t currentValue; esm.getT(currentValue); rawConnections.push_back(currentValue); }
auto rawIt = rawConnections.begin(); size_t pointIndex = 0; mEdges.reserve(edgeCount); for (const auto& point : mPoints) { unsigned char connectionNum = point.mConnectionNum; if (rawConnections.end() - rawIt < connectionNum) esm.fail("Not enough connections"); for (int i = 0; i < connectionNum; ++i) { Edge edge; edge.mV0 = pointIndex; edge.mV1 = *rawIt; ++rawIt; mEdges.push_back(edge); } ++pointIndex; } } break; } case SREC_DELE: esm.skipHSub(); isDeleted = true; break; default: esm.fail("Unknown subrecord"); break; } }
if (!hasData) esm.fail("Missing DATA subrecord"); }
void Pathgrid::save(ESMWriter& esm, bool isDeleted) const { // Correct connection count and sort edges by point // Can probably be optimized PointList correctedPoints = mPoints; std::vector<uint32_t> sortedEdges;
sortedEdges.reserve(mEdges.size());
for (size_t point = 0; point < correctedPoints.size(); ++point) { correctedPoints[point].mConnectionNum = 0;
for (const auto& edge : mEdges) { if (edge.mV0 == point) { sortedEdges.push_back(static_cast<uint32_t>(edge.mV1)); ++correctedPoints[point].mConnectionNum; } } }
// Save esm.writeHNCRefId("NAME", mCell); esm.writeNamedComposite("DATA", mData);
if (isDeleted) { esm.writeHNString("DELE", {}, 3); return; }
if (!correctedPoints.empty()) { esm.startSubRecord("PGRP"); for (const Point& point : correctedPoints) { esm.writeComposite(point); } esm.endRecord("PGRP"); }
if (!sortedEdges.empty()) { esm.startSubRecord("PGRC"); for (const uint32_t& edge : sortedEdges) { esm.writeT(edge); } esm.endRecord("PGRC"); } }
void Pathgrid::blank() { mCell = ESM::RefId(); mData.mX = 0; mData.mY = 0; mData.mGranularity = 0; mData.mPoints = 0; mPoints.clear(); mEdges.clear(); }}