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

#![deny(missing_docs)]

//! Configure a grid on a plot.
//!
//! Grids allow for easier estimating of data values. This module allows the configuration of grids
//! on plots.
//!
//! Grids are created by creating a `Grid` definition, and adding it to a plot:
//!
//! The grid lines for `plotlib` are rendered
//! _underneath_ the data so as to not detract from the data.
//!
//! # Examples
//!
//! ```rust
//! # use plotlib::view::ContinuousView;
//! use plotlib::grid::Grid;
//! # use plotlib::style::LineStyle;
//! # use plotlib::view::View;
//!
//! # let l1 = plotlib::repr::Plot::new(vec![(0., 1.), (2., 1.5), (3., 1.2), (4., 1.1)])
//! #    .line_style(LineStyle::new().colour("burlywood"));
//! // let l1 = Line::new() ...
//! let mut v = ContinuousView::new().add(l1);
//!
//! // 3 vertical lines and 8 horizontal lines
//! v.add_grid(Grid::new(3, 8));
//!
//! // Render plot
//! ```

// Internal type representing the logic of when do we render only horizontal lines, and when do we
// render a full grid
pub(crate) enum GridType<'a> {
    HorizontalOnly(&'a Grid),
    Both(&'a Grid),
}

/// Configuration for the grid on a plot
///
/// Supports changing the number of grid lines for the x and y dimensions.
/// **Note:** for categorical plots, only horizontal lines will be shown.
pub struct Grid {
    /// Number of vertical grid lines (defaults to 3)
    pub nx: u32,
    /// Number of horizontal grid lines (defaults to 3)
    pub ny: u32,
    /// Color of the grid lines (defaults to "darkgrey")
    pub color: String,
}

impl Default for Grid {
    fn default() -> Self {
        Grid::new(3, 3)
    }
}

impl Grid {
    /// Create a new grid with `nx` vertical and `ny` horizontal grid lines
    ///
    /// The default colour is "darkgrey".
    pub fn new(nx: u32, ny: u32) -> Grid {
        Grid {
            nx,
            ny,
            color: "darkgrey".to_owned(),
        }
    }
}