edu.stanford.nlp.trees.TreeLengthComparator Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of stanford-parser Show documentation

Stanford Parser processes raw text in English, Chinese, German, Arabic, and French, and extracts constituency parse trees.

There is a newer version: 3.9.2

Show newest version

package edu.stanford.nlp.trees;

import java.util.Comparator;

/**
 * A TreeLengthComparator orders trees by their yield sentence
 * lengths.
 *
 * @author Christopher Manning
 * @version 2003/03/24
 */
public class TreeLengthComparator implements Comparator {

  /**
   * Create a new TreeLengthComparator.
   */
  public TreeLengthComparator() {
  }


  /**
   * Compare the two objects.
   */
  public int compare(Tree t1, Tree t2) {
    if (t1 == t2) {
      return 0;
    }
    int len1 = t1.yield().size();
    int len2 = t2.yield().size();
    if (len1 > len2) {
      return 1;
    } else if (len1 < len2) {
      return -1;
    } else {
      return 0;
    }
  }

}