mindcraft 0.1.4-0

package/src/agent/tasks/construction_tasks.js

@@ -0,0 +1,1104 @@
+import {Vec3} from 'vec3';
+
+export class ConstructionTaskValidator {
+    constructor(data, agent) {
+        this.blueprint = new Blueprint(data.blueprint);
+        this.agent = agent;
+    }
+    validate() {
+        try {
+            //todo: somehow make this more of a percentage or something
+            // console.log('Validating task...');
+            let valid = false;
+            let score = 0;
+            let result = this.blueprint.check(this.agent.bot);
+            if (result.mismatches.length === 0) {
+                valid = true;
+                console.log('Task is complete');
+            }
+            let total_blocks = result.mismatches.length + result.matches.length;
+            score = (result.matches.length / total_blocks) * 100;
+            console.log(`Task score: ${score}%`);
+            return {
+                "valid": valid, 
+                "score": score
+            };
+        } catch (error) {
+            console.error('Error validating task:', error);
+            return {
+                "valid": false,
+                "score": 0
+            };
+        }
+    }
+}
+
+export function resetConstructionWorld(bot, blueprint) {
+    console.log('Resetting world...');
+    const starting_position = blueprint.levels[0].coordinates;
+    const length = blueprint.levels[0].placement.length + 5;
+    const height = blueprint.levels.length + 5;
+    const width = blueprint.levels[0].placement[0].length + 5;
+    const command = `/fill ${starting_position[0]} ${starting_position[1]} ${starting_position[2]} ${starting_position[0] + width} ${starting_position[1] + height} ${starting_position[2] + length} air`;
+    bot.chat(command);
+    console.log('World reset');
+}
+
+export function checkLevelBlueprint(agent, levelNum) {
+    const blueprint = agent.task.blueprint;
+    const bot = agent.bot;
+    const result = blueprint.checkLevel(bot, levelNum);
+    if (result.mismatches.length === 0) {
+        return `Level ${levelNum} is correct`;
+    } else {
+        let explanation = blueprint.explainLevelDifference(bot, levelNum);
+        return explanation;
+    }
+}
+
+export function checkBlueprint(agent) {
+    console.log('Checking blueprint...');
+    console.log(agent);
+    const blueprint = agent.task.blueprint;
+    const bot = agent.bot;
+    const result = blueprint.check(bot);
+    if (result.mismatches.length === 0) {
+        return "Blueprint is correct";
+    } else {
+        let explanation = blueprint.explainBlueprintDifference(bot);
+        return explanation;
+    }
+}
+
+export class Blueprint {
+    constructor(blueprint) {
+        this.data = blueprint;
+    }
+    explain() {
+        var explanation = "";
+
+        for (let item of this.data.levels) {
+            var coordinates = item.coordinates;
+            explanation += `Level ${item.level}: `;
+            explanation += `Start at coordinates X: ${coordinates[0]}, Y: ${coordinates[1]}, Z: ${coordinates[2]}`;
+            // let placement_string = this._getPlacementString(item.placement);
+            // explanation += `\n${placement_string}\n`;
+        }
+        return explanation;
+    }
+    _getPlacementString(placement) {
+        var placement_string = "[\n";
+        for (let row of placement) {
+            placement_string += "[";
+            for (let i = 0; i < row.length - 1; i++) {
+                let item = row[i];
+                placement_string += `${item}, `;
+            }
+            let final_item = row[row.length - 1];
+            placement_string += `${final_item}],\n`;
+        }
+        placement_string += "]";
+        return placement_string;
+    }
+    explainLevel(levelNum) {
+        const levelData = this.data.levels[levelNum];
+        var explanation = `Level ${levelData.level} `;
+        explanation += `starting at coordinates X: ${levelData.coordinates[0]}, Y: ${levelData.coordinates[1]}, Z: ${levelData.coordinates[2]}`;
+        let placement_string = this._getPlacementString(levelData.placement);
+        explanation += `\n${placement_string}\n`;
+        return explanation;
+    }
+    explainBlueprintDifference(bot) {
+        var explanation = "";
+        const levels = this.data.levels;
+        for (let i = 0; i < levels.length; i++) {
+            let level_explanation = this.explainLevelDifference(bot, i);
+            explanation += level_explanation + "\n";
+        }
+        return explanation;
+    }
+    explainLevelDifference(bot, levelNum) {
+        const results = this.checkLevel(bot, levelNum);
+        const mismatches = results.mismatches;
+        const levelData = this.data.levels[levelNum];
+
+        if (mismatches.length === 0) {
+            return `Level ${levelData.level} is complete`;
+        }
+        var explanation = `Level ${levelData.level} `;
+        // explanation += `at coordinates X: ${levelData.coordinates[0]}, Y: ${levelData.coordinates[1]}, Z: ${levelData.coordinates[2]}`;
+        explanation += " requires the following fixes:\n";
+        for (let item of mismatches) {
+            if (item.actual === 'air') { 
+                explanation += `Place ${item.expected} at coordinates X: ${item.coordinates[0]}, Y: ${item.coordinates[1]}, Z: ${item.coordinates[2]}\n`;
+            } else if (item.expected === 'air') {
+                explanation += `Remove the ${item.actual} at coordinates X: ${item.coordinates[0]}, Y: ${item.coordinates[1]}, Z: ${item.coordinates[2]}\n`;
+            } else {
+                explanation += `Replace the ${item.actual} with a ${item.expected} at coordinates X: ${item.coordinates[0]}, Y: ${item.coordinates[1]}, Z: ${item.coordinates[2]} \n`;
+            }
+        }
+        return explanation;
+    }
+    check(bot) {
+        if (!bot || typeof bot !== 'object' || !bot.hasOwnProperty('blockAt')) {
+            throw new Error('Invalid bot object. Expected a mineflayer bot.');
+        }
+        const levels = this.data.levels;
+        const mismatches = [];
+        const matches = [];
+        for (let i = 0; i < levels.length; i++) {
+            const result = this.checkLevel(bot, i);
+            mismatches.push(...result.mismatches);
+            matches.push(...result.matches);
+        }
+        return {
+            "mismatches": mismatches,
+            "matches": matches
+        };
+    }
+    checkLevel(bot, levelNum) {
+        const levelData = this.data.levels[levelNum];
+        const startCoords = levelData.coordinates;
+        const placement = levelData.placement;
+        const mismatches = [];
+        const matches = [];
+    
+        for (let zOffset = 0; zOffset < placement.length; zOffset++) {
+            const row = placement[zOffset];
+            for (let xOffset = 0; xOffset < row.length; xOffset++) {
+                const blockName = row[xOffset];
+    
+                const x = startCoords[0] + xOffset;
+                const y = startCoords[1];
+                const z = startCoords[2] + zOffset;
+
+                try {
+                    const blockAtLocation = bot.blockAt(new Vec3(x, y, z));
+                    const actualBlockName = blockAtLocation ? bot.registry.blocks[blockAtLocation.type].name : "air";
+
+                    // Skip if both expected and actual block are air
+                    if (blockName === "air" && actualBlockName === "air") {
+                        continue;
+                    }
+
+                    if (actualBlockName !== blockName) {
+                        mismatches.push({
+                            level: levelData.level,
+                            coordinates: [x, y, z],
+                            expected: blockName,
+                            actual: actualBlockName
+                        });
+                    } else {
+                        matches.push({
+                            level: levelData.level,
+                            coordinates: [x, y, z],
+                            expected: blockName,
+                            actual: actualBlockName
+                        });
+                    }
+                } catch (err) {
+                    console.error(`Error getting block at (${x}, ${y}, ${z}):`, err);
+                    return false; // Stop checking if there's an issue getting blocks
+                }
+            }
+        }
+        return {
+            "mismatches": mismatches,
+            "matches": matches
+        };
+    }
+
+    /**
+     * Takes in the blueprint, and then converts it into a set of /setblock commands for the bot to follow
+     * @Returns: An object containing the setblock commands as a list of strings, and a position nearby the blueprint but not in it
+     * @param blueprint
+     */
+    autoBuild() {
+        const commands = [];
+        let blueprint = this.data
+
+        let minX = Infinity, maxX = -Infinity;
+        let minY = Infinity, maxY = -Infinity;
+        let minZ = Infinity, maxZ = -Infinity;
+
+        for (const level of blueprint.levels) {
+            console.log(level.level)
+            const baseX = level.coordinates[0];
+            const baseY = level.coordinates[1];
+            const baseZ = level.coordinates[2];
+            const placement = level.placement;
+
+            // Update bounds
+            minX = Math.min(minX, baseX);
+            maxX = Math.max(maxX, baseX + placement[0].length - 1);
+            minY = Math.min(minY, baseY);
+            maxY = Math.max(maxY, baseY);
+            minZ = Math.min(minZ, baseZ);
+            maxZ = Math.max(maxZ, baseZ + placement.length - 1);
+
+            // Loop through the 2D placement array
+            for (let z = 0; z < placement.length; z++) {
+                for (let x = 0; x < placement[z].length; x++) {
+                    const blockType = placement[z][x];
+                    if (blockType) {
+                        const setblockCommand = `/setblock ${baseX + x} ${baseY} ${baseZ + z} ${blockType}`;
+                        commands.push(setblockCommand);
+                    }
+                }
+            }
+        }
+
+        // Calculate a position nearby the blueprint but not in it
+        const nearbyPosition = {
+            x: maxX + 5, // Move 5 blocks to the right
+            y: minY,     // Stay on the lowest level of the blueprint
+            z: minZ      // Stay aligned with the front of the blueprint
+        };
+
+        return { commands, nearbyPosition };
+    }
+
+
+    /**
+     * Takes in a blueprint, and returns a set of commands to clear up the space.
+     *
+     */
+    autoDelete() {
+        console.log("auto delete called!")
+        const commands = [];
+        let blueprint = this.data
+
+        let minX = Infinity, maxX = -Infinity;
+        let minY = Infinity, maxY = -Infinity;
+        let minZ = Infinity, maxZ = -Infinity;
+
+        for (const level of blueprint.levels) {
+            const baseX = level.coordinates[0];
+            const baseY = level.coordinates[1];
+            const baseZ = level.coordinates[2];
+            const placement = level.placement;
+
+            // Update bounds
+            minX = Math.min(minX, baseX);
+            maxX = Math.max(maxX, baseX + placement[0].length - 1);
+            minY = Math.min(minY, baseY);
+            maxY = Math.max(maxY, baseY);
+            minZ = Math.min(minZ, baseZ);
+            maxZ = Math.max(maxZ, baseZ + placement.length - 1);
+
+            // Loop through the 2D placement array
+            for (let z = 0; z < placement.length; z++) {
+                for (let x = 0; x < placement[z].length; x++) {
+                    const blockType = placement[z][x];
+                    if (blockType) {
+                        const setblockCommand = `/setblock ${baseX + x} ${baseY} ${baseZ + z} air`;
+                        commands.push(setblockCommand);
+                    }
+                }
+            }
+        }
+
+        // Calculate a position nearby the blueprint but not in it
+        const nearbyPosition = {
+            x: maxX + 5, // Move 5 blocks to the right
+            y: minY,     // Stay on the lowest level of the blueprint
+            z: minZ      // Stay aligned with the front of the blueprint
+        };
+
+        return { commands, nearbyPosition };
+    }
+}
+
+
+/**
+ * Systematically builds the houses by placing them next to the already existing rooms. Still uses randomness for what gets placed next.
+ * @param m width of the 3D space
+ * @param n height of the 3D space
+ * @param p depth of the 3D space
+ * @param rooms Number of rooms to attempt to generate
+ * @param minRoomWidth
+ * @param minRoomLength
+ * @param minRoomDepth
+ * @param roomVariance How much the room size will vary
+ * @param wrapping material of wrapping (air, glass, etc...) -> default is air
+ * @param carpetStyle 0,1,2 increasingly more complex
+ * @param windowStyle 0,1,2 increasingly more complex
+ * @param complexity 0,1,2,3,4 for increasingly complex materials for room generation
+ * @param startCoord an array of the x,y,z coordinates to create the blueprint. default = [148,-60,-170]
+ * @returns a blueprint object
+ */
+export function proceduralGeneration(m = 20,
+                                     n = 20,
+                                     p = 20,
+                                     rooms = 8,
+                                     minRoomWidth = 5,
+                                     minRoomLength = 5,
+                                     minRoomDepth = 6,
+                                     roomVariance = 5,
+                                     wrapping = "air",
+                                     carpetStyle = 1,
+                                     windowStyle = 1,
+                                     complexity = 4,
+                                     startCoord = [148,-60,-170]) {
+    // Build 3D space
+    const matrix = Array.from({length: p}, () =>
+        Array.from({length: m}, () =>
+            Array(n).fill('air')
+        )
+    );
+
+    // todo: extrapolate into another param? then have set materials be dynamic? 
+    let roomMaterials = ["stone", "terracotta", "quartz_block", "copper_block", "purpur_block"]
+
+    if (complexity < roomMaterials.length) {
+        roomMaterials = roomMaterials.slice(0, complexity + 1);
+    }
+
+    // Mark entire outer border with 'stone'
+    for (let z = 0; z < p; z++) {
+        for (let x = 0; x < m; x++) {
+            for (let y = 0; y < n; y++) {
+                if (
+                    z === 0 || z === p - 1 || // Top and bottom faces
+                    x === 0 || x === m - 1 || // Front and back faces
+                    y === 0 || y === n - 1    // Left and right faces
+                ) {
+                    matrix[z][x][y] = 'stone';
+                }
+            }
+        }
+    }
+
+    // Replace outer layer with wrap
+    for (let z = 0; z < p; z++) {
+        for (let x = 0; x < m; x++) {
+            for (let y = 0; y < n; y++) {
+                if (
+                    (z === p - 1 || // Top face
+                        x === 0 || x === m - 1 || // Front and back faces
+                        y === 0 || y === n - 1) // Left and right faces
+                ) {
+                    matrix[z][x][y] = wrapping;
+                }
+            }
+        }
+    }
+
+    let placedRooms = 0;
+    let lastRoom = null;
+
+    // Direction probabilities (e.g., 'above': 40%, 'left': 15%, etc.)
+    const directionChances = [
+        {direction: 'above', chance: 0.15},
+        {direction: 'left', chance: 0.15},
+        {direction: 'right', chance: 0.15},
+        {direction: 'forward', chance: 0.15},
+        {direction: 'backward', chance: 0.15},
+    ];
+
+    // Function to pick a random direction based on percentages
+    function getRandomDirection() {
+        const rand = Math.random();
+        let cumulative = 0;
+
+        for (const {direction, chance} of directionChances) {
+            cumulative += chance;
+            if (rand <= cumulative) return direction;
+        }
+        return directionChances[1].direction; // Fallback to the first direction
+    }
+
+    // Ensures no rooms overlap except at edges
+    function isSpaceValid(newX, newY, newZ, newLength, newWidth, newDepth) {
+        for (let di = 0; di < newDepth; di++) {
+            for (let dj = 0; dj < newLength; dj++) {
+                for (let dk = 0; dk < newWidth; dk++) {
+                    const x = newX + dj;
+                    const y = newY + dk;
+                    const z = newZ + di;
+
+                    // Skip checking the outermost borders of the new room (these can overlap with stone)
+                    if (dj === 0 || dj === newLength - 1 ||
+                        dk === 0 || dk === newWidth - 1 ||
+                        di === 0 || di === newDepth - 1) {
+                        continue;
+                    }
+
+                    // For non-border spaces, ensure they're air
+                    if (matrix[z][x][y] !== 'air') {
+                        return false;
+                    }
+                }
+            }
+        }
+        return true;
+    }
+
+    function validateAndBuildBorder(matrix, newX, newY, newZ, newLength, newWidth, newDepth, m, n, p, material) {
+        // Allow rooms to use the matrix edges (note the <= instead of <)
+        if (
+            newX >= 0 && newX + newLength <= m &&
+            newY >= 0 && newY + newWidth <= n &&
+            newZ >= 0 && newZ + newDepth <= p &&
+            isSpaceValid(newX, newY, newZ, newLength, newWidth, newDepth)
+        ) {
+            // console.log(`Placing room at (${newX}, ${newY}, ${newZ}) with dimensions (${newLength}x${newWidth}x${newDepth})`);
+            for (let di = 0; di < newDepth; di++) {
+                for (let dj = 0; dj < newLength; dj++) {
+                    for (let dk = 0; dk < newWidth; dk++) {
+                        const x = newX + dj;
+                        const y = newY + dk;
+                        const z = newZ + di;
+
+                        // If this is at a matrix border, don't modify it
+                        if (z === 0) {
+                            continue;
+                        }
+                        // if (x === 0 || x === m - 1 ||
+                        //     y === 0 || y === n - 1 ||
+                        //     z === 0 || z === p - 1) {
+                        //     continue;
+                        // }
+
+                        // For non-border spaces, check if this is a floor that should be shared
+                        //was: === 'stone'
+                        if (di === 0 && matrix[z - 1][x][y] !== 'air') {
+                            // Skip creating floor if there's a ceiling below
+                            matrix[z][x][y] = 'air';
+                        } else if (di === 0 || di === newDepth - 1 ||
+                            dj === 0 || dj === newLength - 1 ||
+                            dk === 0 || dk === newWidth - 1) {
+                            matrix[z][x][y] = material;
+                        } else {
+                            matrix[z][x][y] = 'air';
+                        }
+
+
+                    }
+                }
+            }
+            return true;
+        }
+        return false;
+    }
+
+    function addDoor(matrix, x, y, z, material) {
+        matrix[z][x][y] = material;
+
+        // Place the lower half of the door
+        // matrix[z + 1][x][y] = 'dark_oak_door[half=lower, hinge=left]';
+
+        matrix[z + 1][x][y] = 'dark_oak_door';
+
+
+        // Place the upper half of the door
+        // matrix[z + 2][x][y] = 'dark_oak_door[half=upper, hinge=left]';
+        matrix[z + 2][x][y] = 'dark_oak_door';
+
+    }
+
+
+    // Takes in a room and randomly converts some faces to be windows
+    function addWindowsAsSquares(matrix, x, y, z, newLength, newWidth, newDepth, material) {
+        // Matrix dimensions
+        const matrixDepth = matrix.length;
+        const matrixLength = matrix[0].length;
+        const matrixWidth = matrix[0][0].length;
+        const windowX = Math.ceil(minRoomWidth / 2)
+        const windowY = Math.ceil(minRoomLength / 2)
+        const windowZ = Math.ceil(minRoomDepth / 2)
+
+        // Helper function to check if coordinates are within bounds
+        function isInBounds(z, x, y) {
+            return z >= 0 && z < matrixDepth &&
+                x >= 0 && x < matrixLength &&
+                y >= 0 && y < matrixWidth;
+        }
+
+        // Front and back faces (z is constant)
+        if (Math.random() < 0.8) {
+            let centerX = x + Math.floor(newLength / 2 - windowX / 2);
+            let centerY = y + Math.floor(newWidth / 2 - windowY / 2);
+
+            for (let dx = 0; dx <= windowX; dx++) {
+                for (let dy = 0; dy <= windowY; dy++) {
+                    let frontZ = z;
+                    let backZ = z + newDepth - 1;
+
+                    if (isInBounds(frontZ, centerX + dx, centerY + dy) &&
+                        matrix[frontZ][centerX + dx][centerY + dy] === material) {
+                        matrix[frontZ][centerX + dx][centerY + dy] = 'glass';
+                    }
+                    if (isInBounds(backZ, centerX + dx, centerY + dy) &&
+                        matrix[backZ][centerX + dx][centerY + dy] === material) {
+                        matrix[backZ][centerX + dx][centerY + dy] = 'glass';
+                    }
+                }
+            }
+        }
+
+        // Left and right faces (x is constant)
+        if (Math.random() < 0.8) {
+            let centerZ = z + Math.floor(newDepth / 2 - windowZ / 2);
+            let centerY = y + Math.floor(newWidth / 2 - windowY / 2);
+
+            for (let dz = 0; dz <= windowZ; dz++) {
+                for (let dy = 0; dy <= windowY; dy++) {
+                    let leftX = x;
+                    let rightX = x + newLength - 1;
+
+                    if (isInBounds(centerZ + dz, leftX, centerY + dy) &&
+                        matrix[centerZ + dz][leftX][centerY + dy] === material) {
+                        matrix[centerZ + dz][leftX][centerY + dy] = 'glass';
+                    }
+                    if (isInBounds(centerZ + dz, rightX, centerY + dy) &&
+                        matrix[centerZ + dz][rightX][centerY + dy] === material) {
+                        matrix[centerZ + dz][rightX][centerY + dy] = 'glass';
+                    }
+                }
+            }
+        }
+
+        // Top and bottom faces (y is constant)
+        if (Math.random() < 0.8) {
+            let centerX = x + Math.floor(newLength / 2 - windowX / 2);
+            let centerZ = z + Math.floor(newDepth / 2 - windowZ / 2);
+
+            for (let dx = 0; dx <= windowX; dx++) {
+                for (let dz = 0; dz <= windowZ; dz++) {
+                    let bottomY = y;
+                    let topY = y + newWidth - 1;
+
+                    if (isInBounds(centerZ + dz, centerX + dx, bottomY) &&
+                        matrix[centerZ + dz][centerX + dx][bottomY] === material) {
+                        matrix[centerZ + dz][centerX + dx][bottomY] = 'glass';
+                    }
+                    if (isInBounds(centerZ + dz, centerX + dx, topY) &&
+                        matrix[centerZ + dz][centerX + dx][topY] === material) {
+                        matrix[centerZ + dz][centerX + dx][topY] = 'glass';
+                    }
+                }
+            }
+        }
+    }
+
+    function addWindowsAsPlane(matrix, x, y, z, newLength, newWidth, newDepth, material) {
+        // Ensure the new dimensions are within bounds
+        const maxX = matrix[0].length;
+        const maxY = matrix[0][0].length;
+        const maxZ = matrix.length;
+
+        // Each face has a 30% chance of becoming a window
+        if (Math.random() < 0.8) {
+            for (let dx = 0; dx < newLength; dx++) {
+                for (let dy = 0; dy < newWidth; dy++) {
+                    let frontZ = z;
+                    let backZ = z + newDepth - 1;
+
+                    // Check bounds before modifying the matrix
+                    if (frontZ >= 0 && frontZ < maxZ && x + dx >= 0 && x + dx < maxX && y + dy >= 0 && y + dy < maxY) {
+                        if (matrix[frontZ][x + dx][y + dy] === material) {
+                            matrix[frontZ][x + dx][y + dy] = 'glass';
+                        }
+                    }
+                    if (backZ >= 0 && backZ < maxZ && x + dx >= 0 && x + dx < maxX && y + dy >= 0 && y + dy < maxY) {
+                        if (matrix[backZ][x + dx][y + dy] === material) {
+                            matrix[backZ][x + dx][y + dy] = 'glass';
+                        }
+                    }
+                }
+            }
+        }
+
+        if (Math.random() < 0.8) {
+            for (let dz = 0; dz < newDepth; dz++) {
+                for (let dy = 0; dy < newWidth; dy++) {
+                    let leftX = x;
+                    let rightX = x + newLength - 1;
+
+                    // Check bounds before modifying the matrix
+                    if (leftX >= 0 && leftX < maxX && z + dz >= 0 && z + dz < maxZ && y + dy >= 0 && y + dy < maxY) {
+                        if (matrix[z + dz][leftX][y + dy] === material) {
+                            matrix[z + dz][leftX][y + dy] = 'glass';
+                        }
+                    }
+                    if (rightX >= 0 && rightX < maxX && z + dz >= 0 && z + dz < maxZ && y + dy >= 0 && y + dy < maxY) {
+                        if (matrix[z + dz][rightX][y + dy] === material) {
+                            matrix[z + dz][rightX][y + dy] = 'glass';
+                        }
+                    }
+                }
+            }
+        }
+    }
+
+
+    //still a little buggy
+    function addStairs(matrix, x, y, z, length, width, material) {
+        let currentZ = z;
+        let currentX = x + 1;
+        let currentY = y + 1;
+        let direction = 0;
+        let stepCount = 0;
+        const maxSteps = length * width; // Safety limit
+
+        while (currentZ >= 0 && currentX < x + length - 1 && currentY < y + width - 1 && stepCount < maxSteps) {
+            // Place stair block
+            matrix[currentZ][currentX][currentY] = material || 'stone';
+
+            // Clear 3 blocks above for headroom
+            for (let i = 1; i <= 3; i++) {
+                if (currentZ + i < matrix.length) {
+                    matrix[currentZ + i][currentX][currentY] = 'air';
+                }
+            }
+
+            // Move to next position based on direction
+            if (direction === 0) {
+                currentX++;
+                if (currentX >= x + length - 1) {
+                    currentX = x + length - 2;
+                    direction = 1;
+                } else {
+                    currentZ--;
+                }
+            } else {
+                currentY++;
+                if (currentY >= y + width - 1) {
+                    currentY = y + width - 2;
+                    direction = 0;
+                } else {
+                    currentZ--;
+                }
+            }
+
+            stepCount++;
+        }
+    }
+
+    function addCarpet(probability, matrix, newX, newY, newZ, newLength, newWidth, material) {
+        let colors = ["blue", "cyan", "light_blue", "lime"];
+
+        // Iterate through the dimensions of the room
+        for (let dx = 1; dx < newLength - 1; dx++) {
+            for (let dy = 1; dy < newWidth - 1; dy++) {
+                let x = newX + dx;
+                let y = newY + dy;
+                let z = newZ; // Start at floor level
+
+                // Check if there is floor (not air)
+                if (matrix[z][x][y] === material) {
+                    // Consider a random probability of adding a carpet
+                    if (Math.random() < probability) {
+                        // Choose a random color for the carpet
+                        let randomColor = colors[Math.floor(Math.random() * colors.length)];
+                        // Add carpet one z position above the floor with a random color
+                        matrix[z + 1][x][y] = `${randomColor}_carpet`;
+                    }
+                }
+            }
+        }
+    }
+
+    function addLadder(matrix, x, y, z) {
+        let currentZ = z + 1;
+
+        // turn the floor into air where person would go up
+        matrix[currentZ][x + 1][y] = 'air';
+
+        // Build the first 3 ladder segments from floor level downwards
+        for (let i = 0; i < 3; i++) {
+            // Place stone block behind ladder
+            matrix[currentZ][x - 1][y] = 'stone';
+            // Place ladder
+            matrix[currentZ][x][y] = 'ladder[facing=north]';
+            currentZ -= 1;
+        }
+
+        // Continue building ladder downwards until a floor is hit or we reach the bottom
+        while (currentZ >= 0 && matrix[currentZ][x][y] === 'air') {
+            // Place stone block behind ladder
+            matrix[currentZ][x - 1][y] = 'stone';
+            // Place ladder
+            matrix[currentZ][x][y] = 'ladder[facing=north]';
+
+            // Move down
+            currentZ--;
+        }
+    }
+
+
+    function embellishments(carpet, windowStyle, matrix, newX, newY, newZ, newLength, newWidth, newDepth, material) {
+
+
+        switch (windowStyle) {
+            case 0:
+                break;
+            case 1:
+                addWindowsAsSquares(matrix, newZ, newY, newZ, newLength, newWidth, newDepth, material)
+                break;
+            case 2:
+                addWindowsAsPlane(matrix, newZ, newY, newZ, newLength, newWidth, newDepth, material)
+        }
+
+
+        switch (carpet) {
+            case 0:
+                break;
+            case 1:
+                addCarpet(0.3, matrix, newX, newY, newZ, newLength, newWidth, material);
+                break;
+            case 2:
+                addCarpet(0.7, matrix, newX, newY, newZ, newLength, newWidth, material)
+                break;
+        }
+
+
+    }
+
+
+    // Places rooms until we can't, or we place all
+    // attempts random configurations of rooms in random directions.
+    while (placedRooms < rooms) {
+        let roomPlaced = false;
+
+        for (let attempt = 0; attempt < 150; attempt++) {
+
+            const material = roomMaterials[Math.floor(Math.random() * roomMaterials.length)];
+
+
+            // dimensions of room
+            const newLength = Math.max(minRoomLength, Math.floor(Math.random() * roomVariance) + minRoomLength);
+            const newWidth = Math.max(minRoomWidth, Math.floor(Math.random() * roomVariance) + minRoomWidth);
+            const newDepth = Math.max(minRoomDepth, Math.floor(Math.random() * Math.floor(roomVariance / 2)) + minRoomDepth);
+            let newX, newY, newZ;
+
+            // first room is special
+            if (placedRooms === 0) {
+                // First room placement
+                newX = Math.floor(Math.random() * (m - newLength - 1)) + 1;
+                newY = Math.floor(Math.random() * (n - newWidth - 1)) + 1;
+                newZ = 0; // Ground floor
+
+                if (validateAndBuildBorder(matrix, newX, newY, newZ, newLength, newWidth, newDepth, m, n, p, material)) {
+                    lastRoom = {x: newX, y: newY, z: newZ, length: newLength, width: newWidth, depth: newDepth};
+                    roomPlaced = true;
+                    placedRooms++;
+
+                    // Add doors to all four sides
+                    // Left side
+                    addDoor(matrix, newX, newY + Math.floor(newWidth / 2), newZ, material);
+                    // Right side
+                    addDoor(matrix, newX + newLength - 1, newY + Math.floor(newWidth / 2), newZ, material);
+                    // Front side
+                    addDoor(matrix, newX + Math.floor(newLength / 2), newY, newZ, material);
+                    // Back side
+                    addDoor(matrix, newX + Math.floor(newLength / 2), newY + newWidth - 1, newZ, material);
+
+                    addCarpet(0.7, matrix, newX, newY, newZ, newLength, newWidth)
+                }
+
+                break;
+            } else {
+                const direction = getRandomDirection();
+
+                switch (direction) {
+                    case 'above':
+                        newX = lastRoom.x;
+                        newY = lastRoom.y;
+                        newZ = lastRoom.z + lastRoom.depth - 1;
+                        if (validateAndBuildBorder(matrix, newX, newY, newZ, newLength, newWidth, newDepth, m, n, p, material)) {
+
+                            embellishments(carpetStyle, windowStyle, matrix, newX, newY, newZ, newLength, newWidth, newDepth, material)
+
+                            // addLadder(matrix, lastRoom.x + Math.floor(lastRoom.length / 2),
+                            //     lastRoom.y + Math.floor(lastRoom.width / 2),
+                            //     newZ); // Adding the ladder
+
+                            addStairs(matrix, newX, newY, newZ, newLength, newWidth, material)
+
+
+                            lastRoom = {x: newX, y: newY, z: newZ, length: newLength, width: newWidth, depth: newDepth};
+                            roomPlaced = true;
+                            placedRooms++;
+                            break;
+                        }
+                        break;
+
+                    case 'left':
+                        newX = lastRoom.x - newLength + 1;
+                        newY = lastRoom.y;
+                        newZ = lastRoom.z;
+                        if (validateAndBuildBorder(matrix, newX, newY, newZ, newLength, newWidth, newDepth, m, n, p, material)) {
+
+
+                            embellishments(carpetStyle, windowStyle, matrix, newX, newY, newZ, newLength, newWidth, newDepth, material)
+
+
+                            addDoor(matrix, lastRoom.x, lastRoom.y + Math.floor(lastRoom.width / 2), lastRoom.z, material);
+
+
+                            lastRoom = {x: newX, y: newY, z: newZ, length: newLength, width: newWidth, depth: newDepth};
+                            roomPlaced = true;
+                            placedRooms++;
+                            break;
+                        }
+                        break;
+
+                    case 'right':
+                        newX = lastRoom.x + lastRoom.length - 1;
+                        newY = lastRoom.y;
+                        newZ = lastRoom.z;
+                        if (validateAndBuildBorder(matrix, newX, newY, newZ, newLength, newWidth, newDepth, m, n, p, material)) {
+
+                            embellishments(carpetStyle, windowStyle, matrix, newX, newY, newZ, newLength, newWidth, newDepth, material)
+
+
+                            addDoor(matrix, lastRoom.x + lastRoom.length - 1,
+                                lastRoom.y + Math.floor(lastRoom.width / 2),
+                                lastRoom.z, material);
+
+
+                            lastRoom = {x: newX, y: newY, z: newZ, length: newLength, width: newWidth, depth: newDepth};
+                            roomPlaced = true;
+                            placedRooms++;
+                            break;
+                        }
+                        break;
+
+                    case 'forward':
+                        newX = lastRoom.x;
+                        newY = lastRoom.y + lastRoom.width - 1;
+                        newZ = lastRoom.z;
+                        if (validateAndBuildBorder(matrix, newX, newY, newZ, newLength, newWidth, newDepth, m, n, p, material)) {
+
+                            embellishments(carpetStyle, windowStyle, matrix, newX, newY, newZ, newLength, newWidth, newDepth, material)
+
+
+                            addDoor(matrix, lastRoom.x + Math.floor(lastRoom.length / 2),
+                                lastRoom.y + lastRoom.width - 1,
+                                lastRoom.z, material);
+
+
+                            lastRoom = {x: newX, y: newY, z: newZ, length: newLength, width: newWidth, depth: newDepth};
+                            roomPlaced = true;
+                            placedRooms++;
+                            break;
+                        }
+                        break;
+
+                    case 'backward':
+                        newX = lastRoom.x;
+                        newY = lastRoom.y - newWidth + 1;
+                        newZ = lastRoom.z;
+                        if (validateAndBuildBorder(matrix, newX, newY, newZ, newLength, newWidth, newDepth, m, n, p, material)) {
+
+                            embellishments(carpetStyle, windowStyle, matrix, newX, newY, newZ, newLength, newWidth, newDepth, material)
+
+
+                            addDoor(matrix, lastRoom.x + Math.floor(lastRoom.length / 2),
+                                lastRoom.y,
+                                lastRoom.z, material);
+
+
+                            lastRoom = {x: newX, y: newY, z: newZ, length: newLength, width: newWidth, depth: newDepth};
+                            roomPlaced = true;
+                            placedRooms++;
+                            break;
+                        }
+                        break;
+                }
+
+                if (roomPlaced) {
+                    break;
+                }
+            }
+        }
+
+        if (!roomPlaced) {
+            console.warn(`Could not place room ${placedRooms + 1}`);
+            break;
+        }
+    }
+
+    // uncomment to visualize blueprint output
+    // printMatrix(matrix)
+
+    return matrixToBlueprint(matrix, startCoord)
+}
+
+
+
+
+/**
+ * for cutesy output
+ * @param matrix
+ */
+function printMatrix(matrix) {
+    matrix.forEach((layer, layerIndex) => {
+        console.log(`Layer ${layerIndex}:`);
+        layer.forEach(row => {
+            console.log(
+                row.map(cell => {
+                    switch (cell) {
+                        case 'stone': return '█';  // Wall
+                        case 'air': return '.';    // Open space
+                        case 'dark_oak_door[half=upper, hinge=left]': return 'D';
+                        case 'dark_oak_door[half=lower, hinge=left]': return 'D';
+                        case 'oak_stairs[facing=north]': return 'S';  // Stairs
+                        case 'oak_stairs[facing=east]': return 'S';  // Stairs
+                        case 'oak_stairs[facing=south]': return 'S';  // Stairs
+                        case 'oak_stairs[facing=west]': return 'S';  // Stairs
+                        case 'glass': return 'W'
+
+
+                        default: return '?';       // Unknown or unmarked space
+                    }
+                }).join(' ')
+            );
+        });
+        console.log('---');
+    });
+}
+
+/**
+ * Converts a 3D matrix into a Minecraft blueprint format
+ * @param {Array<Array<Array<string>>>} matrix - 3D matrix of block types
+ * @param {number[]} startCoord - Starting coordinates [x, y, z]
+ * @returns {Object} a Blueprint object in Minecraft format
+ */
+function matrixToBlueprint(matrix, startCoord) {
+    // Validate inputs
+    if (!Array.isArray(matrix) || !Array.isArray(startCoord) || startCoord.length !== 3) {
+        console.log(matrix)
+        throw new Error('Invalid input format');
+    }
+
+    const [startX, startY, startZ] = startCoord;
+
+
+    // CONSIDER: using blueprint class here?
+    return {
+        levels: matrix.map((level, levelIndex) => ({
+            level: levelIndex,
+            coordinates: [
+                startX,
+                startY + levelIndex,
+                startZ
+            ],
+            placement: level.map(row =>
+                // Ensure each block is a string, default to 'air' if undefined
+                row.map(block => block?.toString() || 'air')
+            )
+        }))
+    };
+}
+
+async function getBlockName(bot, coordinate) {
+    const blockAtLocation = bot.blockAt(new Vec3(coordinate.x, coordinate.y, coordinate.z));
+    return blockAtLocation ? bot.registry.blocks[blockAtLocation.type].name : "air";
+}
+
+/**
+ * Converts a world location to a blueprint. takes some time to ensure that the chunks are loaded before conversion.
+ * @param startCoord - [x,y,z] that signifies the start of the blueprint
+ * @param y_amount - how many spaces you want to register from the start coordinate in the y dimension
+ * @param x_amount - how many spaces in the x direction on minecraft
+ * @param z_amount - how many spaces from the start coordinate in the z direction in minecraft
+ * @param bot - the mineflayer agent (ex. andy)
+ * @returns - a Blueprint object of the converted blueprint
+ */
+export async function worldToBlueprint(startCoord, y_amount, x_amount, z_amount, bot) {
+    await bot.waitForChunksToLoad();
+
+    const materials = {};
+
+    const levels = [];
+    for (let y = 0; y < y_amount; y++) {
+        const placement = [];
+        const coordinates = [startCoord.x, startCoord.y + y, startCoord.z];
+        for (let z = 0; z < z_amount; z++) {
+            const row = [];
+            for (let x = 0; x < x_amount; x++) {
+                const worldCoord = {
+                    x: startCoord.x + x,
+                    y: startCoord.y + y,
+                    z: startCoord.z + z
+                };
+                await bot.waitForChunksToLoad(worldCoord);
+                const blockName = await getBlockName(bot, worldCoord);
+                row.push(blockName);
+                if (blockName !== 'air') {
+                    materials[blockName] = (materials[blockName] || 0) + 1;
+                }
+            }
+            placement.push(row);
+        }
+        levels.push({
+            level: y,
+            coordinates: coordinates,
+            placement: placement
+        })
+    }
+    console.log(levels);
+    const blueprint_data = {
+        materials: materials,
+        levels: levels
+    }
+    return blueprint_data
+}
+
+export function blueprintToTask(blueprint_data, num_agents) {
+    let initialInventory = {}
+    for (let j = 0; j < num_agents; j++) {
+        initialInventory[JSON.stringify(j)] = {"diamond_pickaxe": 1, "diamond_axe": 1, "diamond_shovel": 1};
+    }
+
+    let give_agent = 0;
+    console.log("materials", blueprint_data.materials)
+    for (const key of Object.keys(blueprint_data.materials)) {
+        initialInventory[JSON.stringify(give_agent)][key] = blueprint_data.materials[key];
+        give_agent = (give_agent + 1) % num_agents;
+    }
+
+    const task = {
+        type: "construction",
+        goal: "Make a structure with the blueprint below",
+        conversation: "Let's share materials and make a structure with the blueprint",
+        agent_count: num_agents,
+        blueprint: blueprint_data,
+        initial_inventory: initialInventory,
+    };
+    return task;
+}
+
+// testing code
+
+// let blueprint = proceduralGeneration(20,10,20)
+// const b = new Blueprint(blueprint)
+// const result = b.autoBuild();
+// const commands = result.commands;
+// const nearbyPosition = result.nearbyPosition;
+//
+//
+// import {initBot} from "../../utils/mcdata.js";
+// let bot = initBot("andy");
+
+
+// example usage of world->blueprint function
+
+// bot.once('spawn', async () => {
+//     console.log("nearby position", nearbyPosition);
+//     bot.chat(`/tp @andy ${nearbyPosition.x} ${nearbyPosition.y} ${nearbyPosition.z}`);
+//     for (const command of commands) {
+//         bot.chat(command);
+//     }
+//     const startCoord = {
+//         x: 148,
+//         y: -60,
+//         z: -170
+//     };
+//     // [148,-60,-170] is default start for procedural generation
+//
+//     const worldOutput = await worldToBlueprint(startCoord, 20,10,20, bot)
+// });