From 31fc223517408884d2543b82b01796afb77b1c60 Mon Sep 17 00:00:00 2001
From: Folkert <folkert@folkertdev.nl>
Date: Sat, 23 Apr 2022 20:42:41 +0200
Subject: [PATCH] add recursivity checker

---
 compiler/can/src/reference_matrix.rs | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/compiler/can/src/reference_matrix.rs b/compiler/can/src/reference_matrix.rs
index eaadee556c..ab858d08c3 100644
--- a/compiler/can/src/reference_matrix.rs
+++ b/compiler/can/src/reference_matrix.rs
@@ -39,6 +39,31 @@ impl ReferenceMatrix {
     pub fn get(&self, index: usize) -> bool {
         self.bitvec[index]
     }
+
+    pub fn is_recursive(&self, index: usize) -> bool {
+        let mut scheduled = self.row_slice(index).to_bitvec();
+        let mut visited = self.row_slice(index).to_bitvec();
+
+        // yes this is a bit inefficient because rows are visited repeatedly.
+        while scheduled.any() {
+            for one in scheduled.iter_ones() {
+                if one == index {
+                    return true;
+                }
+
+                visited |= self.row_slice(one)
+            }
+
+            // i.e. visited did not change
+            if visited.count_ones() == scheduled.count_ones() {
+                break;
+            }
+
+            scheduled |= &visited;
+        }
+
+        false
+    }
 }
 
 // Topological sort and strongly-connected components