Preserve augmentation order in module.
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / effective / AbstractEffectiveSimpleDataNodeContainer.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. 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.parser.stmt.rfc6020.effective;
9
10 import com.google.common.collect.ImmutableList;
11 import com.google.common.collect.ImmutableSet;
12 import java.util.ArrayList;
13 import java.util.HashSet;
14 import java.util.List;
15 import java.util.Set;
16 import org.opendaylight.yangtools.yang.common.QName;
17 import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
18 import org.opendaylight.yangtools.yang.model.api.AugmentationTarget;
19 import org.opendaylight.yangtools.yang.model.api.ConstraintDefinition;
20 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
21 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
22 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
23 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
24 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
25 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
27 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.TypeOfCopy;
28
29 abstract class AbstractEffectiveSimpleDataNodeContainer<D extends DeclaredStatement<QName>> extends
30         AbstractEffectiveDocumentedDataNodeContainer<QName, D> implements DataNodeContainer, AugmentationTarget,
31         DataSchemaNode {
32
33     private final SchemaPath path;
34
35     // :FIXME should be private and final
36     boolean augmenting;
37     private final boolean addedByUses;
38     private final boolean configuration;
39     private final ConstraintDefinition constraints;
40
41     private final Set<AugmentationSchema> augmentations;
42     private final List<UnknownSchemaNode> unknownNodes;
43
44     public AbstractEffectiveSimpleDataNodeContainer(final StmtContext<QName, D, ?> ctx) {
45         super(ctx);
46
47         this.path = ctx.getSchemaPath().get();
48         this.constraints = EffectiveConstraintDefinitionImpl.forParent(this);
49
50         ConfigEffectiveStatementImpl configStmt = firstEffective(ConfigEffectiveStatementImpl.class);
51         this.configuration = (configStmt == null) ? true : configStmt.argument();
52
53         // initSubstatementCollectionsAndFields
54
55         List<UnknownSchemaNode> unknownNodesInit = new ArrayList<>();
56         Set<AugmentationSchema> augmentationsInit = new HashSet<>();
57         for (EffectiveStatement<?, ?> effectiveStatement : effectiveSubstatements()) {
58             if (effectiveStatement instanceof UnknownSchemaNode) {
59                 unknownNodesInit.add((UnknownSchemaNode) effectiveStatement);
60             }
61             if (effectiveStatement instanceof AugmentationSchema) {
62                 augmentationsInit.add((AugmentationSchema) effectiveStatement);
63             }
64         }
65         this.unknownNodes = ImmutableList.copyOf(unknownNodesInit);
66         this.augmentations = ImmutableSet.copyOf(augmentationsInit);
67
68         // initCopyType
69         List<TypeOfCopy> copyTypesFromOriginal = ctx.getCopyHistory();
70         if (copyTypesFromOriginal.contains(TypeOfCopy.ADDED_BY_USES_AUGMENTATION)) {
71             this.addedByUses = this.augmenting = true;
72         } else {
73             this.augmenting = copyTypesFromOriginal.contains(TypeOfCopy.ADDED_BY_AUGMENTATION);
74             this.addedByUses = copyTypesFromOriginal.contains(TypeOfCopy.ADDED_BY_USES);
75         }
76     }
77
78     @Override
79     public QName getQName() {
80         return path.getLastComponent();
81     }
82
83     @Override
84     public SchemaPath getPath() {
85         return path;
86     }
87
88     @Override
89     public boolean isAugmenting() {
90         return augmenting;
91     }
92
93     @Override
94     public boolean isAddedByUses() {
95         return addedByUses;
96     }
97
98     @Override
99     public boolean isConfiguration() {
100         return configuration;
101     }
102
103     @Override
104     public ConstraintDefinition getConstraints() {
105         return constraints;
106     }
107
108     @Override
109     public Set<AugmentationSchema> getAvailableAugmentations() {
110         return augmentations;
111     }
112
113     @Override
114     public List<UnknownSchemaNode> getUnknownSchemaNodes() {
115         return unknownNodes;
116     }
117
118 }