add data_mut method

blitzarx1 · Nov 13, 2023 · da21907 · da21907
1 parent d72e9e3
commit da21907
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/elements/edge.rs b/src/elements/edge.rs
@@ -41,6 +41,9 @@ impl<E: Clone> Edge<E> {
         self.data.as_ref()
     }
 
+    pub fn data_mut(&mut self) -> Option<&mut E> {
+        self.data.as_mut()
+    }
     pub fn color(&self, ctx: &Context) -> Color32 {
         if self.selected {
             return ctx.style().visuals.widgets.hovered.fg_stroke.color;

diff --git a/src/elements/node.rs b/src/elements/node.rs
@@ -64,6 +64,10 @@ impl<N: Clone> Node<N> {
         self.data.as_ref()
     }
 
+    pub fn data_mut(&mut self) -> Option<&mut N> {
+        self.data.as_mut()
+    }
+
     pub fn set_data(&mut self, data: Option<N>) {
         self.data = data;
     }