59a3d18e4f21446b494bb1ddbd14118e79e215dd
[mdsal.git] / binding / mdsal-binding-generator-impl / src / main / java / org / opendaylight / mdsal / binding / generator / impl / RuntimeTypeGenerator.java
1 /*
2  * Copyright (c) 2018 Pantheon Technologies, s.r.o.  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.mdsal.binding.generator.impl;
9
10 import com.google.common.collect.BiMap;
11 import com.google.common.collect.HashBiMap;
12 import com.google.common.collect.HashMultimap;
13 import com.google.common.collect.Multimap;
14 import java.util.HashMap;
15 import java.util.Map;
16 import java.util.Set;
17 import org.opendaylight.mdsal.binding.generator.api.BindingRuntimeTypes;
18 import org.opendaylight.mdsal.binding.model.api.Type;
19 import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTypeBuilder;
20 import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTypeBuilderBase;
21 import org.opendaylight.mdsal.binding.model.api.type.builder.TypeMemberBuilder;
22 import org.opendaylight.mdsal.binding.yang.types.TypeProviderImpl;
23 import org.opendaylight.yangtools.yang.common.QName;
24 import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
25 import org.opendaylight.yangtools.yang.model.api.DocumentedNode;
26 import org.opendaylight.yangtools.yang.model.api.DocumentedNode.WithStatus;
27 import org.opendaylight.yangtools.yang.model.api.Module;
28 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
29 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
30
31 final class RuntimeTypeGenerator extends AbstractTypeGenerator {
32     RuntimeTypeGenerator(final SchemaContext context) {
33         // TODO: instantiate a different subclass of TypeProviderImpl
34         super(context, new TypeProviderImpl(context));
35     }
36
37     BindingRuntimeTypes toTypeMapping() {
38         final Map<Type, AugmentationSchemaNode> augmentationToSchema = new HashMap<>();
39         final BiMap<Type, WithStatus> typeToDefiningSchema = HashBiMap.create();
40         final Multimap<Type, Type> choiceToCases = HashMultimap.create();
41         final Map<QName, Type> identities = new HashMap<>();
42
43         for (final ModuleContext ctx : moduleContexts()) {
44             augmentationToSchema.putAll(ctx.getTypeToAugmentation());
45             typeToDefiningSchema.putAll(ctx.getTypeToSchema());
46
47             choiceToCases.putAll(ctx.getChoiceToCases());
48             identities.putAll(ctx.getIdentities());
49         }
50
51         return new BindingRuntimeTypes(augmentationToSchema, typeToDefiningSchema, choiceToCases, identities);
52     }
53
54     @Override
55     void addCodegenInformation(final GeneratedTypeBuilderBase<?> genType, final Module module) {
56         // No-op
57     }
58
59     @Override
60     void addCodegenInformation(final GeneratedTypeBuilderBase<?> genType, final Module module, final SchemaNode node) {
61         // No-op
62     }
63
64     @Override
65     void addCodegenInformation(final GeneratedTypeBuilder interfaceBuilder, final Module module, final String string,
66             final Set<? extends SchemaNode> nodes) {
67         // No-op
68     }
69
70     @Override
71     void addComment(final TypeMemberBuilder<?> genType, final DocumentedNode node) {
72         // No-op
73     }
74 }