Prepare for other years

2020-12-30 18:01:52 +01:00
parent 522618d106
commit 4a90257257
81 changed files with 277 additions and 253 deletions
@@ -0,0 +1,64 @@
+@file:Suppress("MemberVisibilityCanBePrivate")
+
+package be.vandewalleh.aoc.days.geometry
+
+class Grid<T> {
+    val data: ArrayList<ArrayList<T>> = ArrayList()
+
+    constructor(data: Iterable<Iterable<T>>) {
+        data.forEach {
+            this.data.add(ArrayList(it.toList()))
+        }
+    }
+
+    constructor(data: Array<Array<T>>) {
+        data.forEach {
+            this.data.add(ArrayList(it.toList()))
+        }
+    }
+
+    val width get() = data[0].size
+    val lastColumnIndex get() = data[0].size - 1
+
+    val height get() = data.size
+    val lastRowIndex get() = data.size - 1
+
+    operator fun get(x: Int, y: Int): T? {
+        if (y !in 0..lastRowIndex) return null
+        val row = data[y]
+        return if (x !in 0..lastColumnIndex) null
+        else row[x]
+    }
+
+    operator fun set(x: Int, y: Int, value: T) {
+        data[y][x] = value
+    }
+
+    fun row(y: Int): List<T> = data[y]
+    fun firstRow() = row(0)
+    fun lastRow() = row(height - 1)
+
+    fun column(x: Int): List<T> = data.map { it[x] }
+    fun firstColumn() = column(0)
+    fun lastColumn() = column(width - 1)
+
+    fun edges() = listOf(row(0), column(0), row(lastRowIndex), column(lastColumnIndex))
+
+    override fun toString() = buildString {
+        data.forEach { line ->
+            append(line.joinToString(""))
+            appendLine()
+        }
+    }
+
+    override fun equals(other: Any?): Boolean {
+        if (this === other) return true
+        if (javaClass != other?.javaClass) return false
+        other as Grid<*>
+        if (data != other.data) return false
+        return true
+    }
+
+    override fun hashCode() = data.hashCode()
+
+}