From 00000230cf3de760ab568bd9b09d497c9bcfbe30 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Matej=20Jane=C5=BEi=C4=8D?= <janezic.mj@gmail.com>
Date: Wed, 18 Dec 2024 22:01:22 +0100
Subject: [PATCH] feat: add mul<usize> for direction and diagonal directions

---
 src/bin/16.rs             |  2 +-
 src/grid_vec/direction.rs | 32 +++++++++++++++++++++++++++-----
 src/grid_vec/point.rs     |  4 ++++
 3 files changed, 32 insertions(+), 6 deletions(-)

diff --git a/src/bin/16.rs b/src/bin/16.rs
index c2f1cd8..5148653 100644
--- a/src/bin/16.rs
+++ b/src/bin/16.rs
@@ -22,7 +22,7 @@ fn dijkstra(grid: &Grid, start: Point) -> (Visited, Prev) {
     let mut visited: Visited = HashMap::new();
 
     let mut bh = BinaryHeap::new();
-    bh.push(Reverse((0, start, Direction::EAST, start, Direction::EAST)));
+    bh.push(Reverse((0, start, Direction::E, start, Direction::E)));
 
     while let Some(Reverse((len, loc, dir, ploc, pdir))) = bh.pop() {
         let best = visited.entry((loc, dir)).or_insert(usize::MAX);
diff --git a/src/grid_vec/direction.rs b/src/grid_vec/direction.rs
index 4a78d57..43bb91f 100644
--- a/src/grid_vec/direction.rs
+++ b/src/grid_vec/direction.rs
@@ -1,4 +1,4 @@
-use std::fmt::Display;
+use std::{fmt::Display, ops::Mul};
 
 #[derive(Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Clone, Copy)]
 pub struct Direction {
@@ -12,11 +12,33 @@ impl Display for Direction {
     }
 }
 
+impl Mul<usize> for Direction {
+    type Output = Self;
+    fn mul(self, rhs: usize) -> Self::Output {
+        Self {
+            i: self.i * rhs as isize,
+            j: self.j * rhs as isize,
+        }
+    }
+}
+
 impl Direction {
-    pub const NORTH: Self = Direction { i: -1, j: 0 };
-    pub const EAST: Self = Direction { i: 0, j: 1 };
-    pub const SOUTH: Self = Direction { i: 1, j: 0 };
-    pub const WEST: Self = Direction { i: 0, j: -1 };
+    pub const N: Self = Direction { i: -1, j: 0 };
+    pub const E: Self = Direction { i: 0, j: 1 };
+    pub const S: Self = Direction { i: 1, j: 0 };
+    pub const W: Self = Direction { i: 0, j: -1 };
+
+    pub const NE: Self = Direction { i: -1, j: 1 };
+    pub const NW: Self = Direction { i: -1, j: -1 };
+    pub const SE: Self = Direction { i: 1, j: 1 };
+    pub const SW: Self = Direction { i: 1, j: -1 };
+
+    pub const CROSS: [Self; 4] = [Self::N, Self::E, Self::S, Self::W];
+    #[rustfmt::skip]
+    pub const OMNI: [Self; 8] = [
+        Self::N, Self::E, Self::S, Self::W,
+        Self::NE, Self::NW, Self::SE, Self::SW
+    ];
 
     pub fn new(i: isize, j: isize) -> Self {
         Self { i, j }
diff --git a/src/grid_vec/point.rs b/src/grid_vec/point.rs
index 39c702a..dcf2a55 100644
--- a/src/grid_vec/point.rs
+++ b/src/grid_vec/point.rs
@@ -81,6 +81,10 @@ impl Point {
     pub fn sw(self) -> Self {
         self.s().w()
     }
+
+    pub fn bounded_by(&self, other: &Self) -> bool {
+        self.i <= other.i && self.j <= other.j
+    }
 }
 
 #[macro_export]