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.model.api.SchemaPath;
22 final class ModuleContext {
23 private GeneratedTypeBuilder moduleNode;
24 private final List<GeneratedTOBuilder> genTOs = new ArrayList<GeneratedTOBuilder>();
25 private final Map<SchemaPath, Type> typedefs = new HashMap<SchemaPath, Type>();
26 private final Map<SchemaPath, GeneratedTypeBuilder> childNodes = new HashMap<SchemaPath, GeneratedTypeBuilder>();
27 private final Map<SchemaPath, GeneratedTypeBuilder> groupings = new HashMap<SchemaPath, GeneratedTypeBuilder>();
28 private final Map<SchemaPath, GeneratedTypeBuilder> cases = new HashMap<SchemaPath, GeneratedTypeBuilder>();
29 private final Set<GeneratedTOBuilder> identities = new HashSet<GeneratedTOBuilder>();
30 private final Set<GeneratedTypeBuilder> topLevelNodes = new HashSet<GeneratedTypeBuilder>();
31 private final List<GeneratedTypeBuilder> augmentations = new ArrayList<GeneratedTypeBuilder>();
34 List<Type> getGeneratedTypes() {
35 List<Type> result = new ArrayList<>();
37 if (moduleNode != null) {
38 result.add(moduleNode.toInstance());
41 for (GeneratedTOBuilder b : genTOs) {
42 result.add(b.toInstance());
44 for (Type b : typedefs.values()) {
49 for (GeneratedTypeBuilder b : childNodes.values()) {
50 result.add(b.toInstance());
52 for (GeneratedTypeBuilder b : groupings.values()) {
53 result.add(b.toInstance());
55 for (GeneratedTypeBuilder b : cases.values()) {
56 result.add(b.toInstance());
58 for (GeneratedTOBuilder b : identities) {
59 result.add(b.toInstance());
61 for (GeneratedTypeBuilder b : topLevelNodes) {
62 result.add(b.toInstance());
64 for (GeneratedTypeBuilder b : augmentations) {
65 result.add(b.toInstance());
70 public GeneratedTypeBuilder getModuleNode() {
74 public GeneratedTypeBuilder getChildNode(SchemaPath p) {
75 return childNodes.get(p);
78 public GeneratedTypeBuilder getGrouping(SchemaPath p) {
79 return groupings.get(p);
82 public GeneratedTypeBuilder getCase(SchemaPath p) {
86 public void addModuleNode(GeneratedTypeBuilder moduleNode) {
87 this.moduleNode = moduleNode;
90 public void addGeneratedTOBuilder(GeneratedTOBuilder b) {
94 public void addChildNodeType(SchemaPath p, GeneratedTypeBuilder b) {
98 public void addGroupingType(SchemaPath p, GeneratedTypeBuilder b) {
102 public void addTypedefType(SchemaPath p, Type t) {
106 public void addCaseType(SchemaPath p, GeneratedTypeBuilder b) {
110 public void addIdentityType(GeneratedTOBuilder b) {
114 public void addTopLevelNodeType(GeneratedTypeBuilder b) {
115 topLevelNodes.add(b);
118 public void addAugmentType(GeneratedTypeBuilder b) {
119 augmentations.add(b);