com.github.gchudnov.mtg.diagram.Section.scala Maven / Gradle / Ivy

Go to download
package com.github.gchudnov.mtg.diagram

import com.github.gchudnov.mtg.Interval
import com.github.gchudnov.mtg.Domain

/**
 * Section
 */
final case class Section[T](
  title: String,
  intervals: List[Interval[T]],
  annotations: List[String],
):
  def withTitle(newTitle: String): Section[T] =
    this.copy(title = newTitle)

  def addInterval(interval: Interval[T]): Section[T] =
    addInterval(interval, "")

  def addInterval(interval: Interval[T], annotation: String): Section[T] =
    this.copy(
      intervals = intervals :+ interval,
      annotations = annotations :+ (if annotation.isEmpty then annotations.size.toString() else annotation),
    )

  def addInterval(start: T, end: T)(using D: Domain[T]): Section[T] =
    addInterval(start, end, "")

  def addInterval(start: T, end: T, annotation: String)(using D: Domain[T]): Section[T] =
    this.copy(
      intervals = intervals :+ Interval.closed(start, end),
      annotations = annotations :+ (if annotation.isEmpty then annotations.size.toString() else annotation),
    )

/**
 * Section Companion Object
 */
object Section:

  /**
   * An empty section.
   */
  def empty[T]: Section[T] =
    Section("", List.empty[Interval[T]], List.empty[String])