1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47 package org.codehaus.groovy.classgen;
48
49 import groovy.lang.GroovyClassLoader;
50
51 import java.io.OutputStreamWriter;
52 import java.io.PrintWriter;
53 import java.io.StringWriter;
54
55 import org.codehaus.groovy.ast.ClassNode;
56 import org.codehaus.groovy.ast.CompileUnit;
57 import org.codehaus.groovy.control.CompilationUnit;
58 import org.codehaus.groovy.control.CompilerConfiguration;
59 import org.codehaus.groovy.control.SourceUnit;
60 import org.objectweb.asm.ClassVisitor;
61 import org.objectweb.asm.Opcodes;
62 import org.objectweb.asm.util.CheckClassAdapter;
63 import org.objectweb.asm.util.ASMifierClassVisitor;
64
65 /***
66 * A class loader used for debugging the bytecode generation.
67 * This will log all bytecode being loaded
68 *
69 * @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
70 * @version $Revision: 1.14 $
71 */
72 public class DumpingClassLoader extends GroovyClassLoader implements Opcodes {
73
74 protected static boolean CHECK_CLASS = false;
75 protected static boolean DUMP_CLASS = true;
76
77 public DumpingClassLoader(ClassLoader parentLoader) {
78 super(parentLoader);
79 }
80
81
82 protected class DebugCollector extends ClassCollector {
83
84 DebugCollector(GroovyClassLoader cl, CompilationUnit unit, SourceUnit su) {
85 super(new GroovyClassLoader.InnerLoader(cl), unit, su);
86 }
87
88 public void call(ClassVisitor classWriter, ClassNode classNode) {
89
90 if (DUMP_CLASS) {
91 dumper.visitClass(classNode);
92 }
93
94 if (CHECK_CLASS) {
95 checker.visitClass(classNode);
96 }
97
98 super.call(classWriter, classNode);
99 }
100 }
101
102 protected ClassCollector createCollector(CompilationUnit unit) {
103 return new DebugCollector(this,unit, null);
104 }
105
106 protected ASMifierClassVisitor dumpVisitor = new ASMifierClassVisitor(new PrintWriter(new OutputStreamWriter(System.out)));
107 protected ASMifierClassVisitor invisibleDumpVisitor = new ASMifierClassVisitor(new PrintWriter(new StringWriter()));
108 protected CompileUnit unit = new CompileUnit(this, new CompilerConfiguration());
109 protected ClassGenerator checker =
110 new AsmClassGenerator(new GeneratorContext(unit), new CheckClassAdapter(invisibleDumpVisitor), this, null);
111 protected ClassGenerator dumper = new AsmClassGenerator(new GeneratorContext(unit), dumpVisitor, this, null);
112
113 }