78 lines
1.6 KiB
Java
78 lines
1.6 KiB
Java
package org.gcube.nlphub.legacy;
|
|
|
|
import java.util.ArrayList;
|
|
import com.google.gson.JsonArray;
|
|
import com.google.gson.JsonObject;
|
|
|
|
public class NerAnnotationData {
|
|
private String name;
|
|
private ArrayList<NerEntity> nerEntities;
|
|
|
|
/**
|
|
* Class constructor; require the name of the annotation
|
|
* @param name
|
|
*/
|
|
public NerAnnotationData(String name) {
|
|
this.name = name;
|
|
nerEntities = new ArrayList<>();
|
|
}
|
|
|
|
/**
|
|
* add a new NerEntity to the collection
|
|
* @param entity
|
|
*/
|
|
public void addNerEntity(NerEntity entity) {
|
|
nerEntities.add(entity);
|
|
}
|
|
|
|
/**
|
|
* build the proper Json object.
|
|
* @return JsonObject
|
|
*/
|
|
public JsonObject toJson() {
|
|
JsonObject json = new JsonObject();
|
|
JsonArray entities = new JsonArray();
|
|
for(NerEntity e : nerEntities) {
|
|
entities.add(e.toJson());
|
|
}
|
|
json.add(name,entities);
|
|
return json;
|
|
}
|
|
|
|
|
|
public String getName() {
|
|
return name;
|
|
}
|
|
|
|
public void setName(String name) {
|
|
this.name = name;
|
|
}
|
|
|
|
public ArrayList<NerEntity> getNerEntities() {
|
|
return nerEntities;
|
|
}
|
|
|
|
public void setNerEntities(ArrayList<NerEntity> nerEntities) {
|
|
this.nerEntities = nerEntities;
|
|
}
|
|
|
|
/*
|
|
public static void main(String[] args) {
|
|
NerEntity ne1 = new NerEntity(11, 22);
|
|
ne1.addProperty("type", "scalar");
|
|
ne1.addProperty("unit", "hours");
|
|
|
|
NerEntity ne2 = new NerEntity(33, 44);
|
|
ne2.addProperty("type", "scalar");
|
|
ne2.addProperty("unit", "minutes");
|
|
|
|
NerAnnotationData nad = new NerAnnotationData("Measure");
|
|
nad.addNerEntity(ne1);
|
|
nad.addNerEntity(ne2);
|
|
|
|
System.out.println(nad.toJson().toString());
|
|
}*/
|
|
|
|
|
|
}
|