Fix shape modification not updating graphics in testbed

dimforge · Aug 4, 2024 · cd5daae · cd5daae
1 parent 617428e
commit cd5daae
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/examples3d/debug_shape_modification3.rs b/examples3d/debug_shape_modification3.rs
@@ -41,7 +41,7 @@ pub fn init_world(testbed: &mut Testbed) {
     let mut step = 0;
     let snapped_frame = 51;
 
-    testbed.add_callback(move |_, physics, _, _| {
+    testbed.add_callback(move |mut gfx, physics, _, _| {
         step += 1;
 
         // Snap the ball velocity or restore it.
@@ -62,6 +62,10 @@ pub fn init_world(testbed: &mut Testbed) {
 
         let ball_coll = physics.colliders.get_mut(ball_coll_handle).unwrap();
         ball_coll.set_shape(SharedShape::ball(ball_rad * step as f32 * 2.0));
+        if let Some(gfx) = &mut gfx {
+            gfx.remove_collider(ball_coll_handle, &physics.colliders);
+            gfx.add_collider(ball_coll_handle, &physics.colliders);
+        }
     });
 
     /*

diff --git a/src_testbed/graphics.rs b/src_testbed/graphics.rs
@@ -78,14 +78,18 @@ impl GraphicsManager {
         collider: ColliderHandle,
     ) {
         let body = body.unwrap_or(RigidBodyHandle::invalid());
-        if let Some(sns) = self.b2sn.get_mut(&body) {
-            for sn in sns.iter_mut() {
+        if let Some(sns) = self.b2sn.remove(&body) {
+            let mut new_sn = vec![];
+            for sn in sns.into_iter() {
                 if let Some(sn_c) = sn.collider {
                     if sn_c == collider {
                         commands.entity(sn.entity).despawn();
+                    } else {
+                        new_sn.push(sn);
                     }
                 }
             }
+            self.b2sn.insert(body, new_sn);
         }
     }