Search in sources :

Example 1 with CompositeVisitor

use of org.codehaus.groovy.antlr.treewalker.CompositeVisitor in project groovy by apache.

the class AntlrParserPlugin method outputASTInVariousFormsIfNeeded.

private void outputASTInVariousFormsIfNeeded(SourceUnit sourceUnit, SourceBuffer sourceBuffer) {
    // straight xstream output of AST
    // uppercase to hide from jarjar
    String formatProp = System.getProperty("ANTLR.AST".toLowerCase());
    if ("xml".equals(formatProp)) {
        saveAsXML(sourceUnit.getName(), ast);
    }
    // 'pretty printer' output of AST
    if ("groovy".equals(formatProp)) {
        try {
            PrintStream out = new PrintStream(new FileOutputStream(sourceUnit.getName() + ".pretty.groovy"));
            Visitor visitor = new SourcePrinter(out, tokenNames);
            AntlrASTProcessor treewalker = new SourceCodeTraversal(visitor);
            treewalker.process(ast);
        } catch (FileNotFoundException e) {
            System.out.println("Cannot create " + sourceUnit.getName() + ".pretty.groovy");
        }
    }
    // which is a really nice way of seeing the AST, folding nodes etc
    if ("mindmap".equals(formatProp)) {
        try {
            PrintStream out = new PrintStream(new FileOutputStream(sourceUnit.getName() + ".mm"));
            Visitor visitor = new MindMapPrinter(out, tokenNames);
            AntlrASTProcessor treewalker = new PreOrderTraversal(visitor);
            treewalker.process(ast);
        } catch (FileNotFoundException e) {
            System.out.println("Cannot create " + sourceUnit.getName() + ".mm");
        }
    }
    // include original line/col info and source code on the mindmap output
    if ("extendedMindmap".equals(formatProp)) {
        try {
            PrintStream out = new PrintStream(new FileOutputStream(sourceUnit.getName() + ".mm"));
            Visitor visitor = new MindMapPrinter(out, tokenNames, sourceBuffer);
            AntlrASTProcessor treewalker = new PreOrderTraversal(visitor);
            treewalker.process(ast);
        } catch (FileNotFoundException e) {
            System.out.println("Cannot create " + sourceUnit.getName() + ".mm");
        }
    }
    // html output of AST
    if ("html".equals(formatProp)) {
        try {
            PrintStream out = new PrintStream(new FileOutputStream(sourceUnit.getName() + ".html"));
            List<VisitorAdapter> v = new ArrayList<VisitorAdapter>();
            v.add(new NodeAsHTMLPrinter(out, tokenNames));
            v.add(new SourcePrinter(out, tokenNames));
            Visitor visitors = new CompositeVisitor(v);
            AntlrASTProcessor treewalker = new SourceCodeTraversal(visitors);
            treewalker.process(ast);
        } catch (FileNotFoundException e) {
            System.out.println("Cannot create " + sourceUnit.getName() + ".html");
        }
    }
}
Also used : SourcePrinter(org.codehaus.groovy.antlr.treewalker.SourcePrinter) PrintStream(java.io.PrintStream) NodeAsHTMLPrinter(org.codehaus.groovy.antlr.treewalker.NodeAsHTMLPrinter) PreOrderTraversal(org.codehaus.groovy.antlr.treewalker.PreOrderTraversal) CompositeVisitor(org.codehaus.groovy.antlr.treewalker.CompositeVisitor) Visitor(org.codehaus.groovy.antlr.treewalker.Visitor) FileNotFoundException(java.io.FileNotFoundException) ArrayList(java.util.ArrayList) MindMapPrinter(org.codehaus.groovy.antlr.treewalker.MindMapPrinter) FileOutputStream(java.io.FileOutputStream) VisitorAdapter(org.codehaus.groovy.antlr.treewalker.VisitorAdapter) SourceCodeTraversal(org.codehaus.groovy.antlr.treewalker.SourceCodeTraversal) CompositeVisitor(org.codehaus.groovy.antlr.treewalker.CompositeVisitor)

Aggregations

FileNotFoundException (java.io.FileNotFoundException)1 FileOutputStream (java.io.FileOutputStream)1 PrintStream (java.io.PrintStream)1 ArrayList (java.util.ArrayList)1 CompositeVisitor (org.codehaus.groovy.antlr.treewalker.CompositeVisitor)1 MindMapPrinter (org.codehaus.groovy.antlr.treewalker.MindMapPrinter)1 NodeAsHTMLPrinter (org.codehaus.groovy.antlr.treewalker.NodeAsHTMLPrinter)1 PreOrderTraversal (org.codehaus.groovy.antlr.treewalker.PreOrderTraversal)1 SourceCodeTraversal (org.codehaus.groovy.antlr.treewalker.SourceCodeTraversal)1 SourcePrinter (org.codehaus.groovy.antlr.treewalker.SourcePrinter)1 Visitor (org.codehaus.groovy.antlr.treewalker.Visitor)1 VisitorAdapter (org.codehaus.groovy.antlr.treewalker.VisitorAdapter)1