plotters_unstable/series/
point_series.rs

1use crate::element::PointElement;
2use crate::style::{ShapeStyle, SizeDesc};
3
4/// The point plot object, which takes an iterator of points in guest coordinate system
5/// and create an element for each point
6pub struct PointSeries<'a, Coord, I: IntoIterator<Item = Coord>, E, Size: SizeDesc + Clone> {
7    style: ShapeStyle,
8    size: Size,
9    data_iter: I::IntoIter,
10    make_point: &'a dyn Fn(Coord, Size, ShapeStyle) -> E,
11}
12
13impl<'a, Coord, I: IntoIterator<Item = Coord>, E, Size: SizeDesc + Clone> Iterator
14    for PointSeries<'a, Coord, I, E, Size>
15{
16    type Item = E;
17    fn next(&mut self) -> Option<Self::Item> {
18        self.data_iter
19            .next()
20            .map(|x| (self.make_point)(x, self.size.clone(), self.style.clone()))
21    }
22}
23
24impl<'a, Coord, I: IntoIterator<Item = Coord>, E, Size: SizeDesc + Clone>
25    PointSeries<'a, Coord, I, E, Size>
26where
27    E: PointElement<Coord, Size>,
28{
29    /// Create a new point series with the element that implements point trait.
30    /// You may also use a more general way to create a point series with `of_element`
31    /// function which allows a customized element construction function
32    pub fn new<S: Into<ShapeStyle>>(iter: I, size: Size, style: S) -> Self {
33        Self {
34            data_iter: iter.into_iter(),
35            size,
36            style: style.into(),
37            make_point: &|a, b, c| E::make_point(a, b, c),
38        }
39    }
40}
41
42impl<'a, Coord, I: IntoIterator<Item = Coord>, E, Size: SizeDesc + Clone>
43    PointSeries<'a, Coord, I, E, Size>
44{
45    /// Create a new point series. Similar to `PointSeries::new` but it doesn't
46    /// requires the element implements point trait. So instead of using the point
47    /// constructor, it uses the customized function for element creation
48    pub fn of_element<S: Into<ShapeStyle>, F: Fn(Coord, Size, ShapeStyle) -> E>(
49        iter: I,
50        size: Size,
51        style: S,
52        cons: &'a F,
53    ) -> Self {
54        Self {
55            data_iter: iter.into_iter(),
56            size,
57            style: style.into(),
58            make_point: cons,
59        }
60    }
61}