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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

//! Grid helpers.

use crate::input::InputError;

/// Parse 2D grid.
///
/// This function assumes that one byte represents each item in the grid.
///
/// Returns (number of rows, number of columns, data) on success.
///
/// # Examples
///
/// ```
/// # use utils::grid::from_str;
/// assert_eq!(
///     from_str("##.#\n#..#\n#.##", |c| match c {
///         b'#' => Some(true),
///         b'.' => Some(false),
///         _ => None,
///     }).unwrap(),
///     (3, 4, vec![
///         true, true, false, true,
///         true, false, false, true,
///         true, false, true, true,
///     ]),
/// );
/// ```
pub fn from_str<T>(
    input: &str,
    mut func: impl FnMut(u8) -> Option<T>,
) -> Result<(usize, usize, Vec<T>), InputError> {
    let mut data = Vec::with_capacity(input.len());
    let mut lines = input.lines().peekable();

    let Some(&first_line) = lines.peek() else {
        return Err(InputError::new(input, input, "expected grid"));
    };

    let columns = first_line.len().max(1);
    for line in lines {
        if line.len() != columns {
            return Err(InputError::new(
                input,
                line,
                format!("expected {columns} column(s)"),
            ));
        }

        for b in line.bytes() {
            if let Some(v) = func(b) {
                data.push(v);
            } else {
                return Err(InputError::new(input, b as char, "invalid character"));
            }
        }
    }

    let rows = data.len() / columns;
    debug_assert_eq!(rows * columns, data.len());

    Ok((rows, columns, data))
}

/// Parse 2D grid, adding padding around the edges.
///
/// Similar to [`from_str`], but pads the edges of the parsed grid with `padding` rows and columns
/// filled with the default value. This is helpful to avoid bounds checks when considering a
/// location's neighbors.
///
/// Returns (number of rows, number of columns, data) on success (row and column counts include
/// the added padding).
///
/// # Examples
///
/// ```
/// # use utils::grid::from_str_padded;
/// assert_eq!(
///     from_str_padded("##.#\n#..#\n#.##", 2, false, |c| match c {
///         b'#' => Some(true),
///         b'.' => Some(false),
///         _ => None,
///     }).unwrap(),
///     (7, 8, vec![
///         false, false, false, false, false, false, false, false,
///         false, false, false, false, false, false, false, false,
///         false, false, true, true, false, true, false, false,
///         false, false, true, false, false, true, false, false,
///         false, false, true, false, true, true, false, false,
///         false, false, false, false, false, false, false, false,
///         false, false, false, false, false, false, false, false,
///     ]),
/// );
/// ```
pub fn from_str_padded<T: Clone>(
    input: &str,
    padding: usize,
    padding_value: T,
    mut func: impl FnMut(u8) -> Option<T>,
) -> Result<(usize, usize, Vec<T>), InputError> {
    let mut data = Vec::with_capacity(input.len());
    let mut lines = input.lines().peekable();

    let Some(&first_line) = lines.peek() else {
        return Err(InputError::new(input, input, "expected grid"));
    };

    let columns = first_line.len().max(1);
    let padded_columns = columns + 2 * padding;

    // Add initial padding rows + padding for start of first actual row
    data.resize(padded_columns * padding + padding, padding_value.clone());

    for line in lines {
        if line.len() != columns {
            return Err(InputError::new(
                input,
                line,
                format!("expected {columns} column(s)"),
            ));
        }

        for b in line.bytes() {
            if let Some(v) = func(b) {
                data.push(v);
            } else {
                return Err(InputError::new(input, b as char, "invalid character"));
            }
        }

        // Add padding for the end of the current row, and the start of the next row
        data.resize(data.len() + 2 * padding, padding_value.clone());
    }

    // Add final padding rows, minus the already added padding for the start of a row
    data.resize(
        data.len() + padded_columns * padding - padding,
        padding_value,
    );

    let rows = data.len() / padded_columns;
    debug_assert_eq!(rows * padded_columns, data.len());

    Ok((rows, padded_columns, data))
}