dererXII.scalashogi_2.13.5.0.2.source-code.LagTracker.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of scalashogi_2.13 Show documentation

Shogi API written in scala. Immutable and free of side effects.

There is a newer version: 12.1.1

package shogi

final case class LagTracker(
    quotaGain: Centis,
    quota: Centis,
    quotaMax: Centis,
    lagEstimator: DecayingRecorder,
    uncompStats: Stats = EmptyStats,
    lagStats: Stats = EmptyStats,
    // We can remove compEst fields after tuning estimate.
    compEstSqErr: Int = 0,
    compEstOvers: Centis = Centis(0),
    compEstimate: Option[Centis] = None
) {
  def onMove(lag: Centis) = {
    val comp     = lag atMost quota
    val uncomped = lag - comp
    val ceDiff   = (compEstimate | Centis(1)) - comp

    (
      comp,
      copy(
        quota = (quota + quotaGain - comp) atMost quotaMax,
        uncompStats = {
          // start recording after first uncomp.
          if (uncomped == Centis(0) && uncompStats.samples == 0) uncompStats
          else uncompStats record uncomped.centis.toFloat
        },
        lagStats = lagStats record (lag atMost Centis(2000)).centis.toFloat,
        compEstSqErr = compEstSqErr + ceDiff.centis * ceDiff.centis,
        compEstOvers = compEstOvers + ceDiff.nonNeg
      ).recordLag(lag)
    )
  }

  def recordLag(lag: Centis) = {
    val e = lagEstimator.record((lag atMost quotaMax).centis.toFloat)
    copy(
      lagEstimator = e,
      compEstimate = Some(
        Centis(e.mean - .8f * e.deviation).nonNeg atMost quota
      )
    )
  }

  def moves = lagStats.samples

  def lagMean: Option[Centis] = moves > 0 option Centis(lagStats.mean)

  def compEstStdErr: Option[Float] =
    moves > 2 option math.sqrt(compEstSqErr).toFloat / (moves - 2)

  def compAvg: Option[Centis] = totalComp / moves

  def totalComp: Centis = totalLag - totalUncomped

  def totalLag: Centis = Centis(lagStats.total)

  def totalUncomped: Centis = Centis(uncompStats.total)
}

object LagTracker {
  def init(config: Clock.Config) = {
    val quotaGain = Centis(config.estimateTotalSeconds match {
      case i if i >= 180 => 100
      case i if i <= 15  => 35
      case i             => i / 3 + 40
    })
    LagTracker(
      quotaGain = quotaGain,
      quota = quotaGain * 3,
      quotaMax = quotaGain * 7,
      lagEstimator = EmptyDecayingStats(deviation = 4f, decay = 0.85f)
    )
  }
}