diff options
Diffstat (limited to 'Tools/GrownBiomeGenVisualiser/GrownBiomeGenVisualiser.cpp')
| -rw-r--r-- | Tools/GrownBiomeGenVisualiser/GrownBiomeGenVisualiser.cpp | 456 |
1 files changed, 456 insertions, 0 deletions
diff --git a/Tools/GrownBiomeGenVisualiser/GrownBiomeGenVisualiser.cpp b/Tools/GrownBiomeGenVisualiser/GrownBiomeGenVisualiser.cpp new file mode 100644 index 000000000..531e3edd9 --- /dev/null +++ b/Tools/GrownBiomeGenVisualiser/GrownBiomeGenVisualiser.cpp @@ -0,0 +1,456 @@ + +// GrownBiomeGenVisualiser.cpp + +// Implements the main app entrypoint + +#include "Globals.h" +#include <iostream> +#include <fstream> +#include <random> +#define PROT_INT_BUFFER_SIZE (130 * 130) +#include "Generating/ProtIntGen.h" + + + + + +typedef int Color[3]; // Color is an array of 3 ints + + + + +// Forward declarations, needed for GCC and Clang: +void log(const char * a_Fmt, ...) FORMATSTRING(1, 2); +void outputBitmapFile( + const AString & a_FileName, + unsigned a_ImageSizeX, unsigned a_ImageSizeY, + const int * a_ColorIndices, + unsigned a_IndicesSizeX, unsigned a_IndicesSizeY, + const Color * a_Colors, + size_t a_NumColors +); +void initializeBiomeColors(void); +void generateZoomLevels(int a_Seed); +void generateSmoothLevels(int a_Seed); +void generateExamples(int a_Seed); + + + + + + +/** Color palette used for algorithm examples. +No relevance to biomes whatsoever. */ +static const Color spectrumColors[] = +{ + {0, 0, 0}, + {255, 0, 0}, + {0, 255, 0}, + {0, 0, 255}, + {255, 255, 0}, + {255, 0, 255}, + {0, 255, 255}, +}; + + + + + +/** Color palette used for displaying biome groups. */ +static const Color biomeGroupColors[] = +{ + /* bgOcean */ {0x00, 0x00, 0x70}, + /* bgDesert */ {0xfa, 0x94, 0x18}, + /* bgTemperate */ {0x05, 0x66, 0x21}, + /* bgMountains */ {0x60, 0x60, 0x60}, + /* bgIce */ {0xa0, 0xa0, 0xff}, +}; + + + + + +/** Color palette used for outputting biome images. +Initialized from biomeColorMap[] in initializeBiomeColors(). */ +static Color biomeColors[255]; + + + + +/** Map of biome -> color, used to initialize biomeColorMap[]. */ +static const struct +{ + EMCSBiome biome; + Color color; +} +biomeColorMap[] = +{ + { biOcean, { 0x00, 0x00, 0x70 }, }, + { biPlains, { 0x8d, 0xb3, 0x60 }, }, + { biDesert, { 0xfa, 0x94, 0x18 }, }, + { biExtremeHills, { 0x60, 0x60, 0x60 }, }, + { biForest, { 0x05, 0x66, 0x21 }, }, + { biTaiga, { 0x0b, 0x66, 0x59 }, }, + { biSwampland, { 0x2f, 0xff, 0xda }, }, + { biRiver, { 0x30, 0x30, 0xaf }, }, + { biHell, { 0x7f, 0x00, 0x00 }, }, + { biSky, { 0x00, 0x7f, 0xff }, }, + { biFrozenOcean, { 0xa0, 0xa0, 0xdf }, }, + { biFrozenRiver, { 0xa0, 0xa0, 0xff }, }, + { biIcePlains, { 0xff, 0xff, 0xff }, }, + { biIceMountains, { 0xa0, 0xa0, 0xa0 }, }, + { biMushroomIsland, { 0xff, 0x00, 0xff }, }, + { biMushroomShore, { 0xa0, 0x00, 0xff }, }, + { biBeach, { 0xfa, 0xde, 0x55 }, }, + { biDesertHills, { 0xd2, 0x5f, 0x12 }, }, + { biForestHills, { 0x22, 0x55, 0x1c }, }, + { biTaigaHills, { 0x16, 0x39, 0x33 }, }, + { biExtremeHillsEdge, { 0x7f, 0x8f, 0x7f }, }, + { biJungle, { 0x53, 0x7b, 0x09 }, }, + { biJungleHills, { 0x2c, 0x42, 0x05 }, }, + + { biJungleEdge, { 0x62, 0x8b, 0x17 }, }, + { biDeepOcean, { 0x00, 0x00, 0x30 }, }, + { biStoneBeach, { 0xa2, 0xa2, 0x84 }, }, + { biColdBeach, { 0xfa, 0xf0, 0xc0 }, }, + { biBirchForest, { 0x30, 0x74, 0x44 }, }, + { biBirchForestHills, { 0x1f, 0x5f, 0x32 }, }, + { biRoofedForest, { 0x40, 0x51, 0x1a }, }, + { biColdTaiga, { 0x31, 0x55, 0x4a }, }, + { biColdTaigaHills, { 0x59, 0x7d, 0x72 }, }, + { biMegaTaiga, { 0x59, 0x66, 0x51 }, }, + { biMegaTaigaHills, { 0x59, 0x66, 0x59 }, }, + { biExtremeHillsPlus, { 0x50, 0x70, 0x50 }, }, + { biSavanna, { 0xbd, 0xb2, 0x5f }, }, + { biSavannaPlateau, { 0xa7, 0x9d, 0x64 }, }, + { biMesa, { 0xd9, 0x45, 0x15 }, }, + { biMesaPlateauF, { 0xb0, 0x97, 0x65 }, }, + { biMesaPlateau, { 0xca, 0x8c, 0x65 }, }, + + // M variants: + { biSunflowerPlains, { 0xb5, 0xdb, 0x88 }, }, + { biDesertM, { 0xff, 0xbc, 0x40 }, }, + { biExtremeHillsM, { 0x88, 0x88, 0x88 }, }, + { biFlowerForest, { 0x2d, 0x8e, 0x49 }, }, + { biTaigaM, { 0x33, 0x8e, 0x81 }, }, + { biSwamplandM, { 0x07, 0xf9, 0xb2 }, }, + { biIcePlainsSpikes, { 0xb4, 0xdc, 0xdc }, }, + { biJungleM, { 0x7b, 0xa3, 0x31 }, }, + { biJungleEdgeM, { 0x62, 0x8b, 0x17 }, }, + { biBirchForestM, { 0x58, 0x9c, 0x6c }, }, + { biBirchForestHillsM, { 0x47, 0x87, 0x5a }, }, + { biRoofedForestM, { 0x68, 0x79, 0x42 }, }, + { biColdTaigaM, { 0x24, 0x3f, 0x36 }, }, + { biMegaSpruceTaiga, { 0x45, 0x4f, 0x3e }, }, + { biMegaSpruceTaigaHills, { 0x45, 0x4f, 0x4e }, }, + { biExtremeHillsPlusM, { 0x78, 0x98, 0x78 }, }, + { biSavannaM, { 0xe5, 0xda, 0x87 }, }, + { biSavannaPlateauM, { 0xa7, 0x9d, 0x74 }, }, + { biMesaBryce, { 0xff, 0x6d, 0x3d }, }, + { biMesaPlateauFM, { 0xd8, 0xbf, 0x8d }, }, + { biMesaPlateauM, { 0xf2, 0xb4, 0x8d }, }, +}; + + + + + +void log(const char * a_Fmt, ...) +{ + AString buf; + va_list args; + va_start(args, a_Fmt); + AppendVPrintf(buf, a_Fmt, args); + va_end(args); + std::cout << buf << std::endl << std::flush; +} + + + + + +void outputBitmapFile( + const AString & a_FileName, + unsigned a_ImageSizeX, unsigned a_ImageSizeY, + const int * a_ColorIndices, + unsigned a_IndicesSizeX, unsigned a_IndicesSizeY, + const Color * a_Colors, + size_t a_NumColors +) +{ + std::ofstream f(a_FileName, std::ios::out | std::ios::binary); + if (!f.good()) + { + log("Cannot open file %s for writing. Skipping.", a_FileName.c_str()); + return; + } + f << "P3\r\n" << a_ImageSizeX << " " << a_ImageSizeY << "\r\n255\r\n"; + unsigned oldIndY = 0; + for (unsigned y = 0; y < a_ImageSizeY; y++) + { + unsigned indY = y * a_IndicesSizeY / a_ImageSizeY; + if (oldIndY != indY) + { + // Output a horizontal divider line: + for (unsigned x = 0; x < a_ImageSizeX; x++) + { + f << "128 128 128 "; + } + f << "\r\n"; + oldIndY = indY; + continue; + } + unsigned oldIndX = 0; + for (unsigned x = 0; x < a_ImageSizeX; x++) + { + unsigned indX = x * a_IndicesSizeX / a_ImageSizeX; + if (indX == oldIndX) + { + auto & color = a_Colors[a_ColorIndices[indX + a_IndicesSizeX * indY]]; + f << color[0] << " " << color[1] << " " << color[2] << " "; + } + else + { + // vertical divider line: + f << "128 128 128 "; + } + oldIndX = indX; + } + f << "\r\n"; + } +} + + + + + +/** Initializes biomeColors[] using the biomeColorMap[]. */ +void initializeBiomeColors(void) +{ + // Initialize all colors to red, so that anything unassigned is visible: + for (size_t i = 0; i < ARRAYCOUNT(biomeColors); i++) + { + auto & color = biomeColors[i]; + color[0] = 0; + color[1] = 0xff; + color[2] = 0; + } + + // Initialize per-biome: + for(size_t i = 0; i < ARRAYCOUNT(biomeColorMap); i++) + { + auto & dst = biomeColors[biomeColorMap[i].biome]; + const auto & src = biomeColorMap[i].color; + dst[0] = src[0]; + dst[1] = src[1]; + dst[2] = src[2]; + } +} + + + + + +/** Generates a series of images showing the "zoom" effect of the IntGen zoom operation. +Each image is the same size, the integer arrays are scaled to fit the image. */ +void generateZoomLevels(int a_Seed) +{ + log("Generating zoom levels..."); + const unsigned NumImages = 7; ///< Number of images to generate. The more images, the larger they will be + const unsigned maxArrSize = (1 << NumImages) + 1; + for (unsigned i = 1; i <= NumImages; i++) + { + unsigned arrSize = (1 << i) + 1; // Dimensions of the actually generated array + ASSERT(arrSize <= maxArrSize); + int workspace[maxArrSize * maxArrSize]; // Workspace for the generated array + + // Chain the zoom operation as many times as the image number: + std::shared_ptr<cProtIntGen> gen = std::make_shared<cProtIntGenChoice>(a_Seed, static_cast<int>(ARRAYCOUNT(spectrumColors) + 1)); + for (unsigned j = 1; j < i; j++) + { + gen = std::make_shared<cProtIntGenZoom>(a_Seed + static_cast<int>(j), gen); + } + gen->GetInts(0, 0, arrSize, arrSize, workspace); + + // Output to a bitmap file: + AString fnam = Printf("zoomedgrown_%u.pbm", i); + outputBitmapFile(fnam, 257, 257, workspace, arrSize, arrSize, spectrumColors, ARRAYCOUNT(spectrumColors)); + log(" zoom level %u complete", i); + } // for i - Image +} + + + + + +void generateSmoothLevels(int a_Seed) +{ + log("Generating smooth levels..."); + const unsigned NumImages = 7; ///< Number of images to generate. The more images, the larger they will be + const unsigned maxArrSize = 65; ///< Size of the underlying generated array + + // Initialize the underlying generator: + std::shared_ptr<cProtIntGen> underlyingGen = std::make_shared<cProtIntGenChoice>(a_Seed, static_cast<int>(ARRAYCOUNT(spectrumColors) + 1)); + for (int j = 1; j < 4; j++) + { + underlyingGen = std::make_shared<cProtIntGenZoom>(a_Seed + j, underlyingGen); + } + + // Generate smooth levels: + for (unsigned i = 1; i <= NumImages; i++) + { + unsigned arrSize = maxArrSize - 2 * i; // Dimensions of the actually generated array + int workspace[maxArrSize * maxArrSize]; // Workspace for the generated array + + // Chain the zoom operation as many times as the image number: + std::shared_ptr<cProtIntGen> gen = underlyingGen; + for (unsigned j = 1; j < i; j++) + { + gen = std::make_shared<cProtIntGenSmooth>(a_Seed, gen); + } + gen->GetInts(static_cast<int>(i), static_cast<int>(i), arrSize, arrSize, workspace); + + // Output to a bitmap file: + AString fnam = Printf("smoothedgrown_%u.ppm", i); + outputBitmapFile(fnam, 257, 257, workspace, arrSize, arrSize, spectrumColors, ARRAYCOUNT(spectrumColors)); + log(" smooth level %u complete", i); + } // for i - Image +} + + + + + +void generateExamples(int a_Seed) +{ + log("Generating examples"); + + const int maxArrSize = 65; + const int inArrSize = 24; + const int imgSize = 256; + + // Create the inputs: + auto in1 = + std::make_shared<cProtIntGenZoom >(a_Seed + 1, + std::make_shared<cProtIntGenAddIslands >(a_Seed + 2000, 200, + std::make_shared<cProtIntGenSetRandomly >(a_Seed + 9, 300, bgOcean, + std::make_shared<cProtIntGenZoom >(a_Seed + 2, + std::make_shared<cProtIntGenLandOcean >(a_Seed + 1, 30 + ))))); + auto in2 = + std::make_shared<cProtIntGenZoom >(a_Seed + 1, + std::make_shared<cProtIntGenBiomeGroupEdges>(in1 + )); + auto in3 = + std::make_shared<cProtIntGenZoom >(a_Seed + 1, + std::make_shared<cProtIntGenZoom >(a_Seed + 2, + std::make_shared<cProtIntGenBiomes >(a_Seed + 3000, in2 + ))); + auto inAlt = + std::make_shared<cProtIntGenZoom >(a_Seed, + std::make_shared<cProtIntGenLandOcean >(a_Seed, 30 + )); + auto inRiver = std::make_shared<cProtIntGenRiver>(a_Seed, in2); + int workspace[maxArrSize * maxArrSize]; + in1->GetInts(0, 0, inArrSize, inArrSize, workspace); + outputBitmapFile("grownexample_in1.ppm", imgSize, imgSize, workspace, inArrSize, inArrSize, biomeGroupColors, ARRAYCOUNT(biomeGroupColors)); + log(" Created example input 1"); + in2->GetInts(0, 0, inArrSize, inArrSize, workspace); + outputBitmapFile("grownexample_in2.ppm", imgSize, imgSize, workspace, inArrSize, inArrSize, biomeGroupColors, ARRAYCOUNT(biomeGroupColors)); + log(" Created example input 2"); + in3->GetInts(0, 0, inArrSize, inArrSize, workspace); + outputBitmapFile("grownexample_in3.ppm", imgSize, imgSize, workspace, inArrSize, inArrSize, biomeColors, ARRAYCOUNT(biomeColors)); + log(" Created example input 3"); + inAlt->GetInts(0, 0, inArrSize, inArrSize, workspace); + outputBitmapFile("grownexample_in_alt.ppm", imgSize, imgSize, workspace, inArrSize, inArrSize, biomeGroupColors, ARRAYCOUNT(biomeGroupColors)); + log(" Created example input alt"); + inRiver->GetInts(0, 0, inArrSize, inArrSize, workspace); + outputBitmapFile("grownexample_in_river.ppm", imgSize, imgSize, workspace, inArrSize, inArrSize, biomeColors, ARRAYCOUNT(biomeColors)); + log(" Created example input river"); + + // Shortcuts for colormaps used for the outputs: + struct ColorMap + { + const Color * colors; + size_t count; + }; + static const ColorMap cmGroups = { biomeGroupColors, ARRAYCOUNT(biomeGroupColors) }; + static const ColorMap cmBiomes = { biomeColors, ARRAYCOUNT(biomeColors) }; + + // Create the result generators: + struct + { + const char * name; + unsigned size; + int offset; + const ColorMap & colormap; + std::shared_ptr<cProtIntGen> gen; + } + gens[] = + { + {"add_islands", inArrSize, 0, cmGroups, std::make_shared<cProtIntGenAddIslands> (a_Seed, 400, in1)}, + {"alt_biomes", inArrSize, 0, cmBiomes, std::make_shared<cProtIntGenAlternateBiomes>(a_Seed, inAlt, in3)}, + {"beaches", inArrSize - 2, 1, cmBiomes, std::make_shared<cProtIntGenBeaches> (in3)}, + {"biome_edges", inArrSize - 2, 1, cmBiomes, std::make_shared<cProtIntGenBiomeEdges> (a_Seed, in3)}, + {"biomes", inArrSize, 0, cmBiomes, std::make_shared<cProtIntGenBiomes> (a_Seed, in2)}, + {"grp_edges", inArrSize - 2, 0, cmGroups, std::make_shared<cProtIntGenBiomeGroupEdges>(in1)}, + {"m_biomes", inArrSize, 0, cmBiomes, std::make_shared<cProtIntGenMBiomes> (a_Seed, inAlt, in3)}, + {"mix_river", inArrSize, 0, cmBiomes, std::make_shared<cProtIntGenMixRivers> (in3, inRiver)}, + {"river", inArrSize - 2, 1, cmBiomes, inRiver}, + {"set_rnd", inArrSize, 0, cmBiomes, std::make_shared<cProtIntGenSetRandomly> (a_Seed, 500, bgOcean, in3)}, + {"smooth", inArrSize - 2, 1, cmBiomes, std::make_shared<cProtIntGenSmooth> (a_Seed, in3)}, + {"zoom", inArrSize * 2 - 1, 0, cmBiomes, std::make_shared<cProtIntGenZoom> (a_Seed, in3)}, + }; + + // Create the outputs: + for (const auto & gen: gens) + { + gen.gen->GetInts(gen.offset, gen.offset, gen.size, gen.size, workspace); + AString fnam = Printf("grownexample_%s.ppm", gen.name); + outputBitmapFile(fnam, 256, 256, workspace, gen.size, gen.size, gen.colormap.colors, gen.colormap.count); + log(" Created example \"%s\"", gen.name); + } // for gen - gens[] + + log("Examples generated"); +} + + + + + +int main(int argc, char ** argv) +{ + log("GrownBiomeGenVisualiser starting"); + + // Parse the seed from the command line, if present: + int seed; + if (argc > 1) + { + if (!StringToInteger(argv[1], seed)) + { + log("Cannot parse seed from \"%s\", bailing out.", argv[1]); + return 1; + } + } + else + { + // Get a random seed: + std::random_device rd; + seed = static_cast<int>(rd()); + } + + log("Seed = %d", seed); + + initializeBiomeColors(); + + generateZoomLevels(seed); + generateSmoothLevels(seed); + generateExamples(seed); + + log("GrownBiomeGenVisualiser finished"); + return 0; +} + + + + |