import com.google.common.collect.HashMultimap;
import com.google.common.collect.Multimap;
import java.util.Collection;
import com.google.common.collect.HashMultimap;
import com.google.common.collect.Multimap;
import java.util.Collection;
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.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.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.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.SchemaNode;
final class RuntimeTypeGenerator extends AbstractTypeGenerator {
import org.opendaylight.yangtools.yang.model.api.SchemaNode;
final class RuntimeTypeGenerator extends AbstractTypeGenerator {
- RuntimeTypeGenerator(final SchemaContext context, final Map<SchemaNode, JavaTypeName> renames) {
+ RuntimeTypeGenerator(final EffectiveModelContext context, final Map<SchemaNode, JavaTypeName> renames) {
super(context, new RuntimeTypeProvider(context, renames), renames);
}
@NonNull BindingRuntimeTypes toTypeMapping() {
final Map<Type, AugmentationSchemaNode> augmentationToSchema = new HashMap<>();
super(context, new RuntimeTypeProvider(context, renames), renames);
}
@NonNull BindingRuntimeTypes toTypeMapping() {
final Map<Type, AugmentationSchemaNode> augmentationToSchema = new HashMap<>();
final Multimap<Type, Type> choiceToCases = HashMultimap.create();
final Map<QName, Type> identities = new HashMap<>();
final Multimap<Type, Type> choiceToCases = HashMultimap.create();
final Map<QName, Type> identities = new HashMap<>();
+ // 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<>();
+
* builder multiple times, hence we keep a builder->instance cache.
*/
final Map<Type, Type> builderToType = new IdentityHashMap<>();
* builder multiple times, hence we keep a builder->instance cache.
*/
final Map<Type, Type> builderToType = new IdentityHashMap<>();
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()) {
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()) {
- typeToDefiningSchema.put(builtType(builderToType, e.getKey()), e.getValue());
+ 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<Type, Type> e : ctx.getChoiceToCases().entries()) {
choiceToCases.put(builtType(builderToType, e.getKey()), builtType(builderToType, e.getValue()));
- return new BindingRuntimeTypes(schemaContext(), augmentationToSchema, typeToDefiningSchema, choiceToCases,
+ return new BindingRuntimeTypes(schemaContext(), augmentationToSchema, typeToSchema, schemaToType, choiceToCases,