Merge pull request #1062 from jozanek/support-unit-type

Support unit type

Author: kushti

sc/shared/src/main/scala/sigma/compiler/ir/GraphBuilding.scala

@@ -1,29 +1,25 @@
 package sigma.compiler.ir
 
 import org.ergoplatform._
-import sigma.ast.SType.tT
-import sigma.{SigmaException, VersionContext, ast}
 import sigma.Evaluation.stypeToRType
+import sigma.SigmaException
+import sigma.VersionContext
+import sigma.ast.{Ident, Select, Val}
 import sigma.ast.SType.tT
-import sigma.{SigmaException, VersionContext, ast}
 import sigma.ast.TypeCodes.LastConstantCode
 import sigma.ast.Value.Typed
 import sigma.ast.syntax.{SValue, ValueOps}
 import sigma.ast._
 import sigma.compiler.ir.core.MutableLazy
 import sigma.crypto.EcPointType
-import sigma.VersionContext
 import sigma.data.ExactIntegral.{ByteIsExactIntegral, IntIsExactIntegral, LongIsExactIntegral, ShortIsExactIntegral}
 import sigma.data.ExactOrdering.{ByteIsExactOrdering, IntIsExactOrdering, LongIsExactOrdering, ShortIsExactOrdering}
 import sigma.data.{CSigmaDslBuilder, ExactIntegral, ExactNumeric, ExactOrdering, Lazy, Nullable}
-import sigma.util.Extensions.ByteOps
-import sigmastate.interpreter.Interpreter.ScriptEnv
-import sigma.ast.{Ident, Select, Val}
 import sigma.data.UnsignedBigIntNumericOps.{UnsignedBigIntIsExactIntegral, UnsignedBigIntIsExactOrdering}
 import sigma.exceptions.GraphBuildingException
 import sigma.serialization.OpCodes
-import sigma.{SigmaException, ast}
-import sigma.VersionContext
+import sigma.util.Extensions.ByteOps
+import sigmastate.interpreter.Interpreter.ScriptEnv
 
 import scala.collection.mutable.ArrayBuffer
 import scala.language.{existentials,implicitConversions}
@@ -265,6 +261,7 @@ trait GraphBuilding extends Base with DefRewriting { IR: IRContext =>
     case SLong => LongElement
     case SString => StringElement
     case SAny => AnyElement
+    case SUnit => UnitElement
     case SBigInt => bigIntElement
     case SUnsignedBigInt => unsignedBigIntElement
     case SBox => boxElement
@@ -292,6 +289,7 @@ trait GraphBuilding extends Base with DefRewriting { IR: IRContext =>
     case LongElement => SLong
     case StringElement => SString
     case AnyElement => SAny
+    case UnitElement => SUnit
     case _: BigIntElem[_] => SBigInt
     case _: UnsignedBigIntElem[_] => SUnsignedBigInt
     case _: GroupElementElem[_] => SGroupElement
@@ -513,10 +511,10 @@ trait GraphBuilding extends Base with DefRewriting { IR: IRContext =>
       case Ident(n, _) =>
         env.getOrElse(n, !!!(s"Variable $n not found in environment $env"))
 
-      case ast.Upcast(Constant(value, _), toTpe: SNumericType) =>
+      case sigma.ast.Upcast(Constant(value, _), toTpe: SNumericType) =>
         eval(mkConstant(toTpe.upcast(value.asInstanceOf[AnyVal]), toTpe))
 
-      case ast.Downcast(Constant(value, _), toTpe: SNumericType) =>
+      case sigma.ast.Downcast(Constant(value, _), toTpe: SNumericType) =>
         eval(mkConstant(toTpe.downcast(value.asInstanceOf[AnyVal]), toTpe))
 
       // Rule: col.size --> SizeOf(col)
@@ -743,7 +741,7 @@ trait GraphBuilding extends Base with DefRewriting { IR: IRContext =>
         val res = sigmaDslBuilder.sha256(inputV)
         res
 
-      case ast.SizeOf(In(xs)) =>
+      case SizeOf(In(xs)) =>
         xs.elem.asInstanceOf[Any] match {
           case _: CollElem[a,_] =>
             val xsV = asRep[Coll[a]](xs)
@@ -937,11 +935,11 @@ trait GraphBuilding extends Base with DefRewriting { IR: IRContext =>
         val values = colBuilder.fromItems(vs: _*)(eAny)
         values
 
-      case ast.Upcast(In(input), tpe) =>
+      case sigma.ast.Upcast(In(input), tpe) =>
         val elem = stypeToElem(tpe.asNumType)
         upcast(input)(elem)
 
-      case ast.Downcast(In(input), tpe) =>
+      case sigma.ast.Downcast(In(input), tpe) =>
         val elem = stypeToElem(tpe.asNumType)
         downcast(input)(elem)
 

sc/shared/src/test/scala/sigmastate/utxo/BasicOpsSpecification.scala

@@ -3422,20 +3422,16 @@ class BasicOpsSpecification extends CompilerTestingCommons
   }
 
   property("Unit register") {
-    // TODO frontend: implement missing Unit support in compiler
-    //  https://github.com/ScorexFoundation/sigmastate-interpreter/issues/820
     test("R1", env, ext,
-      script = "", /* means cannot be compiled
-                     the corresponding script is { SELF.R4[Unit].isDefined } */
+      script = "{ SELF.R4[Unit].isDefined }",
       ExtractRegisterAs[SUnit.type](Self, reg1)(SUnit).isDefined.toSigmaProp,
       additionalRegistersOpt = Some(Map(
         reg1 -> UnitConstant.instance
       ))
     )
 
     test("R2", env, ext,
-      script = "", /* means cannot be compiled
-                   the corresponding script is "{ SELF.R4[Unit].get == () }" */
+      script = "{ SELF.R4[Unit].get == () }",
       EQ(ExtractRegisterAs[SUnit.type](Self, reg1)(SUnit).get, UnitConstant.instance).toSigmaProp,
       additionalRegistersOpt = Some(Map(
         reg1 -> UnitConstant.instance