-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathArrayCreator.java
More file actions
114 lines (90 loc) · 3.67 KB
/
Copy pathArrayCreator.java
File metadata and controls
114 lines (90 loc) · 3.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package jtree.nodes;
import static jtree.util.Utils.*;
import java.util.List;
import java.util.Objects;
import java.util.function.Consumer;
import jtree.util.Either;
import jtree.util.Utils;
import lombok.EqualsAndHashCode;
import lombok.Getter;
import lombok.Setter;
import lombok.NonNull;
@EqualsAndHashCode
@Getter @Setter
public class ArrayCreator extends Node implements Expression, Dimensioned {
protected @NonNull Type baseType;
protected @NonNull Either<? extends List<Size>, ArrayInitializer<? extends Initializer>> sizesOrInitializer;
private @NonNull List<Dimension> dimensions;
public ArrayCreator(Type baseType, @NonNull List<Size> sizes) {
this(baseType, Either.first(sizes));
}
public ArrayCreator(Type baseType, Size... sizes) {
this(baseType, List.of(sizes));
}
public ArrayCreator(Type baseType, @NonNull ArrayInitializer<? extends Initializer> initializer) {
this(baseType, Either.second(initializer));
}
public ArrayCreator(Type baseType, @NonNull Initializer... initializers) {
this(baseType, Either.second(new ArrayInitializer<>(initializers)));
}
public ArrayCreator(Type baseType, List<Size> sizes, List<Dimension> dimensions) {
this(baseType, Either.first(sizes), dimensions);
}
public ArrayCreator(Type baseType, ArrayInitializer<? extends Initializer> initializer, List<Dimension> dimensions) {
this(baseType, Either.second(initializer), dimensions);
}
public ArrayCreator(Type baseType, Either<? extends List<Size>, ArrayInitializer<? extends Initializer>> sizesOrInitializer) {
this(baseType, sizesOrInitializer, List.of(new Dimension()));
}
public ArrayCreator(Type baseType, Either<? extends List<Size>, ArrayInitializer<? extends Initializer>> sizesOrInitializer, List<Dimension> dimensions) {
setBaseType(baseType);
setSizesOrInitializer(sizesOrInitializer);
setDimensions(dimensions);
}
@Override
public Precedence precedence() {
return Precedence.PRIMARY;
}
@Override
public ArrayCreator clone() {
return new ArrayCreator(getBaseType().clone(), clone(getSizesOrInitializer()), clone(getDimensions()));
}
@Override
public String toCode() {
return "new " + getBaseType().toCode() + getSizesOrInitializer().unravel(
sizes -> joinNodes("", sizes) + dimensionString(),
initializer -> dimensionString() + " " + initializer.toCode()
);
}
public void setSizesOrInitializer(@NonNull Either<? extends List<Size>, ArrayInitializer<? extends Initializer>> sizesOrInitializer) {
this.sizesOrInitializer = sizesOrInitializer.map(Utils::newList, Objects::requireNonNull);
}
public final void setSizes(@NonNull List<Size> sizes) {
setSizesOrInitializer(Either.first(sizes));
}
public final void setSizes(Size... sizes) {
setSizes(List.of(sizes));
}
public final void setInitializer(@NonNull ArrayInitializer<? extends Initializer> initializer) {
setSizesOrInitializer(Either.second(initializer));
}
public final void setInitializer(Initializer... initializers) {
setInitializer(new ArrayInitializer<>(initializers));
}
@Override
public void setDimensions(@NonNull List<Dimension> dimensions) {
this.dimensions = newList(dimensions);
}
@Override
public final void setDimensions(Dimension... dimensions) {
setDimensions(List.of(dimensions));
}
@Override
public <N extends INode> void accept(TreeVisitor visitor, Node parent, Consumer<N> replacer) {
if(visitor.visitArrayCreator(this, parent, cast(replacer))) {
getBaseType().accept(visitor, this, this::setBaseType);
getSizesOrInitializer().accept(sizes -> visitList(visitor, sizes), initializer -> initializer.<ArrayInitializer<? extends Initializer>>accept(visitor, this, this::setInitializer));
visitList(visitor, getDimensions());
}
}
}