FactoryCraft-Server/FactoryCraft/WorldGenerator.cpp

#include "WorldGenerator.h"

#include <Datei.h>
#include <functional>
#include <Logging.h>

#include "Dimension.h"
#include "Game.h"
#include "JsonUtils.h"

using namespace Framework;
using namespace Framework::JSON;
using namespace Framework::Validator;

WorldGenerator::WorldGenerator(int seed)
    : Thread(),
      exit(0),
      seed(seed),
      chunksPerSecond(0.0)
{
    setName("World Generator");
    Framework::Logging::info()
        << "loading world generator configs. Changes at the config files "
           "may lead to a sudden change in landscape.";
    DataValidator* configValidator
        = DataValidator::buildForArray()
              ->removeInvalidEntries()
              ->addAcceptedTypeInArray(Game::INSTANCE->zTypeRegistry()
                      ->getValidator<DimensionGenerator>())
              ->finishArray();
    loadAllJsonsFromDirectory("data/generator",
        [this, configValidator, seed](
            Framework::JSON::JSONValue* zValue, Framework::Text path) {
            Framework::Logging::info()
                << "loading dimension configs from '" << path << "'";
            Framework::RCArray<ValidationResult> invalidParts;
            JSONValue* valid
                = configValidator->getValidParts(zValue, &invalidParts);
            for (ValidationResult* invalidPart : invalidParts)
            {
                Framework::Logging::error() << invalidPart->getInvalidInfo();
            }
            if (valid)
            {
                for (JSONValue* config : *valid->asArray())
                {
                    DimensionGenerator* generator
                        = Game::INSTANCE->zTypeRegistry()
                              ->fromJson<DimensionGenerator>(config);
                    generator->initialize(seed);
                    dimensionGenerators.add(generator);
                }
                valid->release();
            }
        });
    Framework::JSON::JSONObject* schema = configValidator->getJsonSchema();
    Framework::Datei syntaxFile;
    syntaxFile.setDatei("data/syntax/schema/generator.json");
    syntaxFile.erstellen();
    syntaxFile.open(Framework::Datei::Style::schreiben);
    syntaxFile.schreibe(schema->toString(), schema->toString().getLength());
    syntaxFile.close();
    schema->release();
    configValidator->release();

    start();
}

WorldGenerator::~WorldGenerator() {}

Dimension* WorldGenerator::createDimension(int dimensionId)
{
    for (DimensionGenerator* generator : dimensionGenerators)
    {
        if (generator->getDimensionId() == dimensionId)
            return generator->createDimension();
    }
    Framework::Logging::error()
        << "no dimension generator found for dimension " << dimensionId;
    return 0;
}

DimensionGenerator* WorldGenerator::zGenerator(int dimensionId)
{
    for (DimensionGenerator* generator : dimensionGenerators)
    {
        if (generator->getDimensionId() == dimensionId) return generator;
    }
    return 0;
}

void WorldGenerator::thread()
{
    double completeTime = 0.0;
    int generatedChunks = 0;
    while (!exit)
    {
        cs.lock();
        Area next;
        bool hasNext = 0;
        if (requestQueue.getEintragAnzahl() > 0)
        {
            next = requestQueue.get(0);
            requestQueue.remove(0);
            hasNext = 1;
        }
        cs.unlock();
        if (!hasNext)
        {
            chunksPerSecond = 0.0;
            completeTime = 0;
            generatedChunks = 0;
            Sleep(1000);
            continue;
        }
        // TODO: prevent that world loader and world generator create the
        // dimension at the same time
        Dimension* dim = Game::INSTANCE->zDimension(next.dimensionId);
        if (!dim)
        {
            dim = new Dimension(next.dimensionId);
            Game::INSTANCE->addDimension(dim);
        }
        Punkt start = Game::INSTANCE->getChunkCenter(next.startX, next.startY);
        Punkt end = Game::INSTANCE->getChunkCenter(next.endX, next.endY);
        int xDir = start.x > end.x ? -1 : 1;
        int yDir = start.y > end.y ? -1 : 1;
        for (int x = start.x; xDir < 0 ? x >= end.x : x <= end.x;
            x += CHUNK_SIZE * xDir)
        {
            for (int y = start.y; yDir < 0 ? y >= end.y : y <= end.y;
                y += CHUNK_SIZE * yDir)
            {
                if (!Game::INSTANCE->doesChunkExist(x, y, next.dimensionId))
                {
                    ZeitMesser zm;
                    zm.messungStart();
                    Chunk* generatedChunk
                        = zGenerator(next.dimensionId)->generateChunk(x, y);
#ifdef CHUNK_GENERATION_DEBUG_LOG
                    zm.messungEnde();
                    Framework::Logging::trace()
                        << "block generation time: " << zm.getSekunden();
                    completeTime += zm.getSekunden();
                    zm.messungStart();
#endif
                    generatedChunk->initializeLightning();
#ifdef CHUNK_GENERATION_DEBUG_LOG
                    zm.messungEnde();
                    Framework::Logging::trace()
                        << "light calculation: " << zm.getSekunden();
                    completeTime += zm.getSekunden();
                    zm.messungStart();
#endif
                    generatedChunk->removeUnusedBlocks();
#ifdef CHUNK_GENERATION_DEBUG_LOG
                    zm.messungEnde();
                    Framework::Logging::trace()
                        << "unused block removal: " << zm.getSekunden();
                    completeTime += zm.getSekunden();
                    zm.messungStart();
#endif
                    generatedChunk->getThis();
                    dim->setChunk(generatedChunk, Punkt(x, y));
                    generatedChunk->release();
                    zm.messungEnde();
#ifdef CHUNK_GENERATION_DEBUG_LOG
                    Framework::Logging::trace()
                        << "adding chunk to map: " << zm.getSekunden();
#endif
                    completeTime += zm.getSekunden();
                    generatedChunks++;
                    chunksPerSecond = generatedChunks / completeTime;
                }
            }
        }
    }
    Framework::Logging::info() << "World Generator thread exited";
}

void WorldGenerator::requestGeneration(Area request)
{
    cs.lock();
    requestQueue.add(request);
    cs.unlock();
}

void WorldGenerator::exitAndWait()
{
    exit = 1;
    warteAufThread(10000);
    ende();
}

Framework::Either<Block*, int> WorldGenerator::generateSingleBlock(
    Framework::Vec3<int> location, int dimensionId)
{
    return zGenerator(dimensionId)->generateBlock(location);
}

void WorldGenerator::postprocessChunk(Chunk* zChunk)
{
    zGenerator(zChunk->getDimensionId())->postprocessChunk(zChunk);
}

bool WorldGenerator::spawnStructure(Framework::Vec3<int> location,
    int dimensionId,
    std::function<bool(GeneratorTemplate* tmpl)> filter)
{
    return zGenerator(dimensionId)->spawnStructure(location, filter);
}

double WorldGenerator::getGeneratedChunksPerSecond() const
{
    return chunksPerSecond;
}