#include "minecraft/IGameServices.h"

#include "minecraft/world/level/levelgen/structure/MineShaftFeature.h"

#include <stdlib.h>

#include <algorithm>
#include <string>
#include <unordered_map>
#include <utility>

#include "minecraft/GameEnums.h"
#include "app/common/GameRules/LevelGeneration/LevelGenerationOptions.h"
#include "app/linux/LinuxGame.h"
#include "java/Random.h"
#include "minecraft/util/Mth.h"
#include "minecraft/world/level/levelgen/structure/MineShaftStart.h"

const std::string MineShaftFeature::OPTION_CHANCE = "chance";

MineShaftFeature::MineShaftFeature() { chance = 0.01; }

std::string MineShaftFeature::getFeatureName() { return "Mineshaft"; }

MineShaftFeature::MineShaftFeature(
    std::unordered_map<std::string, std::string> options) {
    chance = 0.01;

    for (auto it = options.begin(); it != options.end(); ++it) {
        if (it->first.compare(OPTION_CHANCE) == 0) {
            chance = Mth::getDouble(it->second, chance);
        }
    }
}

bool MineShaftFeature::isFeatureChunk(int x, int z, bool bIsSuperflat) {
    bool forcePlacement = false;
    LevelGenerationOptions* levelGenOptions = gameServices().getLevelGenerationOptions();
    if (levelGenOptions != nullptr) {
        forcePlacement =
            levelGenOptions->isFeatureChunk(x, z, eFeature_Mineshaft);
    }

    return forcePlacement || (random->nextDouble() < chance &&
                              random->nextInt(80) < std::max(abs(x), abs(z)));
}

StructureStart* MineShaftFeature::createStructureStart(int x, int z) {
    // 4J added
    gameServices().addTerrainFeaturePosition(eTerrainFeature_Mineshaft, x, z);

    return new MineShaftStart(level, random, x, z);
}