Added from_parts_with_err method to Array

chandr-andr · chandr-andr · commit c32f38e06cca · 2024-08-24T21:57:20.000+02:00
Signed-off-by: chandr-andr (Kiselev Aleksandr) &lt;chandr@chandr.net&gt;
diff --git a/src/array.rs b/src/array.rs
@@ -1,4 +1,5 @@
 use std::fmt;
+use std::io::{Error, ErrorKind};
 use std::ops::{Index, IndexMut};
 use std::slice;
 use std::vec;
@@ -79,6 +80,22 @@ impl<T> Array<T> {
         }
     }
 
+    pub fn from_parts_with_err(
+        data: Vec<T>,
+        dimensions: Vec<Dimension>,
+    ) -> Result<Array<T>, Error> {
+        if (data.is_empty() && dimensions.is_empty())
+            || data.len() as i32 == dimensions.iter().fold(1, |acc, i| acc * i.len)
+        {
+            return Err(Error::new(ErrorKind::Other, "size mismatch"));
+        }
+
+        return Ok(Array {
+            dims: dimensions,
+            data,
+        });
+    }
+
     /// Creates a new one-dimensional array.
     pub fn from_vec(data: Vec<T>, lower_bound: i32) -> Array<T> {
         Array {
