Make --target=ast dump pre_deserialize and post_deserialize code for …

…CompUnits.
Raku · Oct 23, 2015 · 78a6682 · 78a6682
1 parent 88ef425
commit 78a6682
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/src/QAST/Children.nqp b/src/QAST/Children.nqp
@@ -17,8 +17,8 @@ role QAST::Children {
         @!children := @children;
     }
 
-    method dump_children(int $indent, @onto) {
-        for @!children {
+    method dump_node_list(int $indent, @onto, @node_list) {
+        for @node_list {
             if nqp::istype($_, QAST::Node) {
                 nqp::push(@onto, $_.dump($indent));
             }
@@ -30,4 +30,8 @@ role QAST::Children {
             }
         }
     }
+
+    method dump_children(int $indent, @onto) {
+        self.dump_node_list($indent, @onto, @!children);
+    }
 }
diff --git a/src/QAST/CompUnit.nqp b/src/QAST/CompUnit.nqp
@@ -58,4 +58,21 @@ class QAST::CompUnit is QAST::Node does QAST::Children {
     method code_ref_blocks($value = NO_VALUE) {
         $!code_ref_blocks := $value unless $value =:= NO_VALUE; $!code_ref_blocks
     }
+
+    method dump_children(int $indent, @onto) {
+        if self.pre_deserialize {
+            nqp::push(@onto, nqp::x(' ', $indent) ~ "[pre deserialize]\n");
+            self.dump_node_list($indent, @onto, self.pre_deserialize);
+        }
+
+        if self.post_deserialize {
+            nqp::push(@onto, nqp::x(' ', $indent) ~ "[post deserialize]\n");
+            self.dump_node_list($indent, @onto, self.post_deserialize);
+        }
+
+        if self.pre_deserialize || self.post_deserialize {
+            nqp::push(@onto, nqp::x(' ', $indent) ~ "[children]\n") 
+        }
+        self.dump_node_list($indent, @onto, self.list);
+    }
 }