Populate data/ hierarchy
[yangtools.git] / model / yang-model-ri / src / main / java / org / opendaylight / yangtools / yang / model / ri / stmt / impl / eff / AbstractLeafEffectiveStatement.java
1 /*
2  * Copyright (c) 2020 PANTHEON.tech, s.r.o. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.yangtools.yang.model.ri.stmt.impl.eff;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.collect.ImmutableList;
13 import org.eclipse.jdt.annotation.NonNull;
14 import org.opendaylight.yangtools.concepts.Immutable;
15 import org.opendaylight.yangtools.yang.common.QName;
16 import org.opendaylight.yangtools.yang.model.api.DerivableSchemaNode;
17 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
18 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
19 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
20 import org.opendaylight.yangtools.yang.model.api.stmt.DefaultEffectiveStatement;
21 import org.opendaylight.yangtools.yang.model.api.stmt.DescriptionEffectiveStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.LeafEffectiveStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.LeafStatement;
24 import org.opendaylight.yangtools.yang.model.api.stmt.ReferenceEffectiveStatement;
25 import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
26 import org.opendaylight.yangtools.yang.model.api.stmt.TypeEffectiveStatement;
27 import org.opendaylight.yangtools.yang.model.api.stmt.UnitsEffectiveStatement;
28 import org.opendaylight.yangtools.yang.model.ri.type.ConcreteTypeBuilder;
29 import org.opendaylight.yangtools.yang.model.ri.type.ConcreteTypes;
30 import org.opendaylight.yangtools.yang.model.spi.meta.AbstractDeclaredEffectiveStatement;
31 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.DataSchemaNodeMixin;
32 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.MandatoryMixin;
33 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.MustConstraintMixin;
34
35 public abstract class AbstractLeafEffectiveStatement
36         extends AbstractDeclaredEffectiveStatement.Default<QName, LeafStatement>
37         implements LeafEffectiveStatement, LeafSchemaNode, DerivableSchemaNode,
38             DataSchemaNodeMixin<QName, LeafStatement>, MandatoryMixin<QName, LeafStatement>,
39             MustConstraintMixin<QName, LeafStatement> {
40     private final @NonNull Object substatements;
41     private final @NonNull Immutable path;
42     private final @NonNull TypeDefinition<?> type;
43     private final int flags;
44
45     AbstractLeafEffectiveStatement(final LeafStatement declared, final Immutable path, final int flags,
46             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
47         super(declared);
48         this.path = requireNonNull(path);
49         this.substatements = maskList(substatements);
50         this.flags = flags;
51         // TODO: lazy instantiation?
52         this.type = buildType();
53     }
54
55     AbstractLeafEffectiveStatement(final AbstractLeafEffectiveStatement original, final Immutable path,
56             final int flags) {
57         super(original);
58         this.path = requireNonNull(path);
59         this.substatements = original.substatements;
60         this.flags = flags;
61         // FIXME: share with original?
62         this.type = buildType();
63     }
64
65     @Override
66     public final ImmutableList<? extends EffectiveStatement<?, ?>> effectiveSubstatements() {
67         return unmaskList(substatements);
68     }
69
70     @Override
71     public final int flags() {
72         return flags;
73     }
74
75     @Override
76     public final QName argument() {
77         return getQName();
78     }
79
80     @Override
81     public final Immutable pathObject() {
82         return path;
83     }
84
85     @Override
86     public final TypeDefinition<?> getType() {
87         return type;
88     }
89
90     @Override
91     public final LeafEffectiveStatement asEffectiveStatement() {
92         return this;
93     }
94
95     private TypeDefinition<?> buildType() {
96         final TypeEffectiveStatement<?> typeStmt = findFirstEffectiveSubstatement(TypeEffectiveStatement.class).get();
97         final ConcreteTypeBuilder<?> builder = ConcreteTypes.concreteTypeBuilder(typeStmt.getTypeDefinition(),
98             getQName());
99         for (final EffectiveStatement<?, ?> stmt : effectiveSubstatements()) {
100             if (stmt instanceof DefaultEffectiveStatement) {
101                 builder.setDefaultValue(((DefaultEffectiveStatement)stmt).argument());
102             } else if (stmt instanceof DescriptionEffectiveStatement) {
103                 builder.setDescription(((DescriptionEffectiveStatement)stmt).argument());
104             } else if (stmt instanceof ReferenceEffectiveStatement) {
105                 builder.setReference(((ReferenceEffectiveStatement)stmt).argument());
106             } else if (stmt instanceof StatusEffectiveStatement) {
107                 builder.setStatus(((StatusEffectiveStatement)stmt).argument());
108             } else if (stmt instanceof UnitsEffectiveStatement) {
109                 builder.setUnits(((UnitsEffectiveStatement)stmt).argument());
110             }
111         }
112         return builder.build();
113     }
114 }