*/
package org.opendaylight.mdsal.binding.generator.impl;
-import com.google.common.collect.BiMap;
-import com.google.common.collect.HashBiMap;
import com.google.common.collect.HashMultimap;
import com.google.common.collect.Multimap;
+import java.util.Collection;
import java.util.HashMap;
+import java.util.IdentityHashMap;
import java.util.Map;
-import java.util.Set;
-import org.opendaylight.mdsal.binding.generator.api.BindingRuntimeTypes;
+import java.util.Map.Entry;
+import org.eclipse.jdt.annotation.NonNull;
+import org.opendaylight.mdsal.binding.model.api.JavaTypeName;
import org.opendaylight.mdsal.binding.model.api.Type;
import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTypeBuilder;
import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTypeBuilderBase;
import org.opendaylight.mdsal.binding.model.api.type.builder.TypeMemberBuilder;
-import org.opendaylight.mdsal.binding.yang.types.TypeProviderImpl;
+import org.opendaylight.mdsal.binding.runtime.api.BindingRuntimeTypes;
+import org.opendaylight.mdsal.binding.yang.types.RuntimeTypeProvider;
+import org.opendaylight.yangtools.concepts.Builder;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
import org.opendaylight.yangtools.yang.model.api.DocumentedNode;
import org.opendaylight.yangtools.yang.model.api.DocumentedNode.WithStatus;
+import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
import org.opendaylight.yangtools.yang.model.api.Module;
-import org.opendaylight.yangtools.yang.model.api.SchemaContext;
+import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
import org.opendaylight.yangtools.yang.model.api.SchemaNode;
final class RuntimeTypeGenerator extends AbstractTypeGenerator {
- RuntimeTypeGenerator(final SchemaContext context) {
- // TODO: instantiate a different subclass of TypeProviderImpl
- super(context, new TypeProviderImpl(context));
+ RuntimeTypeGenerator(final EffectiveModelContext context, final Map<SchemaNode, JavaTypeName> renames) {
+ super(context, new RuntimeTypeProvider(context, renames), renames);
}
- BindingRuntimeTypes toTypeMapping() {
+ @NonNull BindingRuntimeTypes toTypeMapping() {
final Map<Type, AugmentationSchemaNode> augmentationToSchema = new HashMap<>();
- final BiMap<Type, WithStatus> typeToDefiningSchema = HashBiMap.create();
+ final Map<Type, WithStatus> typeToSchema = new HashMap<>();
final Multimap<Type, Type> choiceToCases = HashMultimap.create();
final Map<QName, Type> identities = new HashMap<>();
- for (final ModuleContext ctx : moduleContexts()) {
- augmentationToSchema.putAll(ctx.getTypeToAugmentation());
- typeToDefiningSchema.putAll(ctx.getTypeToSchema());
+ // Note: we are keying through WithStatus, but these nodes compare on semantics, so equivalent schema nodes
+ // can result in two distinct types. We certainly need to keep them separate.
+ final Map<WithStatus, Type> schemaToType = new IdentityHashMap<>();
+
+ /*
+ * Fun parts are here. ModuleContext maps have Builders in them, we want plain types. We may encounter each
+ * builder multiple times, hence we keep a builder->instance cache.
+ */
+ final Map<Type, Type> builderToType = new IdentityHashMap<>();
- choiceToCases.putAll(ctx.getChoiceToCases());
- identities.putAll(ctx.getIdentities());
+ for (final ModuleContext ctx : moduleContexts()) {
+ for (Entry<Type, AugmentationSchemaNode> e : ctx.getTypeToAugmentation().entrySet()) {
+ augmentationToSchema.put(builtType(builderToType, e.getKey()), e.getValue());
+ }
+ for (Entry<Type, WithStatus> e : ctx.getTypeToSchema().entrySet()) {
+ final Type type = builtType(builderToType, e.getKey());
+ typeToSchema.put(type, e.getValue());
+ schemaToType.put(e.getValue(), type);
+ }
+ for (Entry<Type, Type> e : ctx.getChoiceToCases().entries()) {
+ choiceToCases.put(builtType(builderToType, e.getKey()), builtType(builderToType, e.getValue()));
+ }
+ for (Entry<QName, GeneratedTypeBuilder> e : ctx.getIdentities().entrySet()) {
+ identities.put(e.getKey(), builtType(builderToType, e.getValue()));
+ }
}
- return new BindingRuntimeTypes(augmentationToSchema, typeToDefiningSchema, choiceToCases, identities);
+ return new BindingRuntimeTypes(schemaContext(), augmentationToSchema, typeToSchema, schemaToType, choiceToCases,
+ identities);
+ }
+
+ private static Type builtType(final Map<Type, Type> knownTypes, final Type type) {
+ if (type instanceof Builder) {
+ final Type existing = knownTypes.get(type);
+ if (existing != null) {
+ return existing;
+ }
+
+ final Type built = (Type) ((Builder<?>)type).build();
+ knownTypes.put(type, built);
+ return built;
+ }
+ return type;
}
@Override
@Override
void addCodegenInformation(final GeneratedTypeBuilder interfaceBuilder, final Module module, final String string,
- final Set<? extends SchemaNode> nodes) {
+ final Collection<? extends SchemaNode> nodes) {
// No-op
}
void addComment(final TypeMemberBuilder<?> genType, final DocumentedNode node) {
// No-op
}
+
+ @Override
+ void addRpcMethodComment(final TypeMemberBuilder<?> genType, final RpcDefinition node) {
+ // No-op
+ }
}