From 64ca30b2f5ff9881b1a9ce27a925ca384b8b91c3 Mon Sep 17 00:00:00 2001
From: yyda <snowmantin@foxmail.com>
Date: Thu, 26 Mar 2026 08:37:23 +0800
Subject: [PATCH] fix: Fix wall array and target state synchronization issue
 when arm picks up objects

---
 src/App.tsx | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/App.tsx b/src/App.tsx
index fe93810c..6c29d585 100644
--- a/src/App.tsx
+++ b/src/App.tsx
@@ -835,6 +835,15 @@ export default function App() {
         if (isAtTarget) {
             if (arm.state === 'picking_down') {
                 if (arm.grabbedObject) {
+                    // Remove from walls array when picked up
+                    const wallIndex = sim.current.walls.indexOf(arm.grabbedObject);
+                    if (wallIndex > -1) {
+                        sim.current.walls.splice(wallIndex, 1);
+                    }
+                    // Check if this is the target and set target to null
+                    if (arm.grabbedObject === sim.current.target) {
+                        sim.current.target = null;
+                    }
                     arm.gripper.add(arm.grabbedObject);
                     arm.grabbedObject.position.set(0, 0.35, 0);
                 }
@@ -851,6 +860,10 @@ export default function App() {
                     arm.grabbedObject.position.copy(worldPos);
                     // Drop to ground level
                     arm.grabbedObject.position.y = 0.25;
+                    // Add back to walls array when dropped
+                    sim.current.walls.push(arm.grabbedObject);
+                    // Set as target when dropped
+                    sim.current.target = arm.grabbedObject;
                     arm.grabbedObject = null;
                 }
                 arm.state = 'dropping_up';