Search in sources :

Example 1 with TreeVisitor

use of edu.stanford.nlp.trees.TreeVisitor in project CoreNLP by stanfordnlp.

the class MWETreeVisitorExternal method main.

/**
   * For debugging.
   * 
   * @param args
   */
public static void main(String[] args) {
    if (args.length != 1) {
        System.err.printf("Usage: java %s atb_tree_file > atb_tree_file.out%n", MWETreeVisitorExternal.class.getName());
        System.exit(-1);
    }
    TreeReaderFactory trf = new ArabicTreeReaderFactory();
    try {
        TreeReader tr = trf.newTreeReader(new BufferedReader(new InputStreamReader(new FileInputStream(args[0]), "UTF-8")));
        TreeVisitor visitor = new MWETreeVisitorExternal();
        int treeId = 0;
        for (Tree tree; (tree = tr.readTree()) != null; ++treeId) {
            if (tree.value().equals("ROOT")) {
                // Skip over the ROOT tag
                tree = tree.firstChild();
            }
            visitor.visitTree(tree);
            System.out.println(tree.toString());
        }
        tr.close();
        System.err.printf("Processed %d trees.%n", treeId);
    } catch (UnsupportedEncodingException e) {
        e.printStackTrace();
    } catch (FileNotFoundException e) {
        e.printStackTrace();
    } catch (IOException e) {
        e.printStackTrace();
    }
}
Also used : InputStreamReader(java.io.InputStreamReader) FileNotFoundException(java.io.FileNotFoundException) TreeReader(edu.stanford.nlp.trees.TreeReader) UnsupportedEncodingException(java.io.UnsupportedEncodingException) IOException(java.io.IOException) FileInputStream(java.io.FileInputStream) ArabicTreeReaderFactory(edu.stanford.nlp.trees.international.arabic.ArabicTreeReaderFactory) TreeVisitor(edu.stanford.nlp.trees.TreeVisitor) BufferedReader(java.io.BufferedReader) Tree(edu.stanford.nlp.trees.Tree) TreeReaderFactory(edu.stanford.nlp.trees.TreeReaderFactory) ArabicTreeReaderFactory(edu.stanford.nlp.trees.international.arabic.ArabicTreeReaderFactory)

Aggregations

Tree (edu.stanford.nlp.trees.Tree)1 TreeReader (edu.stanford.nlp.trees.TreeReader)1 TreeReaderFactory (edu.stanford.nlp.trees.TreeReaderFactory)1 TreeVisitor (edu.stanford.nlp.trees.TreeVisitor)1 ArabicTreeReaderFactory (edu.stanford.nlp.trees.international.arabic.ArabicTreeReaderFactory)1 BufferedReader (java.io.BufferedReader)1 FileInputStream (java.io.FileInputStream)1 FileNotFoundException (java.io.FileNotFoundException)1 IOException (java.io.IOException)1 InputStreamReader (java.io.InputStreamReader)1 UnsupportedEncodingException (java.io.UnsupportedEncodingException)1