Abstract interpretation based analysis rewrite

src/exo/API.py

@@ -19,6 +19,7 @@
 from .pattern_match import match_pattern
 from .prelude import *
 from .new_eff import Check_Aliasing
+from .dataflow import dataflow_analysis
 
 # Moved to new file
 from .proc_eqv import decl_new_proc, derive_proc, assert_eqv_proc, check_eqv_proc
@@ -391,3 +392,6 @@ def is_eq(self, proc: "Procedure"):
 
     def _root(self):
         return IC.Cursor.create(self._loopir_proc)
+
+    def dataflow(self):
+        return dataflow_analysis(self._loopir_proc)

src/exo/LoopIR_pprint.py

@@ -321,33 +321,33 @@ def ptype(self, t):
 # LoopIR Pretty Printing
 
 
-def _format_code(code):
+def format_code(code):
     return FormatCode(code)[0].rstrip("\n")
 
 
 @extclass(LoopIR.proc)
 def __str__(self):
-    return _format_code("\n".join(_print_proc(self, PrintEnv(), "")))
+    return format_code("\n".join(_print_proc(self, PrintEnv(), "")))
 
 
 @extclass(LoopIR.fnarg)
 def __str__(self):
-    return _format_code(_print_fnarg(self, PrintEnv()))
+    return format_code(_print_fnarg(self, PrintEnv()))
 
 
 @extclass(LoopIR.stmt)
 def __str__(self):
-    return _format_code("\n".join(_print_stmt(self, PrintEnv(), "")))
+    return format_code("\n".join(_print_stmt(self, PrintEnv(), "")))
 
 
 @extclass(LoopIR.expr)
 def __str__(self):
-    return _format_code(_print_expr(self, PrintEnv()))
+    return format_code(_print_expr(self, PrintEnv()))
 
 
 @extclass(LoopIR.type)
 def __str__(self):
-    return _format_code(_print_type(self, PrintEnv()))
+    return format_code(_print_type(self, PrintEnv()))
 
 
 del __str__
@@ -588,7 +588,7 @@ def _print_cursor(cur):
 
     root_cur = cur.root()
     lines = _print_cursor_proc(root_cur, cur, PrintEnv(), "")
-    code = _format_code("\n".join(lines))
+    code = format_code("\n".join(lines))
     # need to use "..." for Python parsing, but unquoted ellipses are prettier
     code = code.replace('"..."', "...")
     return code