init material grabber

2025-04-21 00:44:36 -07:00
parent 9a40363c1f
commit 6d7db482c0
6 changed files with 204 additions and 35 deletions
@@ -0,0 +1,69 @@
+class StructureMaterials {
+    instance;
+    materials;
+
+    constructor(instance) {
+        this.instance = instance;
+        this.materials = {};
+    }
+
+    refresh() {
+        this.clear();
+        this.populateActive();
+    }
+
+    populateAll() {
+        for (const layer = 0; layer < this.instance.getMaxLayer(); layer++)
+            this.getLayer(layer)
+    }
+
+    populateLayer(layer) {
+        for (const block of this.instance.getLayerBlocks(layer))
+            this.countBlock(block)
+    }
+
+    populateActive() {
+        for (const block of this.instance.getActiveBlocks())
+            this.countBlock(block)
+    }
+
+    countBlock(block) {
+        const itemStack = block?.getItemStack();
+        const typeId = itemStack?.typeId;
+        if (!typeId) return;
+        if (!this.materials[typeId])
+            this.materials[typeId] = { count: 0, stackSize: itemStack.maxAmount };
+        this.materials[typeId].count++;
+    }
+
+    clear() {
+        for (const key in this.materials)
+            delete this.materials[key];
+    }
+
+    toString() {
+        let message = [];
+        for (const blockType in this.materials) {
+            let count = this.materials[blockType].count;
+            let countStr = '';
+            const stackSize = this.materials[blockType].stackSize;
+            const fullShulker = 27 * stackSize;
+            if (count >= fullShulker)
+                countStr = `${Math.floor(count / fullShulker)} sb`;
+            if (count > fullShulker)
+                countStr += ' + ';
+            count %= fullShulker;
+            if (count >= stackSize)
+                countStr += `${Math.floor(count / stackSize)} stack`;
+            if (count > stackSize)
+                countStr += ' + ';
+            count %= stackSize;
+            if (count > 0)
+                countStr += count;
+            message.push(`  ${blockType}: ${countStr}`);
+        }
+        return message.sort().join('\n');
+    }
+}
+
+export { StructureMaterials };