WhittakerRasterizer.java

/*
 * Copyright 2015 MovingBlocks
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.terasology.polyworld.raster;

import org.terasology.math.TeraMath;
import org.terasology.math.geom.Vector3i;
import org.terasology.polyworld.biome.WhittakerBiome;
import org.terasology.polyworld.biome.WhittakerBiomeFacet;
import org.terasology.registry.CoreRegistry;
import org.terasology.world.block.Block;
import org.terasology.world.block.BlockManager;
import org.terasology.world.chunks.CoreChunk;
import org.terasology.world.generation.Region;
import org.terasology.world.generation.WorldRasterizer;
import org.terasology.world.generation.facets.SeaLevelFacet;
import org.terasology.world.generation.facets.SurfaceHeightFacet;

/**
 */
public class WhittakerRasterizer implements WorldRasterizer {

    private Block water;
    private Block ice;
    private Block stone;
    private Block sand;
    private Block grass;
    private Block snow;
    private Block dirt;

    @Override
    public void initialize() {
        BlockManager blockManager = CoreRegistry.get(BlockManager.class);
        stone = blockManager.getBlock("core:stone");
        water = blockManager.getBlock("core:water");
        ice = blockManager.getBlock("core:Ice");
        sand = blockManager.getBlock("core:Sand");
        grass = blockManager.getBlock("core:Grass");
        snow = blockManager.getBlock("core:Snow");
        dirt = blockManager.getBlock("core:Dirt");
    }

    @Override
    public void generateChunk(CoreChunk chunk, Region chunkRegion) {
        SurfaceHeightFacet surfaceHeightData = chunkRegion.getFacet(SurfaceHeightFacet.class);
        SeaLevelFacet seaLevelFacet = chunkRegion.getFacet(SeaLevelFacet.class);
        WhittakerBiomeFacet biomeFacet = chunkRegion.getFacet(WhittakerBiomeFacet.class);
        int seaLevel = seaLevelFacet.getSeaLevel();

        Vector3i chunkOffset = chunk.getChunkWorldOffset();
        for (int x = 0; x < chunk.getChunkSizeX(); ++x) {
            for (int z = 0; z < chunk.getChunkSizeZ(); ++z) {

                float surfaceHeight = surfaceHeightData.get(x, z);
                int surfaceHeightInt = TeraMath.floorToInt(surfaceHeight);

                WhittakerBiome biome = biomeFacet.get(x, z);

                for (int y = 0; y < chunk.getChunkSizeY(); ++y) {

                    chunk.setBiome(x, y, z, biome);

                    int depth = surfaceHeightInt - y - chunk.getChunkWorldOffsetY();
                    if (depth >= 0) {
                        Block block = getSurfaceBlock(depth, biome);
                        chunk.setBlock(x, y, z, block);
                    } else
                    if (y + chunkOffset.y <= seaLevel) {
                        chunk.setBlock(x, y, z, water);
                    }
                }
            }
        }
    }

    private Block getSurfaceBlock(int depth, WhittakerBiome type) {
        if (depth > 8) {
            return stone;
        }

        if (depth > 1) {
            return dirt;
        }

        switch (type) {
            case TROPICAL_RAIN_FOREST:
            case TROPICAL_SEASONAL_FOREST:
            case GRASSLAND:
            case MARSH:
            case SHRUBLAND:
                return grass;

            case TAIGA:
                return ice;

            case SNOW:
                return snow;

            case TEMPERATE_DECIDUOUS_FOREST:
            case TEMPERATE_RAIN_FOREST:
                return grass;

            case TUNDRA:
            case SCORCHED:
            case BARE:
                return dirt;

            case COAST:
            case BEACH:
            case LAKESHORE:
            case SUBTROPICAL_DESERT:
            case TEMPERATE_DESERT:
                return sand;

            case ICE:
                return sand;

            case LAKE:
            case OCEAN:
                return water;
        }

        return stone;
    }
}