- Added printUnits and printGlobal

1 files changed, 15 insertions, 1 deletions

diff --git a/sources/scalac/atree/ATreePrinter.java b/sources/scalac/atree/ATreePrinter.java
index abbca028c9..6e83b4ae36 100644
--- a/sources/scalac/atree/ATreePrinter.java
+++ b/sources/scalac/atree/ATreePrinter.java
@@ -10,8 +10,9 @@ package scalac.atree;
 
 import ch.epfl.lamp.util.CodePrinter;
 
-import scalac.Unit;
 import scalac.Global;
+import scalac.Phase;
+import scalac.Unit;
 import scalac.symtab.Type;
 import scalac.symtab.Symbol;
 import scalac.symtab.SymbolTablePrinter;
@@ -237,6 +238,19 @@ public class ATreePrinter {
     //########################################################################
     // Public Methods - Printing trees
 
+    /** Prints all global units. */
+    public ATreePrinter printGlobal(Global global) {
+        Phase phase = global.currentPhase;
+        println("[[attributed trees at "+phase+" (after "+phase.prev+")]]");
+        return printUnits(global.units);
+    }
+
+    /** Prints the units. */
+    public ATreePrinter printUnits(Unit[] units) {
+        for (int i = 0; i < units.length; i++) printUnit(units[i]);
+        return this;
+    }
+
     /** Prints the unit. */
     public ATreePrinter printUnit(Unit unit) {
         println("// Scala source: " + unit.source);