71bdc4acb8726b626d82e0abdad946cec3e5ca8a
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / uses / UsesEffectiveStatementImpl.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.rfc7950.stmt.uses;
9
10 import com.google.common.collect.ImmutableList;
11 import com.google.common.collect.ImmutableMap;
12 import com.google.common.collect.ImmutableSet;
13 import java.util.ArrayList;
14 import java.util.HashMap;
15 import java.util.LinkedHashSet;
16 import java.util.List;
17 import java.util.Map;
18 import java.util.Objects;
19 import java.util.Optional;
20 import java.util.Set;
21 import javax.annotation.Nonnull;
22 import org.opendaylight.yangtools.yang.common.QName;
23 import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
24 import org.opendaylight.yangtools.yang.model.api.RevisionAwareXPath;
25 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
26 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
27 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
28 import org.opendaylight.yangtools.yang.model.api.UsesNode;
29 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
30 import org.opendaylight.yangtools.yang.model.api.stmt.GroupingStatement;
31 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
32 import org.opendaylight.yangtools.yang.model.api.stmt.UsesEffectiveStatement;
33 import org.opendaylight.yangtools.yang.model.api.stmt.UsesStatement;
34 import org.opendaylight.yangtools.yang.model.api.stmt.WhenEffectiveStatement;
35 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractEffectiveDocumentedNode;
36 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.refine.RefineEffectiveStatementImpl;
37 import org.opendaylight.yangtools.yang.parser.spi.GroupingNamespace;
38 import org.opendaylight.yangtools.yang.parser.spi.meta.CopyType;
39 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
40
41 final class UsesEffectiveStatementImpl extends AbstractEffectiveDocumentedNode<QName, UsesStatement>
42         implements UsesEffectiveStatement, UsesNode {
43     private final SchemaPath groupingPath;
44     private final boolean addedByUses;
45     private final Map<SchemaPath, SchemaNode> refines;
46     private final Set<AugmentationSchemaNode> augmentations;
47     private final List<UnknownSchemaNode> unknownNodes;
48     private final RevisionAwareXPath whenCondition;
49
50     UsesEffectiveStatementImpl(final StmtContext<QName, UsesStatement, EffectiveStatement<QName, UsesStatement>> ctx) {
51         super(ctx);
52
53         // initGroupingPath
54         final StmtContext<?, GroupingStatement, EffectiveStatement<QName, GroupingStatement>> grpCtx =
55                 ctx.getFromNamespace(GroupingNamespace.class, ctx.getStatementArgument());
56         this.groupingPath = grpCtx.getSchemaPath().get();
57
58         // initCopyType
59         addedByUses = ctx.getCopyHistory().contains(CopyType.ADDED_BY_USES);
60
61         // initSubstatementCollections
62         final List<UnknownSchemaNode> unknownNodesInit = new ArrayList<>();
63         final Set<AugmentationSchemaNode> augmentationsInit = new LinkedHashSet<>();
64         final Map<SchemaPath, SchemaNode> refinesInit = new HashMap<>();
65         for (final EffectiveStatement<?, ?> effectiveStatement : effectiveSubstatements()) {
66             if (effectiveStatement instanceof UnknownSchemaNode) {
67                 final UnknownSchemaNode unknownNode = (UnknownSchemaNode) effectiveStatement;
68                 unknownNodesInit.add(unknownNode);
69             }
70             if (effectiveStatement instanceof AugmentationSchemaNode) {
71                 final AugmentationSchemaNode augmentationSchema = (AugmentationSchemaNode) effectiveStatement;
72                 augmentationsInit.add(augmentationSchema);
73             }
74             if (effectiveStatement instanceof RefineEffectiveStatementImpl) {
75                 final RefineEffectiveStatementImpl refineStmt = (RefineEffectiveStatementImpl) effectiveStatement;
76                 final SchemaNodeIdentifier identifier = refineStmt.argument();
77                 refinesInit.put(identifier.asSchemaPath(), refineStmt.getRefineTargetNode());
78             }
79         }
80         this.unknownNodes = ImmutableList.copyOf(unknownNodesInit);
81         this.augmentations = ImmutableSet.copyOf(augmentationsInit);
82         this.refines = ImmutableMap.copyOf(refinesInit);
83
84         whenCondition = findFirstEffectiveSubstatementArgument(WhenEffectiveStatement.class).orElse(null);
85     }
86
87     @Nonnull
88     @Override
89     public SchemaPath getGroupingPath() {
90         return groupingPath;
91     }
92
93     @Nonnull
94     @Override
95     public Set<AugmentationSchemaNode> getAugmentations() {
96         return augmentations;
97     }
98
99     @Override
100     public boolean isAugmenting() {
101         return false;
102     }
103
104     @Override
105     public boolean isAddedByUses() {
106         return addedByUses;
107     }
108
109     @Nonnull
110     @Override
111     public Map<SchemaPath, SchemaNode> getRefines() {
112         return refines;
113     }
114
115     @Nonnull
116     @Override
117     public List<UnknownSchemaNode> getUnknownSchemaNodes() {
118         return unknownNodes;
119     }
120
121     @Nonnull
122     @Override
123     public Optional<RevisionAwareXPath> getWhenCondition() {
124         return Optional.ofNullable(whenCondition);
125     }
126
127     @Override
128     public int hashCode() {
129         final int prime = 31;
130         int result = 1;
131         result = prime * result + Objects.hashCode(groupingPath);
132         result = prime * result + Objects.hashCode(augmentations);
133         return result;
134     }
135
136     @Override
137     public boolean equals(final Object obj) {
138         if (this == obj) {
139             return true;
140         }
141         if (obj == null) {
142             return false;
143         }
144         if (getClass() != obj.getClass()) {
145             return false;
146         }
147         final UsesEffectiveStatementImpl other = (UsesEffectiveStatementImpl) obj;
148         return Objects.equals(groupingPath, other.groupingPath) && Objects.equals(augmentations, other.augmentations);
149     }
150
151     @Override
152     public String toString() {
153         return UsesEffectiveStatementImpl.class.getSimpleName() + "[groupingPath=" + groupingPath + "]";
154     }
155 }