stackinputs: constructor accepting iterator

0xPolygonMiden · Dec 1, 2023 · 8a300e6 · 8a300e6
1 parent 498943f
commit 8a300e6
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/core/src/stack/inputs.rs b/core/src/stack/inputs.rs
@@ -22,7 +22,12 @@ impl StackInputs {
     // --------------------------------------------------------------------------------------------
 
     /// Returns `[StackInputs]` from a list of values, reversing them into a stack.
-    pub fn new(mut values: Vec<Felt>) -> Result<Self, InputError> {
+    pub fn new<I>(values: I) -> Result<Self, InputError>
+    where
+        I: IntoIterator<Item = Felt>,
+    {
+        let mut values: Vec<Felt> = values.into_iter().collect();
+
         if values.len() > MAX_STACK_INPUTS_SIZE {
             Err(InputError::StackTooBig(values.len()))
         } else {
@@ -37,8 +42,7 @@ impl StackInputs {
     where
         I: IntoIterator<Item = u64>,
     {
-        let values: Vec<Felt> = iter.into_iter().map(Felt::from).collect();
-        Self::new(values)
+        Self::new(iter.into_iter().map(Felt::from))
     }
 
     // PUBLIC ACCESSORS