src/main/java/org/zwobble/couscous/ast/ArrayNode.java


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

package org.zwobble.couscous.ast;

import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper;
import org.zwobble.couscous.ast.visitors.NodeTransformer;
import org.zwobble.couscous.types.Type;
import org.zwobble.couscous.types.Types;

import java.util.List;

public class ArrayNode implements ExpressionNode {
    public static ArrayNode array(Type elementType, List<ExpressionNode> expressions) {
        return new ArrayNode(elementType, expressions);
    }

    private final Type elementType;
    private final List<ExpressionNode> elements;

    public ArrayNode(Type elementType, List<ExpressionNode> elements) {
        this.elementType = elementType;
        this.elements = elements;
    }

    public Type getElementType() {
        return elementType;
    }

    public List<ExpressionNode> getElements() {
        return elements;
    }

    @Override
    public Type getType() {
        return Types.array(elementType);
    }

    @Override
    public <T> T accept(ExpressionNodeMapper<T> visitor) {
        return visitor.visit(this);
    }

    @Override
    public ExpressionNode transform(NodeTransformer transformer) {
        return new ArrayNode(
            transformer.transform(elementType),
            transformer.transformExpressions(elements));
    }

    @Override
    public String toString() {
        return "ArrayNode(" +
            "elementType=" + elementType +
            ", elements=" + elements +
            ')';
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        ArrayNode arrayNode = (ArrayNode) o;

        if (!elementType.equals(arrayNode.elementType)) return false;
        return elements.equals(arrayNode.elements);

    }

    @Override
    public int hashCode() {
        int result = elementType.hashCode();
        result = 31 * result + elements.hashCode();
        return result;
    }
}