From 33548c8d55e7b968fe0bf581223b84efed7e5a0d Mon Sep 17 00:00:00 2001
From: Jesse Boyd <jessepaleg@gmail.com>
Date: Tue, 27 Dec 2016 05:47:21 +1100
Subject: [PATCH] Regen for forge/sponge

---
 .../boydti/fawe/forge/MutableGenLayer.java    | 26 ++++++
 .../boydti/fawe/forge/v0/ForgeQueue_All.java  | 83 ++++++++++++++++++-
 .../fawe/sponge/v1_11/MutableGenLayer.java    | 26 ++++++
 .../fawe/sponge/v1_11/SpongeQueue_1_11.java   | 83 ++++++++++++++++++-
 4 files changed, 211 insertions(+), 7 deletions(-)
 create mode 100644 forge111/src/main/java/com/boydti/fawe/forge/MutableGenLayer.java
 create mode 100644 sponge/src/main/java/com/boydti/fawe/sponge/v1_11/MutableGenLayer.java

diff --git a/forge111/src/main/java/com/boydti/fawe/forge/MutableGenLayer.java b/forge111/src/main/java/com/boydti/fawe/forge/MutableGenLayer.java
new file mode 100644
index 00000000..70b06c13
--- /dev/null
+++ b/forge111/src/main/java/com/boydti/fawe/forge/MutableGenLayer.java
@@ -0,0 +1,26 @@
+package com.boydti.fawe.forge;
+
+import java.util.Arrays;
+import net.minecraft.world.gen.layer.GenLayer;
+import net.minecraft.world.gen.layer.IntCache;
+
+public class MutableGenLayer extends GenLayer {
+
+    private int biome;
+
+    public MutableGenLayer(long seed) {
+        super(seed);
+    }
+
+    public MutableGenLayer set(int biome) {
+        this.biome = biome;
+        return this;
+    }
+
+    @Override
+    public int[] getInts(int areaX, int areaY, int areaWidth, int areaHeight) {
+        int[] biomes = IntCache.getIntCache(areaWidth * areaHeight);
+        Arrays.fill(biomes, biome);
+        return biomes;
+    }
+}
diff --git a/forge111/src/main/java/com/boydti/fawe/forge/v0/ForgeQueue_All.java b/forge111/src/main/java/com/boydti/fawe/forge/v0/ForgeQueue_All.java
index 4314cdb0..1ea5f764 100644
--- a/forge111/src/main/java/com/boydti/fawe/forge/v0/ForgeQueue_All.java
+++ b/forge111/src/main/java/com/boydti/fawe/forge/v0/ForgeQueue_All.java
@@ -4,6 +4,7 @@ import com.boydti.fawe.FaweCache;
 import com.boydti.fawe.example.CharFaweChunk;
 import com.boydti.fawe.example.NMSMappedFaweQueue;
 import com.boydti.fawe.forge.ForgePlayer;
+import com.boydti.fawe.forge.MutableGenLayer;
 import com.boydti.fawe.object.FaweChunk;
 import com.boydti.fawe.object.FawePlayer;
 import com.boydti.fawe.util.MainUtil;
@@ -26,6 +27,7 @@ import java.util.Map;
 import java.util.Set;
 import java.util.UUID;
 import net.minecraft.block.Block;
+import net.minecraft.block.BlockFalling;
 import net.minecraft.block.state.IBlockState;
 import net.minecraft.entity.Entity;
 import net.minecraft.entity.EntityList;
@@ -46,13 +48,17 @@ import net.minecraft.util.math.ChunkPos;
 import net.minecraft.world.EnumSkyBlock;
 import net.minecraft.world.World;
 import net.minecraft.world.WorldServer;
+import net.minecraft.world.biome.BiomeCache;
+import net.minecraft.world.biome.BiomeProvider;
 import net.minecraft.world.chunk.BlockStateContainer;
 import net.minecraft.world.chunk.Chunk;
 import net.minecraft.world.chunk.IChunkGenerator;
 import net.minecraft.world.chunk.IChunkProvider;
 import net.minecraft.world.chunk.NibbleArray;
 import net.minecraft.world.chunk.storage.ExtendedBlockStorage;
+import net.minecraft.world.gen.ChunkProviderOverworld;
 import net.minecraft.world.gen.ChunkProviderServer;
+import net.minecraft.world.storage.WorldInfo;
 import net.minecraftforge.common.DimensionManager;
 
 public class ForgeQueue_All extends NMSMappedFaweQueue<World, Chunk, ExtendedBlockStorage[], ExtendedBlockStorage> {
@@ -62,6 +68,16 @@ public class ForgeQueue_All extends NMSMappedFaweQueue<World, Chunk, ExtendedBlo
     protected final static Field fieldTickingBlockCount;
     protected final static Field fieldNonEmptyBlockCount;
 
+    protected static Field fieldBiomes;
+    protected static Field fieldChunkGenerator;
+    protected static Field fieldSeed;
+    protected static Field fieldBiomeCache;
+    protected static Field fieldBiomes2;
+    protected static Field fieldGenLayer1;
+    protected static Field fieldGenLayer2;
+
+    private static MutableGenLayer genLayer;
+
     static {
         try {
             Class<?> converter = Class.forName("com.sk89q.worldedit.forge.NBTConverter");
@@ -70,6 +86,21 @@ public class ForgeQueue_All extends NMSMappedFaweQueue<World, Chunk, ExtendedBlo
             methodFromNative.setAccessible(true);
             methodToNative.setAccessible(true);
 
+            fieldBiomes = ChunkProviderOverworld.class.getDeclaredField("field_185981_C"); // biomesForGeneration
+            fieldBiomes.setAccessible(true);
+            fieldChunkGenerator = ChunkProviderServer.class.getDeclaredField("field_186029_c"); // chunkGenerator
+            fieldChunkGenerator.setAccessible(true);
+            fieldSeed = WorldInfo.class.getDeclaredField("field_76100_a"); // randomSeed
+            fieldSeed.setAccessible(true);
+            fieldBiomeCache = BiomeProvider.class.getDeclaredField("field_76942_f"); // biomeCache
+            fieldBiomeCache.setAccessible(true);
+            fieldBiomes2 = BiomeProvider.class.getDeclaredField("field_76943_g"); // biomesToSpawnIn
+            fieldBiomes2.setAccessible(true);
+            fieldGenLayer1 = BiomeProvider.class.getDeclaredField("field_76944_d"); // genBiomes
+            fieldGenLayer2 = BiomeProvider.class.getDeclaredField("field_76945_e"); // biomeIndexLayer
+            fieldGenLayer1.setAccessible(true);
+            fieldGenLayer2.setAccessible(true);
+
             fieldTickingBlockCount = ExtendedBlockStorage.class.getDeclaredField("field_76683_c");
             fieldNonEmptyBlockCount = ExtendedBlockStorage.class.getDeclaredField("field_76682_b");
             fieldTickingBlockCount.setAccessible(true);
@@ -169,13 +200,57 @@ public class ForgeQueue_All extends NMSMappedFaweQueue<World, Chunk, ExtendedBlo
     }
 
     @Override
-    public boolean regenerateChunk(World world, int x, int z, BaseBiome biome, Long seed) {
+    public boolean regenerateChunk(net.minecraft.world.World world, int x, int z, BaseBiome biome, Long seed) {
+        if (biome != null) {
+            try {
+                if (seed == null) {
+                    seed = world.getSeed();
+                }
+                nmsWorld.getWorldInfo().getSeed();
+                boolean result;
+                ChunkProviderOverworld generator = new ChunkProviderOverworld(nmsWorld, seed, false, "");
+                net.minecraft.world.biome.Biome base = net.minecraft.world.biome.Biome.getBiome(biome.getId());
+                fieldBiomes.set(generator, new net.minecraft.world.biome.Biome[]{base});
+                boolean cold = base.getTemperature() <= 1;
+                IChunkGenerator existingGenerator = nmsWorld.getChunkProvider().chunkGenerator;
+                long existingSeed = world.getSeed();
+                {
+                    if (genLayer == null) genLayer = new MutableGenLayer(seed);
+                    genLayer.set(biome.getId());
+                    Object existingGenLayer1 = fieldGenLayer1.get(nmsWorld.provider.getBiomeProvider());
+                    Object existingGenLayer2 = fieldGenLayer2.get(nmsWorld.provider.getBiomeProvider());
+                    fieldGenLayer1.set(nmsWorld.provider.getBiomeProvider(), genLayer);
+                    fieldGenLayer2.set(nmsWorld.provider.getBiomeProvider(), genLayer);
+
+                    fieldSeed.set(nmsWorld.getWorldInfo(), seed);
+
+                    ReflectionUtils.setFailsafeFieldValue(fieldBiomeCache, this.nmsWorld.provider.getBiomeProvider(), new BiomeCache(this.nmsWorld.provider.getBiomeProvider()));
+
+                    ReflectionUtils.setFailsafeFieldValue(fieldChunkGenerator, this.nmsWorld.getChunkProvider(), generator);
+
+                    result = regenerateChunk(world, x, z);
+
+                    ReflectionUtils.setFailsafeFieldValue(fieldChunkGenerator, this.nmsWorld.getChunkProvider(), existingGenerator);
+
+                    fieldSeed.set(nmsWorld.getWorldInfo(), existingSeed);
+
+                    fieldGenLayer1.set(nmsWorld.provider.getBiomeProvider(), existingGenLayer1);
+                    fieldGenLayer2.set(nmsWorld.provider.getBiomeProvider(), existingGenLayer2);
+                }
+                return result;
+            } catch (Throwable e) {
+                e.printStackTrace();
+            }
+        }
+        return regenerateChunk(world, x, z);
+    }
+
+    public boolean regenerateChunk(World world, int x, int z) {
         IChunkProvider provider = world.getChunkProvider();
         if (!(provider instanceof ChunkProviderServer)) {
             return false;
         }
-
-
+        BlockFalling.fallInstantly = true;
         try {
             ChunkProviderServer chunkServer = (ChunkProviderServer) provider;
             IChunkGenerator gen = chunkServer.chunkGenerator;
@@ -222,6 +297,8 @@ public class ForgeQueue_All extends NMSMappedFaweQueue<World, Chunk, ExtendedBlo
         } catch (Throwable t) {
             MainUtil.handleError(t);
             return false;
+        } finally {
+            BlockFalling.fallInstantly = false;
         }
     }
 
diff --git a/sponge/src/main/java/com/boydti/fawe/sponge/v1_11/MutableGenLayer.java b/sponge/src/main/java/com/boydti/fawe/sponge/v1_11/MutableGenLayer.java
new file mode 100644
index 00000000..14e13bea
--- /dev/null
+++ b/sponge/src/main/java/com/boydti/fawe/sponge/v1_11/MutableGenLayer.java
@@ -0,0 +1,26 @@
+package com.boydti.fawe.sponge.v1_11;
+
+import java.util.Arrays;
+import net.minecraft.world.gen.layer.GenLayer;
+import net.minecraft.world.gen.layer.IntCache;
+
+public class MutableGenLayer extends GenLayer {
+
+    private int biome;
+
+    public MutableGenLayer(long seed) {
+        super(seed);
+    }
+
+    public MutableGenLayer set(int biome) {
+        this.biome = biome;
+        return this;
+    }
+
+    @Override
+    public int[] getInts(int areaX, int areaY, int areaWidth, int areaHeight) {
+        int[] biomes = IntCache.getIntCache(areaWidth * areaHeight);
+        Arrays.fill(biomes, biome);
+        return biomes;
+    }
+}
diff --git a/sponge/src/main/java/com/boydti/fawe/sponge/v1_11/SpongeQueue_1_11.java b/sponge/src/main/java/com/boydti/fawe/sponge/v1_11/SpongeQueue_1_11.java
index 75a9851b..2aea836d 100644
--- a/sponge/src/main/java/com/boydti/fawe/sponge/v1_11/SpongeQueue_1_11.java
+++ b/sponge/src/main/java/com/boydti/fawe/sponge/v1_11/SpongeQueue_1_11.java
@@ -27,6 +27,7 @@ import java.util.Map;
 import java.util.Set;
 import java.util.UUID;
 import net.minecraft.block.Block;
+import net.minecraft.block.BlockFalling;
 import net.minecraft.block.state.IBlockState;
 import net.minecraft.entity.Entity;
 import net.minecraft.entity.EntityList;
@@ -47,13 +48,17 @@ import net.minecraft.util.math.ChunkPos;
 import net.minecraft.world.EnumSkyBlock;
 import net.minecraft.world.World;
 import net.minecraft.world.WorldServer;
+import net.minecraft.world.biome.BiomeCache;
+import net.minecraft.world.biome.BiomeProvider;
 import net.minecraft.world.chunk.BlockStateContainer;
 import net.minecraft.world.chunk.Chunk;
 import net.minecraft.world.chunk.IChunkGenerator;
 import net.minecraft.world.chunk.IChunkProvider;
 import net.minecraft.world.chunk.NibbleArray;
 import net.minecraft.world.chunk.storage.ExtendedBlockStorage;
+import net.minecraft.world.gen.ChunkProviderOverworld;
 import net.minecraft.world.gen.ChunkProviderServer;
+import net.minecraft.world.storage.WorldInfo;
 import org.spongepowered.api.Sponge;
 
 public class SpongeQueue_1_11 extends NMSMappedFaweQueue<World, net.minecraft.world.chunk.Chunk, ExtendedBlockStorage[], ExtendedBlockStorage> {
@@ -66,6 +71,15 @@ public class SpongeQueue_1_11 extends NMSMappedFaweQueue<World, net.minecraft.wo
     protected final static Field fieldId2ChunkMap;
     protected final static Field fieldChunkGenerator;
 
+    protected static Field fieldBiomes;
+    protected static Field fieldSeed;
+    protected static Field fieldBiomeCache;
+    protected static Field fieldBiomes2;
+    protected static Field fieldGenLayer1;
+    protected static Field fieldGenLayer2;
+
+    private static MutableGenLayer genLayer;
+
     static {
         try {
             Class<?> converter = Class.forName("com.sk89q.worldedit.sponge.nms.NBTConverter");
@@ -79,6 +93,19 @@ public class SpongeQueue_1_11 extends NMSMappedFaweQueue<World, net.minecraft.wo
             fieldId2ChunkMap.setAccessible(true);
             fieldChunkGenerator.setAccessible(true);
 
+            fieldBiomes = ChunkProviderOverworld.class.getDeclaredField("field_185981_C"); // biomesForGeneration
+            fieldBiomes.setAccessible(true);
+            fieldSeed = WorldInfo.class.getDeclaredField("field_76100_a"); // randomSeed
+            fieldSeed.setAccessible(true);
+            fieldBiomeCache = BiomeProvider.class.getDeclaredField("field_76942_f"); // biomeCache
+            fieldBiomeCache.setAccessible(true);
+            fieldBiomes2 = BiomeProvider.class.getDeclaredField("field_76943_g"); // biomesToSpawnIn
+            fieldBiomes2.setAccessible(true);
+            fieldGenLayer1 = BiomeProvider.class.getDeclaredField("field_76944_d"); // genBiomes
+            fieldGenLayer2 = BiomeProvider.class.getDeclaredField("field_76945_e"); // biomeIndexLayer
+            fieldGenLayer1.setAccessible(true);
+            fieldGenLayer2.setAccessible(true);
+
             fieldTickingBlockCount = ExtendedBlockStorage.class.getDeclaredField("field_76683_c");
             fieldNonEmptyBlockCount = ExtendedBlockStorage.class.getDeclaredField("field_76682_b");
             fieldTickingBlockCount.setAccessible(true);
@@ -177,14 +204,12 @@ public class SpongeQueue_1_11 extends NMSMappedFaweQueue<World, net.minecraft.wo
         return getWorld().getChunkProvider().getLoadedChunk(x, z) != null;
     }
 
-    @Override
-    public boolean regenerateChunk(net.minecraft.world.World world, int x, int z, BaseBiome biome, Long seed) {
+    public boolean regenerateChunk(net.minecraft.world.World world, int x, int z) {
         IChunkProvider provider = world.getChunkProvider();
         if (!(provider instanceof ChunkProviderServer)) {
             return false;
         }
-
-
+        BlockFalling.fallInstantly = true;
         try {
             ChunkProviderServer chunkServer = (ChunkProviderServer) provider;
             IChunkGenerator gen = (IChunkGenerator) fieldChunkGenerator.get(chunkServer);
@@ -232,9 +257,59 @@ public class SpongeQueue_1_11 extends NMSMappedFaweQueue<World, net.minecraft.wo
         } catch (Throwable t) {
             MainUtil.handleError(t);
             return false;
+        } finally {
+            BlockFalling.fallInstantly = false;
         }
     }
 
+    @Override
+    public boolean regenerateChunk(net.minecraft.world.World world, int x, int z, BaseBiome biome, Long seed) {
+        if (biome != null) {
+            try {
+                if (seed == null) {
+                    seed = world.getSeed();
+                }
+                nmsWorld.getWorldInfo().getSeed();
+                boolean result;
+                ChunkProviderOverworld generator = new ChunkProviderOverworld(nmsWorld, seed, false, "");
+                net.minecraft.world.biome.Biome base = net.minecraft.world.biome.Biome.getBiome(biome.getId());
+                net.minecraft.world.biome.Biome[] existingBiomes = new net.minecraft.world.biome.Biome[256];
+                Arrays.fill(existingBiomes, base);
+                fieldBiomes.set(generator, existingBiomes);
+                boolean cold = base.getTemperature() <= 1;
+                IChunkGenerator existingGenerator = (IChunkGenerator) fieldChunkGenerator.get(nmsWorld.getChunkProvider());
+                long existingSeed = world.getSeed();
+                {
+                    if (genLayer == null) genLayer = new MutableGenLayer(seed);
+                    genLayer.set(biome.getId());
+                    Object existingGenLayer1 = fieldGenLayer1.get(nmsWorld.provider.getBiomeProvider());
+                    Object existingGenLayer2 = fieldGenLayer2.get(nmsWorld.provider.getBiomeProvider());
+                    fieldGenLayer1.set(nmsWorld.provider.getBiomeProvider(), genLayer);
+                    fieldGenLayer2.set(nmsWorld.provider.getBiomeProvider(), genLayer);
+
+                    fieldSeed.set(nmsWorld.getWorldInfo(), seed);
+
+                    ReflectionUtils.setFailsafeFieldValue(fieldBiomeCache, this.nmsWorld.provider.getBiomeProvider(), new BiomeCache(this.nmsWorld.provider.getBiomeProvider()));
+
+                    ReflectionUtils.setFailsafeFieldValue(fieldChunkGenerator, this.nmsWorld.getChunkProvider(), generator);
+
+                    result = regenerateChunk(world, x, z);
+
+                    ReflectionUtils.setFailsafeFieldValue(fieldChunkGenerator, this.nmsWorld.getChunkProvider(), existingGenerator);
+
+                    fieldSeed.set(nmsWorld.getWorldInfo(), existingSeed);
+
+                    fieldGenLayer1.set(nmsWorld.provider.getBiomeProvider(), existingGenLayer1);
+                    fieldGenLayer2.set(nmsWorld.provider.getBiomeProvider(), existingGenLayer2);
+                }
+                return result;
+            } catch (Throwable e) {
+                e.printStackTrace();
+            }
+        }
+        return regenerateChunk(world, x, z);
+    }
+
     @Override
     public boolean loadChunk(net.minecraft.world.World world, int x, int z, boolean generate) {
         return getCachedSections(world, x, z) != null;