import scala.collection.mutable.ArrayBuffer

object Slices {
  def main(args: Array[String]): Unit = {
    // In Scala, we can use ArrayBuffer for a mutable sequence similar to Go's slices
    var s = ArrayBuffer.empty[String]
    println(s"uninit: $s, ${s.isEmpty}, ${s.length == 0}")

    // To create an empty ArrayBuffer with non-zero length, we can use fill
    s = ArrayBuffer.fill(3)("")
    println(s"emp: $s, len: ${s.length}")

    // We can set and get just like with arrays
    s(0) = "a"
    s(1) = "b"
    s(2) = "c"
    println(s"set: $s")
    println(s"get: ${s(2)}")

    // length returns the length of the ArrayBuffer as expected
    println(s"len: ${s.length}")

    // We can append elements to an ArrayBuffer
    s += "d"
    s ++= Seq("e", "f")
    println(s"apd: $s")

    // We can copy ArrayBuffers
    val c = s.clone()
    println(s"cpy: $c")

    // Scala provides slice operations on collections
    val l = s.slice(2, 5)
    println(s"sl1: $l")

    // This slices up to (but excluding) index 5
    val l2 = s.take(5)
    println(s"sl2: $l2")

    // And this slices from index 2 to the end
    val l3 = s.drop(2)
    println(s"sl3: $l3")

    // We can declare and initialize a sequence in a single line
    val t = Vector("g", "h", "i")
    println(s"dcl: $t")

    // Scala provides many utility functions for collections
    val t2 = Vector("g", "h", "i")
    if (t == t2) {
      println("t == t2")
    }

    // We can create multi-dimensional structures
    val twoD = Array.ofDim[Int](3, 3)
    for (i <- 0 until 3) {
      for (j <- 0 to i) {
        twoD(i)(j) = i + j
      }
    }
    println(s"2d: ${twoD.map(_.mkString("[", ",", "]")).mkString("[", ", ", "]")}")
  }
}