Send edges via context

leo848 · Apr 5, 2024 · 0958f3a · 0958f3a
1 parent 7732757
commit 0958f3a
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/path/create/ilp.rs b/src/path/create/ilp.rs
@@ -96,6 +96,21 @@ pub fn solve<C: CreateContext>(ctx: C) -> C::Path {
     let mut paths = print_paths_solution(&mut model, &x);
 
     while paths.len() > 1 {
+        ctx.send_edges(
+            paths
+                .iter()
+                .flat_map(|path| {
+                    path.iter().tuple_windows::<(_, _)>().chain(
+                        if path[0] == path[path.len() - 1] {
+                            vec![(&path[path.len() - 1], &path[0])]
+                        } else {
+                            vec![]
+                        },
+                    )
+                })
+                .map(|(&i, &j)| (i, j)),
+            None,
+        );
         add_cycle_constraints(&mut model, &paths, &x);
         paths = print_paths_solution(&mut model, &x);
     }