javasharp.SyntaxTreeXmlFilter.java Source code

Introduction

Here is the source code for javasharp.SyntaxTreeXmlFilter.java

Source

/* JavaSharp, a free Java to C# translator based on ANTLRv4
 * Copyright (C) 2014-2015  Philip van Oosten
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 * 
 * https://github.com/pvoosten
 */

package javasharp;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.io.Reader;
import org.antlr.v4.runtime.ANTLRInputStream;
import org.antlr.v4.runtime.BufferedTokenStream;
import org.antlr.v4.runtime.CommonTokenStream;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.XMLReader;
import org.xml.sax.helpers.XMLFilterImpl;

/**
 *
 * @author PvO
 */
class SyntaxTreeXmlFilter extends XMLFilterImpl {

    private final JavaLexer lexer;
    private final JavaParser parser;

    public SyntaxTreeXmlFilter() {
        lexer = new JavaLexer(null);
        parser = new JavaParser(null);
        parser.setBuildParseTree(true);

    }

    private JavaParser.CompilationUnitContext readSourceFile(String javaSourceFile)
            throws FileNotFoundException, IOException {
        return readSourceFile(new File(javaSourceFile));
    }

    private JavaParser.CompilationUnitContext readSourceFile(File javaSourceFile)
            throws FileNotFoundException, IOException {
        Reader fileReader = new FileReader(javaSourceFile);
        ANTLRInputStream is = new ANTLRInputStream(fileReader);
        lexer.setInputStream(is);
        CommonTokenStream tokens = new CommonTokenStream(lexer);
        tokens.fill();
        parser.setTokenStream(tokens);
        JavaParser.CompilationUnitContext ctx = parser.compilationUnit();
        return ctx;
    }

    @Override
    public void parse(InputSource input) throws SAXException, IOException {
        parse(input.getSystemId());
    }

    @Override
    public void parse(String javaSourcePath) throws SAXException, IOException {
        // parse the Java file(s) with the ANTLR generated parser
        JavaParser.CompilationUnitContext ctx = readSourceFile(javaSourcePath);
        // visit the AST with the XmlEmittingVisitor, which generates SAX events
        startDocument();
        XmlEmittingVisitor visitor = new XmlEmittingVisitor(this, (BufferedTokenStream) parser.getTokenStream());
        visitor.visit(ctx);
        endDocument();
    }

    @Override
    public void setParent(XMLReader parent) {
        throw new UnsupportedOperationException(
                "This xml filter emits events that are linked to source code, without XML parser.");
    }
}