1
15
16 package gate.creole.nerc;
17
18 import java.io.Serializable;
19 import java.util.*;
20
21 import gate.*;
22
23
26 public class EntitySet extends AbstractSet implements Set, Serializable {
27
28
29 public EntitySet(String fileName, Document document,
30 AnnotationSet annotationSet) {
31 this.fileName = fileName;
32 myEntities = new HashSet();
33 if(annotationSet != null){
34 Iterator annIter = annotationSet.iterator();
35 while(annIter.hasNext()){
36 myEntities.add(new EntityDescriptor(document,
37 (Annotation)annIter.next()));
38 }
39 }
40 }
41
42
45 public String getTextFileName() {
46 return fileName;
47 }
48
49
53 public String toString() {
54 StringBuffer res = new StringBuffer(gate.Gate.STRINGBUFFER_SIZE);
56
57 res.append("==== FILE: ");
58 res.append(fileName);
59 res.append("\n");
60
61 Iterator entIter = myEntities.iterator();
62 while(entIter.hasNext()){
63 res.append(entIter.next().toString());
65 res.append("\n");
66 }
67 return res.toString();
68 }
69
70 public int size(){ return myEntities.size();}
71
72 public Iterator iterator() {return myEntities.iterator();}
73
74 String fileName;
75 Set myEntities;
76 }