Rework AugmentRuntimeType and Choice/Case linkage
[mdsal.git] / binding / mdsal-binding-generator / src / main / java / org / opendaylight / mdsal / binding / generator / impl / reactor / OpaqueObjectGenerator.java
index f020779dca5b6b679a5245ab53823f4f6b2ce3d8..2e7da2abbb7cb98a78ec4d4ff54ff7b5e1636dc8 100644 (file)
@@ -7,27 +7,65 @@
  */
 package org.opendaylight.mdsal.binding.generator.impl.reactor;
 
-import org.opendaylight.mdsal.binding.generator.impl.tree.SchemaTreeChild;
+import static com.google.common.base.Verify.verify;
+
+import org.eclipse.jdt.annotation.NonNull;
+import org.opendaylight.mdsal.binding.generator.impl.rt.DefaultAnydataRuntimeType;
+import org.opendaylight.mdsal.binding.generator.impl.rt.DefaultAnyxmlRuntimeType;
 import org.opendaylight.mdsal.binding.model.api.GeneratedType;
 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.ri.BindingTypes;
+import org.opendaylight.mdsal.binding.runtime.api.AnydataRuntimeType;
+import org.opendaylight.mdsal.binding.runtime.api.AnyxmlRuntimeType;
+import org.opendaylight.mdsal.binding.runtime.api.OpaqueRuntimeType;
+import org.opendaylight.yangtools.yang.model.api.stmt.AnydataEffectiveStatement;
+import org.opendaylight.yangtools.yang.model.api.stmt.AnyxmlEffectiveStatement;
 import org.opendaylight.yangtools.yang.model.api.stmt.DataTreeEffectiveStatement;
 import org.opendaylight.yangtools.yang.model.util.SchemaInferenceStack;
 
 /**
  * Common generator for {@code anydata} and {@code anyxml}.
  */
-final class OpaqueObjectGenerator<T extends DataTreeEffectiveStatement<?>> extends AbstractExplicitGenerator<T>
-        implements SchemaTreeChild<T, OpaqueObjectGenerator<T>> {
-    OpaqueObjectGenerator(final T statement, final AbstractCompositeGenerator<?> parent) {
-        super(statement, parent);
+abstract class OpaqueObjectGenerator<S extends DataTreeEffectiveStatement<?>, R extends OpaqueRuntimeType>
+        extends AbstractExplicitGenerator<S, R> {
+    static final class Anydata extends OpaqueObjectGenerator<AnydataEffectiveStatement, AnydataRuntimeType> {
+        Anydata(final AnydataEffectiveStatement statement, final AbstractCompositeGenerator<?, ?> parent) {
+            super(statement, parent);
+        }
+
+        @Override
+        AnydataRuntimeType createExternalRuntimeType(final GeneratedType type) {
+            return new DefaultAnydataRuntimeType(type, statement());
+        }
+
+        @Override
+        AnydataRuntimeType createInternalRuntimeType(final AnydataEffectiveStatement statement,
+                final GeneratedType type) {
+            return new DefaultAnydataRuntimeType(type, statement);
+        }
     }
 
-    @Override
-    public OpaqueObjectGenerator<T> generator() {
-        return this;
+    static final class Anyxml extends OpaqueObjectGenerator<AnyxmlEffectiveStatement, AnyxmlRuntimeType> {
+        Anyxml(final AnyxmlEffectiveStatement statement, final AbstractCompositeGenerator<?, ?> parent) {
+            super(statement, parent);
+        }
+
+        @Override
+        AnyxmlRuntimeType createExternalRuntimeType(final GeneratedType type) {
+            return new DefaultAnyxmlRuntimeType(type, statement());
+        }
+
+        @Override
+        AnyxmlRuntimeType createInternalRuntimeType(final AnyxmlEffectiveStatement statement,
+                final GeneratedType type) {
+            return new DefaultAnyxmlRuntimeType(type, statement);
+        }
+    }
+
+    OpaqueObjectGenerator(final S statement, final AbstractCompositeGenerator<?, ?> parent) {
+        super(statement, parent);
     }
 
     @Override
@@ -57,4 +95,25 @@ final class OpaqueObjectGenerator<T extends DataTreeEffectiveStatement<?>> exten
     void constructRequire(final GeneratedTypeBuilderBase<?> builder, final Type returnType) {
         constructRequireImpl(builder, returnType);
     }
+
+    @Override
+    final GeneratedType runtimeJavaType() {
+        return generatedType().orElse(null);
+    }
+
+    @Override
+    final @NonNull R createExternalRuntimeType(final Type type) {
+        verify(type instanceof GeneratedType, "Unexpected type %s", type);
+        return createExternalRuntimeType((GeneratedType) type);
+    }
+
+    abstract @NonNull R createExternalRuntimeType(@NonNull GeneratedType type);
+
+    @Override
+    final R createInternalRuntimeType(final ChildLookup lookup, final S statement, final Type type) {
+        verify(type instanceof GeneratedType, "Unexpected type %s", type);
+        return createInternalRuntimeType(statement, (GeneratedType) type);
+    }
+
+    abstract @NonNull R createInternalRuntimeType(@NonNull S statement, @NonNull GeneratedType type);
 }