#include "TextureLayerDefinition.h"

#include "Zone.h"
#include "NoiseGenerator.h"
#include "SurfaceMaterial.h"
#include "PackStream.h"
#include "Scenery.h"
#include "TerrainDefinition.h"
#include "Color.h"

TextureLayerDefinition::TextureLayerDefinition(DefinitionNode *parent, const string &name)
    : DefinitionNode(parent, name, "texturelayer") {
    terrain_zone = new Zone;
    _displacement_noise = new NoiseGenerator;
    _detail_noise = new NoiseGenerator;
    material = new SurfaceMaterial;

    displacement_height = 0.0;
    displacement_scaling = 1.0;
}

TextureLayerDefinition::~TextureLayerDefinition() {
    delete terrain_zone;
    delete _displacement_noise;
    delete _detail_noise;
    delete material;
}

void TextureLayerDefinition::validate() {
    DefinitionNode::validate();

    if (displacement_scaling < 0.000001) {
        displacement_scaling = 0.000001;
    }

    _displacement_noise->clearLevels();
    _displacement_noise->addLevelsSimple(9, 1.0, -1.0, 1.0, 0.0);
    _displacement_noise->normalizeAmplitude(-1.0, 1.0, 0);
    _displacement_noise->validate();

    _detail_noise->clearLevels();
    _detail_noise->addLevelsSimple(7, 0.01, -1.0, 1.0, 0.0);
    _detail_noise->normalizeAmplitude(-0.008, 0.008, 0);
    _detail_noise->validate();

    material->validate();

    /* Update zone height range */
    const Scenery *scenery = getScenery();
    if (scenery) {
        TerrainDefinition *terrain = scenery->getTerrain();
        HeightInfo height_info = terrain->getHeightInfo();
        terrain_zone->setRelativeHeight(height_info.min_height, height_info.base_height, height_info.max_height);
    }
}

void TextureLayerDefinition::copy(DefinitionNode *destination) const {
    DefinitionNode::copy(destination);

    if (auto tex_destination = static_cast<TextureLayerDefinition *>(destination)) {
        terrain_zone->copy(tex_destination->terrain_zone);

        tex_destination->displacement_scaling = displacement_scaling;
        tex_destination->displacement_height = displacement_height;
        *tex_destination->material = *material;

        _displacement_noise->copy(tex_destination->_displacement_noise);
        _detail_noise->copy(tex_destination->_detail_noise);
    }
}

void TextureLayerDefinition::save(PackStream *stream) const {
    DefinitionNode::save(stream);

    terrain_zone->save(stream);
    stream->write(&displacement_scaling);
    stream->write(&displacement_height);
    material->save(stream);

    _displacement_noise->save(stream);
    _detail_noise->save(stream);
}

void TextureLayerDefinition::load(PackStream *stream) {
    DefinitionNode::load(stream);

    terrain_zone->load(stream);
    stream->read(&displacement_scaling);
    stream->read(&displacement_height);
    material->load(stream);

    _displacement_noise->load(stream);
    _detail_noise->load(stream);
}

void TextureLayerDefinition::applyPreset(TextureLayerPreset preset, RandomGenerator &random) {
    _displacement_noise->randomizeOffsets(random);
    _detail_noise->randomizeOffsets(random);

    terrain_zone->clear();

    switch (preset) {
    case TEXTURES_LAYER_PRESET_MUD:
        displacement_height = 0.02;
        displacement_scaling = 3.0;
        material->setColor(0.015, 0.014, 0.014, 1.0);
        material->reflection = 0.003;
        material->shininess = 4.0;
        break;
    case TEXTURES_LAYER_PRESET_ROCK:
        terrain_zone->addHeightRangeQuick(1.0, 0.6, 0.7, 1.0, 1.0);
        displacement_height = 0.3;
        displacement_scaling = 2.0;
        material->setColor(0.6, 0.55, 0.57, 1.0);
        material->reflection = 0.006;
        material->shininess = 6.0;
        break;
    case TEXTURES_LAYER_PRESET_GRASS:
        terrain_zone->addHeightRangeQuick(1.0, 0.45, 0.5, 0.8, 1.0);
        terrain_zone->addSlopeRangeQuick(1.0, 0.0, 0.0, 0.05, 0.4);
        displacement_height = 0.0;
        displacement_scaling = 1.0;
        material->setColor(0.12, 0.19, 0.035, 1.0);
        material->reflection = 0.001;
        material->shininess = 4.0;
        break;
    case TEXTURES_LAYER_PRESET_SAND:
        terrain_zone->addHeightRangeQuick(1.0, 0.495, 0.505, 0.56, 0.63);
        terrain_zone->addSlopeRangeQuick(1.0, 0.0, 0.0, 0.1, 0.4);
        displacement_height = 0.05;
        displacement_scaling = 5.0;
        material->setColor(1.2, 1.1, 0.9, 1.0);
        material->reflection = 0.008;
        material->shininess = 1.0;
        break;
    case TEXTURES_LAYER_PRESET_SNOW:
        terrain_zone->addHeightRangeQuick(1.0, 0.77, 0.85, 1.0, 1.0);
        terrain_zone->addSlopeRangeQuick(1.0, 0.0, 0.0, 0.2, 1.0);
        displacement_height = 0.1;
        displacement_scaling = 1.0;
        material->setColor(5.0, 5.0, 5.0, 1.0);
        material->reflection = 0.02;
        material->shininess = 0.6;
        break;
    }

    validate();
}