add explicit check for index type (currently u16) capacity

jcmgray · Apr 19, 2024 · 2a60432 · 2a60432
1 parent f3a3e50
commit 2a60432
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -152,6 +152,10 @@ impl ContractionProcessor {
         size_dict: Dict<char, f32>,
         track_flops: bool,
     ) -> ContractionProcessor {
+        if size_dict.len() > Ix::MAX as usize {
+            panic!("cotengrust: too many indices, maximum is {}", Ix::MAX);
+        }
+
         let mut nodes: Dict<Node, Legs> = Dict::default();
         let mut edges: Dict<Ix, BTreeSet<Node>> = Dict::default();
         let mut indmap: Dict<char, Ix> = Dict::default();