enum from bounded

lambdaclass · Aug 28, 2024 · 6131e94 · 6131e94
1 parent 9a23176
commit 6131e94
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/src/vm/enum.rs b/src/vm/enum.rs
@@ -3,8 +3,12 @@ use crate::Value;
 use cairo_lang_sierra::{
     extensions::{
         core::{CoreLibfunc, CoreType, CoreTypeConcrete},
-        enm::{EnumConcreteLibfunc, EnumConcreteType, EnumInitConcreteLibfunc},
+        enm::{
+            EnumConcreteLibfunc, EnumConcreteType, EnumFromBoundedIntConcreteLibfunc,
+            EnumInitConcreteLibfunc,
+        },
         lib_func::SignatureOnlyConcreteLibfunc,
+        ConcreteLibfunc,
     },
     program_registry::ProgramRegistry,
 };
@@ -17,7 +21,7 @@ pub fn eval(
 ) -> EvalAction {
     match selector {
         EnumConcreteLibfunc::Init(info) => eval_init(registry, info, args),
-        EnumConcreteLibfunc::FromBoundedInt(_) => todo!(),
+        EnumConcreteLibfunc::FromBoundedInt(info) => eval_from_bounded_int(registry, info, args),
         EnumConcreteLibfunc::Match(info) => eval_match(registry, info, args),
         EnumConcreteLibfunc::SnapshotMatch(_) => todo!(),
     }
@@ -50,6 +54,24 @@ pub fn eval_init(
     )
 }
 
+pub fn eval_from_bounded_int(
+    _registry: &ProgramRegistry<CoreType, CoreLibfunc>,
+    info: &EnumFromBoundedIntConcreteLibfunc,
+    args: Vec<Value>,
+) -> EvalAction {
+    let [Value::BoundedInt { range: _, value }]: [Value; 1] = args.try_into().unwrap() else {
+        panic!()
+    };
+
+    let enm = Value::Enum {
+        self_ty: info.branch_signatures()[0].vars[0].ty.clone(),
+        index: value.try_into().unwrap(),
+        payload: Box::new(Value::Struct(vec![])),
+    };
+
+    EvalAction::NormalBranch(0, smallvec![enm])
+}
+
 pub fn eval_match(
     registry: &ProgramRegistry<CoreType, CoreLibfunc>,
     info: &SignatureOnlyConcreteLibfunc,