diff options
Diffstat (limited to 'aoc-2022-dotnet/Common/Vec2.fs')
| -rw-r--r-- | aoc-2022-dotnet/Common/Vec2.fs | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/aoc-2022-dotnet/Common/Vec2.fs b/aoc-2022-dotnet/Common/Vec2.fs index bbe3ef2..11c94aa 100644 --- a/aoc-2022-dotnet/Common/Vec2.fs +++ b/aoc-2022-dotnet/Common/Vec2.fs @@ -4,6 +4,16 @@ type Vec2 = | Vec2 of int * int static member zero = Vec2(0, 0) + static member up = Vec2(0, 1) + static member right = Vec2(1, 0) + static member down = Vec2(0, -1) + static member left = Vec2(-1, 0) + + static member directions4 = + [ Vec2.up + Vec2.right + Vec2.down + Vec2.left ] static member inline (~-) = Vec2.map (~-) static member inline (+)(Vec2 (x1, y1), Vec2 (x2, y2)) = Vec2(x1 + x2, y1 + y2) @@ -11,9 +21,17 @@ type Vec2 = static member inline (*)(v1, k) = Vec2.map ((*) k) v1 static member inline dot (Vec2 (x1, y1)) (Vec2 (x2, y2)) = x1 * x2 + y1 * y2 static member inline cross (Vec2 (x1, y1)) (Vec2 (x2, y2)) = x1 * y2 - y1 * x2 - static member map f (Vec2 (x, y)) = Vec2(f x, f y) static member inline sign = Vec2.map sign static member inline lengthSquared(Vec2 (x, y)) = x * x + y * y static member mahattanDist (Vec2 (x1, y1)) (Vec2 (x2, y2)) = abs (x2 - x1) + abs (y2 - y1) static member chebyshevDist (Vec2 (x1, y1)) (Vec2 (x2, y2)) = max (abs <| x2 - x1) (abs <| y2 - y1) + static member neighbours4 v = List.map ((+) v) Vec2.directions4 + + static member inMatrix matrix (Vec2 (col, row)) = + col >= 0 + && col < Array2D.length2 matrix + && row >= 0 + && row < Array2D.length1 matrix + + static member toIndexOf matrix (Vec2 (col, row)) = (Array2D.length2 matrix) * row + col |