From d8dfd864a453a9ead60f970549b2d289d7025364 Mon Sep 17 00:00:00 2001
From: Crozet Sébastien <developer@crozet.re>
Date: Mon, 5 Oct 2020 16:51:16 +0200
Subject: Joint set iteration: also yield the joint handle.

---
 src/dynamics/joint/joint_set.rs | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'src/dynamics')

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.
-- 
cgit