From 1c0f525e57f1d1faeacbd0163c4f5d7f3cf0b976 Mon Sep 17 00:00:00 2001
From: Joe Fioti <jafioti@gmail.com>
Date: Tue, 16 Jan 2024 09:03:41 -0600
Subject: [PATCH] Added looped compiler

---
 src/core/compiler_utils.rs | 30 +++++++++++++++++++++++++++---
 1 file changed, 27 insertions(+), 3 deletions(-)

diff --git a/src/core/compiler_utils.rs b/src/core/compiler_utils.rs
index a0f1c8ec..6ebb3831 100644
--- a/src/core/compiler_utils.rs
+++ b/src/core/compiler_utils.rs
@@ -140,10 +140,34 @@ impl Compiler for () {
     fn compile<T: ToIdsMut>(&self, _: &mut Graph, _: T) {}
 }
 
+/// Wrap this around a compiler to rerun the compiler until it doesn't change the graph anymore
+pub struct Looped<C: Compiler + Debug>(C);
+
+impl<C: Compiler + Debug> Compiler for Looped<C> {
+    fn compile<T: ToIdsMut>(&self, graph: &mut Graph, mut remap: T) {
+        graph.toposort();
+        let mut linearized = graph.linearized_graph.clone();
+        loop {
+            self.0.compile(graph, &mut remap);
+            graph.toposort();
+            if linearized == graph.linearized_graph {
+                break;
+            }
+            linearized = graph.linearized_graph.clone();
+        }
+    }
+}
+
+impl<C: Default + Compiler + Debug> Default for Looped<C> {
+    fn default() -> Self {
+        Self(C::default())
+    }
+}
+
 /// Wrap this around a compiler to measure the time it takes to compile
-pub struct TimedCompiler<C: Compiler + Debug>(C);
+pub struct Timed<C: Compiler + Debug>(C);
 
-impl<C: Compiler + Debug> Compiler for TimedCompiler<C> {
+impl<C: Compiler + Debug> Compiler for Timed<C> {
     fn compile<T: ToIdsMut>(&self, graph: &mut Graph, remap: T) {
         let compiler_name = format!("{:?}", self.0).bold();
         println!("Starting {compiler_name}");
@@ -167,7 +191,7 @@ impl<C: Compiler + Debug> Compiler for TimedCompiler<C> {
     }
 }
 
-impl<C: Default + Compiler + Debug> Default for TimedCompiler<C> {
+impl<C: Default + Compiler + Debug> Default for Timed<C> {
     fn default() -> Self {
         Self(C::default())
     }