71 lines
2.4 KiB
Java
Executable File
71 lines
2.4 KiB
Java
Executable File
package org.gcube.data.analysis.nlphub.mapper;
|
|
|
|
import java.io.BufferedReader;
|
|
import java.io.InputStreamReader;
|
|
import java.net.HttpURLConnection;
|
|
import java.net.URL;
|
|
import java.util.Set;
|
|
|
|
import com.google.gson.JsonObject;
|
|
import com.google.gson.JsonParser;
|
|
import com.google.gson.JsonArray;
|
|
import com.google.gson.JsonElement;
|
|
|
|
import org.gcube.data.analysis.nlphub.legacy.NerAlgorithm;
|
|
import org.gcube.data.analysis.nlphub.legacy.NerAnnotationData;
|
|
import org.gcube.data.analysis.nlphub.legacy.NerEntity;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
public class DefaultMapper implements JsonMapper {
|
|
private static final Logger logger = LoggerFactory.getLogger(DefaultMapper.class);
|
|
|
|
public String getJson(String alg, String link) {
|
|
NerAlgorithm algInfo = new NerAlgorithm(alg);
|
|
try {
|
|
HttpURLConnection connection = (HttpURLConnection) new URL(link).openConnection();
|
|
connection.setRequestMethod("GET");
|
|
BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream()));
|
|
JsonObject jsonRoot = (JsonObject) new JsonParser().parse(reader);
|
|
JsonElement entities = jsonRoot.get("entities");
|
|
if ((entities != null) && (entities.isJsonObject())) {
|
|
JsonObject e = (JsonObject) entities;
|
|
Set<String> properties = e.keySet();
|
|
for (String p : properties) {
|
|
JsonElement property = e.get(p);
|
|
if (!property.isJsonArray())
|
|
continue;
|
|
NerAnnotationData data = new NerAnnotationData(p);
|
|
JsonArray jsonAnnotation = (JsonArray) property;
|
|
int size = jsonAnnotation.size();
|
|
for (int i = 0; i < size; i++) {
|
|
JsonElement ann = jsonAnnotation.get(i);
|
|
if (ann.isJsonObject()) {
|
|
JsonObject annotation = (JsonObject) ann;
|
|
Set<String> fields = annotation.keySet();
|
|
JsonArray indices = (JsonArray) annotation.get("indices");
|
|
NerEntity ne = new NerEntity(indices.get(0).getAsInt(), indices.get(1).getAsInt());
|
|
|
|
for (String f : fields) {
|
|
if (f.equals("indices"))
|
|
continue;
|
|
JsonElement je = annotation.get(f);
|
|
if (!je.isJsonArray() && !je.isJsonObject() && !je.isJsonNull()) {
|
|
ne.addProperty(f, je.getAsString());
|
|
}
|
|
}
|
|
data.addNerEntity(ne);
|
|
}
|
|
}
|
|
algInfo.addAnnotationData(data);
|
|
}
|
|
}
|
|
return algInfo.toJson().toString();
|
|
} catch (Exception e) {
|
|
logger.error(e.getLocalizedMessage(),e);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
}
|