sigma.serialization.ConstantStore.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sigma-state_2.12 Show documentation

Interpreter of a Sigma-State language

The newest version!

package sigma.serialization

import sigma.ast.{Constant, ConstantNode, ConstantPlaceholder, SigmaBuilder}
import debox.Buffer
import sigma.ast.SType

/** HOTSPOT: used in deserialization (don't beautify this code) */
class ConstantStore(private val constants: IndexedSeq[Constant[SType]] = Constant.EmptySeq) {

  private val store: Buffer[Constant[SType]] = Buffer.fromIterable(constants)

  def put[T <: SType](c: Constant[T])(implicit builder: SigmaBuilder): ConstantPlaceholder[T] = {
    store += c.asInstanceOf[Constant[SType]]
    val tpe = c.asInstanceOf[ConstantNode[T]].tpe
    builder.mkConstantPlaceholder[tpe.type](store.length - 1, tpe)
      .asInstanceOf[sigma.ast.ConstantPlaceholder[T]]
  }

  @inline final def get(index: Int): Constant[SType] = store(index)

  @inline final def getAll: IndexedSeq[Constant[SType]] = store.toArray()
}