feat: gen redstone and diamond

natsuneko-laboratory · Aug 7, 2024 · c0ac70f · c0ac70f
1 parent 5e514e8
commit c0ac70f
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/src/main/java/com/natsuneko/floatingisland/FloatingIslandChunkGenerator.java b/src/main/java/com/natsuneko/floatingisland/FloatingIslandChunkGenerator.java
@@ -12,11 +12,13 @@
 import net.minecraft.world.WorldEntitySpawner;
 import net.minecraft.world.WorldType;
 import net.minecraft.world.biome.Biome;
+import net.minecraft.world.biome.BiomeDecorator;
 import net.minecraft.world.chunk.Chunk;
 import net.minecraft.world.chunk.ChunkPrimer;
 import net.minecraft.world.gen.*;
 import net.minecraft.world.gen.feature.WorldGenDungeons;
 import net.minecraft.world.gen.feature.WorldGenLakes;
+import net.minecraft.world.gen.feature.WorldGenMinable;
 import net.minecraft.world.gen.structure.*;
 import net.minecraftforge.event.ForgeEventFactory;
 import net.minecraftforge.event.terraingen.InitMapGenEvent;
@@ -470,6 +472,12 @@ public void populate(int x, int z) {
 
         biome.decorate(this._world, this._rand, new BlockPos(i, 0, j));
 
+        // Adjust Difficulty - RedStone and Diamond
+        if (TerrainGen.populate(this, this._world, this._rand, x, z, hasVillageGenerated, PopulateChunkEvent.Populate.EventType.CUSTOM)) {
+            FloatingIslandBiomeDecorator decorator = new FloatingIslandBiomeDecorator();
+            decorator.generateAdditionalOres(this._world, this._rand, biome, new BlockPos(i, 0, j));
+        }
+
         if (TerrainGen.populate(this, this._world, this._rand, x, z, hasVillageGenerated, PopulateChunkEvent.Populate.EventType.ANIMALS)) {
             WorldEntitySpawner.performWorldGenSpawning(this._world, biome, i + 8, j + 8, 16, 16, this._rand);
         }
@@ -589,4 +597,26 @@ public void recreateStructures(Chunk chunkIn, int x, int z) {
     public boolean isInsideStructure(World world, String s, BlockPos blockPos) {
         return false;
     }
+
+    static class FloatingIslandBiomeDecorator extends BiomeDecorator {
+        public void generateAdditionalOres(World worldIn, Random random, Biome biome, BlockPos pos) {
+            if (this.decorating) {
+                throw new RuntimeException("Already decorating");
+            } else {
+                this.decorating = true;
+                this.chunkPos = pos;
+
+                this.chunkProviderSettings = ChunkGeneratorSettings.Factory.jsonToFactory(worldIn.getWorldInfo().getGeneratorOptions()).build();
+                this.redstoneGen = new WorldGenMinable(Blocks.REDSTONE_ORE.getDefaultState(), this.chunkProviderSettings.redstoneSize);
+                this.diamondGen = new WorldGenMinable(Blocks.DIAMOND_ORE.getDefaultState(), this.chunkProviderSettings.diamondSize);
+
+                if (net.minecraftforge.event.terraingen.TerrainGen.generateOre(worldIn, random, redstoneGen, chunkPos, net.minecraftforge.event.terraingen.OreGenEvent.GenerateMinable.EventType.REDSTONE))
+                    this.genStandardOre1(worldIn, random, this.chunkProviderSettings.redstoneCount, this.redstoneGen, 0, 42);
+                if (net.minecraftforge.event.terraingen.TerrainGen.generateOre(worldIn, random, diamondGen, chunkPos, net.minecraftforge.event.terraingen.OreGenEvent.GenerateMinable.EventType.DIAMOND))
+                    this.genStandardOre1(worldIn, random, this.chunkProviderSettings.diamondCount, this.diamondGen, 0, 42);
+
+                this.decorating = false;
+            }
+        }
+    }
 }