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

use super::color::{Color, Cubehelix, Interpolator, color};

#[derive(Clone)]
pub struct CubehelixInterpolator {
    gamma: f32,
    h: Interpolator,
    s: Interpolator,
    l: Interpolator,
}

impl CubehelixInterpolator {
    pub(crate) fn gamma(self, gamma: f32) -> Self {
        Self { gamma, ..self }
    }

    pub(crate) fn interpolate<T>(&self, t: f32) -> T
    where
        Color: Into<T>,
    {
        Color::from(Cubehelix([
            self.h.interpolate(t),
            self.s.interpolate(t),
            self.l.interpolate(t.powf(self.gamma)),
        ]))
        .into()
    }

    pub(crate) fn warm() -> Self {
        let [sh, ss, sl] = [-100., 0.75, 0.35];
        let [eh, es, el] = [80., 1.50, 0.8];
        let h = color(sh, eh);
        let s = color(ss, es);
        let l = color(sl, el);
        Self { gamma: 1., h, s, l }
    }

    pub(crate) fn cold() -> Self {
        let [sh, ss, sl] = [260., 0.75, 0.35];
        let [eh, es, el] = [80., 1.50, 0.8];
        let h = color(sh, eh);
        let s = color(ss, es);
        let l = color(sl, el);
        Self { gamma: 1., h, s, l }
    }
}

impl Default for CubehelixInterpolator {
    fn default() -> Self {
        let [sh, ss, sl] = [300., 0.5, 0.0];
        let [eh, es, el] = [-240., 0.5, 1.0];
        let h = color(sh, eh);
        let s = color(ss, es);
        let l = color(sl, el);
        Self { gamma: 1., h, s, l }
    }
}

#[cfg(test)]
mod tests {
    use super::CubehelixInterpolator;

    #[test]
    fn test_viridis() {
        let step = 100;
        let variants = [
            ("cold", CubehelixInterpolator::cold()),
            ("warm", CubehelixInterpolator::warm()),
        ];

        for (space, interpolator) in variants {
            let colors: Vec<[f32; 3]> = (0..=step)
                .map(|i| interpolator.interpolate(i as f32 / step as f32))
                .collect();
            for color in colors {
                let [r, g, b] = color;
                assert!(
                    (0. ..=1.).contains(&r),
                    "red must be between [0, 1] (variant: {:?})",
                    space
                );
                assert!(
                    (0. ..=1.).contains(&g),
                    "green must be between [0, 1] (variant: {:?})",
                    space
                );
                assert!(
                    (0. ..=1.).contains(&b),
                    "blue must be between [0, 1] (variant: {:?})",
                    space
                );
            }
        }
    }
}