reorganize a bit

1 files changed, 65 insertions, 0 deletions

diff --git a/src/parse.rs b/src/parse.rs
new file mode 100644
index 0000000..903ad2f
--- /dev/null
+++ b/src/parse.rs
@@ -0,0 +1,65 @@
+use crate::prelude::*;
+
+pub fn data(year: u16, day: u16) -> Result<File> {
+    File::open(format!("data/{}/{}.txt", year, day)).map_err(|e| anyhow!(e))
+}
+
+pub fn lines(fh: File) -> impl Iterator<Item = String> {
+    let fh = std::io::BufReader::new(fh);
+    fh.lines().map(|res| res.unwrap())
+}
+
+pub fn split(fh: File, sep: u8) -> impl Iterator<Item = String> {
+    let fh = std::io::BufReader::new(fh);
+    fh.split(sep)
+        .map(|res| String::from_utf8(res.unwrap()).unwrap())
+}
+
+pub fn ints(iter: impl Iterator<Item = String>) -> impl Iterator<Item = i64> {
+    iter.map(|s| s.trim().parse().unwrap())
+}
+
+pub fn bytes(fh: File) -> impl Iterator<Item = u8> {
+    fh.bytes().map(|res| res.unwrap())
+}
+
+pub fn string(fh: File) -> String {
+    let bytes: Vec<_> = bytes(fh).collect();
+    String::from_utf8(bytes).unwrap()
+}
+
+pub fn bool_grid(
+    lines: impl Iterator<Item = String>,
+    t: u8,
+    f: u8,
+) -> Grid<bool> {
+    lines
+        .map(|s| {
+            s.as_bytes()
+                .iter()
+                .copied()
+                .map(|b| {
+                    if b == f {
+                        false
+                    } else if b == t {
+                        true
+                    } else {
+                        panic!("unrecognized character {}", char::from(b))
+                    }
+                })
+                .collect::<Vec<_>>()
+        })
+        .collect()
+}
+
+pub fn digit_grid(lines: impl Iterator<Item = String>) -> Grid<u8> {
+    lines
+        .map(|s| {
+            s.as_bytes()
+                .iter()
+                .copied()
+                .map(|b| b - b'0')
+                .collect::<Vec<_>>()
+        })
+        .collect()
+}