author | sundar |
Thu, 12 Nov 2015 12:30:15 +0530 | |
changeset 33693 | cc0617a78522 |
parent 26067 | b32ccc3a76c9 |
permissions | -rw-r--r-- |
24283 | 1 |
#// Usage: jjs -fx javaastviewer.js -- <.java files> |
2 |
||
3 |
/* |
|
4 |
* Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved. |
|
26067
b32ccc3a76c9
8055199: Tidy up Nashorn codebase for code standards (August 2014)
attila
parents:
24283
diff
changeset
|
5 |
* |
24283 | 6 |
* Redistribution and use in source and binary forms, with or without |
7 |
* modification, are permitted provided that the following conditions |
|
8 |
* are met: |
|
26067
b32ccc3a76c9
8055199: Tidy up Nashorn codebase for code standards (August 2014)
attila
parents:
24283
diff
changeset
|
9 |
* |
24283 | 10 |
* - Redistributions of source code must retain the above copyright |
11 |
* notice, this list of conditions and the following disclaimer. |
|
26067
b32ccc3a76c9
8055199: Tidy up Nashorn codebase for code standards (August 2014)
attila
parents:
24283
diff
changeset
|
12 |
* |
24283 | 13 |
* - Redistributions in binary form must reproduce the above copyright |
14 |
* notice, this list of conditions and the following disclaimer in the |
|
15 |
* documentation and/or other materials provided with the distribution. |
|
26067
b32ccc3a76c9
8055199: Tidy up Nashorn codebase for code standards (August 2014)
attila
parents:
24283
diff
changeset
|
16 |
* |
24283 | 17 |
* - Neither the name of Oracle nor the names of its |
18 |
* contributors may be used to endorse or promote products derived |
|
19 |
* from this software without specific prior written permission. |
|
26067
b32ccc3a76c9
8055199: Tidy up Nashorn codebase for code standards (August 2014)
attila
parents:
24283
diff
changeset
|
20 |
* |
24283 | 21 |
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS |
22 |
* IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, |
|
23 |
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
|
24 |
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR |
|
25 |
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
|
26 |
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
|
27 |
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
|
28 |
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
|
29 |
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING |
|
30 |
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
|
31 |
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
32 |
*/ |
|
33 |
||
34 |
// This example demonstrates Java subclassing by Java.extend |
|
35 |
// and javac Compiler and Tree API. This example also uses |
|
36 |
// -fx and javafx TreeView to visualize Java AST as TreeView |
|
37 |
||
38 |
if (!$OPTIONS._fx || arguments.length == 0) { |
|
39 |
print("Usage: jjs -fx javaastviewer.js -- <.java files>"); |
|
40 |
exit(1); |
|
41 |
} |
|
42 |
||
43 |
// Java types used |
|
44 |
var Enum = Java.type("java.lang.Enum"); |
|
45 |
var HashSet = Java.type("java.util.HashSet"); |
|
46 |
var Name = Java.type("javax.lang.model.element.Name"); |
|
47 |
var List = Java.type("java.util.List"); |
|
48 |
var Set = Java.type("java.util.Set"); |
|
49 |
var SimpleTreeVisitor = Java.type("com.sun.source.util.SimpleTreeVisitor"); |
|
50 |
var StringArray = Java.type("java.lang.String[]"); |
|
51 |
var ToolProvider = Java.type("javax.tools.ToolProvider"); |
|
52 |
var Tree = Java.type("com.sun.source.tree.Tree"); |
|
53 |
||
54 |
function javaASTToScriptObject(args) { |
|
55 |
// properties ignored (javac implementation class properties) in AST view. |
|
56 |
// may not be exhaustive - any getAbc would become "abc" property or |
|
57 |
// public field becomes a property of same name. |
|
58 |
var ignoredProps = new HashSet(); |
|
26067
b32ccc3a76c9
8055199: Tidy up Nashorn codebase for code standards (August 2014)
attila
parents:
24283
diff
changeset
|
59 |
for each (var word in |
b32ccc3a76c9
8055199: Tidy up Nashorn codebase for code standards (August 2014)
attila
parents:
24283
diff
changeset
|
60 |
['extending', 'implementing', 'init', 'mods', 'clazz', 'defs', |
24283 | 61 |
'expr', 'tag', 'preferredPosition', 'qualid', 'recvparam', |
62 |
'restype', 'params', 'startPosition', 'thrown', |
|
63 |
'tree', 'typarams', 'typetag', 'vartype']) { |
|
64 |
ignoredProps.add(word); |
|
65 |
} |
|
66 |
||
67 |
// get the system compiler tool |
|
68 |
var compiler = ToolProvider.systemJavaCompiler; |
|
69 |
||
70 |
// get standard file manager |
|
71 |
var fileMgr = compiler.getStandardFileManager(null, null, null); |
|
72 |
||
73 |
// make a list of compilation unit from command line argument file names |
|
74 |
// Using Java.to convert script array (arguments) to a Java String[] |
|
75 |
var compUnits = fileMgr.getJavaFileObjects(Java.to(args, StringArray)); |
|
76 |
||
77 |
// create a new compilation task |
|
78 |
var task = compiler.getTask(null, fileMgr, null, null, null, compUnits); |
|
79 |
||
80 |
// subclass SimpleTreeVisitor - converts Java AST node to |
|
81 |
// a simple script object by walking through it |
|
82 |
var ConverterVisitor = Java.extend(SimpleTreeVisitor); |
|
83 |
||
84 |
var visitor = new ConverterVisitor() { |
|
85 |
// convert java AST node to a friendly script object |
|
26067
b32ccc3a76c9
8055199: Tidy up Nashorn codebase for code standards (August 2014)
attila
parents:
24283
diff
changeset
|
86 |
// which can be viewed. Every node ends up in defaultAction |
24283 | 87 |
// method of SimpleTreeVisitor method. |
88 |
||
89 |
defaultAction: function (node, p) { |
|
90 |
var resultObj = {}; |
|
91 |
// Nashorn does not iterate properties and methods of Java objects |
|
92 |
// But, we can bind properties of any object (including java objects) |
|
93 |
// to a script object and iterate it! |
|
94 |
var obj = {}; |
|
95 |
Object.bindProperties(obj, node); |
|
96 |
||
97 |
// we don't want every property, method of java object |
|
98 |
for (var prop in obj) { |
|
99 |
var val = obj[prop]; |
|
100 |
var type = typeof val; |
|
101 |
// ignore 'method' members |
|
102 |
if (type == 'function' || type == 'undefined') { |
|
103 |
continue; |
|
104 |
} |
|
105 |
||
106 |
// ignore properties from Javac implementation |
|
107 |
// classes - hack by name!! |
|
108 |
if (ignoredProps.contains(prop)) { |
|
109 |
continue; |
|
110 |
} |
|
111 |
||
112 |
// subtree - recurse it |
|
113 |
if (val instanceof Tree) { |
|
114 |
resultObj[prop] = visitor.visit(val, p); |
|
115 |
} else if (val instanceof List) { |
|
116 |
// List of trees - recurse each and make an array |
|
117 |
var len = val.size(); |
|
118 |
if (len != 0) { |
|
119 |
var arr = []; |
|
120 |
for (var j = 0; j < len; j++) { |
|
121 |
var e = val[j]; |
|
122 |
if (e instanceof Tree) { |
|
123 |
arr.push(visitor.visit(e, p)); |
|
124 |
} |
|
125 |
} |
|
126 |
resultObj[prop] = arr; |
|
127 |
} |
|
128 |
} else if (val instanceof Set) { |
|
129 |
// Set - used for modifier flags |
|
130 |
// make array |
|
131 |
var len = val.size(); |
|
132 |
if (len != 0) { |
|
133 |
var arr = []; |
|
134 |
for each (var e in val) { |
|
135 |
if (e instanceof Enum || typeof e == 'string') { |
|
136 |
arr.push(e.toString()); |
|
137 |
} |
|
138 |
} |
|
139 |
resultObj[prop] = arr; |
|
140 |
} |
|
141 |
} else if (val instanceof Enum || val instanceof Name) { |
|
142 |
// make string for any Enum or Name |
|
143 |
resultObj[prop] = val.toString(); |
|
144 |
} else if (type != 'object') { |
|
145 |
// primitives 'as is' |
|
146 |
resultObj[prop] = val; |
|
147 |
} |
|
148 |
} |
|
149 |
return resultObj; |
|
150 |
} |
|
151 |
} |
|
152 |
||
153 |
// top level object with one property for each compilation unit |
|
154 |
var scriptObj = {}; |
|
155 |
for each (var cu in task.parse()) { |
|
156 |
scriptObj[cu.sourceFile.name] = cu.accept(visitor, null); |
|
157 |
} |
|
158 |
||
159 |
return scriptObj; |
|
160 |
} |
|
161 |
||
162 |
// JavaFX classes used |
|
163 |
var StackPane = Java.type("javafx.scene.layout.StackPane"); |
|
164 |
var Scene = Java.type("javafx.scene.Scene"); |
|
165 |
var TreeItem = Java.type("javafx.scene.control.TreeItem"); |
|
166 |
var TreeView = Java.type("javafx.scene.control.TreeView"); |
|
167 |
||
168 |
// Create a javafx TreeItem to view a script object |
|
169 |
function treeItemForObject(obj, name) { |
|
170 |
var item = new TreeItem(name); |
|
171 |
for (var prop in obj) { |
|
172 |
var node = obj[prop]; |
|
173 |
if (typeof node == 'object') { |
|
174 |
if (node == null) { |
|
175 |
// skip nulls |
|
176 |
continue; |
|
177 |
} |
|
178 |
var subitem = treeItemForObject(node, prop); |
|
179 |
} else { |
|
180 |
var subitem = new TreeItem(prop + ": " + node); |
|
181 |
} |
|
182 |
item.children.add(subitem); |
|
183 |
} |
|
184 |
||
185 |
item.expanded = true; |
|
186 |
return item; |
|
187 |
} |
|
188 |
||
189 |
var commandArgs = arguments; |
|
190 |
||
191 |
// JavaFX start method |
|
192 |
function start(stage) { |
|
193 |
var obj = javaASTToScriptObject(commandArgs); |
|
194 |
stage.title = "Java AST Viewer" |
|
195 |
var rootItem = treeItemForObject(obj, "AST"); |
|
196 |
rootItem.expanded = true; |
|
197 |
var tree = new TreeView(rootItem); |
|
198 |
var root = new StackPane(); |
|
199 |
root.children.add(tree); |
|
200 |
stage.scene = new Scene(root, 300, 450); |
|
201 |
stage.show(); |
|
202 |
} |