| /* |
| * Copyright (c) 2006, 2013, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. Oracle designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Oracle in the LICENSE file that accompanied this code. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| package com.sun.tools.javac.tree; |
| |
| import com.sun.source.tree.*; |
| import com.sun.tools.javac.tree.JCTree.*; |
| import com.sun.tools.javac.util.List; |
| import com.sun.tools.javac.util.ListBuffer; |
| |
| /** |
| * Creates a copy of a tree, using a given TreeMaker. |
| * Names, literal values, etc are shared with the original. |
| * |
| * <p><b>This is NOT part of any supported API. |
| * If you write code that depends on this, you do so at your own risk. |
| * This code and its internal interfaces are subject to change or |
| * deletion without notice.</b> |
| */ |
| public class TreeCopier<P> implements TreeVisitor<JCTree,P> { |
| private TreeMaker M; |
| |
| /** Creates a new instance of TreeCopier */ |
| public TreeCopier(TreeMaker M) { |
| this.M = M; |
| } |
| |
| public <T extends JCTree> T copy(T tree) { |
| return copy(tree, null); |
| } |
| |
| @SuppressWarnings("unchecked") |
| public <T extends JCTree> T copy(T tree, P p) { |
| if (tree == null) |
| return null; |
| return (T) (tree.accept(this, p)); |
| } |
| |
| public <T extends JCTree> List<T> copy(List<T> trees) { |
| return copy(trees, null); |
| } |
| |
| public <T extends JCTree> List<T> copy(List<T> trees, P p) { |
| if (trees == null) |
| return null; |
| ListBuffer<T> lb = new ListBuffer<T>(); |
| for (T tree: trees) |
| lb.append(copy(tree, p)); |
| return lb.toList(); |
| } |
| |
| public JCTree visitAnnotatedType(AnnotatedTypeTree node, P p) { |
| JCAnnotatedType t = (JCAnnotatedType) node; |
| List<JCAnnotation> annotations = copy(t.annotations, p); |
| JCExpression underlyingType = copy(t.underlyingType, p); |
| return M.at(t.pos).AnnotatedType(annotations, underlyingType); |
| } |
| |
| public JCTree visitAnnotation(AnnotationTree node, P p) { |
| JCAnnotation t = (JCAnnotation) node; |
| JCTree annotationType = copy(t.annotationType, p); |
| List<JCExpression> args = copy(t.args, p); |
| if (t.getKind() == Tree.Kind.TYPE_ANNOTATION) { |
| JCAnnotation newTA = M.at(t.pos).TypeAnnotation(annotationType, args); |
| newTA.attribute = t.attribute; |
| return newTA; |
| } else { |
| JCAnnotation newT = M.at(t.pos).Annotation(annotationType, args); |
| newT.attribute = t.attribute; |
| return newT; |
| } |
| } |
| |
| public JCTree visitAssert(AssertTree node, P p) { |
| JCAssert t = (JCAssert) node; |
| JCExpression cond = copy(t.cond, p); |
| JCExpression detail = copy(t.detail, p); |
| return M.at(t.pos).Assert(cond, detail); |
| } |
| |
| public JCTree visitAssignment(AssignmentTree node, P p) { |
| JCAssign t = (JCAssign) node; |
| JCExpression lhs = copy(t.lhs, p); |
| JCExpression rhs = copy(t.rhs, p); |
| return M.at(t.pos).Assign(lhs, rhs); |
| } |
| |
| public JCTree visitCompoundAssignment(CompoundAssignmentTree node, P p) { |
| JCAssignOp t = (JCAssignOp) node; |
| JCTree lhs = copy(t.lhs, p); |
| JCTree rhs = copy(t.rhs, p); |
| return M.at(t.pos).Assignop(t.getTag(), lhs, rhs); |
| } |
| |
| public JCTree visitBinary(BinaryTree node, P p) { |
| JCBinary t = (JCBinary) node; |
| JCExpression lhs = copy(t.lhs, p); |
| JCExpression rhs = copy(t.rhs, p); |
| return M.at(t.pos).Binary(t.getTag(), lhs, rhs); |
| } |
| |
| public JCTree visitBlock(BlockTree node, P p) { |
| JCBlock t = (JCBlock) node; |
| List<JCStatement> stats = copy(t.stats, p); |
| return M.at(t.pos).Block(t.flags, stats); |
| } |
| |
| public JCTree visitBreak(BreakTree node, P p) { |
| JCBreak t = (JCBreak) node; |
| return M.at(t.pos).Break(t.label); |
| } |
| |
| public JCTree visitCase(CaseTree node, P p) { |
| JCCase t = (JCCase) node; |
| JCExpression pat = copy(t.pat, p); |
| List<JCStatement> stats = copy(t.stats, p); |
| return M.at(t.pos).Case(pat, stats); |
| } |
| |
| public JCTree visitCatch(CatchTree node, P p) { |
| JCCatch t = (JCCatch) node; |
| JCVariableDecl param = copy(t.param, p); |
| JCBlock body = copy(t.body, p); |
| return M.at(t.pos).Catch(param, body); |
| } |
| |
| public JCTree visitClass(ClassTree node, P p) { |
| JCClassDecl t = (JCClassDecl) node; |
| JCModifiers mods = copy(t.mods, p); |
| List<JCTypeParameter> typarams = copy(t.typarams, p); |
| JCExpression extending = copy(t.extending, p); |
| List<JCExpression> implementing = copy(t.implementing, p); |
| List<JCTree> defs = copy(t.defs, p); |
| return M.at(t.pos).ClassDef(mods, t.name, typarams, extending, implementing, defs); |
| } |
| |
| public JCTree visitConditionalExpression(ConditionalExpressionTree node, P p) { |
| JCConditional t = (JCConditional) node; |
| JCExpression cond = copy(t.cond, p); |
| JCExpression truepart = copy(t.truepart, p); |
| JCExpression falsepart = copy(t.falsepart, p); |
| return M.at(t.pos).Conditional(cond, truepart, falsepart); |
| } |
| |
| public JCTree visitContinue(ContinueTree node, P p) { |
| JCContinue t = (JCContinue) node; |
| return M.at(t.pos).Continue(t.label); |
| } |
| |
| public JCTree visitDoWhileLoop(DoWhileLoopTree node, P p) { |
| JCDoWhileLoop t = (JCDoWhileLoop) node; |
| JCStatement body = copy(t.body, p); |
| JCExpression cond = copy(t.cond, p); |
| return M.at(t.pos).DoLoop(body, cond); |
| } |
| |
| public JCTree visitErroneous(ErroneousTree node, P p) { |
| JCErroneous t = (JCErroneous) node; |
| List<? extends JCTree> errs = copy(t.errs, p); |
| return M.at(t.pos).Erroneous(errs); |
| } |
| |
| public JCTree visitExpressionStatement(ExpressionStatementTree node, P p) { |
| JCExpressionStatement t = (JCExpressionStatement) node; |
| JCExpression expr = copy(t.expr, p); |
| return M.at(t.pos).Exec(expr); |
| } |
| |
| public JCTree visitEnhancedForLoop(EnhancedForLoopTree node, P p) { |
| JCEnhancedForLoop t = (JCEnhancedForLoop) node; |
| JCVariableDecl var = copy(t.var, p); |
| JCExpression expr = copy(t.expr, p); |
| JCStatement body = copy(t.body, p); |
| return M.at(t.pos).ForeachLoop(var, expr, body); |
| } |
| |
| public JCTree visitForLoop(ForLoopTree node, P p) { |
| JCForLoop t = (JCForLoop) node; |
| List<JCStatement> init = copy(t.init, p); |
| JCExpression cond = copy(t.cond, p); |
| List<JCExpressionStatement> step = copy(t.step, p); |
| JCStatement body = copy(t.body, p); |
| return M.at(t.pos).ForLoop(init, cond, step, body); |
| } |
| |
| public JCTree visitIdentifier(IdentifierTree node, P p) { |
| JCIdent t = (JCIdent) node; |
| return M.at(t.pos).Ident(t.name); |
| } |
| |
| public JCTree visitIf(IfTree node, P p) { |
| JCIf t = (JCIf) node; |
| JCExpression cond = copy(t.cond, p); |
| JCStatement thenpart = copy(t.thenpart, p); |
| JCStatement elsepart = copy(t.elsepart, p); |
| return M.at(t.pos).If(cond, thenpart, elsepart); |
| } |
| |
| public JCTree visitImport(ImportTree node, P p) { |
| JCImport t = (JCImport) node; |
| JCTree qualid = copy(t.qualid, p); |
| return M.at(t.pos).Import(qualid, t.staticImport); |
| } |
| |
| public JCTree visitArrayAccess(ArrayAccessTree node, P p) { |
| JCArrayAccess t = (JCArrayAccess) node; |
| JCExpression indexed = copy(t.indexed, p); |
| JCExpression index = copy(t.index, p); |
| return M.at(t.pos).Indexed(indexed, index); |
| } |
| |
| public JCTree visitLabeledStatement(LabeledStatementTree node, P p) { |
| JCLabeledStatement t = (JCLabeledStatement) node; |
| JCStatement body = copy(t.body, p); |
| return M.at(t.pos).Labelled(t.label, t.body); |
| } |
| |
| public JCTree visitLiteral(LiteralTree node, P p) { |
| JCLiteral t = (JCLiteral) node; |
| return M.at(t.pos).Literal(t.typetag, t.value); |
| } |
| |
| public JCTree visitMethod(MethodTree node, P p) { |
| JCMethodDecl t = (JCMethodDecl) node; |
| JCModifiers mods = copy(t.mods, p); |
| JCExpression restype = copy(t.restype, p); |
| List<JCTypeParameter> typarams = copy(t.typarams, p); |
| List<JCVariableDecl> params = copy(t.params, p); |
| JCVariableDecl recvparam = copy(t.recvparam, p); |
| List<JCExpression> thrown = copy(t.thrown, p); |
| JCBlock body = copy(t.body, p); |
| JCExpression defaultValue = copy(t.defaultValue, p); |
| return M.at(t.pos).MethodDef(mods, t.name, restype, typarams, recvparam, params, thrown, body, defaultValue); |
| } |
| |
| public JCTree visitMethodInvocation(MethodInvocationTree node, P p) { |
| JCMethodInvocation t = (JCMethodInvocation) node; |
| List<JCExpression> typeargs = copy(t.typeargs, p); |
| JCExpression meth = copy(t.meth, p); |
| List<JCExpression> args = copy(t.args, p); |
| return M.at(t.pos).Apply(typeargs, meth, args); |
| } |
| |
| public JCTree visitModifiers(ModifiersTree node, P p) { |
| JCModifiers t = (JCModifiers) node; |
| List<JCAnnotation> annotations = copy(t.annotations, p); |
| return M.at(t.pos).Modifiers(t.flags, annotations); |
| } |
| |
| public JCTree visitNewArray(NewArrayTree node, P p) { |
| JCNewArray t = (JCNewArray) node; |
| JCExpression elemtype = copy(t.elemtype, p); |
| List<JCExpression> dims = copy(t.dims, p); |
| List<JCExpression> elems = copy(t.elems, p); |
| return M.at(t.pos).NewArray(elemtype, dims, elems); |
| } |
| |
| public JCTree visitNewClass(NewClassTree node, P p) { |
| JCNewClass t = (JCNewClass) node; |
| JCExpression encl = copy(t.encl, p); |
| List<JCExpression> typeargs = copy(t.typeargs, p); |
| JCExpression clazz = copy(t.clazz, p); |
| List<JCExpression> args = copy(t.args, p); |
| JCClassDecl def = copy(t.def, p); |
| return M.at(t.pos).NewClass(encl, typeargs, clazz, args, def); |
| } |
| |
| public JCTree visitLambdaExpression(LambdaExpressionTree node, P p) { |
| JCLambda t = (JCLambda) node; |
| List<JCVariableDecl> params = copy(t.params, p); |
| JCTree body = copy(t.body, p); |
| return M.at(t.pos).Lambda(params, body); |
| } |
| |
| public JCTree visitParenthesized(ParenthesizedTree node, P p) { |
| JCParens t = (JCParens) node; |
| JCExpression expr = copy(t.expr, p); |
| return M.at(t.pos).Parens(expr); |
| } |
| |
| public JCTree visitReturn(ReturnTree node, P p) { |
| JCReturn t = (JCReturn) node; |
| JCExpression expr = copy(t.expr, p); |
| return M.at(t.pos).Return(expr); |
| } |
| |
| public JCTree visitMemberSelect(MemberSelectTree node, P p) { |
| JCFieldAccess t = (JCFieldAccess) node; |
| JCExpression selected = copy(t.selected, p); |
| return M.at(t.pos).Select(selected, t.name); |
| } |
| |
| public JCTree visitMemberReference(MemberReferenceTree node, P p) { |
| JCMemberReference t = (JCMemberReference) node; |
| JCExpression expr = copy(t.expr, p); |
| List<JCExpression> typeargs = copy(t.typeargs, p); |
| return M.at(t.pos).Reference(t.mode, t.name, expr, typeargs); |
| } |
| |
| public JCTree visitEmptyStatement(EmptyStatementTree node, P p) { |
| JCSkip t = (JCSkip) node; |
| return M.at(t.pos).Skip(); |
| } |
| |
| public JCTree visitSwitch(SwitchTree node, P p) { |
| JCSwitch t = (JCSwitch) node; |
| JCExpression selector = copy(t.selector, p); |
| List<JCCase> cases = copy(t.cases, p); |
| return M.at(t.pos).Switch(selector, cases); |
| } |
| |
| public JCTree visitSynchronized(SynchronizedTree node, P p) { |
| JCSynchronized t = (JCSynchronized) node; |
| JCExpression lock = copy(t.lock, p); |
| JCBlock body = copy(t.body, p); |
| return M.at(t.pos).Synchronized(lock, body); |
| } |
| |
| public JCTree visitThrow(ThrowTree node, P p) { |
| JCThrow t = (JCThrow) node; |
| JCTree expr = copy(t.expr, p); |
| return M.at(t.pos).Throw(expr); |
| } |
| |
| public JCTree visitCompilationUnit(CompilationUnitTree node, P p) { |
| JCCompilationUnit t = (JCCompilationUnit) node; |
| List<JCAnnotation> packageAnnotations = copy(t.packageAnnotations, p); |
| JCExpression pid = copy(t.pid, p); |
| List<JCTree> defs = copy(t.defs, p); |
| return M.at(t.pos).TopLevel(packageAnnotations, pid, defs); |
| } |
| |
| public JCTree visitTry(TryTree node, P p) { |
| JCTry t = (JCTry) node; |
| List<JCTree> resources = copy(t.resources, p); |
| JCBlock body = copy(t.body, p); |
| List<JCCatch> catchers = copy(t.catchers, p); |
| JCBlock finalizer = copy(t.finalizer, p); |
| return M.at(t.pos).Try(resources, body, catchers, finalizer); |
| } |
| |
| public JCTree visitParameterizedType(ParameterizedTypeTree node, P p) { |
| JCTypeApply t = (JCTypeApply) node; |
| JCExpression clazz = copy(t.clazz, p); |
| List<JCExpression> arguments = copy(t.arguments, p); |
| return M.at(t.pos).TypeApply(clazz, arguments); |
| } |
| |
| public JCTree visitUnionType(UnionTypeTree node, P p) { |
| JCTypeUnion t = (JCTypeUnion) node; |
| List<JCExpression> components = copy(t.alternatives, p); |
| return M.at(t.pos).TypeUnion(components); |
| } |
| |
| public JCTree visitIntersectionType(IntersectionTypeTree node, P p) { |
| JCTypeIntersection t = (JCTypeIntersection) node; |
| List<JCExpression> bounds = copy(t.bounds, p); |
| return M.at(t.pos).TypeIntersection(bounds); |
| } |
| |
| public JCTree visitArrayType(ArrayTypeTree node, P p) { |
| JCArrayTypeTree t = (JCArrayTypeTree) node; |
| JCExpression elemtype = copy(t.elemtype, p); |
| return M.at(t.pos).TypeArray(elemtype); |
| } |
| |
| public JCTree visitTypeCast(TypeCastTree node, P p) { |
| JCTypeCast t = (JCTypeCast) node; |
| JCTree clazz = copy(t.clazz, p); |
| JCExpression expr = copy(t.expr, p); |
| return M.at(t.pos).TypeCast(clazz, expr); |
| } |
| |
| public JCTree visitPrimitiveType(PrimitiveTypeTree node, P p) { |
| JCPrimitiveTypeTree t = (JCPrimitiveTypeTree) node; |
| return M.at(t.pos).TypeIdent(t.typetag); |
| } |
| |
| public JCTree visitTypeParameter(TypeParameterTree node, P p) { |
| JCTypeParameter t = (JCTypeParameter) node; |
| List<JCAnnotation> annos = copy(t.annotations, p); |
| List<JCExpression> bounds = copy(t.bounds, p); |
| return M.at(t.pos).TypeParameter(t.name, bounds, annos); |
| } |
| |
| public JCTree visitInstanceOf(InstanceOfTree node, P p) { |
| JCInstanceOf t = (JCInstanceOf) node; |
| JCExpression expr = copy(t.expr, p); |
| JCTree clazz = copy(t.clazz, p); |
| return M.at(t.pos).TypeTest(expr, clazz); |
| } |
| |
| public JCTree visitUnary(UnaryTree node, P p) { |
| JCUnary t = (JCUnary) node; |
| JCExpression arg = copy(t.arg, p); |
| return M.at(t.pos).Unary(t.getTag(), arg); |
| } |
| |
| public JCTree visitVariable(VariableTree node, P p) { |
| JCVariableDecl t = (JCVariableDecl) node; |
| JCModifiers mods = copy(t.mods, p); |
| JCExpression vartype = copy(t.vartype, p); |
| JCExpression init = copy(t.init, p); |
| return M.at(t.pos).VarDef(mods, t.name, vartype, init); |
| } |
| |
| public JCTree visitWhileLoop(WhileLoopTree node, P p) { |
| JCWhileLoop t = (JCWhileLoop) node; |
| JCStatement body = copy(t.body, p); |
| JCExpression cond = copy(t.cond, p); |
| return M.at(t.pos).WhileLoop(cond, body); |
| } |
| |
| public JCTree visitWildcard(WildcardTree node, P p) { |
| JCWildcard t = (JCWildcard) node; |
| TypeBoundKind kind = M.at(t.kind.pos).TypeBoundKind(t.kind.kind); |
| JCTree inner = copy(t.inner, p); |
| return M.at(t.pos).Wildcard(kind, inner); |
| } |
| |
| public JCTree visitOther(Tree node, P p) { |
| JCTree tree = (JCTree) node; |
| switch (tree.getTag()) { |
| case LETEXPR: { |
| LetExpr t = (LetExpr) node; |
| List<JCVariableDecl> defs = copy(t.defs, p); |
| JCTree expr = copy(t.expr, p); |
| return M.at(t.pos).LetExpr(defs, expr); |
| } |
| default: |
| throw new AssertionError("unknown tree tag: " + tree.getTag()); |
| } |
| } |
| |
| } |