+ @Override
+ public <X, Z extends EffectiveStatement<X, ?>> @NonNull Optional<X> findSubstatementArgument(
+ final @NonNull Class<Z> type) {
+ if (substatementsInitialized()) {
+ return StmtContextDefaults.findSubstatementArgument(this, type);
+ }
+
+ final Optional<X> templateArg = prototype.findSubstatementArgument(type);
+ if (templateArg.isEmpty()) {
+ return templateArg;
+ }
+ if (SchemaTreeEffectiveStatement.class.isAssignableFrom(type)) {
+ // X is known to be QName
+ return (Optional<X>) templateArg.map(template -> ((QName) template).bindTo(targetModule));
+ }
+ return templateArg;
+ }
+
+ @Override
+ public boolean hasSubstatement(final @NonNull Class<? extends EffectiveStatement<?, ?>> type) {
+ return substatementsInitialized() ? StmtContextDefaults.hasSubstatement(prototype, type)
+ : prototype.hasSubstatement(type);
+ }
+
+ @Override
+ public <D extends DeclaredStatement<QName>, E extends EffectiveStatement<QName, D>>
+ StmtContext<QName, D, E> requestSchemaTreeChild(final QName qname) {
+ LOG.debug("Materializing on lookup of {}", qname);
+ // FIXME: YANGTOOLS-1160: we do not want to force full materialization here
+ ensureEffectiveSubstatements();
+
+ // Now we have to do a lookup as we do not have access to the namespace being populated (yet). Here we are
+ // bypassing additional checks and talk directly to superclass to get the statements.
+ for (StmtContext<?, ?, ?> stmt : super.mutableEffectiveSubstatements()) {
+ if (stmt.producesEffective(SchemaTreeEffectiveStatement.class)
+ && qname.equals(stmt.coerceStatementArgument())) {
+ return (StmtContext<QName, D, E>) stmt;
+ }
+ }
+ return null;
+ }
+