everything everywhere

all at once

1 files changed, 117 insertions, 0 deletions

diff --git a/source/utility/planet_mapgen.cpp b/source/utility/planet_mapgen.cpp
new file mode 100644
index 0000000..7bc9f8d
--- /dev/null
+++ b/source/utility/planet_mapgen.cpp
@@ -0,0 +1,117 @@
+#include "StarFile.hpp"
+#include "StarLexicalCast.hpp"
+#include "StarImage.hpp"
+#include "StarRootLoader.hpp"
+#include "StarCelestialDatabase.hpp"
+#include "StarWorldTemplate.hpp"
+
+using namespace Star;
+
+int main(int argc, char** argv) {
+  try {
+    RootLoader rootLoader({{}, {}, {}, LogLevel::Error, false, {}});
+
+    rootLoader.setSummary("Generate a WorldTemplate and output the data in it to an image");
+
+    rootLoader.addParameter("coordinate", "coordinate", OptionParser::Optional, "coordinate for the celestial world");
+    rootLoader.addParameter("coordseed", "seed", OptionParser::Optional, "seed to use when selecting a random celestial world coordinate");
+    rootLoader.addParameter("size", "size", OptionParser::Optional, "x,y size of the region to be rendered");
+    rootLoader.addSwitch("weighting", "Output instead the region weighting at each point");
+    rootLoader.addSwitch("weightingblocknoise", "apply layout block noise before outputting weighting");
+    rootLoader.addSwitch("transition", "show biome transition regions");
+
+    RootUPtr root;
+    OptionParser::Options options;
+    tie(root, options) = rootLoader.commandInitOrDie(argc, argv);
+
+    CelestialMasterDatabasePtr celestialDatabase = make_shared<CelestialMasterDatabase>();
+
+    Maybe<CelestialCoordinate> coordinate;
+    if (!options.parameters["coordinate"].empty())
+      coordinate = CelestialCoordinate(options.parameters["coordinate"].first());
+    else if (!options.parameters["coordseed"].empty())
+      coordinate = celestialDatabase->findRandomWorld(
+          10, 50, {}, lexicalCast<uint64_t>(options.parameters["coordseed"].first()));
+    else
+      coordinate = celestialDatabase->findRandomWorld();
+
+    if (!coordinate)
+      throw StarException("Could not find world to generate, try again");
+
+    coutf("Generating world with coordinate %s\n", *coordinate);
+
+    WorldTemplate worldTemplate(*coordinate, celestialDatabase);
+    auto size = worldTemplate.size();
+
+    if (!options.parameters["size"].empty()) {
+      auto regionSize = Vec2U(lexicalCast<unsigned>(options.parameters["size"].first().split(",")[0]),
+          lexicalCast<unsigned>(options.parameters["size"].first().split(",")[1]));
+      size = regionSize.piecewiseClamp(Vec2U(0, 0), size);
+    } else if (size[0] > 1000) {
+      size[0] = 1000;
+    }
+
+    coutf("Generating %s size image for world of type '%s'\n", size, worldTemplate.worldParameters()->typeName);
+    auto outputImage = make_shared<Image>(size, PixelFormat::RGB24);
+
+    Color groundColor = Color::rgb(255, 0, 0);
+    Color caveColor = Color::rgb(128, 0, 0);
+    Color blankColor = Color::rgb(0, 0, 0);
+
+    for (size_t x = 0; x < size[0]; ++x) {
+      for (size_t y = 0; y < size[1]; ++y) {
+        if (options.switches.contains("weighting")) {
+          auto layout = worldTemplate.worldLayout();
+          Color color = Color::Black;
+          Vec2I pos(x, y);
+          if (options.switches.contains("weightingblocknoise")) {
+            if (auto blockNoise = layout->blockNoise())
+              pos = blockNoise->apply(pos, size);
+          }
+          auto weightings = layout->getWeighting(pos[0], pos[1]);
+          for (auto const& weighting : weightings) {
+            Color mixColor = Color::rgb(128, 0, 0);
+            mixColor.setHue(staticRandomFloat((uint64_t)weighting.region));
+            color = Color::rgbaf(color.toRgbaF() + mixColor.toRgbaF() * weighting.weight);
+          }
+          outputImage->set(x, y, color.toRgb());
+        } else if (options.switches.contains("transition")) {
+          auto blockInfo = worldTemplate.blockInfo(x, y);
+          if (isRealMaterial(blockInfo.foreground)) {
+            Color color = groundColor;
+            color.setHue(blockInfo.biomeTransition ? 0 : 0.5f);
+            outputImage->set(x, y, color.toRgb());
+          } else if (isRealMaterial(blockInfo.background)) {
+            Color color = caveColor;
+            color.setHue(blockInfo.biomeTransition ? 0 : 0.5f);
+            outputImage->set(x, y, color.toRgb());
+          } else {
+            outputImage->set(x, y, blankColor.toRgb());
+          }
+        } else {
+          // Image y = 0 is the top, so reverse it for the world tile
+          auto blockInfo = worldTemplate.blockInfo(x, y);
+          if (isRealMaterial(blockInfo.foreground)) {
+            Color color = groundColor;
+            color.setHue(staticRandomFloat(blockInfo.foreground));
+            color.setSaturation(staticRandomFloat(blockInfo.foregroundMod));
+            outputImage->set(x, y, color.toRgb());
+          } else if (isRealMaterial(blockInfo.background)) {
+            Color color = caveColor;
+            color.setHue(staticRandomFloat(blockInfo.background));
+            color.setSaturation(staticRandomFloat(blockInfo.backgroundMod));
+            outputImage->set(x, y, color.toRgb());
+          } else {
+            outputImage->set(x, y, blankColor.toRgb());
+          }
+        }
+      }
+    }
+
+    outputImage->writePng(File::open("mapgen.png", IOMode::Write));
+    return 0;
+  } catch (std::exception const& e) {
+    cerrf("exception caught: %s\n", outputException(e, true));
+    return 1;
+  }
+}