liveplot 2.0.1

Realtime interactive plotting library using egui/eframe, with optional gRPC and Parquet export support.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

//! TraceLook: visual styling for plot traces.

use eframe::egui::Color32;
use egui_plot::{LineStyle, MarkerShape};

/// The visual presentation of a trace (color, visibility, line style, markers).
#[derive(Debug, Clone)]
pub struct TraceLook {
    pub color: Color32,
    pub visible: bool,
    pub width: f32,
    pub show_points: bool,
    pub highlight_newest_point: bool,
    pub style: LineStyle,
    pub point_size: f32,
    pub marker: MarkerShape,
}

impl Default for TraceLook {
    fn default() -> Self {
        Self {
            color: Color32::GRAY,
            visible: true,
            width: 1.5,
            show_points: false,
            highlight_newest_point: false,
            style: LineStyle::Solid,
            point_size: 4.0,
            marker: MarkerShape::Circle,
        }
    }
}

impl TraceLook {
    /// Create a new TraceLook with a color allocated based on the trace index.
    pub fn new(index: usize) -> Self {
        Self {
            color: Self::alloc_color(index),
            ..Default::default()
        }
    }

    /// Allocate a distinct color for the given trace index.
    pub fn alloc_color(index: usize) -> Color32 {
        // Consult the global colour palette, which is kept in sync with the
        // currently-applied `ColorScheme`.  Fallback to gray if something went
        // wrong (empty palette or poisoned lock).
        let palette = crate::color_scheme::global_palette();
        if palette.is_empty() {
            Color32::GRAY
        } else {
            palette[index % palette.len()]
        }
    }
}

// --- tests -----------------------------------------------------------------

#[cfg(test)]
mod tests {
    use super::*;
    use crate::color_scheme;
    use egui::Color32;

    #[test]
    fn alloc_color_uses_global_palette() {
        // start with known palette
        color_scheme::set_global_palette(vec![
            Color32::from_rgb(1, 2, 3),
            Color32::from_rgb(4, 5, 6),
        ]);
        assert_eq!(TraceLook::alloc_color(0), Color32::from_rgb(1, 2, 3));
        assert_eq!(TraceLook::alloc_color(1), Color32::from_rgb(4, 5, 6));
        assert_eq!(TraceLook::alloc_color(2), Color32::from_rgb(1, 2, 3));
    }
}