add entities to chunk column

PrismarineJS · Jul 15, 2024 · 4f2b3b3 · 4f2b3b3
1 parent 2dc9a9d
commit 4f2b3b3
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/pc/common/CommonChunkColumn.js b/src/pc/common/CommonChunkColumn.js
@@ -5,6 +5,7 @@ class CommonChunkColumn {
     this.registry = registry
     // Just a collection of deserialized NBTs
     this.blockEntities = {}
+    this.entities = {}
     this.minY = 0
   }
 
@@ -66,6 +67,26 @@ class CommonChunkColumn {
       this.setBlockEntity({ x: entity.x.value >> 4, y: entity.y.value, z: entity.z.value >> 4 }, entity)
     }
   }
+
+  // Entities
+
+  getEntity (id) {
+    return this.entities[id]
+  }
+
+  setEntity (id, entity) {
+    this.entities[id] = entity
+  }
+
+  removeEntity (id) {
+    delete this.entities[id]
+  }
+
+  loadEntities (entities) {
+    for (const id in entities) {
+      this.addEntity(id, entities[id])
+    }
+  }
 }
 
 module.exports = CommonChunkColumn