Initial attempt at restoring regen functionality (#475)

worldedit-forge/src/main/java/com/sk89q/worldedit/forge/ForgeWorld.java

@@ -21,6 +21,7 @@ package com.sk89q.worldedit.forge;
 
 import static com.google.common.base.Preconditions.checkNotNull;
 
+import com.google.common.io.Files;
 import com.sk89q.jnbt.CompoundTag;
 import com.sk89q.worldedit.EditSession;
 import com.sk89q.worldedit.MaxChangedBlocksException;
@@ -33,6 +34,7 @@ import com.sk89q.worldedit.internal.Constants;
 import com.sk89q.worldedit.math.BlockVector2;
 import com.sk89q.worldedit.math.BlockVector3;
 import com.sk89q.worldedit.math.Vector3;
+import com.sk89q.worldedit.regions.CuboidRegion;
 import com.sk89q.worldedit.regions.Region;
 import com.sk89q.worldedit.util.Direction;
 import com.sk89q.worldedit.util.Location;
@@ -55,6 +57,7 @@ import net.minecraft.item.Item;
 import net.minecraft.item.ItemStack;
 import net.minecraft.item.ItemUseContext;
 import net.minecraft.nbt.NBTTagCompound;
+import net.minecraft.server.MinecraftServer;
 import net.minecraft.tileentity.TileEntity;
 import net.minecraft.util.EnumActionResult;
 import net.minecraft.util.ResourceLocation;
@@ -62,6 +65,9 @@ import net.minecraft.util.math.BlockPos;
 import net.minecraft.world.World;
 import net.minecraft.world.WorldServer;
 import net.minecraft.world.chunk.Chunk;
+import net.minecraft.world.chunk.IChunkProvider;
+import net.minecraft.world.chunk.storage.AnvilSaveHandler;
+import net.minecraft.world.gen.ChunkProviderServer;
 import net.minecraft.world.gen.feature.BigBrownMushroomFeature;
 import net.minecraft.world.gen.feature.BigRedMushroomFeature;
 import net.minecraft.world.gen.feature.BigTreeFeature;
@@ -79,7 +85,9 @@ import net.minecraft.world.gen.feature.SwampTreeFeature;
 import net.minecraft.world.gen.feature.TallTaigaTreeFeature;
 import net.minecraft.world.gen.feature.TreeFeature;
 import net.minecraft.world.storage.WorldInfo;
+import net.minecraftforge.common.DimensionManager;
 
+import java.io.File;
 import java.lang.ref.WeakReference;
 import java.util.ArrayList;
 import java.util.List;
@@ -280,46 +288,41 @@ public class ForgeWorld extends AbstractWorld {
 
     @Override
     public boolean regenerate(Region region, EditSession editSession) {
-        // TODO Fix for 1.13
+        // Don't even try to regen if it's going to fail.
+        IChunkProvider provider = getWorld().getChunkProvider();
+        if (!(provider instanceof ChunkProviderServer)) {
             return false;
-//        // Don't even try to regen if it's going to fail.
+        }
-//        IChunkProvider provider = getWorld().getChunkProvider();
+
-//        if (!(provider instanceof ChunkProviderServer)) {
+        File saveFolder = Files.createTempDir();
-//            return false;
+        // register this just in case something goes wrong
-//        }
+        // normally it should be deleted at the end of this method
-//
+        saveFolder.deleteOnExit();
-//        File saveFolder = Files.createTempDir();
+        try {
-//        // register this just in case something goes wrong
+            WorldServer originalWorld = (WorldServer) getWorld();
-//        // normally it should be deleted at the end of this method
+
-//        saveFolder.deleteOnExit();
+            MinecraftServer server = originalWorld.getServer();
-//
+            AnvilSaveHandler saveHandler = new AnvilSaveHandler(saveFolder, originalWorld.getSaveHandler().getWorldDirectory().getName(), server, server.getDataFixer());
-//        WorldServer originalWorld = (WorldServer) getWorld();
+            World freshWorld = new WorldServer(server, saveHandler, originalWorld.getSavedDataStorage(), originalWorld.getWorldInfo(), originalWorld.dimension.getType(), originalWorld.profiler).init();
-//
+
-//        MinecraftServer server = originalWorld.getServer();
+            // Pre-gen all the chunks
-//        AnvilSaveHandler saveHandler = new AnvilSaveHandler(saveFolder, originalWorld.getSaveHandler().getWorldDirectory().getName(), server, server.getDataFixer());
+            // We need to also pull one more chunk in every direction
-//        World freshWorld = new WorldServer(server, saveHandler, originalWorld.getSavedDataStorage(), originalWorld.getWorldInfo(), originalWorld.dimension.getType(), originalWorld.profiler).func_212251_i__();
+            CuboidRegion expandedPreGen = new CuboidRegion(region.getMinimumPoint().subtract(16, 0, 16), region.getMaximumPoint().add(16, 0, 16));
-//
+            for (BlockVector2 chunk : expandedPreGen.getChunks()) {
-//        // Pre-gen all the chunks
+                freshWorld.getChunk(chunk.getBlockX(), chunk.getBlockZ());
-//        // We need to also pull one more chunk in every direction
+            }
-//        CuboidRegion expandedPreGen = new CuboidRegion(region.getMinimumPoint().subtract(16, 0, 16), region.getMaximumPoint().add(16, 0, 16));
+
-//        for (BlockVector2 chunk : expandedPreGen.getChunks()) {
+            ForgeWorld from = new ForgeWorld(freshWorld);
-//            freshWorld.getChunk(chunk.getBlockX(), chunk.getBlockZ());
+            for (BlockVector3 vec : region) {
-//        }
+                editSession.setBlock(vec, from.getFullBlock(vec));
-//
+            }
-//        ForgeWorld from = new ForgeWorld(freshWorld);
+        } catch (MaxChangedBlocksException e) {
-//        try {
+            throw new RuntimeException(e);
-//            for (BlockVector3 vec : region) {
+        } finally {
-//                editSession.setBlock(vec, from.getFullBlock(vec));
+            saveFolder.delete();
-//            }
+        }
-//        } catch (MaxChangedBlocksException e) {
+
-//            throw new RuntimeException(e);
+        return true;
-//        } finally {
-//            saveFolder.delete();
-//            DimensionManager.setWorld(originalWorld.dimension.getType(), null, server);
-//            DimensionManager.setWorld(originalWorld.dimension.getType(), originalWorld, server);
-//        }
-//
-//        return true;
     }
 
     @Nullable