v0.7.4 (#67) Edge customization, node & edge coloring

Author: blitzarx1

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "egui_graphs"
-version = "0.7.3"
+version = "0.7.4"
 authors = ["Dmitrii Samsonov <blitzarx1@gmail.com>"]
 license = "MIT"
 homepage = "https://github.com/blitzarx1/egui_graphs"

Cargo.toml

@@ -21,6 +21,7 @@ mod settings;
 const SIMULATION_DT: f32 = 0.035;
 const FPS_LINE_COLOR: Color32 = Color32::from_rgb(128, 128, 128);
 const CHANGES_LIMIT: usize = 100;
+const EDGE_COLOR: Color32 = Color32::from_rgba_premultiplied(128, 128, 128, 64);
 
 pub struct ConfigurableApp {
     g: Graph<(), (), Directed>,
@@ -654,7 +655,8 @@ impl App for ConfigurableApp {
             let settings_style = &egui_graphs::SettingsStyle::new()
                 .with_labels_always(self.settings_style.labels_always)
                 .with_edge_radius_weight(self.settings_style.edge_radius_weight)
-                .with_folded_radius_weight(self.settings_style.edge_radius_weight);
+                .with_folded_radius_weight(self.settings_style.edge_radius_weight)
+                .with_edge_color(EDGE_COLOR);
             ui.add(
                 &mut GraphView::new(&mut self.g)
                     .with_interactions(settings_interaction)

examples/configurable/src/main.rs

@@ -17,6 +17,6 @@ pub use self::settings::{SettingsInteraction, SettingsNavigation, SettingsStyle}
 pub use self::state_computed::StateComputedNode;
 pub use self::subgraphs::SubGraph;
 pub use self::transform::{
-    add_edge, add_node, add_node_custom, default_edge_transform, default_node_transform,
-    to_input_graph, to_input_graph_custom,
+    add_edge, add_edge_custom, add_node, add_node_custom, default_edge_transform,
+    default_node_transform, to_input_graph, to_input_graph_custom,
 };

src/lib.rs

@@ -233,6 +233,18 @@ impl SettingsStyle {
         self
     }
 
+    /// Sets default color for node.
+    pub fn with_node_color(mut self, color: Color32) -> Self {
+        self.color_node = color;
+        self
+    }
+
+    /// Sets default color for edge.
+    pub fn with_edge_color(mut self, color: Color32) -> Self {
+        self.color_edge = color;
+        self
+    }
+
     pub(crate) fn color_node<N: Clone>(&self, ctx: &egui::Context, n: &Node<N>) -> Color32 {
         if n.color().is_some() {
             return n.color().unwrap();

src/settings.rs

@@ -1,27 +1,23 @@
 use crate::{graph_view::Graph, Edge, Node};
 use egui::Vec2;
 use petgraph::{
-    data::Build,
     stable_graph::{EdgeIndex, NodeIndex, StableGraph},
     visit::IntoNodeReferences,
     EdgeType,
 };
-use rand::{seq::IteratorRandom, Rng};
+use rand::Rng;
 use std::collections::HashMap;
 
 pub const DEFAULT_SPAWN_SIZE: f32 = 250.;
 
-/// Helper function which adds user's node to the [`egui_graphs::Graph`] instance.
+/// Helper function which adds user's node to the [`super::Graph`] instance.
 ///
 /// If graph is not empty it picks any node position and adds new node in the vicinity of it.
 pub fn add_node<N: Clone, E: Clone, Ty: EdgeType>(g: &mut Graph<N, E, Ty>, n: &N) -> NodeIndex {
-    g.add_node(default_node_transform(
-        NodeIndex::new(g.node_count() + 1),
-        n,
-    ))
+    add_node_custom(g, n, default_node_transform)
 }
 
-/// Helper function which adds user's node to the [`egui_graphs::Graph`] instance.
+/// Helper function which adds user's node to the [`super::Graph`] instance with custom node transform function.
 ///
 /// If graph is not empty it picks any node position and adds new node in the vicinity of it.
 pub fn add_node_custom<N: Clone, E: Clone, Ty: EdgeType>(
@@ -32,17 +28,28 @@ pub fn add_node_custom<N: Clone, E: Clone, Ty: EdgeType>(
     g.add_node(node_transform(NodeIndex::new(g.node_count() + 1), n))
 }
 
-/// Helper function which adds user's edge to the [`egui_graphs::Graph`] instance.
+/// Helper function which adds user's edge to the [`super::Graph`] instance.
 pub fn add_edge<N: Clone, E: Clone, Ty: EdgeType>(
     g: &mut Graph<N, E, Ty>,
     start: NodeIndex,
     end: NodeIndex,
     e: &E,
+) -> EdgeIndex {
+    add_edge_custom(g, start, end, e, default_edge_transform)
+}
+
+/// Helper function which adds user's edge to the [`super::Graph`] instance with custom edge transform function.
+pub fn add_edge_custom<N: Clone, E: Clone, Ty: EdgeType>(
+    g: &mut Graph<N, E, Ty>,
+    start: NodeIndex,
+    end: NodeIndex,
+    e: &E,
+    edge_transform: impl Fn(EdgeIndex, &E) -> Edge<E>,
 ) -> EdgeIndex {
     g.add_edge(
         start,
         end,
-        default_edge_transform(EdgeIndex::new(g.edge_count() + 1), e),
+        edge_transform(EdgeIndex::new(g.edge_count() + 1), e),
     )
 }