alexandrepiveteau / epfl-datalog-kotlin Goto Github PK

package io.github.alexandrepiveteau.datalog.dsl.domains

import io.github.alexandrepiveteau.datalog.core.rule.Value
import io.github.alexandrepiveteau.datalog.dsl.Domain

/** An implementation of [Domain] for [String] values. */
private object StringDomain : Domain<String> {
  private val one = Value("1")
  override fun unit(): Value<String> = one
  override fun min(
      a: Value<String>,
      b: Value<String>,
  ) = if (a.value.toInt() < b.value.toInt()) a else b
  override fun max(
      a: Value<String>,
      b: Value<String>,
  ) = if (a.value.toInt() > b.value.toInt()) a else b
  override fun sum(
      a: Value<String>,
      b: Value<String>,
  ) = Value((a.value.toInt() + b.value.toInt()).toString())
}

// Domains.

/** Returns the [Domain] for [String] values. */
fun String.Companion.domain(): Domain<String> = StringDomain