lang: small fix

Author: tcoratger

crates/leanVm/src/lang/const_expr.rs

@@ -1,6 +1,6 @@
 use std::fmt;
 
-use p3_field::PrimeCharacteristicRing;
+use p3_field::{PrimeCharacteristicRing, PrimeField};
 
 use crate::{
     constant::F,
@@ -64,6 +64,14 @@ impl ConstExpression {
             _ => None,
         })
     }
+
+    #[must_use]
+    pub fn try_naive_simplification(&self) -> Self {
+        self.naive_eval().map_or_else(
+            || self.clone(),
+            |value| Self::scalar(value.as_canonical_biguint().try_into().unwrap()),
+        )
+    }
 }
 
 impl From<usize> for ConstExpression {

crates/leanVm/src/lang/simple_expr.rs

@@ -35,6 +35,13 @@ impl SimpleExpr {
         matches!(self, Self::Constant(_))
     }
 
+    #[must_use] pub fn simplify_if_const(&self) -> Self {
+        if let Self::Constant(constant) = self {
+            return constant.try_naive_simplification().into();
+        }
+        self.clone()
+    }
+
     #[must_use]
     pub fn as_constant(&self) -> Option<ConstExpression> {
         match self {