mypyc: save the ops as well

mypyc/emitmodule.py

@@ -15,7 +15,7 @@
 from mypyc.emitwrapper import (
     generate_wrapper_function, wrapper_function_header,
 )
-from mypyc.ops import FuncIR, ClassIR, ModuleIR
+from mypyc.ops import FuncIR, ClassIR, ModuleIR, format_func
 from mypyc.refcount import insert_ref_count_opcodes
 from mypyc.exceptions import insert_exception_handling
 from mypyc.emit import EmitterContext, Emitter, HeaderDeclaration
@@ -31,7 +31,8 @@ def __init__(self, declaration: HeaderDeclaration, mark: bool) -> None:
 
 def compile_modules_to_c(sources: List[BuildSource], module_names: List[str], options: Options,
                          use_shared_lib: bool,
-                         alt_lib_path: Optional[str] = None) -> str:
+                         alt_lib_path: Optional[str] = None,
+                         ops: Optional[List[str]] = None) -> str:
     """Compile Python module(s) to C that can be used from Python C extension modules."""
     assert options.strict_optional, 'strict_optional must be turned on'
     result = build(sources=sources,
@@ -51,6 +52,12 @@ def compile_modules_to_c(sources: List[BuildSource], module_names: List[str], op
     for _, module in modules:
         for fn in module.functions:
             insert_ref_count_opcodes(fn)
+    # Format ops for debugging
+    if ops is not None:
+        for _, module in modules:
+            for fn in module.functions:
+                ops.extend(format_func(fn))
+                ops.append('')
     # Generate C code.
     source_paths = {module_name: result.files[module_name].path
                     for module_name in module_names}

scripts/mypyc

@@ -97,24 +97,34 @@ def main() -> None:
     # around with making the single module code handle packages.)
     use_shared_lib = len(module_names) > 1 or any('.' in x for x in module_names)
 
+    # XXX: This is not the right way to pass in command line args
+    save_dir = os.getenv("MYPYC_DIR", None)
+
+    ops = [] if save_dir else None
+
     try:
         ctext = emitmodule.compile_modules_to_c(
             sources=sources,
             module_names=module_names,
             options=options,
-            use_shared_lib=use_shared_lib)
+            use_shared_lib=use_shared_lib,
+            ops=ops)
     except CompileError as e:
         for line in e.messages:
             print(line)
         sys.exit(1)
 
-    # XXX: This is not the right way to pass in command line args
-    save_dir = os.getenv("MYPYC_DIR", None)
+    print("C generated!")
+
     if save_dir:
         try:
             os.mkdir(save_dir)
         except FileExistsError:
             pass
+        if ops:
+            with open(os.path.join(save_dir, 'ops.txt'), 'w') as f:
+                f.write('\n'.join(ops))
+
         dir = os.path.abspath(save_dir)
     else:
         dir = tempfile.mkdtemp(prefix='mypyc-')