0f7179e791eeb761cf1b7528e94a96728eff348c
[controller.git] / opendaylight / sal / yang-prototype / code-generator / yang-model-parser-impl / src / main / java / org / opendaylight / controller / yang / parser / builder / impl / GroupingBuilderImpl.java
1 /*
2  * Copyright (c) 2013 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.controller.yang.parser.builder.impl;
9
10 import java.util.ArrayList;
11 import java.util.Collections;
12 import java.util.HashMap;
13 import java.util.HashSet;
14 import java.util.List;
15 import java.util.Map;
16 import java.util.Set;
17 import java.util.TreeSet;
18
19 import org.opendaylight.controller.yang.common.QName;
20 import org.opendaylight.controller.yang.model.api.DataSchemaNode;
21 import org.opendaylight.controller.yang.model.api.GroupingDefinition;
22 import org.opendaylight.controller.yang.model.api.SchemaPath;
23 import org.opendaylight.controller.yang.model.api.Status;
24 import org.opendaylight.controller.yang.model.api.TypeDefinition;
25 import org.opendaylight.controller.yang.model.api.UnknownSchemaNode;
26 import org.opendaylight.controller.yang.model.api.UsesNode;
27 import org.opendaylight.controller.yang.parser.builder.api.Builder;
28 import org.opendaylight.controller.yang.parser.builder.api.DataSchemaNodeBuilder;
29 import org.opendaylight.controller.yang.parser.builder.api.GroupingBuilder;
30 import org.opendaylight.controller.yang.parser.builder.api.TypeDefinitionBuilder;
31 import org.opendaylight.controller.yang.parser.builder.api.UsesNodeBuilder;
32 import org.opendaylight.controller.yang.parser.util.Comparators;
33
34 public final class GroupingBuilderImpl implements GroupingBuilder {
35     private Builder parent;
36     private boolean isBuilt;
37     private final GroupingDefinitionImpl instance;
38     private final int line;
39     private final QName qname;
40     private SchemaPath schemaPath;
41     private String description;
42     private String reference;
43     private Status status = Status.CURRENT;
44     private boolean addedByUses;
45
46     private Set<DataSchemaNode> childNodes;
47     private final Set<DataSchemaNodeBuilder> addedChildNodes = new HashSet<DataSchemaNodeBuilder>();
48
49     private Set<GroupingDefinition> groupings;
50     private final Set<GroupingBuilder> addedGroupings = new HashSet<GroupingBuilder>();
51
52     private Set<TypeDefinition<?>> typedefs;
53     private final Set<TypeDefinitionBuilder> addedTypedefs = new HashSet<TypeDefinitionBuilder>();
54
55     private Set<UsesNode> usesNodes;
56     private final Set<UsesNodeBuilder> addedUsesNodes = new HashSet<UsesNodeBuilder>();
57
58     private List<UnknownSchemaNode> unknownNodes;
59     private final List<UnknownSchemaNodeBuilder> addedUnknownNodes = new ArrayList<UnknownSchemaNodeBuilder>();
60
61     public GroupingBuilderImpl(final QName qname, final int line) {
62         this.qname = qname;
63         instance = new GroupingDefinitionImpl(qname);
64         this.line = line;
65     }
66
67     public GroupingBuilderImpl(GroupingBuilder builder) {
68         qname = builder.getQName();
69         parent = builder.getParent();
70         instance = new GroupingDefinitionImpl(qname);
71         line = builder.getLine();
72         schemaPath = builder.getPath();
73         description = builder.getDescription();
74         reference = builder.getReference();
75         status = builder.getStatus();
76         addedByUses = builder.isAddedByUses();
77         childNodes = builder.getChildNodes();
78         addedChildNodes.addAll(builder.getChildNodeBuilders());
79         groupings = builder.getGroupings();
80         addedGroupings.addAll(builder.getGroupingBuilders());
81         addedUsesNodes.addAll(builder.getUses());
82         addedUnknownNodes.addAll(builder.getUnknownNodes());
83     }
84
85     @Override
86     public GroupingDefinition build() {
87         if (!isBuilt) {
88             instance.setPath(schemaPath);
89             instance.setDescription(description);
90             instance.setReference(reference);
91             instance.setStatus(status);
92             instance.setAddedByUses(addedByUses);
93
94             // CHILD NODES
95             final Map<QName, DataSchemaNode> childs = new HashMap<QName, DataSchemaNode>();
96             if (childNodes == null) {
97                 for (DataSchemaNodeBuilder node : addedChildNodes) {
98                     childs.put(node.getQName(), node.build());
99                 }
100             } else {
101                 for (DataSchemaNode node : childNodes) {
102                     childs.put(node.getQName(), node);
103                 }
104             }
105             instance.setChildNodes(childs);
106
107             // GROUPINGS
108             if (groupings == null) {
109                 groupings = new TreeSet<GroupingDefinition>(Comparators.SCHEMA_NODE_COMP);
110                 for (GroupingBuilder builder : addedGroupings) {
111                     groupings.add(builder.build());
112                 }
113             }
114             instance.setGroupings(groupings);
115
116             // TYPEDEFS
117             if (typedefs == null) {
118                 typedefs = new TreeSet<TypeDefinition<?>>(Comparators.SCHEMA_NODE_COMP);
119                 for (TypeDefinitionBuilder entry : addedTypedefs) {
120                     typedefs.add(entry.build());
121                 }
122             }
123             instance.setTypeDefinitions(typedefs);
124
125             // USES
126             if (usesNodes == null) {
127                 usesNodes = new HashSet<UsesNode>();
128                 for (UsesNodeBuilder builder : addedUsesNodes) {
129                     usesNodes.add(builder.build());
130                 }
131             }
132             instance.setUses(usesNodes);
133
134             // UNKNOWN NODES
135             if (unknownNodes == null) {
136                 unknownNodes = new ArrayList<UnknownSchemaNode>();
137                 for (UnknownSchemaNodeBuilder b : addedUnknownNodes) {
138                     unknownNodes.add(b.build());
139                 }
140                 Collections.sort(unknownNodes, Comparators.SCHEMA_NODE_COMP);
141             }
142             instance.setUnknownSchemaNodes(unknownNodes);
143
144             isBuilt = true;
145         }
146
147         return instance;
148     }
149
150     @Override
151     public int getLine() {
152         return line;
153     }
154
155     @Override
156     public Builder getParent() {
157         return parent;
158     }
159
160     @Override
161     public void setParent(final Builder parent) {
162         this.parent = parent;
163     }
164
165     @Override
166     public QName getQName() {
167         return qname;
168     }
169
170     @Override
171     public Set<TypeDefinitionBuilder> getTypeDefinitionBuilders() {
172         return addedTypedefs;
173     }
174
175     @Override
176     public void addTypedef(final TypeDefinitionBuilder type) {
177         addedTypedefs.add(type);
178     }
179
180     public void setTypedefs(final Set<TypeDefinition<?>> typedefs) {
181         this.typedefs = typedefs;
182     }
183
184     @Override
185     public SchemaPath getPath() {
186         return schemaPath;
187     }
188
189     @Override
190     public void setPath(SchemaPath schemaPath) {
191         this.schemaPath = schemaPath;
192     }
193
194     @Override
195     public String getDescription() {
196         return description;
197     }
198
199     @Override
200     public void setDescription(final String description) {
201         this.description = description;
202     }
203
204     @Override
205     public String getReference() {
206         return reference;
207     }
208
209     @Override
210     public void setReference(final String reference) {
211         this.reference = reference;
212     }
213
214     @Override
215     public Status getStatus() {
216         return status;
217     }
218
219     @Override
220     public void setStatus(final Status status) {
221         this.status = status;
222     }
223
224     @Override
225     public boolean isAddedByUses() {
226         return addedByUses;
227     }
228
229     @Override
230     public void setAddedByUses(final boolean addedByUses) {
231         this.addedByUses = addedByUses;
232     }
233
234     @Override
235     public DataSchemaNodeBuilder getChildNode(String name) {
236         DataSchemaNodeBuilder result = null;
237         for (DataSchemaNodeBuilder node : addedChildNodes) {
238             if (node.getQName().getLocalName().equals(name)) {
239                 result = node;
240                 break;
241             }
242         }
243         return result;
244     }
245
246     @Override
247     public Set<DataSchemaNode> getChildNodes() {
248         return childNodes;
249     }
250
251     @Override
252     public void addChildNode(final DataSchemaNodeBuilder childNode) {
253         addedChildNodes.add(childNode);
254     }
255
256     @Override
257     public Set<DataSchemaNodeBuilder> getChildNodeBuilders() {
258         return addedChildNodes;
259     }
260
261     public void setChildNodes(final Set<DataSchemaNode> childNodes) {
262         this.childNodes = childNodes;
263     }
264
265     @Override
266     public Set<GroupingDefinition> getGroupings() {
267         return Collections.emptySet();
268     }
269
270     @Override
271     public Set<GroupingBuilder> getGroupingBuilders() {
272         return addedGroupings;
273     }
274
275     @Override
276     public void addGrouping(final GroupingBuilder grouping) {
277         addedGroupings.add(grouping);
278     }
279
280     public void setGroupings(final Set<GroupingDefinition> groupings) {
281         this.groupings = groupings;
282     }
283
284     @Override
285     public Set<UsesNodeBuilder> getUses() {
286         return addedUsesNodes;
287     }
288
289     @Override
290     public void addUsesNode(final UsesNodeBuilder usesBuilder) {
291         addedUsesNodes.add(usesBuilder);
292     }
293
294     public void setUsesnodes(final Set<UsesNode> usesNodes) {
295         this.usesNodes = usesNodes;
296     }
297
298     @Override
299     public List<UnknownSchemaNodeBuilder> getUnknownNodes() {
300         return addedUnknownNodes;
301     }
302
303     @Override
304     public void addUnknownSchemaNode(final UnknownSchemaNodeBuilder unknownNode) {
305         addedUnknownNodes.add(unknownNode);
306     }
307
308     public void setUnknownNodes(List<UnknownSchemaNode> unknownNodes) {
309         this.unknownNodes = unknownNodes;
310     }
311
312     private final class GroupingDefinitionImpl implements GroupingDefinition {
313         private final QName qname;
314         private SchemaPath path;
315         private String description;
316         private String reference;
317         private Status status;
318         private boolean addedByUses;
319         private Map<QName, DataSchemaNode> childNodes = Collections.emptyMap();
320         private Set<GroupingDefinition> groupings = Collections.emptySet();
321         private Set<TypeDefinition<?>> typeDefinitions = Collections.emptySet();
322         private Set<UsesNode> uses = Collections.emptySet();
323         private List<UnknownSchemaNode> unknownNodes = Collections.emptyList();
324
325         private GroupingDefinitionImpl(final QName qname) {
326             this.qname = qname;
327         }
328
329         @Override
330         public QName getQName() {
331             return qname;
332         }
333
334         @Override
335         public SchemaPath getPath() {
336             return path;
337         }
338
339         private void setPath(SchemaPath path) {
340             this.path = path;
341         }
342
343         @Override
344         public String getDescription() {
345             return description;
346         }
347
348         private void setDescription(String description) {
349             this.description = description;
350         }
351
352         @Override
353         public String getReference() {
354             return reference;
355         }
356
357         private void setReference(String reference) {
358             this.reference = reference;
359         }
360
361         @Override
362         public Status getStatus() {
363             return status;
364         }
365
366         private void setStatus(Status status) {
367             this.status = status;
368         }
369
370         @Override
371         public boolean isAddedByUses() {
372             return addedByUses;
373         }
374
375         private void setAddedByUses(final boolean addedByUses) {
376             this.addedByUses = addedByUses;
377         }
378
379         @Override
380         public Set<DataSchemaNode> getChildNodes() {
381             final Set<DataSchemaNode> result = new TreeSet<DataSchemaNode>(Comparators.SCHEMA_NODE_COMP);
382             result.addAll(childNodes.values());
383             return result;
384         }
385
386         private void setChildNodes(Map<QName, DataSchemaNode> childNodes) {
387             this.childNodes = childNodes;
388         }
389
390         @Override
391         public Set<GroupingDefinition> getGroupings() {
392             return groupings;
393         }
394
395         private void setGroupings(Set<GroupingDefinition> groupings) {
396             this.groupings = groupings;
397         }
398
399         @Override
400         public Set<UsesNode> getUses() {
401             return uses;
402         }
403
404         private void setUses(Set<UsesNode> uses) {
405             this.uses = uses;
406         }
407
408         @Override
409         public Set<TypeDefinition<?>> getTypeDefinitions() {
410             return typeDefinitions;
411         }
412
413         private void setTypeDefinitions(Set<TypeDefinition<?>> typeDefinitions) {
414             this.typeDefinitions = typeDefinitions;
415         }
416
417         @Override
418         public List<UnknownSchemaNode> getUnknownSchemaNodes() {
419             return unknownNodes;
420         }
421
422         private void setUnknownSchemaNodes(List<UnknownSchemaNode> unknownNodes) {
423             if (unknownNodes != null) {
424                 this.unknownNodes = unknownNodes;
425             }
426         }
427
428         @Override
429         public DataSchemaNode getDataChildByName(QName name) {
430             return childNodes.get(name);
431         }
432
433         @Override
434         public DataSchemaNode getDataChildByName(String name) {
435             DataSchemaNode result = null;
436             for (Map.Entry<QName, DataSchemaNode> entry : childNodes.entrySet()) {
437                 if (entry.getKey().getLocalName().equals(name)) {
438                     result = entry.getValue();
439                     break;
440                 }
441             }
442             return result;
443         }
444
445         @Override
446         public int hashCode() {
447             final int prime = 31;
448             int result = 1;
449             result = prime * result + ((qname == null) ? 0 : qname.hashCode());
450             result = prime * result + ((path == null) ? 0 : path.hashCode());
451             return result;
452         }
453
454         @Override
455         public boolean equals(Object obj) {
456             if (this == obj) {
457                 return true;
458             }
459             if (obj == null) {
460                 return false;
461             }
462             if (getClass() != obj.getClass()) {
463                 return false;
464             }
465             final GroupingDefinitionImpl other = (GroupingDefinitionImpl) obj;
466             if (qname == null) {
467                 if (other.qname != null) {
468                     return false;
469                 }
470             } else if (!qname.equals(other.qname)) {
471                 return false;
472             }
473             if (path == null) {
474                 if (other.path != null) {
475                     return false;
476                 }
477             } else if (!path.equals(other.path)) {
478                 return false;
479             }
480             return true;
481         }
482
483         @Override
484         public String toString() {
485             StringBuilder sb = new StringBuilder(GroupingDefinitionImpl.class.getSimpleName());
486             sb.append("[");
487             sb.append("qname=" + qname);
488             sb.append("]");
489             return sb.toString();
490         }
491     }
492
493 }