diff options
Diffstat (limited to 'javaparser-symbol-solver-testing/src/test/test_sourcecode/javaparser_new_src/javaparser-core/com/github/javaparser/ast/body/VariableDeclarator.java')
-rw-r--r-- | javaparser-symbol-solver-testing/src/test/test_sourcecode/javaparser_new_src/javaparser-core/com/github/javaparser/ast/body/VariableDeclarator.java | 147 |
1 files changed, 147 insertions, 0 deletions
diff --git a/javaparser-symbol-solver-testing/src/test/test_sourcecode/javaparser_new_src/javaparser-core/com/github/javaparser/ast/body/VariableDeclarator.java b/javaparser-symbol-solver-testing/src/test/test_sourcecode/javaparser_new_src/javaparser-core/com/github/javaparser/ast/body/VariableDeclarator.java new file mode 100644 index 000000000..ff78d6e68 --- /dev/null +++ b/javaparser-symbol-solver-testing/src/test/test_sourcecode/javaparser_new_src/javaparser-core/com/github/javaparser/ast/body/VariableDeclarator.java @@ -0,0 +1,147 @@ +/* + * Copyright (C) 2007-2010 JĂșlio Vilmar Gesser. + * Copyright (C) 2011, 2013-2016 The JavaParser Team. + * + * This file is part of JavaParser. + * + * JavaParser can be used either under the terms of + * a) the GNU Lesser General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * b) the terms of the Apache License + * + * You should have received a copy of both licenses in LICENCE.LGPL and + * LICENCE.APACHE. Please refer to those files for details. + * + * JavaParser 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 Lesser General Public License for more details. + */ + +package com.github.javaparser.ast.body; + +import com.github.javaparser.Range; +import com.github.javaparser.ast.ArrayBracketPair; +import com.github.javaparser.ast.Node; +import com.github.javaparser.ast.expr.Expression; +import com.github.javaparser.ast.expr.NameExpr; +import com.github.javaparser.ast.nodeTypes.NodeWithElementType; +import com.github.javaparser.ast.nodeTypes.NodeWithType; +import com.github.javaparser.ast.type.ArrayType; +import com.github.javaparser.ast.type.Type; +import com.github.javaparser.ast.visitor.GenericVisitor; +import com.github.javaparser.ast.visitor.VoidVisitor; +import com.github.javaparser.utils.Pair; + +import java.util.List; + +import static com.github.javaparser.ast.type.ArrayType.wrapInArrayTypes; + +/** + * @author Julio Vilmar Gesser + */ +public final class VariableDeclarator extends Node implements + NodeWithType<VariableDeclarator> { + + private VariableDeclaratorId id; + + private Expression init; + + public VariableDeclarator() { + } + + public VariableDeclarator(VariableDeclaratorId id) { + setId(id); + } + + public VariableDeclarator(String variableName) { + setId(new VariableDeclaratorId(variableName)); + } + + /** + * Defines the declaration of a variable. + * + * @param id The identifier for this variable. IE. The variables name. + * @param init What this variable should be initialized to. + * An {@link com.github.javaparser.ast.expr.AssignExpr} is unnecessary as the <code>=</code> operator is + * already added. + */ + public VariableDeclarator(VariableDeclaratorId id, Expression init) { + setId(id); + setInit(init); + } + + public VariableDeclarator(String variableName, Expression init) { + setId(new VariableDeclaratorId(variableName)); + setInit(init); + } + + public VariableDeclarator(Range range, VariableDeclaratorId id, Expression init) { + super(range); + setId(id); + setInit(init); + } + + @Override + public <R, A> R accept(GenericVisitor<R, A> v, A arg) { + return v.visit(this, arg); + } + + @Override + public <A> void accept(VoidVisitor<A> v, A arg) { + v.visit(this, arg); + } + + public VariableDeclaratorId getId() { + return id; + } + + public Expression getInit() { + return init; + } + + public VariableDeclarator setId(VariableDeclaratorId id) { + this.id = id; + setAsParentNodeOf(this.id); + return this; + } + + public VariableDeclarator setInit(Expression init) { + this.init = init; + setAsParentNodeOf(this.init); + return this; + } + + /** + * Will create a {@link NameExpr} with the init param + */ + public VariableDeclarator setInit(String init) { + this.init = new NameExpr(init); + setAsParentNodeOf(this.init); + return this; + } + + + @Override + public Type getType() { + NodeWithElementType<?> elementType = getParentNodeOfType(NodeWithElementType.class); + + return wrapInArrayTypes(elementType.getElementType(), + elementType.getArrayBracketPairsAfterElementType(), + getId().getArrayBracketPairsAfterId()); + } + + @Override + public VariableDeclarator setType(Type type) { + Pair<Type, List<ArrayBracketPair>> unwrapped = ArrayType.unwrapArrayTypes(type); + NodeWithElementType<?> nodeWithElementType = getParentNodeOfType(NodeWithElementType.class); + if (nodeWithElementType == null) { + throw new IllegalStateException("Cannot set type without a parent"); + } + nodeWithElementType.setElementType(unwrapped.a); + nodeWithElementType.setArrayBracketPairsAfterElementType(null); + getId().setArrayBracketPairsAfterId(unwrapped.b); + return this; + } +} |