src/share/classes/com/sun/tools/javac/tree/TreeInfo.java

Sat, 01 Dec 2007 00:00:00 +0000

author
duke
date
Sat, 01 Dec 2007 00:00:00 +0000
changeset 1
9a66ca7c79fa
child 104
5e89c4ca637c
permissions
-rw-r--r--

Initial load

duke@1 1 /*
duke@1 2 * Copyright 1999-2006 Sun Microsystems, Inc. All Rights Reserved.
duke@1 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
duke@1 4 *
duke@1 5 * This code is free software; you can redistribute it and/or modify it
duke@1 6 * under the terms of the GNU General Public License version 2 only, as
duke@1 7 * published by the Free Software Foundation. Sun designates this
duke@1 8 * particular file as subject to the "Classpath" exception as provided
duke@1 9 * by Sun in the LICENSE file that accompanied this code.
duke@1 10 *
duke@1 11 * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
duke@1 14 * version 2 for more details (a copy is included in the LICENSE file that
duke@1 15 * accompanied this code).
duke@1 16 *
duke@1 17 * You should have received a copy of the GNU General Public License version
duke@1 18 * 2 along with this work; if not, write to the Free Software Foundation,
duke@1 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1 20 *
duke@1 21 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
duke@1 22 * CA 95054 USA or visit www.sun.com if you need additional information or
duke@1 23 * have any questions.
duke@1 24 */
duke@1 25
duke@1 26 package com.sun.tools.javac.tree;
duke@1 27
duke@1 28 import com.sun.source.tree.Tree;
duke@1 29 import com.sun.tools.javac.comp.AttrContext;
duke@1 30 import com.sun.tools.javac.comp.Env;
duke@1 31 import java.util.Map;
duke@1 32 import com.sun.tools.javac.util.*;
duke@1 33 import com.sun.tools.javac.util.JCDiagnostic.DiagnosticPosition;
duke@1 34 import com.sun.tools.javac.code.*;
duke@1 35 import com.sun.tools.javac.tree.JCTree.*;
duke@1 36
duke@1 37 import static com.sun.tools.javac.code.Flags.*;
duke@1 38 import com.sun.tools.javac.util.JCDiagnostic.SimpleDiagnosticPosition;
duke@1 39
duke@1 40 /** Utility class containing inspector methods for trees.
duke@1 41 *
duke@1 42 * <p><b>This is NOT part of any API supported by Sun Microsystems. If
duke@1 43 * you write code that depends on this, you do so at your own risk.
duke@1 44 * This code and its internal interfaces are subject to change or
duke@1 45 * deletion without notice.</b>
duke@1 46 */
duke@1 47 public class TreeInfo {
duke@1 48 protected static final Context.Key<TreeInfo> treeInfoKey =
duke@1 49 new Context.Key<TreeInfo>();
duke@1 50
duke@1 51 public static TreeInfo instance(Context context) {
duke@1 52 TreeInfo instance = context.get(treeInfoKey);
duke@1 53 if (instance == null)
duke@1 54 instance = new TreeInfo(context);
duke@1 55 return instance;
duke@1 56 }
duke@1 57
duke@1 58 /** The names of all operators.
duke@1 59 */
duke@1 60 private Name[] opname = new Name[JCTree.MOD - JCTree.POS + 1];
duke@1 61
duke@1 62 private TreeInfo(Context context) {
duke@1 63 context.put(treeInfoKey, this);
duke@1 64
duke@1 65 Name.Table names = Name.Table.instance(context);
duke@1 66 opname[JCTree.POS - JCTree.POS] = names.fromString("+");
duke@1 67 opname[JCTree.NEG - JCTree.POS] = names.hyphen;
duke@1 68 opname[JCTree.NOT - JCTree.POS] = names.fromString("!");
duke@1 69 opname[JCTree.COMPL - JCTree.POS] = names.fromString("~");
duke@1 70 opname[JCTree.PREINC - JCTree.POS] = names.fromString("++");
duke@1 71 opname[JCTree.PREDEC - JCTree.POS] = names.fromString("--");
duke@1 72 opname[JCTree.POSTINC - JCTree.POS] = names.fromString("++");
duke@1 73 opname[JCTree.POSTDEC - JCTree.POS] = names.fromString("--");
duke@1 74 opname[JCTree.NULLCHK - JCTree.POS] = names.fromString("<*nullchk*>");
duke@1 75 opname[JCTree.OR - JCTree.POS] = names.fromString("||");
duke@1 76 opname[JCTree.AND - JCTree.POS] = names.fromString("&&");
duke@1 77 opname[JCTree.EQ - JCTree.POS] = names.fromString("==");
duke@1 78 opname[JCTree.NE - JCTree.POS] = names.fromString("!=");
duke@1 79 opname[JCTree.LT - JCTree.POS] = names.fromString("<");
duke@1 80 opname[JCTree.GT - JCTree.POS] = names.fromString(">");
duke@1 81 opname[JCTree.LE - JCTree.POS] = names.fromString("<=");
duke@1 82 opname[JCTree.GE - JCTree.POS] = names.fromString(">=");
duke@1 83 opname[JCTree.BITOR - JCTree.POS] = names.fromString("|");
duke@1 84 opname[JCTree.BITXOR - JCTree.POS] = names.fromString("^");
duke@1 85 opname[JCTree.BITAND - JCTree.POS] = names.fromString("&");
duke@1 86 opname[JCTree.SL - JCTree.POS] = names.fromString("<<");
duke@1 87 opname[JCTree.SR - JCTree.POS] = names.fromString(">>");
duke@1 88 opname[JCTree.USR - JCTree.POS] = names.fromString(">>>");
duke@1 89 opname[JCTree.PLUS - JCTree.POS] = names.fromString("+");
duke@1 90 opname[JCTree.MINUS - JCTree.POS] = names.hyphen;
duke@1 91 opname[JCTree.MUL - JCTree.POS] = names.asterisk;
duke@1 92 opname[JCTree.DIV - JCTree.POS] = names.slash;
duke@1 93 opname[JCTree.MOD - JCTree.POS] = names.fromString("%");
duke@1 94 }
duke@1 95
duke@1 96
duke@1 97 /** Return name of operator with given tree tag.
duke@1 98 */
duke@1 99 public Name operatorName(int tag) {
duke@1 100 return opname[tag - JCTree.POS];
duke@1 101 }
duke@1 102
duke@1 103 /** Is tree a constructor declaration?
duke@1 104 */
duke@1 105 public static boolean isConstructor(JCTree tree) {
duke@1 106 if (tree.getTag() == JCTree.METHODDEF) {
duke@1 107 Name name = ((JCMethodDecl) tree).name;
duke@1 108 return name == name.table.init;
duke@1 109 } else {
duke@1 110 return false;
duke@1 111 }
duke@1 112 }
duke@1 113
duke@1 114 /** Is there a constructor declaration in the given list of trees?
duke@1 115 */
duke@1 116 public static boolean hasConstructors(List<JCTree> trees) {
duke@1 117 for (List<JCTree> l = trees; l.nonEmpty(); l = l.tail)
duke@1 118 if (isConstructor(l.head)) return true;
duke@1 119 return false;
duke@1 120 }
duke@1 121
duke@1 122 /** Is statement an initializer for a synthetic field?
duke@1 123 */
duke@1 124 public static boolean isSyntheticInit(JCTree stat) {
duke@1 125 if (stat.getTag() == JCTree.EXEC) {
duke@1 126 JCExpressionStatement exec = (JCExpressionStatement)stat;
duke@1 127 if (exec.expr.getTag() == JCTree.ASSIGN) {
duke@1 128 JCAssign assign = (JCAssign)exec.expr;
duke@1 129 if (assign.lhs.getTag() == JCTree.SELECT) {
duke@1 130 JCFieldAccess select = (JCFieldAccess)assign.lhs;
duke@1 131 if (select.sym != null &&
duke@1 132 (select.sym.flags() & SYNTHETIC) != 0) {
duke@1 133 Name selected = name(select.selected);
duke@1 134 if (selected != null && selected == selected.table._this)
duke@1 135 return true;
duke@1 136 }
duke@1 137 }
duke@1 138 }
duke@1 139 }
duke@1 140 return false;
duke@1 141 }
duke@1 142
duke@1 143 /** If the expression is a method call, return the method name, null
duke@1 144 * otherwise. */
duke@1 145 public static Name calledMethodName(JCTree tree) {
duke@1 146 if (tree.getTag() == JCTree.EXEC) {
duke@1 147 JCExpressionStatement exec = (JCExpressionStatement)tree;
duke@1 148 if (exec.expr.getTag() == JCTree.APPLY) {
duke@1 149 Name mname = TreeInfo.name(((JCMethodInvocation) exec.expr).meth);
duke@1 150 return mname;
duke@1 151 }
duke@1 152 }
duke@1 153 return null;
duke@1 154 }
duke@1 155
duke@1 156 /** Is this a call to this or super?
duke@1 157 */
duke@1 158 public static boolean isSelfCall(JCTree tree) {
duke@1 159 Name name = calledMethodName(tree);
duke@1 160 if (name != null) {
duke@1 161 Name.Table names = name.table;
duke@1 162 return name==names._this || name==names._super;
duke@1 163 } else {
duke@1 164 return false;
duke@1 165 }
duke@1 166 }
duke@1 167
duke@1 168 /** Is this a call to super?
duke@1 169 */
duke@1 170 public static boolean isSuperCall(JCTree tree) {
duke@1 171 Name name = calledMethodName(tree);
duke@1 172 if (name != null) {
duke@1 173 Name.Table names = name.table;
duke@1 174 return name==names._super;
duke@1 175 } else {
duke@1 176 return false;
duke@1 177 }
duke@1 178 }
duke@1 179
duke@1 180 /** Is this a constructor whose first (non-synthetic) statement is not
duke@1 181 * of the form this(...)?
duke@1 182 */
duke@1 183 public static boolean isInitialConstructor(JCTree tree) {
duke@1 184 JCMethodInvocation app = firstConstructorCall(tree);
duke@1 185 if (app == null) return false;
duke@1 186 Name meth = name(app.meth);
duke@1 187 return meth == null || meth != meth.table._this;
duke@1 188 }
duke@1 189
duke@1 190 /** Return the first call in a constructor definition. */
duke@1 191 public static JCMethodInvocation firstConstructorCall(JCTree tree) {
duke@1 192 if (tree.getTag() != JCTree.METHODDEF) return null;
duke@1 193 JCMethodDecl md = (JCMethodDecl) tree;
duke@1 194 Name.Table names = md.name.table;
duke@1 195 if (md.name != names.init) return null;
duke@1 196 if (md.body == null) return null;
duke@1 197 List<JCStatement> stats = md.body.stats;
duke@1 198 // Synthetic initializations can appear before the super call.
duke@1 199 while (stats.nonEmpty() && isSyntheticInit(stats.head))
duke@1 200 stats = stats.tail;
duke@1 201 if (stats.isEmpty()) return null;
duke@1 202 if (stats.head.getTag() != JCTree.EXEC) return null;
duke@1 203 JCExpressionStatement exec = (JCExpressionStatement) stats.head;
duke@1 204 if (exec.expr.getTag() != JCTree.APPLY) return null;
duke@1 205 return (JCMethodInvocation)exec.expr;
duke@1 206 }
duke@1 207
duke@1 208 /** Return true if a tree represents the null literal. */
duke@1 209 public static boolean isNull(JCTree tree) {
duke@1 210 if (tree.getTag() != JCTree.LITERAL)
duke@1 211 return false;
duke@1 212 JCLiteral lit = (JCLiteral) tree;
duke@1 213 return (lit.typetag == TypeTags.BOT);
duke@1 214 }
duke@1 215
duke@1 216 /** The position of the first statement in a block, or the position of
duke@1 217 * the block itself if it is empty.
duke@1 218 */
duke@1 219 public static int firstStatPos(JCTree tree) {
duke@1 220 if (tree.getTag() == JCTree.BLOCK && ((JCBlock) tree).stats.nonEmpty())
duke@1 221 return ((JCBlock) tree).stats.head.pos;
duke@1 222 else
duke@1 223 return tree.pos;
duke@1 224 }
duke@1 225
duke@1 226 /** The end position of given tree, if it is a block with
duke@1 227 * defined endpos.
duke@1 228 */
duke@1 229 public static int endPos(JCTree tree) {
duke@1 230 if (tree.getTag() == JCTree.BLOCK && ((JCBlock) tree).endpos != Position.NOPOS)
duke@1 231 return ((JCBlock) tree).endpos;
duke@1 232 else if (tree.getTag() == JCTree.SYNCHRONIZED)
duke@1 233 return endPos(((JCSynchronized) tree).body);
duke@1 234 else if (tree.getTag() == JCTree.TRY) {
duke@1 235 JCTry t = (JCTry) tree;
duke@1 236 return endPos((t.finalizer != null)
duke@1 237 ? t.finalizer
duke@1 238 : t.catchers.last().body);
duke@1 239 } else
duke@1 240 return tree.pos;
duke@1 241 }
duke@1 242
duke@1 243
duke@1 244 /** Get the start position for a tree node. The start position is
duke@1 245 * defined to be the position of the first character of the first
duke@1 246 * token of the node's source text.
duke@1 247 * @param tree The tree node
duke@1 248 */
duke@1 249 public static int getStartPos(JCTree tree) {
duke@1 250 if (tree == null)
duke@1 251 return Position.NOPOS;
duke@1 252
duke@1 253 switch(tree.getTag()) {
duke@1 254 case(JCTree.APPLY):
duke@1 255 return getStartPos(((JCMethodInvocation) tree).meth);
duke@1 256 case(JCTree.ASSIGN):
duke@1 257 return getStartPos(((JCAssign) tree).lhs);
duke@1 258 case(JCTree.BITOR_ASG): case(JCTree.BITXOR_ASG): case(JCTree.BITAND_ASG):
duke@1 259 case(JCTree.SL_ASG): case(JCTree.SR_ASG): case(JCTree.USR_ASG):
duke@1 260 case(JCTree.PLUS_ASG): case(JCTree.MINUS_ASG): case(JCTree.MUL_ASG):
duke@1 261 case(JCTree.DIV_ASG): case(JCTree.MOD_ASG):
duke@1 262 return getStartPos(((JCAssignOp) tree).lhs);
duke@1 263 case(JCTree.OR): case(JCTree.AND): case(JCTree.BITOR):
duke@1 264 case(JCTree.BITXOR): case(JCTree.BITAND): case(JCTree.EQ):
duke@1 265 case(JCTree.NE): case(JCTree.LT): case(JCTree.GT):
duke@1 266 case(JCTree.LE): case(JCTree.GE): case(JCTree.SL):
duke@1 267 case(JCTree.SR): case(JCTree.USR): case(JCTree.PLUS):
duke@1 268 case(JCTree.MINUS): case(JCTree.MUL): case(JCTree.DIV):
duke@1 269 case(JCTree.MOD):
duke@1 270 return getStartPos(((JCBinary) tree).lhs);
duke@1 271 case(JCTree.CLASSDEF): {
duke@1 272 JCClassDecl node = (JCClassDecl)tree;
duke@1 273 if (node.mods.pos != Position.NOPOS)
duke@1 274 return node.mods.pos;
duke@1 275 break;
duke@1 276 }
duke@1 277 case(JCTree.CONDEXPR):
duke@1 278 return getStartPos(((JCConditional) tree).cond);
duke@1 279 case(JCTree.EXEC):
duke@1 280 return getStartPos(((JCExpressionStatement) tree).expr);
duke@1 281 case(JCTree.INDEXED):
duke@1 282 return getStartPos(((JCArrayAccess) tree).indexed);
duke@1 283 case(JCTree.METHODDEF): {
duke@1 284 JCMethodDecl node = (JCMethodDecl)tree;
duke@1 285 if (node.mods.pos != Position.NOPOS)
duke@1 286 return node.mods.pos;
duke@1 287 if (node.typarams.nonEmpty()) // List.nil() used for no typarams
duke@1 288 return getStartPos(node.typarams.head);
duke@1 289 return node.restype == null ? node.pos : getStartPos(node.restype);
duke@1 290 }
duke@1 291 case(JCTree.SELECT):
duke@1 292 return getStartPos(((JCFieldAccess) tree).selected);
duke@1 293 case(JCTree.TYPEAPPLY):
duke@1 294 return getStartPos(((JCTypeApply) tree).clazz);
duke@1 295 case(JCTree.TYPEARRAY):
duke@1 296 return getStartPos(((JCArrayTypeTree) tree).elemtype);
duke@1 297 case(JCTree.TYPETEST):
duke@1 298 return getStartPos(((JCInstanceOf) tree).expr);
duke@1 299 case(JCTree.POSTINC):
duke@1 300 case(JCTree.POSTDEC):
duke@1 301 return getStartPos(((JCUnary) tree).arg);
duke@1 302 case(JCTree.VARDEF): {
duke@1 303 JCVariableDecl node = (JCVariableDecl)tree;
duke@1 304 if (node.mods.pos != Position.NOPOS) {
duke@1 305 return node.mods.pos;
duke@1 306 } else {
duke@1 307 return getStartPos(node.vartype);
duke@1 308 }
duke@1 309 }
duke@1 310 case(JCTree.ERRONEOUS): {
duke@1 311 JCErroneous node = (JCErroneous)tree;
duke@1 312 if (node.errs != null && node.errs.nonEmpty())
duke@1 313 return getStartPos(node.errs.head);
duke@1 314 }
duke@1 315 }
duke@1 316 return tree.pos;
duke@1 317 }
duke@1 318
duke@1 319 /** The end position of given tree, given a table of end positions generated by the parser
duke@1 320 */
duke@1 321 public static int getEndPos(JCTree tree, Map<JCTree, Integer> endPositions) {
duke@1 322 if (tree == null)
duke@1 323 return Position.NOPOS;
duke@1 324
duke@1 325 if (endPositions == null) {
duke@1 326 // fall back on limited info in the tree
duke@1 327 return endPos(tree);
duke@1 328 }
duke@1 329
duke@1 330 Integer mapPos = endPositions.get(tree);
duke@1 331 if (mapPos != null)
duke@1 332 return mapPos;
duke@1 333
duke@1 334 switch(tree.getTag()) {
duke@1 335 case(JCTree.BITOR_ASG): case(JCTree.BITXOR_ASG): case(JCTree.BITAND_ASG):
duke@1 336 case(JCTree.SL_ASG): case(JCTree.SR_ASG): case(JCTree.USR_ASG):
duke@1 337 case(JCTree.PLUS_ASG): case(JCTree.MINUS_ASG): case(JCTree.MUL_ASG):
duke@1 338 case(JCTree.DIV_ASG): case(JCTree.MOD_ASG):
duke@1 339 return getEndPos(((JCAssignOp) tree).rhs, endPositions);
duke@1 340 case(JCTree.OR): case(JCTree.AND): case(JCTree.BITOR):
duke@1 341 case(JCTree.BITXOR): case(JCTree.BITAND): case(JCTree.EQ):
duke@1 342 case(JCTree.NE): case(JCTree.LT): case(JCTree.GT):
duke@1 343 case(JCTree.LE): case(JCTree.GE): case(JCTree.SL):
duke@1 344 case(JCTree.SR): case(JCTree.USR): case(JCTree.PLUS):
duke@1 345 case(JCTree.MINUS): case(JCTree.MUL): case(JCTree.DIV):
duke@1 346 case(JCTree.MOD):
duke@1 347 return getEndPos(((JCBinary) tree).rhs, endPositions);
duke@1 348 case(JCTree.CASE):
duke@1 349 return getEndPos(((JCCase) tree).stats.last(), endPositions);
duke@1 350 case(JCTree.CATCH):
duke@1 351 return getEndPos(((JCCatch) tree).body, endPositions);
duke@1 352 case(JCTree.CONDEXPR):
duke@1 353 return getEndPos(((JCConditional) tree).falsepart, endPositions);
duke@1 354 case(JCTree.FORLOOP):
duke@1 355 return getEndPos(((JCForLoop) tree).body, endPositions);
duke@1 356 case(JCTree.FOREACHLOOP):
duke@1 357 return getEndPos(((JCEnhancedForLoop) tree).body, endPositions);
duke@1 358 case(JCTree.IF): {
duke@1 359 JCIf node = (JCIf)tree;
duke@1 360 if (node.elsepart == null) {
duke@1 361 return getEndPos(node.thenpart, endPositions);
duke@1 362 } else {
duke@1 363 return getEndPos(node.elsepart, endPositions);
duke@1 364 }
duke@1 365 }
duke@1 366 case(JCTree.LABELLED):
duke@1 367 return getEndPos(((JCLabeledStatement) tree).body, endPositions);
duke@1 368 case(JCTree.MODIFIERS):
duke@1 369 return getEndPos(((JCModifiers) tree).annotations.last(), endPositions);
duke@1 370 case(JCTree.SYNCHRONIZED):
duke@1 371 return getEndPos(((JCSynchronized) tree).body, endPositions);
duke@1 372 case(JCTree.TOPLEVEL):
duke@1 373 return getEndPos(((JCCompilationUnit) tree).defs.last(), endPositions);
duke@1 374 case(JCTree.TRY): {
duke@1 375 JCTry node = (JCTry)tree;
duke@1 376 if (node.finalizer != null) {
duke@1 377 return getEndPos(node.finalizer, endPositions);
duke@1 378 } else if (!node.catchers.isEmpty()) {
duke@1 379 return getEndPos(node.catchers.last(), endPositions);
duke@1 380 } else {
duke@1 381 return getEndPos(node.body, endPositions);
duke@1 382 }
duke@1 383 }
duke@1 384 case(JCTree.WILDCARD):
duke@1 385 return getEndPos(((JCWildcard) tree).inner, endPositions);
duke@1 386 case(JCTree.TYPECAST):
duke@1 387 return getEndPos(((JCTypeCast) tree).expr, endPositions);
duke@1 388 case(JCTree.TYPETEST):
duke@1 389 return getEndPos(((JCInstanceOf) tree).clazz, endPositions);
duke@1 390 case(JCTree.POS):
duke@1 391 case(JCTree.NEG):
duke@1 392 case(JCTree.NOT):
duke@1 393 case(JCTree.COMPL):
duke@1 394 case(JCTree.PREINC):
duke@1 395 case(JCTree.PREDEC):
duke@1 396 return getEndPos(((JCUnary) tree).arg, endPositions);
duke@1 397 case(JCTree.WHILELOOP):
duke@1 398 return getEndPos(((JCWhileLoop) tree).body, endPositions);
duke@1 399 case(JCTree.ERRONEOUS): {
duke@1 400 JCErroneous node = (JCErroneous)tree;
duke@1 401 if (node.errs != null && node.errs.nonEmpty())
duke@1 402 return getEndPos(node.errs.last(), endPositions);
duke@1 403 }
duke@1 404 }
duke@1 405 return Position.NOPOS;
duke@1 406 }
duke@1 407
duke@1 408
duke@1 409 /** A DiagnosticPosition with the preferred position set to the
duke@1 410 * end position of given tree, if it is a block with
duke@1 411 * defined endpos.
duke@1 412 */
duke@1 413 public static DiagnosticPosition diagEndPos(final JCTree tree) {
duke@1 414 final int endPos = TreeInfo.endPos(tree);
duke@1 415 return new DiagnosticPosition() {
duke@1 416 public JCTree getTree() { return tree; }
duke@1 417 public int getStartPosition() { return TreeInfo.getStartPos(tree); }
duke@1 418 public int getPreferredPosition() { return endPos; }
duke@1 419 public int getEndPosition(Map<JCTree, Integer> endPosTable) {
duke@1 420 return TreeInfo.getEndPos(tree, endPosTable);
duke@1 421 }
duke@1 422 };
duke@1 423 }
duke@1 424
duke@1 425 /** The position of the finalizer of given try/synchronized statement.
duke@1 426 */
duke@1 427 public static int finalizerPos(JCTree tree) {
duke@1 428 if (tree.getTag() == JCTree.TRY) {
duke@1 429 JCTry t = (JCTry) tree;
duke@1 430 assert t.finalizer != null;
duke@1 431 return firstStatPos(t.finalizer);
duke@1 432 } else if (tree.getTag() == JCTree.SYNCHRONIZED) {
duke@1 433 return endPos(((JCSynchronized) tree).body);
duke@1 434 } else {
duke@1 435 throw new AssertionError();
duke@1 436 }
duke@1 437 }
duke@1 438
duke@1 439 /** Find the position for reporting an error about a symbol, where
duke@1 440 * that symbol is defined somewhere in the given tree. */
duke@1 441 public static int positionFor(final Symbol sym, final JCTree tree) {
duke@1 442 JCTree decl = declarationFor(sym, tree);
duke@1 443 return ((decl != null) ? decl : tree).pos;
duke@1 444 }
duke@1 445
duke@1 446 /** Find the position for reporting an error about a symbol, where
duke@1 447 * that symbol is defined somewhere in the given tree. */
duke@1 448 public static DiagnosticPosition diagnosticPositionFor(final Symbol sym, final JCTree tree) {
duke@1 449 JCTree decl = declarationFor(sym, tree);
duke@1 450 return ((decl != null) ? decl : tree).pos();
duke@1 451 }
duke@1 452
duke@1 453 /** Find the declaration for a symbol, where
duke@1 454 * that symbol is defined somewhere in the given tree. */
duke@1 455 public static JCTree declarationFor(final Symbol sym, final JCTree tree) {
duke@1 456 class DeclScanner extends TreeScanner {
duke@1 457 JCTree result = null;
duke@1 458 public void scan(JCTree tree) {
duke@1 459 if (tree!=null && result==null)
duke@1 460 tree.accept(this);
duke@1 461 }
duke@1 462 public void visitTopLevel(JCCompilationUnit that) {
duke@1 463 if (that.packge == sym) result = that;
duke@1 464 else super.visitTopLevel(that);
duke@1 465 }
duke@1 466 public void visitClassDef(JCClassDecl that) {
duke@1 467 if (that.sym == sym) result = that;
duke@1 468 else super.visitClassDef(that);
duke@1 469 }
duke@1 470 public void visitMethodDef(JCMethodDecl that) {
duke@1 471 if (that.sym == sym) result = that;
duke@1 472 else super.visitMethodDef(that);
duke@1 473 }
duke@1 474 public void visitVarDef(JCVariableDecl that) {
duke@1 475 if (that.sym == sym) result = that;
duke@1 476 else super.visitVarDef(that);
duke@1 477 }
duke@1 478 }
duke@1 479 DeclScanner s = new DeclScanner();
duke@1 480 tree.accept(s);
duke@1 481 return s.result;
duke@1 482 }
duke@1 483
duke@1 484 public static Env<AttrContext> scopeFor(JCTree node, JCCompilationUnit unit) {
duke@1 485 return scopeFor(pathFor(node, unit));
duke@1 486 }
duke@1 487
duke@1 488 public static Env<AttrContext> scopeFor(List<JCTree> path) {
duke@1 489 // TODO: not implemented yet
duke@1 490 throw new UnsupportedOperationException("not implemented yet");
duke@1 491 }
duke@1 492
duke@1 493 public static List<JCTree> pathFor(final JCTree node, final JCCompilationUnit unit) {
duke@1 494 class Result extends Error {
duke@1 495 static final long serialVersionUID = -5942088234594905625L;
duke@1 496 List<JCTree> path;
duke@1 497 Result(List<JCTree> path) {
duke@1 498 this.path = path;
duke@1 499 }
duke@1 500 }
duke@1 501 class PathFinder extends TreeScanner {
duke@1 502 List<JCTree> path = List.nil();
duke@1 503 public void scan(JCTree tree) {
duke@1 504 if (tree != null) {
duke@1 505 path = path.prepend(tree);
duke@1 506 if (tree == node)
duke@1 507 throw new Result(path);
duke@1 508 super.scan(tree);
duke@1 509 path = path.tail;
duke@1 510 }
duke@1 511 }
duke@1 512 }
duke@1 513 try {
duke@1 514 new PathFinder().scan(unit);
duke@1 515 } catch (Result result) {
duke@1 516 return result.path;
duke@1 517 }
duke@1 518 return List.nil();
duke@1 519 }
duke@1 520
duke@1 521 /** Return the statement referenced by a label.
duke@1 522 * If the label refers to a loop or switch, return that switch
duke@1 523 * otherwise return the labelled statement itself
duke@1 524 */
duke@1 525 public static JCTree referencedStatement(JCLabeledStatement tree) {
duke@1 526 JCTree t = tree;
duke@1 527 do t = ((JCLabeledStatement) t).body;
duke@1 528 while (t.getTag() == JCTree.LABELLED);
duke@1 529 switch (t.getTag()) {
duke@1 530 case JCTree.DOLOOP: case JCTree.WHILELOOP: case JCTree.FORLOOP: case JCTree.FOREACHLOOP: case JCTree.SWITCH:
duke@1 531 return t;
duke@1 532 default:
duke@1 533 return tree;
duke@1 534 }
duke@1 535 }
duke@1 536
duke@1 537 /** Skip parens and return the enclosed expression
duke@1 538 */
duke@1 539 public static JCExpression skipParens(JCExpression tree) {
duke@1 540 while (tree.getTag() == JCTree.PARENS) {
duke@1 541 tree = ((JCParens) tree).expr;
duke@1 542 }
duke@1 543 return tree;
duke@1 544 }
duke@1 545
duke@1 546 /** Skip parens and return the enclosed expression
duke@1 547 */
duke@1 548 public static JCTree skipParens(JCTree tree) {
duke@1 549 if (tree.getTag() == JCTree.PARENS)
duke@1 550 return skipParens((JCParens)tree);
duke@1 551 else
duke@1 552 return tree;
duke@1 553 }
duke@1 554
duke@1 555 /** Return the types of a list of trees.
duke@1 556 */
duke@1 557 public static List<Type> types(List<? extends JCTree> trees) {
duke@1 558 ListBuffer<Type> ts = new ListBuffer<Type>();
duke@1 559 for (List<? extends JCTree> l = trees; l.nonEmpty(); l = l.tail)
duke@1 560 ts.append(l.head.type);
duke@1 561 return ts.toList();
duke@1 562 }
duke@1 563
duke@1 564 /** If this tree is an identifier or a field or a parameterized type,
duke@1 565 * return its name, otherwise return null.
duke@1 566 */
duke@1 567 public static Name name(JCTree tree) {
duke@1 568 switch (tree.getTag()) {
duke@1 569 case JCTree.IDENT:
duke@1 570 return ((JCIdent) tree).name;
duke@1 571 case JCTree.SELECT:
duke@1 572 return ((JCFieldAccess) tree).name;
duke@1 573 case JCTree.TYPEAPPLY:
duke@1 574 return name(((JCTypeApply) tree).clazz);
duke@1 575 default:
duke@1 576 return null;
duke@1 577 }
duke@1 578 }
duke@1 579
duke@1 580 /** If this tree is a qualified identifier, its return fully qualified name,
duke@1 581 * otherwise return null.
duke@1 582 */
duke@1 583 public static Name fullName(JCTree tree) {
duke@1 584 tree = skipParens(tree);
duke@1 585 switch (tree.getTag()) {
duke@1 586 case JCTree.IDENT:
duke@1 587 return ((JCIdent) tree).name;
duke@1 588 case JCTree.SELECT:
duke@1 589 Name sname = fullName(((JCFieldAccess) tree).selected);
duke@1 590 return sname == null ? null : sname.append('.', name(tree));
duke@1 591 default:
duke@1 592 return null;
duke@1 593 }
duke@1 594 }
duke@1 595
duke@1 596 public static Symbol symbolFor(JCTree node) {
duke@1 597 node = skipParens(node);
duke@1 598 switch (node.getTag()) {
duke@1 599 case JCTree.CLASSDEF:
duke@1 600 return ((JCClassDecl) node).sym;
duke@1 601 case JCTree.METHODDEF:
duke@1 602 return ((JCMethodDecl) node).sym;
duke@1 603 case JCTree.VARDEF:
duke@1 604 return ((JCVariableDecl) node).sym;
duke@1 605 default:
duke@1 606 return null;
duke@1 607 }
duke@1 608 }
duke@1 609
duke@1 610 /** If this tree is an identifier or a field, return its symbol,
duke@1 611 * otherwise return null.
duke@1 612 */
duke@1 613 public static Symbol symbol(JCTree tree) {
duke@1 614 tree = skipParens(tree);
duke@1 615 switch (tree.getTag()) {
duke@1 616 case JCTree.IDENT:
duke@1 617 return ((JCIdent) tree).sym;
duke@1 618 case JCTree.SELECT:
duke@1 619 return ((JCFieldAccess) tree).sym;
duke@1 620 case JCTree.TYPEAPPLY:
duke@1 621 return symbol(((JCTypeApply) tree).clazz);
duke@1 622 default:
duke@1 623 return null;
duke@1 624 }
duke@1 625 }
duke@1 626
duke@1 627 /** Return true if this is a nonstatic selection. */
duke@1 628 public static boolean nonstaticSelect(JCTree tree) {
duke@1 629 tree = skipParens(tree);
duke@1 630 if (tree.getTag() != JCTree.SELECT) return false;
duke@1 631 JCFieldAccess s = (JCFieldAccess) tree;
duke@1 632 Symbol e = symbol(s.selected);
duke@1 633 return e == null || (e.kind != Kinds.PCK && e.kind != Kinds.TYP);
duke@1 634 }
duke@1 635
duke@1 636 /** If this tree is an identifier or a field, set its symbol, otherwise skip.
duke@1 637 */
duke@1 638 public static void setSymbol(JCTree tree, Symbol sym) {
duke@1 639 tree = skipParens(tree);
duke@1 640 switch (tree.getTag()) {
duke@1 641 case JCTree.IDENT:
duke@1 642 ((JCIdent) tree).sym = sym; break;
duke@1 643 case JCTree.SELECT:
duke@1 644 ((JCFieldAccess) tree).sym = sym; break;
duke@1 645 default:
duke@1 646 }
duke@1 647 }
duke@1 648
duke@1 649 /** If this tree is a declaration or a block, return its flags field,
duke@1 650 * otherwise return 0.
duke@1 651 */
duke@1 652 public static long flags(JCTree tree) {
duke@1 653 switch (tree.getTag()) {
duke@1 654 case JCTree.VARDEF:
duke@1 655 return ((JCVariableDecl) tree).mods.flags;
duke@1 656 case JCTree.METHODDEF:
duke@1 657 return ((JCMethodDecl) tree).mods.flags;
duke@1 658 case JCTree.CLASSDEF:
duke@1 659 return ((JCClassDecl) tree).mods.flags;
duke@1 660 case JCTree.BLOCK:
duke@1 661 return ((JCBlock) tree).flags;
duke@1 662 default:
duke@1 663 return 0;
duke@1 664 }
duke@1 665 }
duke@1 666
duke@1 667 /** Return first (smallest) flag in `flags':
duke@1 668 * pre: flags != 0
duke@1 669 */
duke@1 670 public static long firstFlag(long flags) {
duke@1 671 int flag = 1;
duke@1 672 while ((flag & StandardFlags) != 0 && (flag & flags) == 0)
duke@1 673 flag = flag << 1;
duke@1 674 return flag;
duke@1 675 }
duke@1 676
duke@1 677 /** Return flags as a string, separated by " ".
duke@1 678 */
duke@1 679 public static String flagNames(long flags) {
duke@1 680 return Flags.toString(flags & StandardFlags).trim();
duke@1 681 }
duke@1 682
duke@1 683 /** Operator precedences values.
duke@1 684 */
duke@1 685 public static final int
duke@1 686 notExpression = -1, // not an expression
duke@1 687 noPrec = 0, // no enclosing expression
duke@1 688 assignPrec = 1,
duke@1 689 assignopPrec = 2,
duke@1 690 condPrec = 3,
duke@1 691 orPrec = 4,
duke@1 692 andPrec = 5,
duke@1 693 bitorPrec = 6,
duke@1 694 bitxorPrec = 7,
duke@1 695 bitandPrec = 8,
duke@1 696 eqPrec = 9,
duke@1 697 ordPrec = 10,
duke@1 698 shiftPrec = 11,
duke@1 699 addPrec = 12,
duke@1 700 mulPrec = 13,
duke@1 701 prefixPrec = 14,
duke@1 702 postfixPrec = 15,
duke@1 703 precCount = 16;
duke@1 704
duke@1 705
duke@1 706 /** Map operators to their precedence levels.
duke@1 707 */
duke@1 708 public static int opPrec(int op) {
duke@1 709 switch(op) {
duke@1 710 case JCTree.POS:
duke@1 711 case JCTree.NEG:
duke@1 712 case JCTree.NOT:
duke@1 713 case JCTree.COMPL:
duke@1 714 case JCTree.PREINC:
duke@1 715 case JCTree.PREDEC: return prefixPrec;
duke@1 716 case JCTree.POSTINC:
duke@1 717 case JCTree.POSTDEC:
duke@1 718 case JCTree.NULLCHK: return postfixPrec;
duke@1 719 case JCTree.ASSIGN: return assignPrec;
duke@1 720 case JCTree.BITOR_ASG:
duke@1 721 case JCTree.BITXOR_ASG:
duke@1 722 case JCTree.BITAND_ASG:
duke@1 723 case JCTree.SL_ASG:
duke@1 724 case JCTree.SR_ASG:
duke@1 725 case JCTree.USR_ASG:
duke@1 726 case JCTree.PLUS_ASG:
duke@1 727 case JCTree.MINUS_ASG:
duke@1 728 case JCTree.MUL_ASG:
duke@1 729 case JCTree.DIV_ASG:
duke@1 730 case JCTree.MOD_ASG: return assignopPrec;
duke@1 731 case JCTree.OR: return orPrec;
duke@1 732 case JCTree.AND: return andPrec;
duke@1 733 case JCTree.EQ:
duke@1 734 case JCTree.NE: return eqPrec;
duke@1 735 case JCTree.LT:
duke@1 736 case JCTree.GT:
duke@1 737 case JCTree.LE:
duke@1 738 case JCTree.GE: return ordPrec;
duke@1 739 case JCTree.BITOR: return bitorPrec;
duke@1 740 case JCTree.BITXOR: return bitxorPrec;
duke@1 741 case JCTree.BITAND: return bitandPrec;
duke@1 742 case JCTree.SL:
duke@1 743 case JCTree.SR:
duke@1 744 case JCTree.USR: return shiftPrec;
duke@1 745 case JCTree.PLUS:
duke@1 746 case JCTree.MINUS: return addPrec;
duke@1 747 case JCTree.MUL:
duke@1 748 case JCTree.DIV:
duke@1 749 case JCTree.MOD: return mulPrec;
duke@1 750 case JCTree.TYPETEST: return ordPrec;
duke@1 751 default: throw new AssertionError();
duke@1 752 }
duke@1 753 }
duke@1 754
duke@1 755 static Tree.Kind tagToKind(int tag) {
duke@1 756 switch (tag) {
duke@1 757 // Postfix expressions
duke@1 758 case JCTree.POSTINC: // _ ++
duke@1 759 return Tree.Kind.POSTFIX_INCREMENT;
duke@1 760 case JCTree.POSTDEC: // _ --
duke@1 761 return Tree.Kind.POSTFIX_DECREMENT;
duke@1 762
duke@1 763 // Unary operators
duke@1 764 case JCTree.PREINC: // ++ _
duke@1 765 return Tree.Kind.PREFIX_INCREMENT;
duke@1 766 case JCTree.PREDEC: // -- _
duke@1 767 return Tree.Kind.PREFIX_DECREMENT;
duke@1 768 case JCTree.POS: // +
duke@1 769 return Tree.Kind.UNARY_PLUS;
duke@1 770 case JCTree.NEG: // -
duke@1 771 return Tree.Kind.UNARY_MINUS;
duke@1 772 case JCTree.COMPL: // ~
duke@1 773 return Tree.Kind.BITWISE_COMPLEMENT;
duke@1 774 case JCTree.NOT: // !
duke@1 775 return Tree.Kind.LOGICAL_COMPLEMENT;
duke@1 776
duke@1 777 // Binary operators
duke@1 778
duke@1 779 // Multiplicative operators
duke@1 780 case JCTree.MUL: // *
duke@1 781 return Tree.Kind.MULTIPLY;
duke@1 782 case JCTree.DIV: // /
duke@1 783 return Tree.Kind.DIVIDE;
duke@1 784 case JCTree.MOD: // %
duke@1 785 return Tree.Kind.REMAINDER;
duke@1 786
duke@1 787 // Additive operators
duke@1 788 case JCTree.PLUS: // +
duke@1 789 return Tree.Kind.PLUS;
duke@1 790 case JCTree.MINUS: // -
duke@1 791 return Tree.Kind.MINUS;
duke@1 792
duke@1 793 // Shift operators
duke@1 794 case JCTree.SL: // <<
duke@1 795 return Tree.Kind.LEFT_SHIFT;
duke@1 796 case JCTree.SR: // >>
duke@1 797 return Tree.Kind.RIGHT_SHIFT;
duke@1 798 case JCTree.USR: // >>>
duke@1 799 return Tree.Kind.UNSIGNED_RIGHT_SHIFT;
duke@1 800
duke@1 801 // Relational operators
duke@1 802 case JCTree.LT: // <
duke@1 803 return Tree.Kind.LESS_THAN;
duke@1 804 case JCTree.GT: // >
duke@1 805 return Tree.Kind.GREATER_THAN;
duke@1 806 case JCTree.LE: // <=
duke@1 807 return Tree.Kind.LESS_THAN_EQUAL;
duke@1 808 case JCTree.GE: // >=
duke@1 809 return Tree.Kind.GREATER_THAN_EQUAL;
duke@1 810
duke@1 811 // Equality operators
duke@1 812 case JCTree.EQ: // ==
duke@1 813 return Tree.Kind.EQUAL_TO;
duke@1 814 case JCTree.NE: // !=
duke@1 815 return Tree.Kind.NOT_EQUAL_TO;
duke@1 816
duke@1 817 // Bitwise and logical operators
duke@1 818 case JCTree.BITAND: // &
duke@1 819 return Tree.Kind.AND;
duke@1 820 case JCTree.BITXOR: // ^
duke@1 821 return Tree.Kind.XOR;
duke@1 822 case JCTree.BITOR: // |
duke@1 823 return Tree.Kind.OR;
duke@1 824
duke@1 825 // Conditional operators
duke@1 826 case JCTree.AND: // &&
duke@1 827 return Tree.Kind.CONDITIONAL_AND;
duke@1 828 case JCTree.OR: // ||
duke@1 829 return Tree.Kind.CONDITIONAL_OR;
duke@1 830
duke@1 831 // Assignment operators
duke@1 832 case JCTree.MUL_ASG: // *=
duke@1 833 return Tree.Kind.MULTIPLY_ASSIGNMENT;
duke@1 834 case JCTree.DIV_ASG: // /=
duke@1 835 return Tree.Kind.DIVIDE_ASSIGNMENT;
duke@1 836 case JCTree.MOD_ASG: // %=
duke@1 837 return Tree.Kind.REMAINDER_ASSIGNMENT;
duke@1 838 case JCTree.PLUS_ASG: // +=
duke@1 839 return Tree.Kind.PLUS_ASSIGNMENT;
duke@1 840 case JCTree.MINUS_ASG: // -=
duke@1 841 return Tree.Kind.MINUS_ASSIGNMENT;
duke@1 842 case JCTree.SL_ASG: // <<=
duke@1 843 return Tree.Kind.LEFT_SHIFT_ASSIGNMENT;
duke@1 844 case JCTree.SR_ASG: // >>=
duke@1 845 return Tree.Kind.RIGHT_SHIFT_ASSIGNMENT;
duke@1 846 case JCTree.USR_ASG: // >>>=
duke@1 847 return Tree.Kind.UNSIGNED_RIGHT_SHIFT_ASSIGNMENT;
duke@1 848 case JCTree.BITAND_ASG: // &=
duke@1 849 return Tree.Kind.AND_ASSIGNMENT;
duke@1 850 case JCTree.BITXOR_ASG: // ^=
duke@1 851 return Tree.Kind.XOR_ASSIGNMENT;
duke@1 852 case JCTree.BITOR_ASG: // |=
duke@1 853 return Tree.Kind.OR_ASSIGNMENT;
duke@1 854
duke@1 855 // Null check (implementation detail), for example, __.getClass()
duke@1 856 case JCTree.NULLCHK:
duke@1 857 return Tree.Kind.OTHER;
duke@1 858
duke@1 859 default:
duke@1 860 return null;
duke@1 861 }
duke@1 862 }
duke@1 863 }

mercurial