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();