dear_implot/plots/
scatter.rs

1//! Scatter plot implementation
2
3use super::{
4    Plot, PlotDataLayout, PlotDataOffset, PlotDataStride, PlotError, PlotItemStyle,
5    plot_spec_with_style, validate_data_lengths, with_plot_str_or_empty,
6};
7use crate::{ItemFlags, Marker, ScatterFlags, sys};
8
9/// Builder for scatter plots with customization options
10pub struct ScatterPlot<'a> {
11    label: &'a str,
12    x_data: &'a [f64],
13    y_data: &'a [f64],
14    style: PlotItemStyle,
15    flags: ScatterFlags,
16    item_flags: ItemFlags,
17    layout: PlotDataLayout,
18}
19
20impl<'a> super::PlotItemStyled for ScatterPlot<'a> {
21    fn style_mut(&mut self) -> &mut PlotItemStyle {
22        &mut self.style
23    }
24}
25
26impl<'a> ScatterPlot<'a> {
27    /// Create a new scatter plot with the given label and data
28    pub fn new(label: &'a str, x_data: &'a [f64], y_data: &'a [f64]) -> Self {
29        Self {
30            label,
31            x_data,
32            y_data,
33            style: PlotItemStyle::default(),
34            flags: ScatterFlags::NONE,
35            item_flags: ItemFlags::NONE,
36            layout: PlotDataLayout::DEFAULT,
37        }
38    }
39
40    /// Replace the entire item style override for this scatter plot.
41    pub fn with_style(mut self, style: PlotItemStyle) -> Self {
42        self.style = style;
43        self
44    }
45
46    /// Set the scatter line color. Use the alpha channel to control transparency.
47    pub fn with_line_color(mut self, color: [f32; 4]) -> Self {
48        self.style = self.style.with_line_color(color);
49        self
50    }
51
52    /// Set the outline width in pixels.
53    pub fn with_line_weight(mut self, weight: f32) -> Self {
54        self.style = self.style.with_line_weight(weight);
55        self
56    }
57
58    /// Set the marker type for the scatter plot.
59    pub fn with_marker(mut self, marker: Marker) -> Self {
60        self.style = self.style.with_marker(marker);
61        self
62    }
63
64    /// Set the marker size in pixels.
65    pub fn with_marker_size(mut self, size: f32) -> Self {
66        self.style = self.style.with_marker_size(size);
67        self
68    }
69
70    /// Set the marker outline color.
71    pub fn with_marker_line_color(mut self, color: [f32; 4]) -> Self {
72        self.style = self.style.with_marker_line_color(color);
73        self
74    }
75
76    /// Set the marker fill color.
77    pub fn with_marker_fill_color(mut self, color: [f32; 4]) -> Self {
78        self.style = self.style.with_marker_fill_color(color);
79        self
80    }
81
82    /// Set scatter flags for customization
83    pub fn with_flags(mut self, flags: ScatterFlags) -> Self {
84        self.flags = flags;
85        self
86    }
87
88    /// Set common item flags for this plot item (applies to all plot types)
89    pub fn with_item_flags(mut self, flags: ItemFlags) -> Self {
90        self.item_flags = flags;
91        self
92    }
93
94    /// Set the data layout used to read X/Y samples.
95    pub fn with_data_layout(mut self, layout: PlotDataLayout) -> Self {
96        self.layout = layout;
97        self
98    }
99
100    /// Set the sample-index offset used to read X/Y samples.
101    pub fn with_offset(mut self, offset: PlotDataOffset) -> Self {
102        self.layout = self.layout.with_offset(offset);
103        self
104    }
105
106    /// Set the byte stride used to read X/Y samples.
107    pub fn with_stride(mut self, stride: PlotDataStride) -> Self {
108        self.layout = self.layout.with_stride(stride);
109        self
110    }
111
112    /// Validate the plot data
113    pub fn validate(&self) -> Result<(), PlotError> {
114        validate_data_lengths(self.x_data, self.y_data)
115    }
116}
117
118impl<'a> Plot for ScatterPlot<'a> {
119    fn plot(&self) {
120        if self.validate().is_err() {
121            return; // Skip plotting if data is invalid
122        }
123        let Ok(count) = i32::try_from(self.x_data.len()) else {
124            return;
125        };
126
127        with_plot_str_or_empty(self.label, |label_ptr| unsafe {
128            let spec = plot_spec_with_style(
129                self.style,
130                self.flags.bits() | self.item_flags.bits(),
131                self.layout,
132            );
133            sys::ImPlot_PlotScatter_doublePtrdoublePtr(
134                label_ptr,
135                self.x_data.as_ptr(),
136                self.y_data.as_ptr(),
137                count,
138                spec,
139            );
140        })
141    }
142
143    fn label(&self) -> &str {
144        self.label
145    }
146}
147
148/// Simple scatter plot for quick plotting without builder pattern
149pub struct SimpleScatterPlot<'a> {
150    label: &'a str,
151    values: &'a [f64],
152    style: PlotItemStyle,
153    flags: ScatterFlags,
154    item_flags: ItemFlags,
155    x_scale: f64,
156    x_start: f64,
157}
158
159impl<'a> super::PlotItemStyled for SimpleScatterPlot<'a> {
160    fn style_mut(&mut self) -> &mut PlotItemStyle {
161        &mut self.style
162    }
163}
164
165impl<'a> SimpleScatterPlot<'a> {
166    /// Create a simple scatter plot with Y values only (X will be indices)
167    pub fn new(label: &'a str, values: &'a [f64]) -> Self {
168        Self {
169            label,
170            values,
171            style: PlotItemStyle::default(),
172            flags: ScatterFlags::NONE,
173            item_flags: ItemFlags::NONE,
174            x_scale: 1.0,
175            x_start: 0.0,
176        }
177    }
178
179    /// Replace the entire item style override for this scatter plot.
180    pub fn with_style(mut self, style: PlotItemStyle) -> Self {
181        self.style = style;
182        self
183    }
184
185    /// Set the scatter line color. Use the alpha channel to control transparency.
186    pub fn with_line_color(mut self, color: [f32; 4]) -> Self {
187        self.style = self.style.with_line_color(color);
188        self
189    }
190
191    /// Set the outline width in pixels.
192    pub fn with_line_weight(mut self, weight: f32) -> Self {
193        self.style = self.style.with_line_weight(weight);
194        self
195    }
196
197    /// Set the marker type for the scatter plot.
198    pub fn with_marker(mut self, marker: Marker) -> Self {
199        self.style = self.style.with_marker(marker);
200        self
201    }
202
203    /// Set the marker size in pixels.
204    pub fn with_marker_size(mut self, size: f32) -> Self {
205        self.style = self.style.with_marker_size(size);
206        self
207    }
208
209    /// Set the marker outline color.
210    pub fn with_marker_line_color(mut self, color: [f32; 4]) -> Self {
211        self.style = self.style.with_marker_line_color(color);
212        self
213    }
214
215    /// Set the marker fill color.
216    pub fn with_marker_fill_color(mut self, color: [f32; 4]) -> Self {
217        self.style = self.style.with_marker_fill_color(color);
218        self
219    }
220
221    /// Set scatter flags for customization
222    pub fn with_flags(mut self, flags: ScatterFlags) -> Self {
223        self.flags = flags;
224        self
225    }
226
227    /// Set common item flags for this plot item (applies to all plot types)
228    pub fn with_item_flags(mut self, flags: ItemFlags) -> Self {
229        self.item_flags = flags;
230        self
231    }
232
233    /// Set X scale factor
234    pub fn with_x_scale(mut self, scale: f64) -> Self {
235        self.x_scale = scale;
236        self
237    }
238
239    /// Set X start value
240    pub fn with_x_start(mut self, start: f64) -> Self {
241        self.x_start = start;
242        self
243    }
244}
245
246impl<'a> Plot for SimpleScatterPlot<'a> {
247    fn plot(&self) {
248        if self.values.is_empty() {
249            return;
250        }
251        let Ok(count) = i32::try_from(self.values.len()) else {
252            return;
253        };
254
255        with_plot_str_or_empty(self.label, |label_ptr| unsafe {
256            let spec = plot_spec_with_style(
257                self.style,
258                self.flags.bits() | self.item_flags.bits(),
259                PlotDataLayout::DEFAULT,
260            );
261            sys::ImPlot_PlotScatter_doublePtrInt(
262                label_ptr,
263                self.values.as_ptr(),
264                count,
265                self.x_scale,
266                self.x_start,
267                spec,
268            );
269        })
270    }
271
272    fn label(&self) -> &str {
273        self.label
274    }
275}
276
277/// Convenience functions for quick scatter plotting
278impl<'ui> crate::PlotUi<'ui> {
279    /// Plot a scatter plot with X and Y data
280    pub fn scatter_plot(
281        &self,
282        label: &str,
283        x_data: &[f64],
284        y_data: &[f64],
285    ) -> Result<(), PlotError> {
286        let plot = ScatterPlot::new(label, x_data, y_data);
287        plot.validate()?;
288        self.bind();
289        plot.plot();
290        Ok(())
291    }
292
293    /// Plot a simple scatter plot with Y values only (X will be indices)
294    pub fn simple_scatter_plot(&self, label: &str, values: &[f64]) -> Result<(), PlotError> {
295        if values.is_empty() {
296            return Err(PlotError::EmptyData);
297        }
298        let plot = SimpleScatterPlot::new(label, values);
299        self.bind();
300        plot.plot();
301        Ok(())
302    }
303}
304
305#[cfg(test)]
306mod tests {
307    use super::*;
308
309    #[test]
310    fn test_scatter_plot_creation() {
311        let x_data = [1.0, 2.0, 3.0, 4.0];
312        let y_data = [1.0, 4.0, 2.0, 3.0];
313
314        let plot = ScatterPlot::new("test", &x_data, &y_data);
315        assert_eq!(plot.label(), "test");
316        assert!(plot.validate().is_ok());
317    }
318
319    #[test]
320    fn test_scatter_plot_validation() {
321        let x_data = [1.0, 2.0, 3.0];
322        let y_data = [1.0, 4.0]; // Different length
323
324        let plot = ScatterPlot::new("test", &x_data, &y_data);
325        assert!(plot.validate().is_err());
326    }
327
328    #[test]
329    fn test_simple_scatter_plot() {
330        let values = [1.0, 2.0, 3.0, 4.0];
331        let plot = SimpleScatterPlot::new("test", &values)
332            .with_flags(ScatterFlags::NO_CLIP)
333            .with_item_flags(ItemFlags::NO_FIT);
334        assert_eq!(plot.label(), "test");
335        assert_eq!(plot.flags.bits(), ScatterFlags::NO_CLIP.bits());
336        assert_eq!(plot.item_flags, ItemFlags::NO_FIT);
337    }
338
339    #[test]
340    fn test_scatter_plot_style_builders() {
341        let x_data = [1.0, 2.0, 3.0, 4.0];
342        let y_data = [1.0, 4.0, 2.0, 3.0];
343
344        let plot = ScatterPlot::new("styled", &x_data, &y_data)
345            .with_line_color([0.1, 0.2, 0.3, 0.4])
346            .with_line_weight(1.5)
347            .with_marker(Marker::Square)
348            .with_marker_size(8.0)
349            .with_marker_line_color([0.9, 0.8, 0.7, 0.6])
350            .with_marker_fill_color([0.6, 0.7, 0.8, 0.9]);
351
352        assert_eq!(
353            plot.style.line_color,
354            Some(sys::ImVec4_c {
355                x: 0.1,
356                y: 0.2,
357                z: 0.3,
358                w: 0.4,
359            })
360        );
361        assert_eq!(plot.style.line_weight, Some(1.5));
362        assert_eq!(plot.style.marker, Some(Marker::Square as sys::ImPlotMarker));
363        assert_eq!(plot.style.marker_size, Some(8.0));
364        assert_eq!(
365            plot.style.marker_line_color,
366            Some(sys::ImVec4_c {
367                x: 0.9,
368                y: 0.8,
369                z: 0.7,
370                w: 0.6,
371            })
372        );
373        assert_eq!(
374            plot.style.marker_fill_color,
375            Some(sys::ImVec4_c {
376                x: 0.6,
377                y: 0.7,
378                z: 0.8,
379                w: 0.9,
380            })
381        );
382    }
383}
dear_implot/plots/scatter.rs

dear_implot/plots/
scatter.rs
