Joint set iteration: also yield the joint handle.

1 files changed, 8 insertions, 4 deletions

diff --git a/src/dynamics/joint/joint_set.rs b/src/dynamics/joint/joint_set.rs
index 90f5190..2b1895f 100644
--- a/src/dynamics/joint/joint_set.rs
+++ b/src/dynamics/joint/joint_set.rs
@@ -66,17 +66,21 @@ impl JointSet {
     }
 
     /// Iterates through all the joint on this set.
-    pub fn iter(&self) -> impl Iterator<Item = &Joint> {
-        self.joint_graph.graph.edges.iter().map(|e| &e.weight)
+    pub fn iter(&self) -> impl Iterator<Item = (JointHandle, &Joint)> {
+        self.joint_graph
+            .graph
+            .edges
+            .iter()
+            .map(|e| (e.weight.handle, &e.weight))
     }
 
     /// Iterates mutably through all the joint on this set.
-    pub fn iter_mut(&mut self) -> impl Iterator<Item = &mut Joint> {
+    pub fn iter_mut(&mut self) -> impl Iterator<Item = (JointHandle, &mut Joint)> {
         self.joint_graph
             .graph
             .edges
             .iter_mut()
-            .map(|e| &mut e.weight)
+            .map(|e| (e.weight.handle, &mut e.weight))
     }
 
     // /// The set of joints as an array.