Optimize entity get (lazy nbt)

worldedit-bukkit/src/main/java/com/boydti/fawe/bukkit/adapter/v1_13_1/Spigot_v1_13_R2.java

@@ -29,6 +29,7 @@ import com.sk89q.worldedit.bukkit.BukkitAdapter;
 import com.sk89q.worldedit.bukkit.adapter.BukkitImplAdapter;
 import com.sk89q.worldedit.bukkit.adapter.CachedBukkitAdapter;
 import com.sk89q.worldedit.entity.BaseEntity;
+import com.sk89q.worldedit.entity.LazyBaseEntity;
 import com.sk89q.worldedit.internal.Constants;
 import com.sk89q.worldedit.math.BlockVector3;
 import com.sk89q.worldedit.registry.state.*;
@@ -36,6 +37,7 @@ import com.sk89q.worldedit.util.Direction;
 import com.sk89q.worldedit.world.biome.BiomeType;
 import com.sk89q.worldedit.world.block.BlockState;
 import com.sk89q.worldedit.world.block.*;
+import com.sk89q.worldedit.world.entity.EntityType;
 import com.sk89q.worldedit.world.registry.BlockMaterial;
 import net.minecraft.server.v1_13_R2.*;
 import org.bukkit.Bukkit;
@@ -57,6 +59,7 @@ import javax.annotation.Nullable;
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
 import java.util.*;
+import java.util.function.Supplier;
 import java.util.logging.Level;
 import java.util.stream.Collectors;
 
@@ -287,9 +290,16 @@ public final class Spigot_v1_13_R2 extends CachedBukkitAdapter implements Bukkit
         String id = getEntityId(mcEntity);
 
         if (id != null) {
+            EntityType type = com.sk89q.worldedit.world.entity.EntityTypes.get(id);
+            Supplier<CompoundTag> saveTag = new Supplier<CompoundTag>() {
+                @Override
+                public CompoundTag get() {
                     NBTTagCompound tag = new NBTTagCompound();
                     readEntityIntoTag(mcEntity, tag);
-            return new BaseEntity(com.sk89q.worldedit.world.entity.EntityTypes.get(id), (CompoundTag) toNative(tag));
+                    return (CompoundTag) toNative(tag);
+                }
+            };
+            return new LazyBaseEntity(type, saveTag);
         } else {
             return null;
         }

worldedit-bukkit/src/main/java/com/boydti/fawe/bukkit/v1_13/BukkitChunk_1_13_Copy.java

@@ -22,6 +22,9 @@ public class BukkitChunk_1_13_Copy extends BukkitChunk_1_13 {
 
     @Override
     public int[][] getCombinedIdArrays() {
+        if (this.sectionPalettes == null) {
+            return this.ids;
+        }
         for (int i = 0; i < ids.length; i++) {
             getIdArray(i);
         }
@@ -30,6 +33,7 @@ public class BukkitChunk_1_13_Copy extends BukkitChunk_1_13 {
 
     @Override
     public int[] getIdArray(int layer) {
+        if (this.sectionPalettes != null) {
             ChunkSection section = this.sectionPalettes[layer];
             int[] idsArray = this.ids[layer];
             if (section != null && idsArray == null) {
@@ -61,6 +65,8 @@ public class BukkitChunk_1_13_Copy extends BukkitChunk_1_13 {
             }
             return idsArray;
         }
+        return null;
+    }
 
     @Override
     public void setTile(int x, int y, int z, CompoundTag tile) {

worldedit-bukkit/src/main/java/com/sk89q/worldedit/bukkit/BukkitEntity.java

@@ -21,6 +21,7 @@ package com.sk89q.worldedit.bukkit;
 
 import static com.google.common.base.Preconditions.checkNotNull;
 
+import com.boydti.fawe.util.TaskManager;
 import com.sk89q.worldedit.bukkit.adapter.BukkitImplAdapter;
 import com.sk89q.worldedit.entity.BaseEntity;
 import com.sk89q.worldedit.entity.Entity;
@@ -29,6 +30,8 @@ import com.sk89q.worldedit.entity.metadata.EntityProperties;
 import com.sk89q.worldedit.extent.Extent;
 import com.sk89q.worldedit.util.Location;
 import com.sk89q.worldedit.world.NullWorld;
+import com.sk89q.worldedit.world.entity.EntityTypes;
+import org.bukkit.entity.EntityType;
 
 import java.lang.ref.WeakReference;
 
@@ -40,6 +43,7 @@ import javax.annotation.Nullable;
 public class BukkitEntity implements Entity {
 
     private final WeakReference<org.bukkit.entity.Entity> entityRef;
+    private final EntityType type;
 
     /**
      * Create a new instance.
@@ -48,6 +52,7 @@ public class BukkitEntity implements Entity {
      */
     public BukkitEntity(org.bukkit.entity.Entity entity) {
         checkNotNull(entity);
+        this.type = entity.getType();
         this.entityRef = new WeakReference<>(entity);
     }
 
@@ -81,6 +86,11 @@ public class BukkitEntity implements Entity {
         }
     }
 
+    @Override
+    public com.sk89q.worldedit.world.entity.EntityType getType() {
+        return EntityTypes.get(type.getName().toUpperCase());
+    }
+
     @Override
     public BaseEntity getState() {
         org.bukkit.entity.Entity entity = entityRef.get();

worldedit-bukkit/src/main/java/com/sk89q/worldedit/bukkit/BukkitWorld.java

@@ -18,6 +18,7 @@
 
 package com.sk89q.worldedit.bukkit;
 
+import com.boydti.fawe.Fawe;
 import com.sk89q.worldedit.EditSession;
 import com.sk89q.worldedit.WorldEdit;
 import com.sk89q.worldedit.WorldEditException;
@@ -82,6 +83,7 @@ public class BukkitWorld extends AbstractWorld {
 
     @Override
     public List<com.sk89q.worldedit.entity.Entity> getEntities(Region region) {
+        System.out.println(Fawe.isMainThread());
         World world = getWorld();
 
         List<Entity> ents = world.getEntities();
@@ -96,6 +98,7 @@ public class BukkitWorld extends AbstractWorld {
 
     @Override
     public List<com.sk89q.worldedit.entity.Entity> getEntities() {
+        System.out.println(Fawe.isMainThread());
         List<com.sk89q.worldedit.entity.Entity> list = new ArrayList<>();
         for (Entity entity : getWorld().getEntities()) {
             list.add(BukkitAdapter.adapt(entity));

worldedit-core/src/main/java/com/boydti/fawe/object/extent/FastWorldEditExtent.java

@@ -149,11 +149,13 @@ public class FastWorldEditExtent extends AbstractDelegateExtent implements HasFa
 
     @Override
     public List<? extends Entity> getEntities() {
+        System.out.println("World cp * " + world + " | " + world.getClass());
         return world.getEntities();
     }
 
     @Override
     public List<? extends Entity> getEntities(final Region region) {
+        System.out.println("World cp rg " + world + " | " + world.getClass());
         return world.getEntities(region);
     }
 

worldedit-core/src/main/java/com/sk89q/worldedit/entity/Entity.java

@@ -22,6 +22,7 @@ package com.sk89q.worldedit.entity;
 import com.sk89q.worldedit.extent.Extent;
 import com.sk89q.worldedit.util.Faceted;
 import com.sk89q.worldedit.util.Location;
+import com.sk89q.worldedit.world.entity.EntityType;
 
 import javax.annotation.Nullable;
 
@@ -62,6 +63,11 @@ public interface Entity extends Faceted {
      */
     boolean setLocation(Location location);
 
+    default EntityType getType() {
+        BaseEntity state = getState();
+        return state != null ? state.getType() : null;
+    }
+
     /**
      * Get the extent that this entity is on.
      *

worldedit-core/src/main/java/com/sk89q/worldedit/entity/LazyBaseEntity.java

@@ -0,0 +1,32 @@
+package com.sk89q.worldedit.entity;
+
+import com.boydti.fawe.Fawe;
+import com.boydti.fawe.util.TaskManager;
+import com.sk89q.jnbt.CompoundTag;
+import com.sk89q.worldedit.world.entity.EntityType;
+
+import javax.annotation.Nullable;
+import java.util.function.Supplier;
+
+public class LazyBaseEntity extends BaseEntity {
+    private Supplier<CompoundTag> saveTag;
+    public LazyBaseEntity(EntityType type, Supplier<CompoundTag> saveTag) {
+        super(type, null);
+        this.saveTag = saveTag;
+    }
+
+    @Nullable
+    @Override
+    public CompoundTag getNbtData() {
+        Supplier<CompoundTag> tmp = saveTag;
+        if (tmp != null) {
+            saveTag = null;
+            if (Fawe.isMainThread()) {
+                setNbtData(tmp.get());
+            } else {
+                setNbtData(TaskManager.IMP.sync(tmp));
+            }
+        }
+        return super.getNbtData();
+    }
+}

worldedit-core/src/main/java/com/sk89q/worldedit/function/operation/ForwardExtentCopy.java

@@ -27,12 +27,14 @@ import com.boydti.fawe.object.function.block.BiomeCopy;
 import com.boydti.fawe.object.function.block.CombinedBlockCopy;
 import com.boydti.fawe.object.function.block.SimpleBlockCopy;
 import com.boydti.fawe.util.MaskTraverser;
+import com.google.common.base.Predicate;
 import com.sk89q.worldedit.EditSession;
 import static com.google.common.base.Preconditions.checkArgument;
 import static com.google.common.base.Preconditions.checkNotNull;
 
 import com.google.common.collect.Lists;
 import com.sk89q.worldedit.WorldEditException;
+import com.sk89q.worldedit.entity.BaseEntity;
 import com.sk89q.worldedit.entity.Entity;
 import com.sk89q.worldedit.extent.Extent;
 import com.sk89q.worldedit.extent.clipboard.BlockArrayClipboard;
@@ -52,6 +54,9 @@ import com.sk89q.worldedit.math.BlockVector3;
 import com.sk89q.worldedit.math.transform.Identity;
 import com.sk89q.worldedit.math.transform.Transform;
 import com.sk89q.worldedit.regions.Region;
+import com.sk89q.worldedit.world.entity.EntityTypes;
+
+import javax.annotation.Nullable;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.stream.Collectors;
@@ -353,11 +358,15 @@ public class ForwardExtentCopy implements Operation {
         List<? extends Entity> entities;
         if (isCopyingEntities()) {
             // filter players since they can't be copied
-            entities = source.getEntities()
+            entities = source.getEntities(region)
                     .stream()
-                    .filter(entity -> entity.getState() != null &&
+                    .filter(new Predicate<Entity>() {
-                            !entity.getState().getType().getId().equals("minecraft:player") &&
+                        @Override
-                            region.contains(entity.getLocation().toBlockPoint()))
+                        public boolean apply(@Nullable Entity input) {
+                            BaseEntity state = input.getState();
+                            return state != null && state.getType() != EntityTypes.PLAYER;
+                        }
+                    })
                     .collect(Collectors.toList());
         } else {
             entities = new ArrayList<>();