feat: custom data with id and hashmap for userdata in plot

Author: bircni

demo/src/plot_demo.rs

@@ -1,8 +1,10 @@
+use std::collections::HashMap;
 use std::ops::RangeInclusive;
 use std::{f64::consts::TAU, sync::Arc};
 
+use egui::mutex::Mutex;
 use egui::{
-    Checkbox, Color32, ComboBox, NumExt as _, Pos2, Response, ScrollArea, Stroke, TextWrapMode,
+    Checkbox, Color32, ComboBox, Id, NumExt as _, Pos2, Response, ScrollArea, Stroke, TextWrapMode,
     Vec2b, WidgetInfo, WidgetType, remap, vec2,
 };
 
@@ -24,6 +26,7 @@ enum Panel {
     Interaction,
     CustomAxes,
     LinkedAxes,
+    Userdata,
 }
 
 impl Default for Panel {
@@ -44,6 +47,7 @@ pub struct PlotDemo {
     interaction_demo: InteractionDemo,
     custom_axes_demo: CustomAxesDemo,
     linked_axes_demo: LinkedAxesDemo,
+    userdata_demo: UserdataDemo,
     open_panel: Panel,
 }
 
@@ -131,6 +135,7 @@ impl PlotDemo {
                     ui.selectable_value(&mut self.open_panel, Panel::Interaction, "Interaction");
                     ui.selectable_value(&mut self.open_panel, Panel::CustomAxes, "Custom Axes");
                     ui.selectable_value(&mut self.open_panel, Panel::LinkedAxes, "Linked Axes");
+                    ui.selectable_value(&mut self.open_panel, Panel::Userdata, "Userdata");
                 });
         });
         ui.separator();
@@ -160,6 +165,9 @@ impl PlotDemo {
             Panel::LinkedAxes => {
                 self.linked_axes_demo.ui(ui);
             }
+            Panel::Userdata => {
+                self.userdata_demo.ui(ui);
+            }
         }
     }
 }
@@ -636,7 +644,7 @@ impl CustomAxesDemo {
             }
         };
 
-        let label_fmt = |_s: &str, val: &PlotPoint| {
+        let label_fmt = |_s: &str, val: &PlotPoint, _| {
             format!(
                 "Day {d}, {h}:{m:02}\n{p:.2}%",
                 d = day(val.x),
@@ -1191,6 +1199,62 @@ impl ChartsDemo {
     }
 }
 
+#[derive(PartialEq, serde::Deserialize, serde::Serialize, Default)]
+struct UserdataDemo {}
+
+struct DemoPoint {
+    x: f64,
+    y: f64,
+    custom_info: bool,
+}
+
+impl UserdataDemo {
+    #[allow(clippy::unused_self, clippy::significant_drop_tightening)]
+    fn ui(&self, ui: &mut egui::Ui) -> Response {
+        let points = (1..=1000)
+            .map(|i| DemoPoint {
+                x: i as f64 / 1000.0,
+                y: ((i as f64) / 1000.0 * std::f64::consts::PI * 2.0).sin(),
+                custom_info: i % 2 == 0,
+            })
+            .collect::<Vec<_>>();
+
+        let custom_things = Arc::new(Mutex::new(HashMap::<Id, Vec<bool>>::new()));
+
+        let custom_things_ = custom_things.clone();
+        Plot::new("Userdata Plot Demo")
+            .legend(Legend::default())
+            .label_formatter(|_, _, item| {
+                format!(
+                    "item: {:?}\ncustom_thing: {:?}",
+                    item,
+                    item.and_then(|(id, index)| custom_things_
+                        .lock()
+                        .get(&id)
+                        .and_then(|vec| vec.get(index).copied()))
+                )
+            })
+            .show(ui, |plot_ui| {
+                let id = Id::new(1234);
+                let mut lock = custom_things.lock();
+                let entry = lock.entry(id).or_default();
+
+                for p in &points {
+                    entry.push(p.custom_info);
+                }
+
+                plot_ui.line(
+                    Line::new(
+                        "test",
+                        points.iter().map(|p| [p.x, p.y]).collect::<Vec<_>>(),
+                    )
+                    .id(id),
+                );
+            })
+            .response
+    }
+}
+
 fn is_approx_zero(val: f64) -> bool {
     val.abs() < 1e-6
 }