Terrain2.h

Go to the documentation of this file.

#pragma once
 
#include <tdme/tdme.h>
#include <tdme/utilities/fwd-tdme.h>
 
#include <unordered_map>
#include <unordered_set>
#include <vector>
 
#include <tdme/engine/fileio/textures/fwd-tdme.h>
#include <tdme/engine/model/fwd-tdme.h>
#include <tdme/engine/primitives/BoundingBox.h>
#include <tdme/engine/Transformations.h>
#include <tdme/math/Vector2.h>
#include <tdme/math/Vector3.h>
 
using std::unordered_map;
using std::unordered_set;
using std::vector;
 
using tdme::engine::fileio::textures::Texture;
using tdme::engine::model::Model;
using tdme::engine::primitives::BoundingBox;
using tdme::engine::Transformations;
using tdme::math::Vector2;
using tdme::math::Vector3;
 
/**
 * Terrain 2 utility
 * @author Andreas Drewke
 */
class tdme::utilities::Terrain2
{
public:
    static constexpr float STEP_SIZE { 1.0f };
    static constexpr float PARTITION_SIZE { 64.0f };
 
    struct FoliageBrush {
        Texture* brushTexture;
        float brushScale;
        float brushDensity;
    };
 
    struct FoliageBrushPrototype {
        int prototypeId;
        float count;
        bool normalAlign;
        float rotationXMin;
        float rotationXMax;
        float rotationYMin;
        float rotationYMax;
        float rotationZMin;
        float rotationZMax;
        float scaleMin;
        float scaleMax;
        float heightMin;
        float heightMax;
        float slopeMin;
        float slopeMax;
    };
 
private:
    /**
     * Get the terrain vertex for given x and z position without providing y component
     * @param x x
     * @param z z
     * @return terrain vertex
     */
    inline static void getTerrainVertex(int x, int z, Vector3& vertex) {
        vertex.set(
            static_cast<float>(x) * STEP_SIZE,
            0.0f,
            static_cast<float>(z) * STEP_SIZE
        );
    }
    /**
     * Get the terrain vertex for given x and z position
     * @param terrainHeightVector terrain height vector
     * @param verticesPerX vertices per x
     * @param verticesPerZ vertices per z
     * @param x x
     * @param z z
     * @return terrain vertex
     */
    inline static bool getTerrainVertex(const vector<float>& terrainHeightVector, int verticesPerX, int verticesPerZ, int x, int z, Vector3& vertex) {
        vertex.set(
            static_cast<float>(x) * STEP_SIZE,
            0.0f,
            static_cast<float>(z) * STEP_SIZE
        );
        if (x < 0 || x >= verticesPerX) return false;
        if (z < 0 || z >= verticesPerZ) return false;
        vertex.set(
            static_cast<float>(x) * STEP_SIZE,
            terrainHeightVector[z * verticesPerX + x],
            static_cast<float>(z) * STEP_SIZE
        );
        return true;
    }
 
    /**
     * Get the terrain vertex for given x and z position
     * @param x x
     * @param z z
     * @param waterHeight water height
     * @param vertex vertex
     */
    inline static void getWaterVertex(int x, int z, float waterHeight, Vector3& vertex) {
        vertex.set(
            static_cast<float>(x) * STEP_SIZE,
            waterHeight,
            static_cast<float>(z) * STEP_SIZE
        );
    }
 
    /**
     * @returns if water includes given position
     * @param waterPositionSet water position set
     * @param x x
     * @param z z
     */
    inline static bool hasWaterPosition(const unordered_map<int, unordered_set<int>>& waterPositionSet, int x, int z) {
        auto waterPositionSetIt = waterPositionSet.find(z);
        if (waterPositionSetIt == waterPositionSet.end()) return false;
        auto waterXPositionSetIt = waterPositionSetIt->second.find(x);
        if (waterXPositionSetIt == waterPositionSetIt->second.end()) return false;
        return true;
    }
 
    /**
     * Compute terrain vertex normal for given x and z position
     * @param terrainHeightVector terrain height vector
     * @param verticesPerX vertices per x
     * @param verticesPerZ vertices per z
     * @param x x
     * @param z z
     * @return normal for given vertex index
     */
    static const Vector3 computeTerrainVertexNormal(const vector<float>& terrainHeightVector, int verticesPerX, int verticesPerZ, int x, int z);
 
    /**
     * Determine if water can be generated
     * @param terrainHeightVector terrain height vector
     * @param verticesPerX vertices per x
     * @param verticesPerZ vertices per z
     * @param x x
     * @param z z
     * @param waterHeight water height
     * @return if water can be generated at given position
     */
    inline static bool determineWater(const vector<float>& terrainHeightVector, int verticesPerX, int verticesPerZ, int x, int z, float waterHeight) {
        if (x < 0 || x >= verticesPerX ||
            z < 0 || z >= verticesPerZ) return false;
        auto vertexIdx = z * verticesPerX + x;
        auto terrainVertexHeight = terrainHeightVector[vertexIdx];
        if (terrainVertexHeight >= waterHeight) return false;
        return true;
    }
 
    /**
     * Determine if water can be generated from left to right starting with x and z
     * @param terrainHeightVector terrain height vector
     * @param verticesPerX vertices per x
     * @param verticesPerZ vertices per z
     * @param x x
     * @param z z
     * @param waterHeight water height
     * @param waterXPositionSet water x position set
     * @return if water can be generated at given position
     */
    inline static void determineWaterXPositionSet(const vector<float>& terrainHeightVector, int verticesPerX, int verticesPerZ, int x, int z, float waterHeight, unordered_set<int>& waterXPositionSet) {
        auto xMin = -1;
        auto xMax = +1;
        while(true == true) {
            auto terrainHeightVectorX = x + xMin;
            if (terrainHeightVectorX < 0 || terrainHeightVectorX >= verticesPerX ||
                z < 0 || z >= verticesPerZ) break;
            auto vertexIdx = z * verticesPerX + terrainHeightVectorX;
            auto terrainVertexHeight = terrainHeightVector[vertexIdx];
            if (terrainVertexHeight >= waterHeight) break;
            waterXPositionSet.insert(terrainHeightVectorX);
            xMin--;
        }
        while(true == true) {
            auto terrainHeightVectorX = x + xMax;
            if (terrainHeightVectorX < 0 || terrainHeightVectorX >= verticesPerX ||
                z < 0 || z >= verticesPerZ) break;
            auto vertexIdx = z * verticesPerX + terrainHeightVectorX;
            auto terrainVertexHeight = terrainHeightVector[vertexIdx];
            if (terrainVertexHeight >= waterHeight) break;
            waterXPositionSet.insert(terrainHeightVectorX);
            xMax++;
        }
        if (waterXPositionSet.size() > 2) {
            waterXPositionSet.insert(x + xMin - 0);
            waterXPositionSet.insert(x + xMin - 1);
            waterXPositionSet.insert(x + xMin - 2);
            waterXPositionSet.insert(x + xMax + 0);
            waterXPositionSet.insert(x + xMax + 1);
            waterXPositionSet.insert(x + xMax + 2);
        }
    }
 
public:
    enum BrushOperation {
        BRUSHOPERATION_NONE,
        BRUSHOPERATION_ADD,
        BRUSHOPERATION_SUBTRACT,
        BRUSHOPERATION_FLATTEN,
        BRUSHOPERATION_DELETE,
        BRUSHOPERATION_SMOOTH,
        BRUSHOPERATION_WATER_ADD,
        BRUSHOPERATION_WATER_DELETE,
        BRUSHOPERATION_RAMP,
    };
 
    /**
     * Create terrain models
     * @param width width
     * @param depth depth
     * @param y float y
     * @param terrainHeightVector terrain height vector
     * @param terrainBoundingBox terrain bounding box
     * @param terrainModels terrain models vector
     * @param createLODLevels create LOD Levels
     */
    static void createTerrainModels(float width, float depth, float y, vector<float>& terrainHeightVector, BoundingBox& terrainBoundingBox, vector<Model*>& terrainModels, bool createLODLevels = false);
 
    /**
     * Apply brush to given terrain models
     * @param terrainBoundingBox terrain bounding box
     * @param terrainModels terrain models vector
     * @param terrainHeightVector terrain height vector
     * @param brushCenterPosition brush center position
     * @param brushTexture brush texture
     * @param brushScale brush scale
     * @param brushStrength brush strength
     * @param brushOperation brush operation
     * @param flattenHeight flatten height
     *
     */
    static void applyBrushToTerrainModels(
        BoundingBox& terrainBoundingBox, // TODO: constness
        vector<Model*>& terrainModels,
        vector<float>& terrainHeightVector,
        const Vector3& brushCenterPosition,
        Texture* brushTexture,
        float brushScale,
        float brushStrength,
        BrushOperation brushOperation,
        float flattenHeight = 0.0f
    );
 
    /**
     * Apply ramp brush to given terrain models
     * @param terrainBoundingBox terrain bounding box
     * @param terrainModels terrain models vector
     * @param terrainHeightVector terrain height vector
     * @param brushCenterPosition brush center position
     * @param brushTexture brush texture
     * @param brushRotation brush rotation
     * @param brushScale brush scale
     * @param flattenHeightMin minimum flatten height
     * @param flattenHeightMax maximum flatten height
     *
     */
    static void applyRampBrushToTerrainModels(
        BoundingBox& terrainBoundingBox, // TODO: constness
        vector<Model*>& terrainModels,
        vector<float>& terrainHeightVector,
        const Vector3& brushCenterPosition,
        Texture* brushTexture,
        float brushRotation,
        const Vector2& brushScale,
        float flattenHeightMin,
        float flattenHeightMax
    );
 
    /**
     * Update foliage after using terrain brush
     * @param terrainBoundingBox terrain bounding box
     * @param terrainHeightVector terrain height vector
     * @param brushCenterPosition brush center position
     * @param foliageBrush foliage brush
     * @param foliageMaps foliage maps
     * @param updateFoliagePartitions update foliage partitions
     */
    static void updateFoliageTerrainBrush(
        BoundingBox& terrainBoundingBox, // TODO: constness
        vector<float>& terrainHeightVector,
        const Vector3& brushCenterPosition,
        const FoliageBrush& foliageBrush,
        vector<unordered_map<int, vector<Transformations>>>& foliageMaps,
        unordered_set<int>& updateFoliagePartitions
    );
 
    /**
     * Update foliage after using terrain ramp brush
     * @param terrainBoundingBox terrain bounding box
     * @param terrainHeightVector terrain height vector
     * @param brushCenterPosition brush center position
     * @param brushTexture brush texture
     * @param brushRotation brush rotation
     * @param brushScale brush scale
     * @param foliageMaps foliage maps
     * @param updateFoliagePartitions update foliage partitions
     */
    static void updateFoliageTerrainRampBrush(
        BoundingBox& terrainBoundingBox, // TODO: constness
        vector<float>& terrainHeightVector,
        const Vector3& brushCenterPosition,
        Texture* brushTexture,
        float brushRotation,
        const Vector2& brushScale,
        vector<unordered_map<int, vector<Transformations>>>& foliageMaps,
        unordered_set<int>& updateFoliagePartitions
    );
 
    /**
     * Compute water positions map using a auto fill like algorithm at given brush center position
     * @param terrainBoundingBox terrain bounding box
     * @param terrainHeightVector terrain height vector
     * @param brushCenterPosition brush center position
     * @param waterHeight waterHeight
     * @param waterPositionMap water position map
     */
    static bool computeWaterPositionMap(
        BoundingBox& terrainBoundingBox,
        const vector<float>& terrainHeightVector,
        const Vector3& brushCenterPosition,
        float waterHeight,
        unordered_map<int, unordered_set<int>>& waterPositionMap
    );
 
    /**
     * Compute water reflection environment mapping position
     * @param waterPositionMap water position map
     * @param waterHeight water height
     * @return water reflection environment mapping position
     *
     */
    static Vector3 computeWaterReflectionEnvironmentMappingPosition(
        const unordered_map<int, unordered_set<int>>& waterPositionMap,
        float waterHeight
    );
 
    /**
     * Create partitioned water models using given water position map
     * @param terrainBoundingBox terrain bounding box
     * @param waterPositionMap water position map
     * @param waterHeight waterHeight
     * @param waterModelIdx water model index
     * @param waterModels water models
     *
     */
    static void createWaterModels(
        BoundingBox& terrainBoundingBox,
        const unordered_map<int, unordered_set<int>>& waterPositionMap,
        float waterHeight,
        int waterModelIdx,
        vector<Model*>& waterModels
    );
 
    /**
     * Get terrain models height for e.g. flatten
     * @param terrainBoundingBox terrain bounding box
     * @param terrainModels terrain models vector
     * @param terrainHeightVector terrain height vector
     * @param brushHeight brush height
     *
     */
    static bool getTerrainModelsHeight(
        BoundingBox& terrainBoundingBox,  // TODO: constness
        vector<Model*>& terrainModels,
        vector<float>& terrainHeightVector,
        const Vector3& brushCenterPosition,
        float& brushHeight
    );
 
    /**
     * Create foliage maps
     * @param terrainBoundingBox terrain bounding box
     * @param foliageMaps foliage maps
     */
    static void createFoliageMaps(
        BoundingBox& terrainBoundingBox, // TODO: constness
        vector<unordered_map<int, vector<Transformations>>>& foliageMaps
    );
 
    /**
     * Create foliage maps
     * @param terrainWidth terrain width
     * @param terrainDepth terrain depth
     * @param foliageMaps foliage maps
     */
    static void createFoliageMaps(
        float terrainWidth,
        float terrainDepth,
        vector<unordered_map<int, vector<Transformations>>>& foliageMaps
    );
 
    /**
     * Empty foliage maps
     * @param foliageMaps foliage maps
     */
    static void emptyFoliageMaps(
        vector<unordered_map<int, vector<Transformations>>>& foliageMaps
    );
 
    /**
     * Apply foliage brush
     * @param terrainBoundingBox terrain bounding box
     * @param terrainHeightVector terrain height vector
     * @param brushCenterPosition brush center position
     * @param foliageBrush foliage brush
     * @param foliageBrushPrototypes foliage brush prototypes
     * @param brushOperation brush operation
     * @param foliageMaps foliage maps
     * @param newFoliageMaps new foliage maps
     * @param prototypeScale prototype scale
     */
    static void applyFoliageBrush(
        BoundingBox& terrainBoundingBox, // TODO: constness
        vector<float>& terrainHeightVector,
        const Vector3& brushCenterPosition,
        const FoliageBrush& foliageBrush,
        const vector<FoliageBrushPrototype>& foliageBrushPrototypes,
        BrushOperation brushOperation,
        vector<unordered_map<int, vector<Transformations>>>& foliageMaps,
        vector<unordered_map<int, vector<Transformations>>>& newFoliageMaps
    );
 
    /**
     * Apply foliage delete brush
     * @param terrainBoundingBox terrain bounding box
     * @param brushCenterPosition brush center position
     * @param foliageBrush foliage brush
     * @param foliageBrushPrototypes foliage brush prototypes
     * @param brushOperation brush operation
     * @param foliageMaps foliage maps
     * @param recreateFoliagePartitions recreate foliage partitions
     */
    static void applyFoliageDeleteBrush(
        BoundingBox& terrainBoundingBox, // TODO: constness
        const Vector3& brushCenterPosition,
        const FoliageBrush& foliageBrush,
        const vector<FoliageBrushPrototype>& foliageBrushPrototypes,
        BrushOperation brushOperation,
        vector<unordered_map<int, vector<Transformations>>>& foliageMaps,
        unordered_set<int>& recreateFoliagePartitions
    );
 
    /**
     * Mirror terrain around X axis
     * @param flipZ flip Z
     * @param width width
     * @param depth depth
     * @param terrainHeightVector terrain height vector
     * @param waterPositionMapsHeight water position maps heights
     * @param waterPositionMaps water position maps
     * @param foliageMaps foliage maps
     */
    static void mirrorXAxis(
        bool flipZ,
        float width,
        float depth,
        vector<float>& terrainHeightVector,
        unordered_map<int, float>& waterPositionMapsHeight,
        unordered_map<int, unordered_map<int, unordered_set<int>>>& waterPositionMaps,
        vector<unordered_map<int, vector<Transformations>>>& foliageMaps
    );
 
    /**
     * Mirror terrain around Z axis
     * @param flipX flip X
     * @param width width
     * @param depth depth
     * @param terrainHeightVector terrain height vector
     * @param waterPositionMapsHeight water position maps heights
     * @param waterPositionMaps water position maps
     * @param foliageMaps foliage maps
     */
    static void mirrorZAxis(
        bool flipX,
        float width,
        float depth,
        vector<float>& terrainHeightVector,
        unordered_map<int, float>& waterPositionMapsHeight,
        unordered_map<int, unordered_map<int, unordered_set<int>>>& waterPositionMaps,
        vector<unordered_map<int, vector<Transformations>>>& foliageMaps
    );
 
};