Add limit to AnnotationIterator

Author: Angel Chang

src/edu/stanford/nlp/pipeline/AnnotationIterator.java

@@ -26,6 +26,7 @@ public class AnnotationIterator extends AbstractIterator<Annotation> implements
     JSONAnnotationReader jsonReader = new JSONAnnotationReader();
     String format;
     int docCnt = 0;
+    int limit = 0;
 
     public AnnotationIterator(String filename) throws IOException {
         this.filename = filename;
@@ -45,6 +46,11 @@ public AnnotationIterator(String filename) throws IOException {
         nextDoc = readNextDocument();
     }
 
+    public AnnotationIterator(String filename, int limit) throws IOException {
+        this(filename);
+        this.limit = limit;
+    }
+
     @Override
     public boolean hasNext() {
         return nextDoc != null;
@@ -68,6 +74,9 @@ public Annotation readNextDocument() {
         if (br == null && input == null) {
             return null;
         }
+        if (limit > 0 && docCnt >= limit) {
+            return null;
+        }
         try {
             if (serializer != null) {
                 if (input.available() > 0) {