2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.yangtools.yang.parser.builder.impl;
10 import com.google.common.base.Preconditions;
11 import com.google.common.collect.ImmutableList;
12 import com.google.common.collect.ImmutableSet;
13 import java.util.ArrayList;
14 import java.util.HashSet;
15 import java.util.List;
17 import java.util.TreeSet;
18 import org.opendaylight.yangtools.yang.common.QName;
19 import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
20 import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;
21 import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
22 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
23 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
24 import org.opendaylight.yangtools.yang.parser.builder.api.AugmentationSchemaBuilder;
25 import org.opendaylight.yangtools.yang.parser.builder.api.AugmentationTargetBuilder;
26 import org.opendaylight.yangtools.yang.parser.builder.api.ConstraintsBuilder;
27 import org.opendaylight.yangtools.yang.parser.builder.api.DataSchemaNodeBuilder;
28 import org.opendaylight.yangtools.yang.parser.builder.api.SchemaNodeBuilder;
29 import org.opendaylight.yangtools.yang.parser.builder.api.UnknownSchemaNodeBuilder;
30 import org.opendaylight.yangtools.yang.parser.builder.util.AbstractSchemaNodeBuilder;
31 import org.opendaylight.yangtools.yang.parser.builder.util.Comparators;
32 import org.opendaylight.yangtools.yang.parser.util.YangParseException;
34 public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements DataSchemaNodeBuilder,
35 AugmentationTargetBuilder {
36 private ChoiceNodeImpl instance;
38 // DataSchemaNode args
39 private boolean augmenting;
40 private boolean addedByUses;
41 private boolean configuration;
42 private ChoiceSchemaNode originalNode;
43 private ChoiceBuilder originalBuilder;
44 private final ConstraintsBuilder constraints;
45 // AugmentationTarget args
46 private final Set<AugmentationSchema> augmentations = new HashSet<>();
47 private final List<AugmentationSchemaBuilder> augmentationBuilders = new ArrayList<>();
49 private final Set<ChoiceCaseBuilder> caseBuilders = new HashSet<>();
50 private String defaultCase;
52 public ChoiceBuilder(final String moduleName, final int line, final QName qname, final SchemaPath path) {
53 super(moduleName, line, qname);
54 this.schemaPath = path;
55 constraints = new ConstraintsBuilderImpl(moduleName, line);
58 public ChoiceBuilder(final String moduleName, final int line, final QName qname, final SchemaPath path,
59 final ChoiceSchemaNode base) {
60 super(moduleName, line, qname);
61 this.schemaPath = path;
62 constraints = new ConstraintsBuilderImpl(moduleName, line, base.getConstraints());
64 description = base.getDescription();
65 reference = base.getReference();
66 status = base.getStatus();
67 augmenting = base.isAugmenting();
68 addedByUses = base.isAddedByUses();
70 configuration = base.isConfiguration();
71 augmentations.addAll(base.getAvailableAugmentations());
73 Set<DataSchemaNodeBuilder> wrapped = BuilderUtils.wrapChildNodes(moduleName, line, new HashSet<DataSchemaNode>(
74 base.getCases()), path, qname);
75 for (DataSchemaNodeBuilder wrap : wrapped) {
76 if (wrap instanceof ChoiceCaseBuilder) {
77 caseBuilders.add((ChoiceCaseBuilder) wrap);
81 addedUnknownNodes.addAll(BuilderUtils.wrapUnknownNodes(moduleName, line, base.getUnknownSchemaNodes(), path,
86 public ChoiceSchemaNode build() {
87 if (instance != null) {
91 instance = new ChoiceNodeImpl(qname, schemaPath);
93 instance.description = description;
94 instance.reference = reference;
95 instance.status = status;
96 instance.augmenting = augmenting;
97 instance.addedByUses = addedByUses;
98 instance.configuration = configuration;
100 instance.constraints = constraints.toInstance();
101 instance.defaultCase = defaultCase;
104 if (originalNode == null && originalBuilder != null) {
105 originalNode = originalBuilder.build();
107 instance.original = originalNode;
110 final Set<ChoiceCaseNode> cases = new TreeSet<>(Comparators.SCHEMA_NODE_COMP);
111 for (ChoiceCaseBuilder caseBuilder : caseBuilders) {
112 cases.add(caseBuilder.build());
114 instance.cases = ImmutableSet.copyOf(cases);
117 for (AugmentationSchemaBuilder builder : augmentationBuilders) {
118 augmentations.add(builder.build());
120 instance.augmentations = ImmutableSet.copyOf(augmentations);
123 for (UnknownSchemaNodeBuilder b : addedUnknownNodes) {
124 unknownNodes.add(b.build());
126 instance.unknownNodes = ImmutableList.copyOf(unknownNodes);
131 public Set<ChoiceCaseBuilder> getCases() {
139 * name of case to search
140 * @return case with given name if present, null otherwise
142 public ChoiceCaseBuilder getCaseNodeByName(final String caseName) {
143 for (ChoiceCaseBuilder addedCase : caseBuilders) {
144 if (addedCase.getQName().getLocalName().equals(caseName)) {
152 * Add case node to this choice.
154 * If node is not declared with 'case' keyword, create new case builder and
155 * make this node child of newly created case.
160 public void addCase(final DataSchemaNodeBuilder caseNode) {
161 QName caseQName = caseNode.getQName();
162 String caseName = caseQName.getLocalName();
164 for (ChoiceCaseBuilder existingCase : caseBuilders) {
165 if (existingCase.getQName().getLocalName().equals(caseName)) {
166 throw new YangParseException(caseNode.getModuleName(), caseNode.getLine(), "Can not add '" + caseNode
167 + "' to node '" + qname.getLocalName() + "' in module '" + getModuleName()
168 + "': case with same name already declared at line " + existingCase.getLine());
172 if (caseNode instanceof ChoiceCaseBuilder) {
173 caseBuilders.add((ChoiceCaseBuilder) caseNode);
175 ChoiceCaseBuilder caseBuilder = new ChoiceCaseBuilder(caseNode.getModuleName(), caseNode.getLine(),
176 caseQName, caseNode.getPath());
177 if (caseNode.isAugmenting()) {
178 // if node is added by augmentation, set case builder augmenting
179 // as true and node augmenting as false
180 caseBuilder.setAugmenting(true);
181 caseNode.setAugmenting(false);
183 SchemaPath newPath = caseNode.getPath().createChild(caseQName);
184 caseNode.setPath(newPath);
185 caseBuilder.addChildNode(caseNode);
186 caseBuilders.add(caseBuilder);
191 public boolean isAugmenting() {
196 public void setAugmenting(final boolean augmenting) {
197 this.augmenting = augmenting;
201 public boolean isAddedByUses() {
206 public void setAddedByUses(final boolean addedByUses) {
207 this.addedByUses = addedByUses;
211 public ChoiceBuilder getOriginal() {
212 return originalBuilder;
216 public void setOriginal(final SchemaNodeBuilder builder) {
217 Preconditions.checkArgument(builder instanceof ChoiceBuilder, "Original of choice cannot be " + builder);
218 this.originalBuilder = (ChoiceBuilder) builder;
222 public boolean isConfiguration() {
223 return configuration;
227 public void setConfiguration(final boolean configuration) {
228 this.configuration = configuration;
232 public ConstraintsBuilder getConstraints() {
237 public void addAugmentation(final AugmentationSchemaBuilder augment) {
238 augmentationBuilders.add(augment);
241 public List<AugmentationSchemaBuilder> getAugmentationBuilders() {
242 return augmentationBuilders;
245 public String getDefaultCase() {
249 public void setDefaultCase(final String defaultCase) {
250 this.defaultCase = defaultCase;
254 public int hashCode() {
255 final int prime = 31;
257 result = prime * result + ((schemaPath == null) ? 0 : schemaPath.hashCode());
262 public boolean equals(final Object obj) {
269 if (getClass() != obj.getClass()) {
272 ChoiceBuilder other = (ChoiceBuilder) obj;
273 if (schemaPath == null) {
274 if (other.schemaPath != null) {
277 } else if (!schemaPath.equals(other.schemaPath)) {
280 if (getParent() == null) {
281 if (other.getParent() != null) {
284 } else if (!getParent().equals(other.getParent())) {
291 public String toString() {
292 return "choice " + qname.getLocalName();