1 files changed, 129 insertions, 0 deletions

diff --git a/src/Generating/PieceStructuresGen.cpp b/src/Generating/PieceStructuresGen.cpp
new file mode 100644
index 000000000..4cbf8b032
--- /dev/null
+++ b/src/Generating/PieceStructuresGen.cpp
@@ -0,0 +1,129 @@
+
+// PieceStructuresGen.cpp
+
+// Declares the cPieceStructuresGen class representing the PieceStructures finisher generator
+
+#include "Globals.h"
+#include "PieceStructuresGen.h"
+#include "PrefabStructure.h"
+#include "IniFile.h"
+#include "../Stopwatch.h"
+
+
+
+
+
+
+cPieceStructuresGen::cPieceStructuresGen(int a_Seed):
+	m_Seed(a_Seed)
+{
+}
+
+
+
+
+
+bool cPieceStructuresGen::Initialize(const AString & a_Prefabs, int a_SeaLevel, cBiomeGenPtr a_BiomeGen, cTerrainHeightGenPtr a_HeightGen)
+{
+	// Load each piecepool:
+	auto structures = StringSplitAndTrim(a_Prefabs, "|");
+	for (const auto & s: structures)
+	{
+		auto fileName = Printf("Prefabs%cPieceStructures%c%s.cubeset", cFile::PathSeparator, cFile::PathSeparator, s.c_str());
+		if (!cFile::IsFile(fileName))
+		{
+			fileName.append(".gz");
+			if (!cFile::IsFile(fileName))
+			{
+				LOGWARNING("Cannot load PieceStructures cubeset file %s", fileName.c_str());
+				continue;
+			}
+		}
+		auto gen = std::make_shared<cGen>(m_Seed, a_BiomeGen, a_HeightGen, a_SeaLevel, s);
+		if (gen->LoadFromFile(fileName))
+		{
+			m_Gens.push_back(gen);
+		}
+	}
+
+	// Report a warning if no generators available:
+	if (m_Gens.empty())
+	{
+		LOGWARNING("The PieceStructures generator was asked to generate \"%s\", but none of them are valid.", a_Prefabs.c_str());
+		return false;
+	}
+	return true;
+}
+
+
+
+
+
+void cPieceStructuresGen::GenFinish(cChunkDesc & a_Chunk)
+{
+	for (auto & gen: m_Gens)
+	{
+		gen->GenFinish(a_Chunk);
+	}
+}
+
+
+
+
+
+////////////////////////////////////////////////////////////////////////////////
+// cPieceStructuresGen::cGen:
+
+cPieceStructuresGen::cGen::cGen(int a_Seed, cBiomeGenPtr a_BiomeGen, cTerrainHeightGenPtr a_HeightGen, int a_SeaLevel, const AString & a_Name):
+	Super(a_Seed),
+	m_BiomeGen(a_BiomeGen),
+	m_HeightGen(a_HeightGen),
+	m_SeaLevel(a_SeaLevel),
+	m_Name(a_Name),
+	m_MaxDepth(5)
+{
+}
+
+
+
+
+
+bool cPieceStructuresGen::cGen::LoadFromFile(const AString & a_FileName)
+{
+	// Load the piecepool from the file, log any warnings:
+	if (!m_Pool.LoadFromFile(a_FileName, true))
+	{
+		return false;
+	}
+	if (NoCaseCompare(m_Pool.GetIntendedUse(), "PieceStructures") != 0)
+	{
+		LOGWARNING("PieceStructures generator: File %s is intended for use in \"%s\", rather than piece structures. Loading the file, but the generator may behave unexpectedly.",
+			a_FileName.c_str(), m_Pool.GetIntendedUse().c_str()
+		);
+	}
+	m_Pool.AssignGens(m_Seed, m_BiomeGen, m_HeightGen, m_SeaLevel);
+
+	// Apply generator params from the piecepool (in the metadata) into the generator:
+	auto & generatorParams = m_Pool.GetAllMetadata();
+	SetGeneratorParams(generatorParams);
+	m_MaxDepth = GetStringMapInteger<int>(generatorParams, "MaxDepth", m_MaxDepth);
+
+	return true;
+}
+
+
+
+
+
+cGridStructGen::cStructurePtr cPieceStructuresGen::cGen::CreateStructure(int a_GridX, int a_GridZ, int a_OriginX, int a_OriginZ)
+{
+	cStopwatch sw(Printf("CreateStructure for %s at <%d, %d>", m_Name.c_str(), a_GridX, a_GridZ));
+	cBFSPieceGenerator pg(m_Pool, m_Seed);
+	cPlacedPieces outPieces;
+	pg.PlacePieces(a_OriginX, a_OriginZ, m_MaxDepth, outPieces);
+	return std::make_shared<cPrefabStructure>(a_GridX, a_GridZ, a_OriginX, a_OriginZ, outPieces, m_HeightGen);
+}
+
+
+
+