Populate SubmoduleEffectiveModule with import namespaces

[yangtools.git] / yang / yang-model-export / src / main / java / org / opendaylight / yangtools / yang / model / export / DeclaredStatementFormatter.java

diff --git a/yang/yang-model-export/src/main/java/org/opendaylight/yangtools/yang/model/export/DeclaredStatementFormatter.java b/yang/yang-model-export/src/main/java/org/opendaylight/yangtools/yang/model/export/DeclaredStatementFormatter.java
index 040f0c9a9d5c3f6e58bf87228749e57ef8ffaf01..e0efa40e7bd7b65bf282af1a80d0dd5d65369e6f 100644 (file)
--- a/yang/yang-model-export/src/main/java/org/opendaylight/yangtools/yang/model/export/DeclaredStatementFormatter.java
+++ b/yang/yang-model-export/src/main/java/org/opendaylight/yangtools/yang/model/export/DeclaredStatementFormatter.java
@@ -18,7 +18,7 @@ import org.opendaylight.yangtools.concepts.Immutable;
 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
 import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
 import org.opendaylight.yangtools.yang.model.api.stmt.ModuleEffectiveStatement;
-import org.opendaylight.yangtools.yang.model.api.stmt.ModuleEffectiveStatement.QNameModuleToPrefixNamespace;
+import org.opendaylight.yangtools.yang.model.api.stmt.SubmoduleEffectiveStatement;
 
 /**
  * Utility class for formatting {@link DeclaredStatement}s.
@@ -48,7 +48,13 @@ public final class DeclaredStatementFormatter implements Immutable {
      */
     public YangTextSnippet toYangTextSnippet(final ModuleEffectiveStatement module,
             final DeclaredStatement<?> statement) {
-        return new YangTextSnippet(statement, module.findAll(QNameModuleToPrefixNamespace.class), ignoredStatements,
+        return new YangTextSnippet(statement, StatementPrefixResolver.forModule(module), ignoredStatements,
+            omitDefaultStatements);
+    }
+
+    public YangTextSnippet toYangTextSnippet(final SubmoduleEffectiveStatement submodule,
+            final DeclaredStatement<?> statement) {
+        return new YangTextSnippet(statement, StatementPrefixResolver.forSubmodule(submodule), ignoredStatements,
             omitDefaultStatements);
     }