author | jlahoda |
Tue, 02 Apr 2019 16:51:25 +0200 | |
branch | jlahoda-tree-builder |
changeset 57300 | c79e191854e4 |
parent 57298 | 72d5f7163f12 |
permissions | -rw-r--r-- |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
1 |
/* |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
2 |
* Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved. |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
3 |
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
4 |
* |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
5 |
* This code is free software; you can redistribute it and/or modify it |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
6 |
* under the terms of the GNU General Public License version 2 only, as |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
7 |
* published by the Free Software Foundation. Oracle designates this |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
8 |
* particular file as subject to the "Classpath" exception as provided |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
9 |
* by Oracle in the LICENSE file that accompanied this code. |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
10 |
* |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
11 |
* This code is distributed in the hope that it will be useful, but WITHOUT |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
12 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
13 |
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
14 |
* version 2 for more details (a copy is included in the LICENSE file that |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
15 |
* accompanied this code). |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
16 |
* |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
17 |
* You should have received a copy of the GNU General Public License version |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
18 |
* 2 along with this work; if not, write to the Free Software Foundation, |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
19 |
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
20 |
* |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
21 |
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
22 |
* or visit www.oracle.com if you need additional information or have any |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
23 |
* questions. |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
24 |
*/package com.sun.tools.javac.api; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
25 |
|
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
26 |
import java.nio.CharBuffer; |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
27 |
import java.util.function.Consumer; |
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
28 |
import java.util.function.Function; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
29 |
|
57300 | 30 |
import javax.lang.model.element.Modifier; |
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
31 |
import javax.tools.JavaFileObject; |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
32 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
33 |
import com.sun.source.doctree.DocTree; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
34 |
import com.sun.source.tree.CompilationUnitTree; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
35 |
import com.sun.source.util.TreeBuilder; |
57300 | 36 |
import com.sun.tools.javac.code.Flags; |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
37 |
import com.sun.tools.javac.code.TypeTag; |
57297 | 38 |
import com.sun.tools.javac.tree.JCTree; |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
39 |
import com.sun.tools.javac.tree.JCTree.JCClassDecl; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
40 |
import com.sun.tools.javac.tree.JCTree.JCCompilationUnit; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
41 |
import com.sun.tools.javac.tree.JCTree.JCExpression; |
57297 | 42 |
import com.sun.tools.javac.tree.JCTree.JCMethodDecl; |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
43 |
import com.sun.tools.javac.tree.JCTree.JCVariableDecl; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
44 |
import com.sun.tools.javac.tree.JCTree.Tag; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
45 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
46 |
import com.sun.tools.javac.tree.TreeMaker; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
47 |
import com.sun.tools.javac.util.List; |
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
48 |
import com.sun.tools.javac.main.JavaCompiler; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
49 |
import com.sun.tools.javac.parser.Parser; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
50 |
import com.sun.tools.javac.parser.ParserFactory; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
51 |
import com.sun.tools.javac.tree.JCTree.JCBlock; |
57300 | 52 |
import com.sun.tools.javac.tree.JCTree.JCModifiers; |
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
53 |
import com.sun.tools.javac.tree.JCTree.JCStatement; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
54 |
import com.sun.tools.javac.util.JCDiagnostic; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
55 |
import com.sun.tools.javac.util.Log.DiagnosticHandler; |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
56 |
import com.sun.tools.javac.util.Names; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
57 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
58 |
/** |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
59 |
* Implementation for TreeBuilder. |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
60 |
* Note: this is only a crude experiment. |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
61 |
*/ |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
62 |
public class TreeBuilderImpl implements TreeBuilder { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
63 |
|
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
64 |
private final JavaCompiler compiler; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
65 |
private final ParserFactory parserFactory; |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
66 |
private final TreeMaker make; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
67 |
private final Names names; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
68 |
|
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
69 |
public TreeBuilderImpl(JavaCompiler compiler, ParserFactory parserFactory, TreeMaker make, Names names) { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
70 |
this.compiler = compiler; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
71 |
this.parserFactory = parserFactory; |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
72 |
this.make = make; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
73 |
this.names = names; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
74 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
75 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
76 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
77 |
public CompilationUnitTree createCompilationUnitTree(Consumer<CompilationUnit> unit) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
78 |
CompilationUnitImpl cui = new CompilationUnitImpl(); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
79 |
unit.accept(cui); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
80 |
return cui.result; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
81 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
82 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
83 |
private final class CompilationUnitImpl implements CompilationUnit { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
84 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
85 |
private final JCCompilationUnit result; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
86 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
87 |
public CompilationUnitImpl() { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
88 |
this.result = make.TopLevel(List.nil()); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
89 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
90 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
91 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
92 |
public CompilationUnit _package(String... qname) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
93 |
JCExpression qualIdent = make.Ident(names.fromString(qname[0])); //XXX: should check qname.length > 0! |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
94 |
for (int i = 1; i < qname.length; i++) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
95 |
qualIdent = make.Select(qualIdent, names.fromString(qname[i])); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
96 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
97 |
result.defs = result.defs.stream().filter(t -> !t.hasTag(Tag.PACKAGEDEF)).collect(List.collector()) //XXX: what should be the behavior if already filled? |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
98 |
.prepend(make.PackageDecl(List.nil(), qualIdent)); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
99 |
return this; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
100 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
101 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
102 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
103 |
public CompilationUnit _class(String name, Consumer<Class> clazz) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
104 |
ClassImpl ci = new ClassImpl(name); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
105 |
clazz.accept(ci); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
106 |
result.defs = result.defs.append(ci.result); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
107 |
return this; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
108 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
109 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
110 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
111 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
112 |
private final class ClassImpl implements Class { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
113 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
114 |
private final JCClassDecl result; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
115 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
116 |
public ClassImpl(String name) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
117 |
this.result = make.ClassDef(make.Modifiers(0), names.fromString(name), List.nil(), null, List.nil(), List.nil()); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
118 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
119 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
120 |
@Override |
57296 | 121 |
public Class superclass(Consumer<Type> sup) { |
122 |
result.extending = visitType(sup); //TODO: check extending not filled! |
|
123 |
return this; |
|
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
124 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
125 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
126 |
@Override |
57296 | 127 |
public Class superinterface(Consumer<Type> sup) { |
128 |
result.implementing = result.implementing.append(visitType(sup)); |
|
129 |
return this; |
|
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
130 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
131 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
132 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
133 |
public Class field(String name, Consumer<Type> type, Consumer<Variable> field) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
134 |
TypeImpl ti = new TypeImpl(); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
135 |
type.accept(ti); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
136 |
if (ti.type == null) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
137 |
throw new IllegalStateException("Type not provided!"); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
138 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
139 |
VariableImpl vi = new VariableImpl(ti.type, name); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
140 |
field.accept(vi); |
57295
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
141 |
result.defs = result.defs.append(vi.result); |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
142 |
return this; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
143 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
144 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
145 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
146 |
public Class method(String name, Consumer<Type> restype, Consumer<Method> method) { |
57297 | 147 |
TypeImpl ti = new TypeImpl(); |
148 |
restype.accept(ti); |
|
149 |
if (ti.type == null) { |
|
150 |
throw new IllegalStateException("Type not provided!"); |
|
151 |
} |
|
152 |
MethodImpl vi = new MethodImpl(ti.type, name); |
|
153 |
method.accept(vi); |
|
154 |
result.defs = result.defs.append(vi.result); |
|
155 |
return this; |
|
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
156 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
157 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
158 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
159 |
public Class _class(String name, Consumer<Class> clazz) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
160 |
throw new UnsupportedOperationException("Not supported yet."); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
161 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
162 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
163 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
164 |
public Class modifiers(Consumer<Modifiers> modifiers) { |
57300 | 165 |
modifiers.accept(new ModifiersImpl(result.mods)); |
166 |
return this; |
|
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
167 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
168 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
169 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
170 |
public Class javadoc(DocTree doc) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
171 |
throw new UnsupportedOperationException("Not supported yet."); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
172 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
173 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
174 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
175 |
public Class javadoc(String doc) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
176 |
throw new UnsupportedOperationException("Not supported yet."); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
177 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
178 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
179 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
180 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
181 |
private final class TypeImpl implements Type { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
182 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
183 |
private JCExpression type; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
184 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
185 |
@Override |
57296 | 186 |
public void _class(Consumer<QualifiedName> className, Consumer<TypeArguments> typeArguments) { |
187 |
JCExpression clazz = visitQualifiedName(className); |
|
188 |
TypeArgumentsImpl ta = new TypeArgumentsImpl(); |
|
189 |
typeArguments.accept(ta); |
|
190 |
if (ta.types.isEmpty()) { |
|
191 |
type = clazz; |
|
192 |
} else { |
|
193 |
type = make.TypeApply(clazz, ta.types); |
|
194 |
} |
|
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
195 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
196 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
197 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
198 |
public void _int() { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
199 |
//XXX: check empty! |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
200 |
type = make.TypeIdent(TypeTag.INT); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
201 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
202 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
203 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
204 |
public void _float() { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
205 |
throw new UnsupportedOperationException("Not supported yet."); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
206 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
207 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
208 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
209 |
public void _void() { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
210 |
throw new UnsupportedOperationException("Not supported yet."); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
211 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
212 |
|
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
213 |
@Override |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
214 |
public void type(String typeSpec) { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
215 |
type = parse(typeSpec, Parser::parseType); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
216 |
} |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
217 |
|
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
218 |
} |
57296 | 219 |
|
220 |
private final class TypeArgumentsImpl implements TypeArguments { |
|
221 |
private List<JCExpression> types = List.nil(); |
|
222 |
@Override |
|
223 |
public TypeArguments type(Consumer<Type> t) { |
|
224 |
TypeImpl type = new TypeImpl(); |
|
225 |
t.accept(type); |
|
226 |
types = types.append(type.type); |
|
227 |
return this; |
|
228 |
} |
|
229 |
} |
|
230 |
||
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
231 |
private final class VariableImpl implements Variable { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
232 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
233 |
private final JCVariableDecl result; |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
234 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
235 |
public VariableImpl(JCExpression type, String name) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
236 |
result = make.VarDef(make.Modifiers(0), names.fromString(name), type, null); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
237 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
238 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
239 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
240 |
public Variable init(Consumer<Expression> init) { |
57296 | 241 |
result.init = visitExpression(init); |
57295
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
242 |
return this; |
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
243 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
244 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
245 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
246 |
public Variable modifiers(Consumer<Modifiers> modifiers) { |
57300 | 247 |
modifiers.accept(new ModifiersImpl(result.mods)); |
248 |
return this; |
|
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
249 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
250 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
251 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
252 |
public Variable javadoc(DocTree doc) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
253 |
throw new UnsupportedOperationException("Not supported yet."); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
254 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
255 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
256 |
@Override |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
257 |
public Variable javadoc(String doc) { |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
258 |
throw new UnsupportedOperationException("Not supported yet."); |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
259 |
} |
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
260 |
|
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
261 |
} |
57295
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
262 |
|
57297 | 263 |
private final class MethodImpl implements Method { |
264 |
||
265 |
private final JCMethodDecl result; |
|
266 |
||
267 |
public MethodImpl(JCExpression restype, String name) { |
|
268 |
result = make.MethodDef(make.Modifiers(0), names.fromString(name), restype, List.nil(), List.nil(), List.nil(), null, null); |
|
269 |
} |
|
270 |
||
271 |
@Override |
|
272 |
public Method parameter(Consumer<Type> type, Consumer<Parameter> parameter) { |
|
273 |
ParameterImpl paramImpl = new ParameterImpl(visitType(type)); |
|
274 |
parameter.accept(paramImpl); |
|
275 |
result.params = result.params.append(paramImpl.result); |
|
276 |
return this; |
|
277 |
} |
|
278 |
||
279 |
@Override |
|
280 |
public Method body(Consumer<Block> statements) { |
|
281 |
BlockImpl block = new BlockImpl(); |
|
282 |
statements.accept(block); |
|
283 |
result.body = make.Block(0, block.statements); |
|
284 |
return this; |
|
285 |
} |
|
286 |
||
287 |
@Override |
|
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
288 |
public Method body(String body) { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
289 |
JCStatement parsedBody = parse(body, Parser::parseStatement); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
290 |
if (!parsedBody.hasTag(Tag.BLOCK)) { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
291 |
throw new IllegalArgumentException("Block not provided!"); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
292 |
} |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
293 |
result.body = (JCBlock) parsedBody; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
294 |
return this; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
295 |
} |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
296 |
|
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
297 |
@Override |
57297 | 298 |
public Method modifiers(Consumer<Modifiers> modifiers) { |
57300 | 299 |
modifiers.accept(new ModifiersImpl(result.mods)); |
300 |
return this; |
|
57297 | 301 |
} |
302 |
||
303 |
@Override |
|
304 |
public Method javadoc(DocTree doc) { |
|
305 |
throw new UnsupportedOperationException("Not supported yet."); |
|
306 |
} |
|
307 |
||
308 |
@Override |
|
309 |
public Method javadoc(String doc) { |
|
310 |
throw new UnsupportedOperationException("Not supported yet."); |
|
311 |
} |
|
312 |
||
313 |
} |
|
314 |
||
315 |
private final class ParameterImpl implements Parameter { |
|
316 |
||
317 |
private final JCVariableDecl result; |
|
318 |
||
319 |
public ParameterImpl(JCExpression type) { |
|
320 |
//TODO: infer name |
|
321 |
result = make.VarDef(make.Modifiers(0), null, type, null); |
|
322 |
} |
|
323 |
||
324 |
@Override |
|
325 |
public Parameter modifiers(Consumer<Modifiers> modifiers) { |
|
326 |
throw new UnsupportedOperationException("Not supported yet."); |
|
327 |
} |
|
328 |
||
329 |
@Override |
|
330 |
public Parameter name(String name) { |
|
331 |
result.name = names.fromString(name); //XXX: check not set yet. |
|
332 |
return this; |
|
333 |
} |
|
334 |
||
335 |
} |
|
336 |
||
337 |
private final class BlockImpl extends StatementBaseImpl<Block> implements Block { |
|
338 |
||
339 |
private List<JCStatement> statements = List.nil(); |
|
340 |
||
341 |
@Override |
|
342 |
protected Block addStatement(JCStatement stat) { |
|
343 |
statements = statements.append(stat); |
|
344 |
return this; |
|
345 |
} |
|
346 |
||
347 |
} |
|
348 |
||
349 |
private final class StatementImpl extends StatementBaseImpl<Void> implements Statement { |
|
350 |
private JCStatement result; |
|
351 |
||
352 |
@Override |
|
353 |
protected Void addStatement(JCStatement stat) { |
|
354 |
if (result != null) { |
|
355 |
throw new IllegalStateException(); |
|
356 |
} |
|
357 |
result = stat; |
|
358 |
return null; |
|
359 |
} |
|
360 |
} |
|
361 |
||
362 |
private abstract class StatementBaseImpl<S> implements StatementBase<S> { |
|
363 |
||
364 |
@Override |
|
365 |
public S _if(Consumer<Expression> cond, Consumer<Statement> ifPart) { |
|
366 |
JCExpression expr = visitExpression(cond); |
|
367 |
//TODO: should this automatic wrapping with parenthesized be here? |
|
368 |
expr = make.Parens(expr); |
|
369 |
StatementImpl ifStatement = new StatementImpl(); |
|
370 |
ifPart.accept(ifStatement); |
|
371 |
//TODO: check ifPart filled! |
|
372 |
return addStatement(make.If(expr, ifStatement.result, null)); |
|
373 |
} |
|
374 |
||
375 |
@Override |
|
376 |
public S _if(Consumer<Expression> cond, Consumer<Statement> ifPart, Consumer<Statement> elsePart) { |
|
377 |
JCExpression expr = visitExpression(cond); |
|
378 |
//TODO: should this automatic wrapping with parenthesized be here? |
|
379 |
expr = make.Parens(expr); |
|
380 |
StatementImpl ifStatement = new StatementImpl(); |
|
381 |
ifPart.accept(ifStatement); |
|
382 |
//TODO: check ifPart filled! |
|
383 |
StatementImpl elseStatement = new StatementImpl(); |
|
384 |
elsePart.accept(elseStatement); |
|
385 |
return addStatement(make.If(expr, ifStatement.result, elseStatement.result)); |
|
386 |
} |
|
387 |
||
388 |
@Override |
|
389 |
public S _return() { |
|
390 |
throw new UnsupportedOperationException("Not supported yet."); |
|
391 |
} |
|
392 |
||
393 |
@Override |
|
394 |
public S _return(Consumer<Expression> expr) { |
|
395 |
return addStatement(make.Return(visitExpression(expr))); |
|
396 |
} |
|
397 |
||
398 |
@Override |
|
399 |
public S expr(Consumer<Expression> expr) { |
|
400 |
throw new UnsupportedOperationException("Not supported yet."); |
|
401 |
} |
|
402 |
||
403 |
@Override |
|
404 |
public S skip() { |
|
405 |
throw new UnsupportedOperationException("Not supported yet."); |
|
406 |
} |
|
407 |
||
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
408 |
@Override |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
409 |
public S statement(String statement) { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
410 |
return addStatement(parse(statement, Parser::parseStatement)); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
411 |
} |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
412 |
|
57297 | 413 |
protected abstract S addStatement(JCStatement stat); |
414 |
} |
|
415 |
||
57295
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
416 |
private final class ExpressionImpl implements Expression { |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
417 |
|
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
418 |
private JCExpression expr; |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
419 |
|
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
420 |
@Override |
57297 | 421 |
public void equal_to(Consumer<Expression> lhs, Consumer<Expression> rhs) { |
422 |
expr = make.Binary(Tag.EQ, |
|
423 |
visitExpression(lhs), |
|
424 |
visitExpression(rhs)); |
|
425 |
} |
|
426 |
||
427 |
@Override |
|
57295
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
428 |
public void minusminus(Consumer<Expression> expr) { |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
429 |
throw new UnsupportedOperationException("Not supported yet."); |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
430 |
} |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
431 |
|
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
432 |
@Override |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
433 |
public void plus(Consumer<Expression> lhs, Consumer<Expression> rhs) { |
57296 | 434 |
expr = make.Binary(Tag.PLUS, |
435 |
visitExpression(lhs), |
|
436 |
visitExpression(rhs)); |
|
57295
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
437 |
} |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
438 |
|
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
439 |
@Override |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
440 |
public void cond(Consumer<Expression> cond, Consumer<Expression> truePart, Consumer<Expression> falsePart) { |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
441 |
throw new UnsupportedOperationException("Not supported yet."); |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
442 |
} |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
443 |
|
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
444 |
@Override |
57296 | 445 |
public void select(Consumer<Expression> selected, String name) { |
446 |
expr = make.Select(visitExpression(selected), names.fromString(name)); |
|
447 |
} |
|
448 |
||
449 |
@Override |
|
450 |
public void ident(String ident) { |
|
451 |
expr = make.Ident(names.fromString(ident)); //XXX |
|
57295
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
452 |
} |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
453 |
|
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
454 |
@Override |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
455 |
public void literal(Object value) { |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
456 |
expr = make.Literal(value); |
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
457 |
} |
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
458 |
|
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
459 |
@Override |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
460 |
public void expression(String expression) { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
461 |
expr = parse(expression, Parser::parseExpression); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
462 |
} |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
463 |
|
57295
5497ee9d40f4
Adding a prototype of CodeBuilder to build a sequence of TreeBuilder calls to construct a given AST.
jlahoda
parents:
57267
diff
changeset
|
464 |
} |
57296 | 465 |
|
466 |
private final class QualifiedNameImpl implements QualifiedName { |
|
467 |
||
468 |
private JCExpression expr; |
|
469 |
||
470 |
@Override |
|
471 |
public void select(Consumer<QualifiedName> selected, String name) { |
|
472 |
expr = make.Select(visitQualifiedName(selected), names.fromString(name)); |
|
473 |
} |
|
474 |
||
475 |
@Override |
|
476 |
public void ident(String ident) { |
|
477 |
expr = make.Ident(names.fromString(ident)); |
|
478 |
} |
|
479 |
||
480 |
@Override |
|
481 |
public void ident(String... qnames) { |
|
482 |
expr = make.Ident(names.fromString(qnames[0])); |
|
483 |
for (int i = 1; i < qnames.length; i++) { |
|
484 |
expr = make.Select(expr, names.fromString(qnames[i])); |
|
485 |
} |
|
486 |
} |
|
487 |
||
488 |
} |
|
489 |
||
57300 | 490 |
private final class ModifiersImpl implements Modifiers { |
491 |
||
492 |
private final JCModifiers mods; |
|
493 |
||
494 |
public ModifiersImpl(JCModifiers mods) { |
|
495 |
this.mods = mods; |
|
496 |
} |
|
497 |
||
498 |
@Override |
|
499 |
public Modifiers modifier(Modifier modifier) { |
|
500 |
long flag; |
|
501 |
switch (modifier) { |
|
502 |
case PUBLIC: flag = Flags.PUBLIC; break; |
|
503 |
case PROTECTED: flag = Flags.PROTECTED; break; |
|
504 |
case PRIVATE: flag = Flags.PRIVATE; break; |
|
505 |
case ABSTRACT: flag = Flags.ABSTRACT; break; |
|
506 |
case STATIC: flag = Flags.STATIC; break; |
|
507 |
case FINAL: flag = Flags.FINAL; break; |
|
508 |
case TRANSIENT: flag = Flags.TRANSIENT; break; |
|
509 |
case VOLATILE: flag = Flags.VOLATILE; break; |
|
510 |
case SYNCHRONIZED: flag = Flags.SYNCHRONIZED; break; |
|
511 |
case NATIVE: flag = Flags.NATIVE; break; |
|
512 |
case STRICTFP: flag = Flags.STRICTFP; break; |
|
513 |
case DEFAULT: flag = Flags.DEFAULT; break; |
|
514 |
default: |
|
515 |
throw new IllegalArgumentException("Unknown modifier: " + modifier); |
|
516 |
} |
|
517 |
mods.flags |= flag; |
|
518 |
return this; |
|
519 |
} |
|
520 |
||
521 |
@Override |
|
522 |
public Modifiers annotation(Consumer<Annotation> annotation) { |
|
523 |
throw new UnsupportedOperationException("Not supported yet."); |
|
524 |
} |
|
525 |
||
526 |
} |
|
527 |
||
57296 | 528 |
private JCExpression visitExpression(Consumer<Expression> c) { |
529 |
ExpressionImpl expr = new ExpressionImpl(); |
|
530 |
||
531 |
c.accept(expr); |
|
532 |
||
533 |
if (expr.expr == null) { |
|
534 |
throw new IllegalStateException("Expression not provided!"); |
|
535 |
} |
|
536 |
||
537 |
return expr.expr; |
|
538 |
} |
|
539 |
||
540 |
private JCExpression visitQualifiedName(Consumer<QualifiedName> c) { |
|
541 |
QualifiedNameImpl expr = new QualifiedNameImpl(); |
|
542 |
||
543 |
c.accept(expr); |
|
544 |
||
545 |
if (expr.expr == null) { |
|
546 |
throw new IllegalStateException("Name not provided!"); |
|
547 |
} |
|
548 |
||
549 |
return expr.expr; |
|
550 |
} |
|
551 |
||
552 |
private JCExpression visitType(Consumer<Type> c) { |
|
553 |
TypeImpl type = new TypeImpl(); |
|
554 |
||
555 |
c.accept(type); |
|
556 |
||
557 |
if (type.type == null) { |
|
558 |
throw new IllegalStateException("Expression not provided!"); |
|
559 |
} |
|
560 |
||
561 |
return type.type; |
|
562 |
} |
|
57298
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
563 |
|
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
564 |
private <T extends JCTree> T parse(String toParse, Function<Parser, T> runParse) { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
565 |
if (toParse == null || toParse.equals("")) |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
566 |
throw new IllegalArgumentException(); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
567 |
JavaFileObject prev = compiler.log.useSource(null); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
568 |
DiagnosticHandler h = null; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
569 |
try { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
570 |
h = new DiagnosticHandler() { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
571 |
{ |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
572 |
install(compiler.log); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
573 |
} |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
574 |
@Override |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
575 |
public void report(JCDiagnostic err) { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
576 |
if (err.getKind() == JCDiagnostic.Kind.ERROR) { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
577 |
throw new IllegalArgumentException("Cannot parse input: " + err.getMessage(null)); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
578 |
} |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
579 |
} |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
580 |
}; |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
581 |
CharBuffer buf = CharBuffer.wrap((toParse+"\u0000").toCharArray(), 0, toParse.length()); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
582 |
Parser parser = parserFactory.newParser(buf, false, false, false); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
583 |
return runParse.apply(parser); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
584 |
} finally { |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
585 |
compiler.log.popDiagnosticHandler(h); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
586 |
compiler.log.useSource(prev); |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
587 |
} |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
588 |
} |
72d5f7163f12
Adding ability to supply code as text instead of Trees on some places.
jlahoda
parents:
57297
diff
changeset
|
589 |
|
57267
97aaf02ed830
A first very crude experiment on a tree builder for javac.
jlahoda
parents:
diff
changeset
|
590 |
} |