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.sal.binding.generator.impl;
10 import java.util.ArrayList;
11 import java.util.HashMap;
12 import java.util.HashSet;
13 import java.util.List;
17 import org.opendaylight.yangtools.sal.binding.model.api.Type;
18 import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTOBuilder;
19 import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTypeBuilder;
20 import org.opendaylight.yangtools.yang.common.QName;
21 import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
22 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
24 import com.google.common.collect.BiMap;
25 import com.google.common.collect.HashBiMap;
27 public final class ModuleContext {
28 private GeneratedTypeBuilder moduleNode;
29 private final List<GeneratedTOBuilder> genTOs = new ArrayList<GeneratedTOBuilder>();
30 private final Map<SchemaPath, Type> typedefs = new HashMap<SchemaPath, Type>();
31 private final Map<SchemaPath, GeneratedTypeBuilder> childNodes = new HashMap<SchemaPath, GeneratedTypeBuilder>();
32 private final Map<SchemaPath, GeneratedTypeBuilder> groupings = new HashMap<SchemaPath, GeneratedTypeBuilder>();
33 private final Map<SchemaPath, GeneratedTypeBuilder> cases = new HashMap<SchemaPath, GeneratedTypeBuilder>();
34 private final Map<QName,GeneratedTOBuilder> identities = new HashMap<>();
35 private final Set<GeneratedTypeBuilder> topLevelNodes = new HashSet<GeneratedTypeBuilder>();
36 private final List<GeneratedTypeBuilder> augmentations = new ArrayList<GeneratedTypeBuilder>();
37 private final BiMap<Type,AugmentationSchema> typeToAugmentation = HashBiMap.create();
41 List<Type> getGeneratedTypes() {
42 List<Type> result = new ArrayList<>();
44 if (moduleNode != null) {
45 result.add(moduleNode.toInstance());
48 for (GeneratedTOBuilder b : genTOs) {
49 result.add(b.toInstance());
51 for (Type b : typedefs.values()) {
56 for (GeneratedTypeBuilder b : childNodes.values()) {
57 result.add(b.toInstance());
59 for (GeneratedTypeBuilder b : groupings.values()) {
60 result.add(b.toInstance());
62 for (GeneratedTypeBuilder b : cases.values()) {
63 result.add(b.toInstance());
65 for (GeneratedTOBuilder b : identities.values()) {
66 result.add(b.toInstance());
68 for (GeneratedTypeBuilder b : topLevelNodes) {
69 result.add(b.toInstance());
71 for (GeneratedTypeBuilder b : augmentations) {
72 result.add(b.toInstance());
77 public GeneratedTypeBuilder getModuleNode() {
81 public GeneratedTypeBuilder getChildNode(SchemaPath p) {
82 return childNodes.get(p);
85 public GeneratedTypeBuilder getGrouping(SchemaPath p) {
86 return groupings.get(p);
89 public GeneratedTypeBuilder getCase(SchemaPath p) {
93 public void addModuleNode(GeneratedTypeBuilder moduleNode) {
94 this.moduleNode = moduleNode;
97 public void addGeneratedTOBuilder(GeneratedTOBuilder b) {
101 public void addChildNodeType(SchemaPath p, GeneratedTypeBuilder b) {
102 childNodes.put(p, b);
105 public void addGroupingType(SchemaPath p, GeneratedTypeBuilder b) {
109 public void addTypedefType(SchemaPath p, Type t) {
113 public void addCaseType(SchemaPath p, GeneratedTypeBuilder b) {
117 public void addIdentityType(QName name,GeneratedTOBuilder b) {
118 identities.put(name,b);
121 public void addTopLevelNodeType(GeneratedTypeBuilder b) {
122 topLevelNodes.add(b);
125 public void addAugmentType(GeneratedTypeBuilder b) {
126 augmentations.add(b);
129 public Map<SchemaPath, Type> getTypedefs() {
133 public Map<SchemaPath, GeneratedTypeBuilder> getChildNodes() {
137 public Map<SchemaPath, GeneratedTypeBuilder> getGroupings() {
141 public Map<SchemaPath, GeneratedTypeBuilder> getCases() {
145 public Map<QName,GeneratedTOBuilder> getIdentities() {
149 public Set<GeneratedTypeBuilder> getTopLevelNodes() {
150 return topLevelNodes;
153 public List<GeneratedTypeBuilder> getAugmentations() {
154 return augmentations;
157 public BiMap<Type, AugmentationSchema> getTypeToAugmentation() {
158 return typeToAugmentation;
161 public void addTypeToAugmentation(GeneratedTypeBuilder builder, AugmentationSchema schema) {
162 typeToAugmentation.put(builder, schema);