teststate.data.VectorTree.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of core_3 Show documentation

core

There is a newer version: 3.1.0

package teststate.data

import teststate.data.VectorTree.Node

final case class VectorTree[+A](elements: Vector[Node[A]]) {
  def map[B](f: A => B): VectorTree[B] =
    VectorTree(elements.map(_.map(f)))

  def shallow: Vector[A] =
    elements.map(_.value)

  def iteratorWithLevel(currentLevel: Int = 0): Iterator[(Int, A)] =
    elements.iterator.flatMap(_.iteratorWithLevel(currentLevel))

  def prettyPrintedLines(indent: String = "  "): Iterator[String] =
    iteratorWithLevel().map(x => (indent * x._1) + x._2)

  def ana[B](f: (A, Vector[B]) => B): Vector[B] =
    elements.map(n => f(n.value, n.children.ana(f)))

  override def toString: String =
    prettyPrintedLines().mkString("\n")

  def deepSize: Int =
    elements.foldLeft(0)(_ + _.deepSize)
}

object VectorTree {
  val empty = apply(Vector.empty)

  final case class Node[+A](value: A, children: VectorTree[A]) {
    def map[B](f: A => B): Node[B] =
      Node(f(value), children.map(f))

    def iteratorWithLevel(currentLevel: Int = 0): Iterator[(Int, A)] =
      Iterator.single(currentLevel -> value) ++ children.iteratorWithLevel(currentLevel + 1)

    def deepSize: Int =
      1 + children.deepSize
  }

  def flat[A](as: Vector[A]): VectorTree[A] =
    apply(as.map(Node[A](_, empty)))

  def lift[A](a: Node[A]): VectorTree[A] =
    apply(Vector.empty[Node[A]] :+ a)

  def one[A](value: A): VectorTree[A] =
    one(value, empty)

  def one[A](value: A, children: VectorTree[A]): VectorTree[A] =
    lift(Node(value, children))
}