001    /* Copyright (c) 2014, Jesper Öqvist <jesper.oqvist@cs.lth.se>
002     * All rights reserved.
003     *
004     * Redistribution and use in source and binary forms, with or without
005     * modification, are permitted provided that the following conditions are met:
006     *
007     * 1. Redistributions of source code must retain the above copyright notice,
008     * this list of conditions and the following disclaimer.
009     *
010     * 2. Redistributions in binary form must reproduce the above copyright notice,
011     * this list of conditions and the following disclaimer in the documentation
012     * and/or other materials provided with the distribution.
013     *
014     * 3. Neither the name of the copyright holder nor the names of its
015     * contributors may be used to endorse or promote products derived from this
016     * software without specific prior written permission.
017     *
018     * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
019     * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
020     * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
021     * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
022     * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
023     * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
024     * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
025     * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
026     * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
027     * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
028     * POSSIBILITY OF SUCH DAMAGE.
029     */
030    
031    aspect StructuredPrettyPrint {
032      public String ASTNode.structuredPrettyPrint() throws IOException {
033        ByteArrayOutputStream out = new ByteArrayOutputStream();
034        // First, transform the tree by wrapping all expressions in ParExpr.
035        wrapExprs();
036        prettyPrint(new PrintStream(out, false, "UTF8"));
037        return out.toString().trim();
038      }
039    
040      /**
041       * Hacky way of inserting parens around all expressions.
042       */
043      private void ASTNode.wrapExprs() {
044        for (int i = 0; i < getNumChildNoTransform(); ++i) {
045          ASTNode child = getChildNoTransform(i);
046          if (child instanceof Expr &&
047              !(child instanceof ParExpr) &&
048              !(child instanceof Access) &&
049              !(child instanceof Literal)) {
050            child.setParent(null);
051            ParExpr parExpr = new ParExpr((Expr) child);
052            setChild(parExpr, i);
053          }
054          child.wrapExprs();
055        }
056      }
057    }