vizkit 0.1.2

A rendering-agnostic kit for data visualization
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use super::{CircleProperties, ShapeAttrs};

/// Creates an iterator of [`CircleProperties`] for drawing circles.
///
/// # Example
///
/// ```
/// use vizkit::{
///     chromatic::Color,
///     draw::{CircleProperties, ShapeAttrs, circle_iter},
///     scale::ScaleContinuous,
/// };
///
/// let x = ScaleContinuous::linear().domain([0., 0.5]);
/// let y = ScaleContinuous::linear().domain([0., 140_000.]);
/// let r = ScaleContinuous::sqrt().domain([0., 3000.]).range([4., 40.]);
///
/// let data = [[0.1, 12_000., 400.], [0.4, 80_000., 800.]];
///
/// let circles: Vec<CircleProperties> = circle_iter(
///     &data,
///     |d| x.scale(d[0]),
///     |d| y.scale(d[1]),
///     |d| r.scale(d[2]),
///     |d| ShapeAttrs {
///         fill_color: Some(Color(if d[2] < 1500. {
///             [1., 0., 0.] // red
///         } else {
///             [0., 0., 1.] // blue
///         })),
///         fill_opacity: 0.5,
///         ..Default::default()
///     },
/// ).collect();
/// ```
pub fn circle_iter<Data>(
    values: &[Data],
    x: impl Fn(&Data) -> f32,
    y: impl Fn(&Data) -> f32,
    r: impl Fn(&Data) -> f32,
    shape_attrs: impl Fn(&Data) -> ShapeAttrs,
) -> impl Iterator<Item = CircleProperties> {
    values.iter().map(move |value| {
        let circle_values = (shape_attrs)(value);
        CircleProperties {
            center: [x(value), y(value)],
            radius: r(value),
            fill_color: circle_values.fill_color,
            fill_opacity: circle_values.fill_opacity,
            stroke_color: circle_values.stroke_color,
            stroke_width: circle_values.stroke_width,
            stroke_opacity: circle_values.stroke_opacity,
        }
    })
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::scale::ScaleContinuous;

    #[test]
    fn test_rect_iter() {
        let width = 928.;
        let height = 500.;
        let margin_top = 30.;
        let margin_right = 0.;
        let margin_bottom = 30.;
        let margin_left = 40.;
        let x = ScaleContinuous::linear()
            .domain([0., 50.])
            .range([margin_left, width - margin_right]);

        let y = ScaleContinuous::linear()
            .domain([0., 200.])
            .range([height - margin_bottom, margin_top]);

        let values = [
            (5., 90.),
            (20., 30.),
            (30., 120.),
            (40., 20.),
            (45., 50.),
            (50., 200.),
        ];

        for circle_property in circle_iter(
            &values,
            |d| x.scale(d.0),
            |d| y.scale(d.1),
            |_| 5.,
            |_| ShapeAttrs::stroke_default(),
        ) {
            assert!(
                circle_property.center[0] >= margin_left
                    && circle_property.center[0] <= width - margin_right
            );
            assert!(
                circle_property.center[1] >= margin_top
                    && circle_property.center[1] <= height - margin_bottom
            );
            assert_eq!(circle_property.radius, 5.);
        }
    }
}