MagicBees/src/main/java/magicbees/bees/HiveDescription.java

package magicbees.bees;

import forestry.api.apiculture.hives.HiveManager;
import forestry.api.apiculture.hives.IHiveDescription;
import forestry.api.apiculture.hives.IHiveGen;
import forestry.api.core.EnumHumidity;
import forestry.api.core.EnumTemperature;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import java.util.Random;
import magicbees.block.types.HiveType;
import magicbees.main.Config;
import magicbees.main.utils.BlockUtil;
import magicbees.main.utils.LogHelper;
import magicbees.world.feature.FeatureOreVein;
import magicbees.world.feature.HiveGenNether;
import magicbees.world.feature.HiveGenOblivion;
import magicbees.world.feature.HiveGenUnderground;
import net.minecraft.block.Block;
import net.minecraft.init.Blocks;
import net.minecraft.world.World;
import net.minecraft.world.biome.BiomeGenBase;
import net.minecraftforge.common.BiomeDictionary;

public enum HiveDescription implements IHiveDescription {
    CURIOUS(
            HiveType.CURIOUS,
            3.0f,
            HiveManager.genHelper.tree(),
            false,
            BiomeDictionary.Type.FOREST,
            BiomeDictionary.Type.PLAINS,
            BiomeDictionary.Type.HILLS),
    UNUSUAL(
            HiveType.UNUSUAL,
            1.0f,
            HiveManager.genHelper.ground(Blocks.dirt, Blocks.grass),
            false,
            BiomeDictionary.Type.PLAINS,
            BiomeDictionary.Type.MOUNTAIN,
            BiomeDictionary.Type.HILLS,
            BiomeDictionary.Type.RIVER),
    RESONANT(
            HiveType.RESONANT,
            0.9f,
            HiveManager.genHelper.ground(Blocks.sand, Blocks.sandstone),
            false,
            BiomeDictionary.Type.SANDY,
            BiomeDictionary.Type.MESA,
            BiomeDictionary.Type.HOT,
            BiomeDictionary.Type.MAGICAL),
    DEEP(
            HiveType.DEEP,
            5.0f,
            new HiveGenUnderground(10, 15, 5),
            true,
            BiomeDictionary.Type.HILLS,
            BiomeDictionary.Type.MOUNTAIN,
            BiomeDictionary.Type.MAGICAL) {
        @Override
        public void postGen(World world, int x, int y, int z) {
            super.postGen(world, x, y, z);
            Random random = world.rand;
            FeatureOreVein.redstoneGen.generateVein(world, random, x + 1, y, z, 5);
            FeatureOreVein.redstoneGen.generateVein(world, random, x - 1, y, z, 5);
            FeatureOreVein.redstoneGen.generateVein(world, random, x, y + 1, z, 5);
            FeatureOreVein.redstoneGen.generateVein(world, random, x, y - 1, z, 5);
            FeatureOreVein.redstoneGen.generateVein(world, random, x, y, z + 1, 5);
            FeatureOreVein.redstoneGen.generateVein(world, random, x, y, z - 1, 5);
        }
    },
    INFERNAL(HiveType.INFERNAL, 50.0f, new HiveGenNether(0, 175, 6), true, BiomeDictionary.Type.NETHER) {
        @Override
        public void postGen(World world, int x, int y, int z) {
            super.postGen(world, x, y, z);
            Random random = world.rand;
            FeatureOreVein.netherQuartzGen.generateVein(world, random, x + 1, y, z, 4);
            FeatureOreVein.netherQuartzGen.generateVein(world, random, x - 1, y, z, 4);
            FeatureOreVein.netherQuartzGen.generateVein(world, random, x, y + 1, z, 4);
            FeatureOreVein.netherQuartzGen.generateVein(world, random, x, y - 1, z, 4);
            FeatureOreVein.netherQuartzGen.generateVein(world, random, x, y, z + 1, 4);
            FeatureOreVein.netherQuartzGen.generateVein(world, random, x, y, z - 1, 4);
        }
    },
    INFERNAL_OVERWORLD(
            HiveType.INFERNAL,
            0.95f,
            new HiveGenUnderground(5, 13, 6),
            true,
            BiomeDictionary.Type.MAGICAL,
            BiomeDictionary.Type.HOT) {
        @Override
        public void postGen(World world, int x, int y, int z) {
            super.postGen(world, x, y, z);
            Random random = world.rand;
            FeatureOreVein.glowstoneGen.generateVein(world, random, x + 1, y, z, world.rand.nextInt(4) + 1);
            FeatureOreVein.glowstoneGen.generateVein(world, random, x - 1, y, z, world.rand.nextInt(4) + 1);

            if (BlockUtil.canBlockReplaceAt(world, x, y + 1, z, Blocks.stone)) {
                world.setBlock(x, y + 1, z, Blocks.glowstone, 0, 3);
            }

            if (BlockUtil.canBlockReplaceAt(world, x, y - 1, z, Blocks.stone)) {
                world.setBlock(x, y - 1, z, Blocks.glowstone, 0, 3);
            }

            FeatureOreVein.glowstoneGen.generateVein(world, random, x, y, z + 1, world.rand.nextInt(4) + 1);
            FeatureOreVein.glowstoneGen.generateVein(world, random, x, y, z - 1, world.rand.nextInt(4) + 1);
        }
    },
    OBLIVION(HiveType.OBLIVION, 20.0f, new HiveGenOblivion(), true, BiomeDictionary.Type.END) {
        @Override
        public void postGen(World world, int x, int y, int z) {
            super.postGen(world, x, y, z);
            int obsidianSpikeHeight = world.rand.nextInt(8) + 3;

            for (int i = 1; i < obsidianSpikeHeight && y - i > 0; ++i) {
                world.setBlock(x, y - i, z, Blocks.obsidian, 0, 2);
            }
        }
    },
    OBLIVION_OVERWORLD(
            HiveType.OBLIVION,
            0.87f,
            new HiveGenUnderground(5, 5, 5),
            true,
            BiomeDictionary.Type.MAGICAL,
            BiomeDictionary.Type.COLD) {
        @Override
        public void postGen(World world, int x, int y, int z) {
            super.postGen(world, x, y, z);
            Random random = world.rand;

            FeatureOreVein.endStoneGen.generateVein(world, random, x + 1, y, z, world.rand.nextInt(6) + 1);
            FeatureOreVein.endStoneGen.generateVein(world, random, x - 1, y, z, world.rand.nextInt(6) + 1);
            FeatureOreVein.endStoneGen.generateVein(world, random, x, y + 1, z, world.rand.nextInt(3) + 1);
            FeatureOreVein.endStoneGen.generateVein(world, random, x, y - 1, z, world.rand.nextInt(3) + 1);
            FeatureOreVein.endStoneGen.generateVein(world, random, x, y, z + 1, world.rand.nextInt(6) + 1);
            FeatureOreVein.endStoneGen.generateVein(world, random, x, y, z - 1, world.rand.nextInt(6) + 1);
        }
    };

    private static boolean logSpawns = Config.logHiveSpawns;

    private final HiveType hiveType;
    private final float genChance;
    private final List<BiomeDictionary.Type> biomes = new ArrayList<BiomeDictionary.Type>();
    private final IHiveGen hiveGen;
    private final boolean spawnsIgnoreClimate;

    private HiveDescription(
            HiveType type,
            float chance,
            IHiveGen genType,
            boolean ignoreClimate,
            BiomeDictionary.Type... goodBiomeTypes) {
        hiveType = type;
        genChance = chance;
        hiveGen = genType;
        spawnsIgnoreClimate = ignoreClimate;
        for (BiomeDictionary.Type biomeType : goodBiomeTypes) {
            biomes.add(biomeType);
        }
    }

    @Override
    public IHiveGen getHiveGen() {
        return hiveGen;
    }

    @Override
    public Block getBlock() {
        return Config.hive;
    }

    @Override
    public int getMeta() {
        return hiveType.ordinal();
    }

    @Override
    public boolean isGoodBiome(BiomeGenBase biome) {
        BiomeDictionary.Type[] types = BiomeDictionary.getTypesForBiome(biome);
        for (BiomeDictionary.Type type : types) {
            if (biomes.contains(type)) {
                return true;
            }
        }

        return false;
    }

    @Override
    public boolean isGoodHumidity(EnumHumidity humidity) {
        return spawnsIgnoreClimate || hiveType.getOccupant().canWorkInHumidity(humidity);
    }

    @Override
    public boolean isGoodTemperature(EnumTemperature temperature) {
        return spawnsIgnoreClimate || hiveType.getOccupant().canWorkInTemperature(temperature);
    }

    @Override
    public float getGenChance() {
        return genChance;
    }

    @Override
    public void postGen(World world, int x, int y, int z) {
        if (logSpawns) {
            LogHelper.info("Spawned " + this.toString().toLowerCase(Locale.ENGLISH) + " hive at: X " + x + ", Y: " + y
                    + ", Z: " + z);
        }
    }
}