Prune optrees after saving units

sorear · Jan 17, 2011 · b1c6b40 · b1c6b40
1 parent 0fd012e
commit b1c6b40
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/Metamodel.pm6 b/src/Metamodel.pm6
@@ -494,6 +494,11 @@ class StaticSub is RefTarget {
     method add_child($z) { push $.zyg, $z }
     method children() { @$.zyg }
 
+    method clear_optree() {
+        $.code = Any;
+        $.ltm = Any;
+    }
+
     method create_static_pad() {
         return Nil if $.spad_exists;
         $.spad_exists = True;
@@ -647,6 +652,10 @@ class Unit {
         }
     }
 
+    method clear_optrees() {
+        self.visit_local_subs_postorder({ $_.clear_optree })
+    }
+
     method visit_local_subs_postorder($cb) {
         sub rec {
             for $_.children { rec($_) }

diff --git a/src/NieczaCompiler.pm6 b/src/NieczaCompiler.pm6
@@ -25,7 +25,8 @@ method !compile($unitname, $filename, $modtime, $source, $main, $run, $end) {
             :$filename, :$modtime, :$source); },
         (map -> $st { $st.typename => { $ast = $st.invoke($ast) } }, @$.stages),
         ($.backend.typename ~ "-save") =>
-            { $.backend.save_unit($unitname, $ast); $ast = Any },
+            { $.backend.save_unit($unitname, $ast);
+              $ast.clear_optrees; $ast = Any },
         ($.backend.typename ~ "-post") =>
             { $.backend.post_save($unitname, :$main); },
         ($.backend.typename ~ "-run") =>