import static com.google.common.base.Verify.verify;
import static com.google.common.base.Verify.verifyNotNull;
+import java.util.ArrayList;
+import java.util.List;
import java.util.Map;
import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.mdsal.binding.generator.impl.reactor.CollisionDomain.Member;
+import org.opendaylight.mdsal.binding.generator.impl.rt.DefaultModuleRuntimeType;
import org.opendaylight.mdsal.binding.model.api.GeneratedType;
import org.opendaylight.mdsal.binding.model.api.JavaTypeName;
-import org.opendaylight.mdsal.binding.model.api.YangSourceDefinition;
-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.spec.naming.BindingMapping;
+import org.opendaylight.mdsal.binding.runtime.api.AugmentRuntimeType;
+import org.opendaylight.mdsal.binding.runtime.api.ModuleRuntimeType;
+import org.opendaylight.mdsal.binding.runtime.api.RuntimeType;
+import org.opendaylight.mdsal.binding.runtime.api.YangDataRuntimeType;
+import org.opendaylight.yangtools.yang.binding.contract.Naming;
+import org.opendaylight.yangtools.yang.binding.contract.StatementNamespace;
import org.opendaylight.yangtools.yang.common.AbstractQName;
import org.opendaylight.yangtools.yang.common.QNameModule;
-import org.opendaylight.yangtools.yang.model.api.Module;
-import org.opendaylight.yangtools.yang.model.api.stmt.ChoiceEffectiveStatement;
-import org.opendaylight.yangtools.yang.model.api.stmt.DataTreeEffectiveStatement;
+import org.opendaylight.yangtools.yang.common.YangDataName;
import org.opendaylight.yangtools.yang.model.api.stmt.ModuleEffectiveStatement;
import org.opendaylight.yangtools.yang.model.util.SchemaInferenceStack;
* Generator corresponding to a {@code module} statement. These generators are roots for generating types for a
* particular {@link QNameModule} as mapped into the root package.
*/
-public final class ModuleGenerator extends AbstractCompositeGenerator<ModuleEffectiveStatement> {
+public final class ModuleGenerator extends AbstractCompositeGenerator<ModuleEffectiveStatement, ModuleRuntimeType> {
private final @NonNull JavaTypeName yangModuleInfo;
- private final @NonNull ClassPlacement placement;
/**
* Note that for sake of simplicity of lookup and child mapping, this instance serves as the root for all child
* generators, but mapping to {@link CollisionDomain}s and their {@link Member}s is rather weird. This generator
- * actually produces one of <em>secondary</em> members, more precisely the {@link BindingMapping#DATA_ROOT_SUFFIX}
- * one. Counter-intuitively the other secondary members live as children of this generator. To support this we have
- * also have this field, which is actually the <em>primary</em> derived from the module's name.
+ * actually produces one of <em>secondary</em> members, more precisely the {@link Naming#DATA_ROOT_SUFFIX} one.
+ * Counter-intuitively the other secondary members live as children of this generator. To support this we also have
+ * this field, which is actually the <em>primary</em> derived from the module's name.
*/
private final Member prefixMember;
ModuleGenerator(final ModuleEffectiveStatement statement) {
super(statement);
- yangModuleInfo = JavaTypeName.create(javaPackage(), BindingMapping.MODULE_INFO_CLASS_NAME);
- placement = computePlacement();
- prefixMember = placement != ClassPlacement.NONE || haveSecondary()
- ? domain().addPrefix(this, new ModuleNamingStrategy(statement.argument())) : null;
+ yangModuleInfo = JavaTypeName.create(javaPackage(), Naming.MODULE_INFO_CLASS_NAME);
+ prefixMember = domain().addPrefix(this, new ModuleNamingStrategy(statement.argument()));
}
- private @NonNull ClassPlacement computePlacement() {
- return statement().findFirstEffectiveSubstatement(DataTreeEffectiveStatement.class).isPresent()
- || statement().findFirstEffectiveSubstatement(ChoiceEffectiveStatement.class).isPresent()
- ? ClassPlacement.TOP_LEVEL : ClassPlacement.NONE;
- }
-
- private boolean haveSecondary() {
- for (Generator child : this) {
- if (child instanceof AbstractImplicitGenerator) {
- return true;
- }
- }
- return false;
+ @Override
+ StatementNamespace namespace() {
+ return StatementNamespace.DATA_ROOT;
}
@Override
@Override
String createJavaPackage() {
- return BindingMapping.getRootPackageName(statement().localQNameModule());
+ return Naming.getRootPackageName(statement().localQNameModule());
}
@Override
- AbstractCompositeGenerator<?> getPackageParent() {
+ AbstractCompositeGenerator<?, ?> getPackageParent() {
return this;
}
return domain();
}
- @Override
- ClassPlacement classPlacement() {
- return placement;
- }
-
@Override
Member createMember(final CollisionDomain domain) {
- return domain.addSecondary(this, prefixMember, BindingMapping.DATA_ROOT_SUFFIX);
+ return domain.addSecondary(this, prefixMember, Naming.DATA_ROOT_SUFFIX);
}
@Override
GeneratedType createTypeImpl(final TypeBuilderFactory builderFactory) {
- final GeneratedTypeBuilder builder = builderFactory.newGeneratedTypeBuilder(typeName());
+ final var builder = builderFactory.newGeneratedTypeBuilder(typeName());
builder.setModuleName(statement().argument().getLocalName());
builder.addImplementsType(BindingTypes.DATA_ROOT);
}
addGetterMethods(builder, builderFactory);
-
- if (builderFactory instanceof TypeBuilderFactory.Codegen) {
- final ModuleEffectiveStatement stmt = statement();
- verify(stmt instanceof Module, "Unexpected module %s", stmt);
- final Module module = (Module) stmt;
-
- YangSourceDefinition.of(module).ifPresent(builder::setYangSourceDefinition);
- TypeComments.description(module).ifPresent(builder::addComment);
- module.getDescription().ifPresent(builder::setDescription);
- module.getReference().ifPresent(builder::setReference);
- }
+ builderFactory.addCodegenInformation(statement(), builder);
return builder.build();
}
}
void addQNameConstant(final GeneratedTypeBuilderBase<?> builder, final AbstractQName localName) {
- builder.addConstant(BindingTypes.QNAME, BindingMapping.QNAME_STATIC_FIELD_NAME,
+ builder.addConstant(BindingTypes.QNAME, Naming.QNAME_STATIC_FIELD_NAME,
Map.entry(yangModuleInfo, localName.getLocalName()));
}
+
+ void addNameConstant(final GeneratedTypeBuilderBase<?> builder, final YangDataName templateName) {
+ builder.addConstant(BindingTypes.YANG_DATA_NAME, Naming.NAME_STATIC_FIELD_NAME,
+ Map.entry(yangModuleInfo, templateName));
+ }
+
+ @Override
+ CompositeRuntimeTypeBuilder<ModuleEffectiveStatement, ModuleRuntimeType> createBuilder(
+ final ModuleEffectiveStatement statement) {
+ return new CompositeRuntimeTypeBuilder<>(statement) {
+ @Override
+ ModuleRuntimeType build(final GeneratedType type, final ModuleEffectiveStatement statement,
+ final List<RuntimeType> children, final List<AugmentRuntimeType> augments) {
+ verify(augments.isEmpty(), "Unexpected augments %s", augments);
+
+ final var yangDataChildren = new ArrayList<YangDataRuntimeType>();
+ for (var child : ModuleGenerator.this) {
+ if (child instanceof YangDataGenerator yangDataGen) {
+ yangDataChildren.add(yangDataGen.getRuntimeType());
+ }
+ }
+
+ return new DefaultModuleRuntimeType(type, statement, children, yangDataChildren);
+ }
+ };
+ }
}