обработка NBT в чанках

anvil-loader/src/main/java/mc/world/anvil/AnvilBlock.java

@@ -1,5 +1,6 @@
 package mc.world.anvil;
 
+import com.flowpowered.nbt.CompoundTag;
 import lombok.extern.slf4j.Slf4j;
 import mc.core.world.block.Block;
 import mc.core.world.block.BlockLocation;
@@ -49,6 +50,20 @@ public class AnvilBlock implements Block {
         return globalLocation;
     }
 
+    @Override
+    public CompoundTag getNBTData() {
+        CompoundTag compoundTag = chunkSection.getParent().getNbtByGlobalXYZ(
+                (chunkSection.getX() << 4) + location.getX(),
+                (chunkSection.getY() << 4) + location.getY(),
+                (chunkSection.getZ() << 4) + location.getZ()
+        );
+
+        compoundTag.getValue().remove("Items");
+        compoundTag.getValue().remove("Lock");
+
+        return compoundTag;
+    }
+
     @Override
     public String toString() {
         return "AnvilBlock{" +

anvil-loader/src/main/java/mc/world/anvil/AnvilChunk.java

@@ -20,6 +20,7 @@ public class AnvilChunk implements Chunk {
     private int z;
     private TByteList biomes = new TByteArrayList(256);
     private List<ChunkSection> sections;
+    private ListTag<CompoundTag> tileEntities;
 
     @SuppressWarnings("unchecked")
     AnvilChunk(CompoundTag chunkTag) {
@@ -29,6 +30,7 @@ public class AnvilChunk implements Chunk {
         this.z = ((IntTag) levelTagMap.get("zPos")).getValue();
 
         biomes.add(((ByteArrayTag) levelTagMap.get("Biomes")).getValue());
+        tileEntities = (ListTag<CompoundTag>) levelTagMap.get("TileEntities");
 
         List<CompoundTag> sections = ((ListTag<CompoundTag>) levelTagMap.get("Sections")).getValue();
         this.sections = new ArrayList<>(sections.size());
@@ -49,6 +51,18 @@ public class AnvilChunk implements Chunk {
         }
     }
 
+    CompoundTag getNbtByGlobalXYZ(int x, int y, int z) {
+        for (CompoundTag compoundTag : tileEntities.getValue()) {
+            CompoundMap compoundMap = compoundTag.getValue();
+            if (((IntTag)compoundMap.get("x")).getValue() == x
+                    && ((IntTag)compoundMap.get("y")).getValue() == y
+                    && ((IntTag)compoundMap.get("z")).getValue() == z) {
+                return compoundTag;
+            }
+        }
+        return null;
+    }
+
     @Override
     public ChunkSection getChunkSection(int height) {
         if (height > sections.size()-1) return null;

anvil-loader/src/test/java/mc/world/anvil/RegionTest.java

@@ -1,5 +1,9 @@
 package mc.world.anvil;
 
+import com.flowpowered.nbt.CompoundMap;
+import com.flowpowered.nbt.CompoundTag;
+import com.flowpowered.nbt.IntTag;
+import com.flowpowered.nbt.StringTag;
 import lombok.SneakyThrows;
 import mc.core.world.block.Block;
 import mc.core.world.block.BlockType;
@@ -60,6 +64,16 @@ class RegionTest {
         }
     }
 
+    private CompoundTag createExceptedNBT(Block block) {
+        CompoundMap compoundMap = new CompoundMap();
+        compoundMap.put(new IntTag("x", block.getLocation().getX()));
+        compoundMap.put(new IntTag("y", block.getLocation().getY()));
+        compoundMap.put(new IntTag("z", block.getLocation().getZ()));
+        compoundMap.put(new StringTag("id", block.getBlockType().getNamedId()));
+
+        return new CompoundTag("", compoundMap);
+    }
+
     private void checkSection2(ChunkSection chunkSection) {
         for (int y = 0; y < 16; y++) {
             for (int x = 0; x < 16; x++) {
@@ -70,6 +84,23 @@ class RegionTest {
                     // @formatter:off
                     if      (y == 0) assertEquals(BlockType.DIRT, block.getBlockType(), msg);
                     else if (y == 1) assertEquals(BlockType.GRASS, block.getBlockType(), msg);
+                    else if (y == 2) {
+                        if      ((x == 2 || x == 4 || x == 5) && z == 1) {
+                            assertEquals(BlockType.CHEST_NORTH, block.getBlockType(), msg);
+                            assertEquals(createExceptedNBT(block), block.getNBTData());
+                        } else if ((x == 2 || x == 3 || x == 5) && z == 6) {
+                            assertEquals(BlockType.CHEST_SOUTH, block.getBlockType(), msg);
+                            assertEquals(createExceptedNBT(block), block.getNBTData());
+                        } else if (x == 1 && (z == 2 || z == 3 || z == 5)) {
+                            assertEquals(BlockType.CHEST_WEST, block.getBlockType(), msg);
+                            assertEquals(createExceptedNBT(block), block.getNBTData());
+                        } else if (x == 6 && (z == 2 || z == 4 || z == 5)) {
+                            assertEquals(BlockType.CHEST_EAST, block.getBlockType(), msg);
+                            assertEquals(createExceptedNBT(block), block.getNBTData());
+                        } else {
+                            assertEquals(BlockType.AIR, block.getBlockType(), msg);
+                        }
+                    }
                     else             assertEquals(BlockType.AIR, block.getBlockType(), msg);
                     // @formatter:on
                 }