private static View readView(JsonObject json, TextAnnotation ta) throws Exception {
String viewClass = readString("viewType", json);
String viewName = readString("viewName", json);
String viewGenerator = readString("generator", json);
double score = 0;
if (json.has("score")) score = readDouble("score", json);
View view = createEmptyView(ta, viewClass, viewName, viewGenerator, score);
List<Constituent> constituents = new ArrayList<>();
if (json.has("constituents")) {
JsonArray cJson = json.getAsJsonArray("constituents");
for (int i = 0; i < cJson.size(); i++) {
JsonObject cJ = (JsonObject) cJson.get(i);
Constituent c = readConstituent(cJ, ta, viewName);
constituents.add(c);
view.addConstituent(c);
}
}
if (json.has("relations")) {
JsonArray rJson = json.getAsJsonArray("relations");
for (int i = 0; i < rJson.size(); i++) {
JsonObject rJ = (JsonObject) rJson.get(i);
String name = readString("relationName", rJ);
double s = 0;
if (rJ.has("score")) s = readDouble("score", rJ);
int src = readInt("srcConstituent", rJ);
int tgt = readInt("targetConstituent", rJ);
Map<String, Double> labelsToScores = null;
if (rJ.has(LABEL_SCORE_MAP)) {
labelsToScores = new HashMap<>();
readLabelsToScores(labelsToScores, rJ);
}
Relation rel = null;
if (null == labelsToScores)
rel = new Relation(name, constituents.get(src), constituents.get(tgt), s);
else rel = new Relation(labelsToScores, constituents.get(src), constituents.get(tgt));
readAttributes(rel, rJ);
view.addRelation(rel);
}
}
return view;
}
private static void writeView(View view, JsonObject json) {
writeString("viewType", view.getClass().getCanonicalName(), json);
writeString("viewName", view.getViewName(), json);
writeString("generator", view.getViewGenerator(), json);
if (view.getScore() != 0) writeDouble("score", view.getScore(), json);
List<Constituent> constituents = view.getConstituents();
if (constituents.size() > 0) {
JsonArray cJson = new JsonArray();
for (int i = 0; i < view.getNumberOfConstituents(); i++) {
Constituent constituent = constituents.get(i);
JsonObject c = new JsonObject();
writeConstituent(constituent, c);
cJson.add(c);
}
json.add("constituents", cJson);
}
List<Relation> relations = view.getRelations();
if (relations.size() > 0) {
JsonArray rJson = new JsonArray();
for (Relation r : relations) {
Constituent src = r.getSource();
Constituent tgt = r.getTarget();
int srcId = constituents.indexOf(src);
int tgtId = constituents.indexOf(tgt);
JsonObject rJ = new JsonObject();
writeString("relationName", r.getRelationName(), rJ);
if (r.getScore() != 0) writeDouble("score", r.getScore(), rJ);
writeInt("srcConstituent", srcId, rJ);
writeInt("targetConstituent", tgtId, rJ);
writeAttributes(r, rJ);
Map<String, Double> labelsToScores = r.getLabelsToScores();
if (null != labelsToScores) writeLabelsToScores(labelsToScores, rJ);
rJson.add(rJ);
}
json.add("relations", rJson);
}
}