Indent PolyAlgebra debug outputs

polypheny · Apr 3, 2024 · a8c7bde · a8c7bde
1 parent 3888e85
commit a8c7bde
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 7 deletions.
diff --git a/core/src/main/java/org/polypheny/db/algebra/AbstractAlgNode.java b/core/src/main/java/org/polypheny/db/algebra/AbstractAlgNode.java
@@ -357,9 +357,12 @@ public AlgWriter explainTerms( AlgWriter pw ) {
 
 
     @Override
-    public void buildPolyAlgebra( StringBuilder sb ) {
+    public void buildPolyAlgebra( StringBuilder sb, String prefix ) {
+        final String INDENT = " ";
+        String nextPrefix = prefix == null ? null : prefix + INDENT;
+
         PolyAlgDeclaration decl = getPolyAlgDeclaration();
-        sb.append( decl.opName );
+        sb.append( prefix == null ? "" : prefix ).append( decl.opName );
         sb.append( collectAttributes().serializeArguments( this ) );
 
         int size = getInputs().size();
@@ -375,13 +378,14 @@ public void buildPolyAlgebra( StringBuilder sb ) {
 
             StringBuilder csb = new StringBuilder();
             if ( projections.isEmpty() ) {
-                child.buildPolyAlgebra( csb );
+                child.buildPolyAlgebra( csb, nextPrefix );
             } else {
-                csb.append( PolyAlgRegistry.getDeclaration( LogicalRelProject.class ).opName )
+                csb.append( nextPrefix )
+                        .append( PolyAlgRegistry.getDeclaration( LogicalRelProject.class ).opName )
                         .append( "#[" )
                         .append( PolyAlgUtils.joinMultiValued( projections, true ) )
                         .append( "](\n" );
-                child.buildPolyAlgebra( csb );
+                child.buildPolyAlgebra( csb, nextPrefix == null ? null : nextPrefix + INDENT );
                 csb.append( ")" );
             }
             sb.append( csb );

diff --git a/core/src/main/java/org/polypheny/db/algebra/AlgNode.java b/core/src/main/java/org/polypheny/db/algebra/AlgNode.java
@@ -245,8 +245,13 @@ public interface AlgNode extends AlgOptNode, Cloneable {
      * The basic structure of PolyAlgebra is {@code OPERATOR[attributes](children)}.
      *
      * @param sb StringBuilder for building the representation
+     * @param prefix Prefix to be added in front of each operator (resulting in operators having a visual indentation) or null if not desired
      */
-    void buildPolyAlgebra( StringBuilder sb );
+    void buildPolyAlgebra( StringBuilder sb, String prefix );
+
+    default void buildPolyAlgebra( StringBuilder sb) {
+        buildPolyAlgebra( sb, null );
+    }
 
     /**
      * Retrieves the PolyAlgDeclaration for this AlgNode implementation.

diff --git a/core/src/main/java/org/polypheny/db/plan/AlgOptUtil.java b/core/src/main/java/org/polypheny/db/plan/AlgOptUtil.java
@@ -1365,7 +1365,7 @@ public static void registerAbstractAlgs( AlgPlanner planner ) {
     public static String dumpPlan( String header, AlgNode alg, ExplainFormat format, ExplainLevel detailLevel ) {
         //TODO: Delete lines for testing PolyAlg serialization
         StringBuilder sb = new StringBuilder();
-        alg.buildPolyAlgebra( sb );
+        alg.buildPolyAlgebra( sb, "" );
         System.out.println( "===== " + header + " =====" );
         System.out.println( sb );
         System.out.print( "----> " );