/******************************************************************************
* Copyright (C) 2006-2013  IFS Institute for Software and others
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Original authors:
*     Dennis Hunziker
*     Ueli Kistler
* Contributors:
*     Fabio Zadrozny <fabiofz@gmail.com> - initial implementation
******************************************************************************/
/* 
 * Copyright (C) 2006, 2007  Dennis Hunziker, Ueli Kistler 
 */

package ch.hsr.ukistler.astgraph;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;

import org.eclipse.jface.text.Document;
import org.eclipse.jface.text.IDocument;
import org.jgraph.graph.DefaultGraphCell;
import org.python.pydev.core.IPythonNature;
import org.python.pydev.parser.PyParser;
import org.python.pydev.parser.jython.SimpleNode;
import org.python.pydev.shared_core.parsing.BaseParser.ParseOutput;

/**
 * Starts the parser and its visitor (GraphVisitor)
 * 
 * @author Ueli Kistler
 * 
 */
public class ASTGraph {

    public ParseOutput parseFile(String fileName) throws FileNotFoundException, IOException,
            Throwable {
        File pythonSource = new File(fileName);
        BufferedReader in = new BufferedReader(new FileReader(pythonSource));
        StringBuilder source = new StringBuilder();
        try {
            String line = "";
            while ((line = in.readLine()) != null) {
                source.append(line);
                source.append("\n");
            }
        } finally {
            in.close();
        }

        IDocument doc = new Document(source.toString());
        ParseOutput objects = PyParser.reparseDocument(new PyParser.ParserInfo(doc,
                IPythonNature.LATEST_GRAMMAR_VERSION, null));
        if (objects.error != null) {
            throw objects.error;
        }
        return objects;
    }

    public DefaultGraphCell[] generateTree(SimpleNode node) throws IOException, Exception {
        GraphVisitor visitor = new GraphVisitor();
        node.accept(visitor);
        DefaultGraphCell[] cells = visitor.getCells();

        return cells;
    }

}