diff options
Diffstat (limited to 'javaparser-core/src/main/java/com/github/javaparser/ast/expr/ArrayCreationExpr.java')
| -rw-r--r-- | javaparser-core/src/main/java/com/github/javaparser/ast/expr/ArrayCreationExpr.java | 281 |
1 files changed, 281 insertions, 0 deletions
diff --git a/javaparser-core/src/main/java/com/github/javaparser/ast/expr/ArrayCreationExpr.java b/javaparser-core/src/main/java/com/github/javaparser/ast/expr/ArrayCreationExpr.java new file mode 100644 index 000000000..509925803 --- /dev/null +++ b/javaparser-core/src/main/java/com/github/javaparser/ast/expr/ArrayCreationExpr.java @@ -0,0 +1,281 @@ +/* + * 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.expr; + +import com.github.javaparser.Range; +import com.github.javaparser.ast.*; +import com.github.javaparser.ast.observer.ObservableProperty; +import com.github.javaparser.ast.type.ArrayType; +import com.github.javaparser.ast.type.ClassOrInterfaceType; +import com.github.javaparser.ast.type.Type; +import com.github.javaparser.ast.visitor.CloneVisitor; +import com.github.javaparser.ast.visitor.GenericVisitor; +import com.github.javaparser.ast.visitor.VoidVisitor; +import com.github.javaparser.metamodel.ArrayCreationExprMetaModel; +import com.github.javaparser.metamodel.JavaParserMetaModel; +import com.github.javaparser.metamodel.NonEmptyProperty; +import java.util.Arrays; +import java.util.List; +import java.util.Optional; +import static com.github.javaparser.JavaParser.parseType; +import static com.github.javaparser.utils.Utils.assertNotNull; +import javax.annotation.Generated; +import com.github.javaparser.TokenRange; +import com.github.javaparser.ast.Node; +import com.github.javaparser.metamodel.OptionalProperty; +import java.util.function.Consumer; + +/** + * <code>new int[5][4][][]</code> or <code>new int[][]{{1},{2,3}}</code>. + * + * <br/>"int" is the element type. + * <br/>All the brackets are stored in the levels field, from left to right. + * + * @author Julio Vilmar Gesser + */ +public final class ArrayCreationExpr extends Expression { + + @NonEmptyProperty + private NodeList<ArrayCreationLevel> levels; + + private Type elementType; + + @OptionalProperty + private ArrayInitializerExpr initializer; + + public ArrayCreationExpr() { + this(null, new ClassOrInterfaceType(), new NodeList<>(), new ArrayInitializerExpr()); + } + + @AllFieldsConstructor + public ArrayCreationExpr(Type elementType, NodeList<ArrayCreationLevel> levels, ArrayInitializerExpr initializer) { + this(null, elementType, levels, initializer); + } + + public ArrayCreationExpr(Type elementType) { + this(null, elementType, new NodeList<>(), new ArrayInitializerExpr()); + } + + /** + * @deprecated range shouldn't be in utility constructors. + */ + @Deprecated + public ArrayCreationExpr(Range range, Type elementType) { + this(null, elementType, new NodeList<>(), new ArrayInitializerExpr()); + setRange(range); + } + + /** + * This constructor is used by the parser and is considered private. + */ + @Generated("com.github.javaparser.generator.core.node.MainConstructorGenerator") + public ArrayCreationExpr(TokenRange tokenRange, Type elementType, NodeList<ArrayCreationLevel> levels, ArrayInitializerExpr initializer) { + super(tokenRange); + setElementType(elementType); + setLevels(levels); + setInitializer(initializer); + customInitialization(); + } + + @Override + @Generated("com.github.javaparser.generator.core.node.AcceptGenerator") + public <R, A> R accept(final GenericVisitor<R, A> v, final A arg) { + return v.visit(this, arg); + } + + @Override + @Generated("com.github.javaparser.generator.core.node.AcceptGenerator") + public <A> void accept(final VoidVisitor<A> v, final A arg) { + v.visit(this, arg); + } + + @Generated("com.github.javaparser.generator.core.node.PropertyGenerator") + public Optional<ArrayInitializerExpr> getInitializer() { + return Optional.ofNullable(initializer); + } + + @Generated("com.github.javaparser.generator.core.node.PropertyGenerator") + public Type getElementType() { + return elementType; + } + + /** + * Sets the initializer + * + * @param initializer the initializer, can be null + * @return this, the ArrayCreationExpr + */ + @Generated("com.github.javaparser.generator.core.node.PropertyGenerator") + public ArrayCreationExpr setInitializer(final ArrayInitializerExpr initializer) { + if (initializer == this.initializer) { + return (ArrayCreationExpr) this; + } + notifyPropertyChange(ObservableProperty.INITIALIZER, this.initializer, initializer); + if (this.initializer != null) + this.initializer.setParentNode(null); + this.initializer = initializer; + setAsParentNodeOf(initializer); + return this; + } + + @Generated("com.github.javaparser.generator.core.node.PropertyGenerator") + public ArrayCreationExpr setElementType(final Type elementType) { + assertNotNull(elementType); + if (elementType == this.elementType) { + return (ArrayCreationExpr) this; + } + notifyPropertyChange(ObservableProperty.ELEMENT_TYPE, this.elementType, elementType); + if (this.elementType != null) + this.elementType.setParentNode(null); + this.elementType = elementType; + setAsParentNodeOf(elementType); + return this; + } + + @Generated("com.github.javaparser.generator.core.node.PropertyGenerator") + public NodeList<ArrayCreationLevel> getLevels() { + return levels; + } + + @Generated("com.github.javaparser.generator.core.node.PropertyGenerator") + public ArrayCreationExpr setLevels(final NodeList<ArrayCreationLevel> levels) { + assertNotNull(levels); + if (levels == this.levels) { + return (ArrayCreationExpr) this; + } + notifyPropertyChange(ObservableProperty.LEVELS, this.levels, levels); + if (this.levels != null) + this.levels.setParentNode(null); + this.levels = levels; + setAsParentNodeOf(levels); + return this; + } + + /** + * Takes the element type and wraps it in an ArrayType for every array creation level. + */ + public Type createdType() { + Type result = elementType; + for (int i = 0; i < levels.size(); i++) { + result = new ArrayType(result, ArrayType.Origin.TYPE, new NodeList<>()); + } + return result; + } + + /** + * Sets this type to this class and try to import it to the {@link CompilationUnit} if needed + * + * @param typeClass the type + * @return this + */ + public ArrayCreationExpr setElementType(Class<?> typeClass) { + tryAddImportToParentCompilationUnit(typeClass); + return setElementType(parseType(typeClass.getSimpleName())); + } + + public ArrayCreationExpr setElementType(final String type) { + return setElementType(parseType(type)); + } + + @Override + @Generated("com.github.javaparser.generator.core.node.RemoveMethodGenerator") + public boolean remove(Node node) { + if (node == null) + return false; + if (initializer != null) { + if (node == initializer) { + removeInitializer(); + return true; + } + } + for (int i = 0; i < levels.size(); i++) { + if (levels.get(i) == node) { + levels.remove(i); + return true; + } + } + return super.remove(node); + } + + @Generated("com.github.javaparser.generator.core.node.RemoveMethodGenerator") + public ArrayCreationExpr removeInitializer() { + return setInitializer((ArrayInitializerExpr) null); + } + + @Override + @Generated("com.github.javaparser.generator.core.node.CloneGenerator") + public ArrayCreationExpr clone() { + return (ArrayCreationExpr) accept(new CloneVisitor(), null); + } + + @Override + @Generated("com.github.javaparser.generator.core.node.GetMetaModelGenerator") + public ArrayCreationExprMetaModel getMetaModel() { + return JavaParserMetaModel.arrayCreationExprMetaModel; + } + + @Override + @Generated("com.github.javaparser.generator.core.node.ReplaceMethodGenerator") + public boolean replace(Node node, Node replacementNode) { + if (node == null) + return false; + if (node == elementType) { + setElementType((Type) replacementNode); + return true; + } + if (initializer != null) { + if (node == initializer) { + setInitializer((ArrayInitializerExpr) replacementNode); + return true; + } + } + for (int i = 0; i < levels.size(); i++) { + if (levels.get(i) == node) { + levels.set(i, (ArrayCreationLevel) replacementNode); + return true; + } + } + return super.replace(node, replacementNode); + } + + @Override + @Generated("com.github.javaparser.generator.core.node.TypeCastingGenerator") + public boolean isArrayCreationExpr() { + return true; + } + + @Override + @Generated("com.github.javaparser.generator.core.node.TypeCastingGenerator") + public ArrayCreationExpr asArrayCreationExpr() { + return this; + } + + @Generated("com.github.javaparser.generator.core.node.TypeCastingGenerator") + public void ifArrayCreationExpr(Consumer<ArrayCreationExpr> action) { + action.accept(this); + } + + @Override + @Generated("com.github.javaparser.generator.core.node.TypeCastingGenerator") + public Optional<ArrayCreationExpr> toArrayCreationExpr() { + return Optional.of(this); + } +} |