sigma.data.CSigmaProp.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.data

import sigma.ast.SSigmaProp
import sigma.serialization.CoreSerializer
import sigma.util.Extensions.SigmaBooleanOps
import sigma.{Coll, Colls, GroupElement, SigmaProp}

/** A default implementation of [[SigmaProp]] interface.
  *
  * @see [[SigmaProp]] for detailed descriptions
  */
case class CSigmaProp(sigmaTree: SigmaBoolean) extends SigmaProp with WrapperOf[SigmaBoolean] {
  override def wrappedValue: SigmaBoolean = sigmaTree

  // TODO refactor: remove this (it shouldn't be used in interpreter)
  override def isValid: Boolean = sigmaTree match {
    case p: TrivialProp => p.condition
    case _ => sys.error(s"Method CostingSigmaProp.isValid is not defined for $sigmaTree")
  }

  override def propBytes: Coll[Byte] = {
    // in order to have comparisons like  `box.propositionBytes == pk.propBytes` we need to make sure
    // the same serialization method is used in both cases
    // TODO v6.0: add `pk.propBytes(version)` (see https://github.com/ScorexFoundation/sigmastate-interpreter/issues/903)
    val w = CoreSerializer.startWriter()
    w.put(0)  // ErgoTree.header
    w.putType(SSigmaProp)
    SigmaBoolean.serializer.serialize(wrappedValue, w)
    Colls.fromArray(w.toBytes)
  }

  override def &&(other: SigmaProp): SigmaProp = other match {
    case other: CSigmaProp =>
      CSigmaProp(CAND.normalized(Array(sigmaTree, other.sigmaTree)))
  }

  override def ||(other: SigmaProp): SigmaProp = other match {
    case other: CSigmaProp =>
      CSigmaProp(COR.normalized(Array(sigmaTree, other.sigmaTree)))
  }

  override def toString: String = s"SigmaProp(${wrappedValue.showToString})"
}

object CSigmaProp {
  /** Create trivial sigma proposition with given boolean value. */
  def apply(b: Boolean): CSigmaProp =
    CSigmaProp(TrivialProp(b))
    
  /** Create SigmaProp value with underlying ProveDlog proposition. */
  def withProveDlog(ge: GroupElement) =
    CSigmaProp(ProveDlog(ge.asInstanceOf[CGroupElement].wrappedValue))

  /** Create SigmaProp value with underlying ProveDHTuple proposition. */
  def withProveDHTuple(g: GroupElement, h: GroupElement, u: GroupElement, v: GroupElement) =
    CSigmaProp(ProveDHTuple(
      g.asInstanceOf[CGroupElement].wrappedValue,
      h.asInstanceOf[CGroupElement].wrappedValue,
      u.asInstanceOf[CGroupElement].wrappedValue,
      v.asInstanceOf[CGroupElement].wrappedValue))
}