use of org.corpus_tools.salt.common.SPointingRelation in project ANNIS by korpling.
the class CorefVisualizer method getAllTokens.
/**
* collects all Tokens of the component
* @param n SStructuredNode to start with
* @param name String that determines which component we search for
* @param c componenttype, that will include its Tokens
* @param cnr Number of the component
* @return List of Tokens
*/
private List<String> getAllTokens(SStructuredNode n, String name, TComponenttype c, long cnr, String namespace) {
List<String> result = null;
if (!visitedNodes.contains(n.getId())) {
result = new LinkedList<String>();
visitedNodes.add(n.getId());
if (tokensOfNode.containsKey(n.getId())) {
for (String t : tokensOfNode.get(n.getId())) {
result.add(t);
if (componentOfToken.get(t) == null) {
List<Long> newlist = new LinkedList<Long>();
newlist.add(cnr);
componentOfToken.put(t, newlist);
} else {
if (!componentOfToken.get(t).contains(cnr)) {
componentOfToken.get(t).add(cnr);
}
}
}
} else {
result = searchTokens(n, cnr);
if (result != null) {
tokensOfNode.put(n.getId(), result);
}
}
// get "P"-Edges!
List<SRelation<SNode, SNode>> outEdges = n.getGraph().getOutRelations(n.getId());
if (outEdges != null) {
for (SRelation<? extends SNode, ? extends SNode> e : outEdges) {
if (includeEdge(e, namespace)) {
SPointingRelation rel = (SPointingRelation) e;
if (name.equals(componentNameForRelation(rel)) && !visitedNodes.contains(rel.getTarget().getId())) {
c.nodeList.add(rel.getTarget().getId());
List<String> Med = getAllTokens(rel.getTarget(), name, c, cnr, namespace);
for (String l : Med) {
if (result != null && !result.contains(l)) {
result.add(l);
}
}
}
}
}
}
List<SRelation<SNode, SNode>> inEdges = n.getGraph().getInRelations(n.getId());
if (inEdges != null) {
for (SRelation<? extends SNode, ? extends SNode> e : inEdges) {
if (includeEdge(e, namespace)) {
SPointingRelation rel = (SPointingRelation) e;
if (name.equals(componentNameForRelation(rel)) && !visitedNodes.contains(rel.getSource().getId())) {
c.nodeList.add(rel.getSource().getId());
List<String> Med = getAllTokens(rel.getSource(), name, c, cnr, namespace);
for (String s : Med) {
if (result != null && !result.contains(s)) {
result.add(s);
}
}
}
}
}
}
}
return result;
}
use of org.corpus_tools.salt.common.SPointingRelation in project ANNIS by korpling.
the class CorefVisualizer method writeOutput.
/**
* writes Output for the CorefVisualizer
* @param writer writer to write with
*/
@Override
public void writeOutput(VisualizerInput input, Writer w) {
// root html element
Html html = new Html();
Head head = new Head();
Body body = new Body();
html.removeXmlns();
html.appendChild(head);
html.appendChild(body);
try {
LinkedList<String> fonts = new LinkedList<String>();
if (input.getFont() != null) {
Link linkFont = new Link();
linkFont.setHref(input.getFont().getUrl());
head.appendChild(linkFont);
fonts.add(input.getFont().getName());
}
fonts.add("serif");
Link linkJQueryUI = new Link();
linkJQueryUI.setHref(input.getResourcePath("coref/jquery-ui-1.11.4.custom/jquery-ui.min.css"));
linkJQueryUI.setRel("stylesheet");
linkJQueryUI.setType("text/css");
head.appendChild(linkJQueryUI);
Link linkJQueryUIStructure = new Link();
linkJQueryUIStructure.setHref(input.getResourcePath("coref/jquery-ui-1.11.4.custom/jquery-ui.structure.min.css"));
linkJQueryUIStructure.setRel("stylesheet");
linkJQueryUIStructure.setType("text/css");
head.appendChild(linkJQueryUIStructure);
Script scriptJquery = new Script("text/javascript");
scriptJquery.setSrc(input.getResourcePath("coref/jquery.js"));
head.appendChild(scriptJquery);
Script scriptUI = new Script("text/javascript");
scriptUI.setSrc(input.getResourcePath("coref/jquery-ui-1.11.4.custom/jquery-ui.min.js"));
head.appendChild(scriptUI);
Link linkCoref = new Link();
linkCoref.setHref(input.getResourcePath("coref/coref.css"));
linkCoref.setRel("stylesheet");
linkCoref.setType("text/css");
head.appendChild(linkCoref);
Script scriptCoref = new Script("text/javascript");
scriptCoref.setSrc(input.getResourcePath("coref/CorefVisualizer.js"));
head.appendChild(scriptCoref);
body.setStyle("font-family: '" + StringUtils.join(fonts, "', '") + "';");
// get Info
globalIndex = 0;
tokensOfNode = new HashMap<String, List<String>>();
referentList = new LinkedList<TReferent>();
komponent = new LinkedList<TComponent>();
referentOfToken = new HashMap<String, HashMap<Long, Integer>>();
componentOfToken = new HashMap<String, List<Long>>();
componenttype = new LinkedList<TComponenttype>();
SDocument saltDoc = input.getDocument();
SDocumentGraph saltGraph = saltDoc.getDocumentGraph();
if (saltGraph == null) {
body.setText("An Error occured: Could not get Graph of Result (Graph == null).");
return;
}
List<SRelation<SNode, SNode>> edgeList = saltGraph.getRelations();
if (edgeList == null) {
return;
}
for (SRelation rawRel : edgeList) {
if (includeEdge(rawRel, input.getNamespace())) {
SPointingRelation rel = (SPointingRelation) rawRel;
String relType = componentNameForRelation(rel);
visitedNodes = new LinkedList<String>();
// got type for this?
boolean gotIt = false;
int componentnr;
for (componentnr = 0; componentnr < componenttype.size(); componentnr++) {
if (componenttype.get(componentnr) != null && componenttype.get(componentnr).type != null && componenttype.get(componentnr).nodeList != null && componenttype.get(componentnr).type.equals(relType) && componenttype.get(componentnr).nodeList.contains(rel.getSource().getId())) {
gotIt = true;
break;
}
}
TComponent currentComponent;
TComponenttype currentComponenttype;
if (gotIt) {
currentComponent = komponent.get(componentnr);
currentComponenttype = componenttype.get(componentnr);
} else {
currentComponenttype = new TComponenttype();
currentComponenttype.type = relType;
componenttype.add(currentComponenttype);
componentnr = komponent.size();
currentComponent = new TComponent();
currentComponent.type = relType;
currentComponent.tokenList = new LinkedList<String>();
komponent.add(currentComponent);
currentComponenttype.nodeList.add(rel.getSource().getId());
}
TReferent ref = new TReferent();
ref.annotations = new HashSet<SerializableAnnotation>();
for (SAnnotation anno : rel.getAnnotations()) {
ref.annotations.add(new SerializableAnnotation(anno));
}
ref.component = componentnr;
referentList.add(ref);
List<String> currentTokens = getAllTokens(rel.getSource(), componentNameForRelation(rel), currentComponenttype, componentnr, input.getNamespace());
// neu
setReferent(rel.getTarget(), globalIndex, 0);
// neu
setReferent(rel.getSource(), globalIndex, 1);
for (String s : currentTokens) {
if (!currentComponent.tokenList.contains(s)) {
currentComponent.tokenList.add(s);
}
}
globalIndex++;
}
}
colorlist = new HashMap<Integer, Integer>();
// A list containing all the generated HTML elements, one list entry
// for each text.
List<List<Node>> nodesPerText = new LinkedList<List<Node>>();
// write output for each text separatly
List<STextualDS> texts = saltGraph.getTextualDSs();
if (texts != null && !texts.isEmpty()) {
for (STextualDS t : texts) {
DataSourceSequence<Integer> sequence = new DataSourceSequence<>(t, 0, (t.getText() != null) ? t.getText().length() : 0);
List<SToken> token = saltGraph.getSortedTokenByText(saltGraph.getTokensBySequence(sequence));
if (token != null) {
boolean validText = true;
if (Boolean.parseBoolean(input.getMappings().getProperty("hide_empty", "false"))) {
validText = false;
// check if the text contains any matching annotations
for (SToken tok : token) {
/*
* The token is only added to this map if an valid edge
* (according to the resolver trigger) conntected to
* this token was found.
*/
if (referentOfToken.get(tok.getId()) != null && !referentOfToken.get(tok.getId()).isEmpty()) {
validText = true;
break;
}
}
}
if (validText) {
List<Node> nodes = outputSingleText(token, input);
nodesPerText.add(nodes);
}
}
}
// end for each STexutalDS
/*
* Append the generated output to the body, wrap in table if necessary.
*/
// present all texts as columns side by side if using multiple texts
Table tableTexts = new Table();
Tr trTextRow = new Tr();
trTextRow.setCSSClass("textRow");
// only append wrapper table if we have multiple texts
if (nodesPerText.size() > 1) {
body.appendChild(tableTexts);
tableTexts.appendChild(trTextRow);
}
for (List<Node> nodes : nodesPerText) {
// multi-text mode?
if (nodesPerText.size() > 1) {
Td tdSingleText = new Td();
trTextRow.appendChild(tdSingleText);
tdSingleText.setCSSClass("text");
tdSingleText.appendChild(nodes);
} else {
body.appendChild(nodes);
}
}
} else {
Text errorTxt = new Text("Could not find any texts for the " + input.getNamespace() + " node namespace (layer).");
body.appendChild(errorTxt);
}
// write HTML4 transitional doctype
w.append(new Doctype(DocumentType.HTMLTransitional).write());
// append the html tree
w.append(html.write());
} catch (IOException ex) {
log.error(null, ex);
}
}
use of org.corpus_tools.salt.common.SPointingRelation in project ANNIS by korpling.
the class VakyarthaDependencyTree method printHTMLOutput.
public void printHTMLOutput(VisualizerInput input, Writer writer, Map<SNode, Integer> selectedNodes) {
SDocumentGraph sDocumentGraph = input.getSResult().getDocumentGraph();
for (SNode n : sDocumentGraph.getNodes()) {
if (selectNode(n)) {
RelannisNodeFeature feat = (RelannisNodeFeature) n.getFeature(ANNIS_NS, FEAT_RELANNIS_NODE).getValue();
int tokenIdx = feat != null ? (int) feat.getTokenIndex() : -1;
selectedNodes.put(n, tokenIdx);
}
}
Map<SNode, Integer> node2Int = new HashMap<SNode, Integer>();
int count = 0;
for (SNode tok : selectedNodes.keySet()) {
node2Int.put(tok, count++);
}
try {
println("<html>", writer);
println("<head>", writer);
LinkedList<String> fontsText = new LinkedList<String>();
LinkedList<String> fontsDep = new LinkedList<String>();
if (input.getFont() != null) {
fontsText.add(input.getFont().getName());
fontsDep.add(input.getFont().getName());
println("<link href=\"" + input.getFont().getUrl() + "\" rel=\"stylesheet\" type=\"text/css\" >", writer);
}
fontsText.add("sans-serif");
fontsDep.add("serif");
println("<script type=\"text/javascript\" src=\"" + input.getResourcePath("vakyartha/jquery.js") + "\"></script>", writer);
println("<script type=\"text/javascript\" src=\"" + input.getResourcePath("vakyartha/raphael-min.js") + "\"></script>", writer);
println("<script type=\"text/javascript\" src=\"" + input.getResourcePath("vakyartha/vakyarthaDependency.js") + "\"></script>", writer);
// output the data for the javascript
println("<script type=\"text/javascript\">", writer);
println("fcolors={};", writer);
println("shownfeatures=[\"t\"];", writer);
println("tokens=new Object();", writer);
count = 0;
for (SNode node : selectedNodes.keySet()) {
JSONObject vakyarthaObject = new JSONObject();
String completeAnnotation = getAnnotation(node);
String annotationValue = completeAnnotation.replaceFirst(".*=", "");
String text = getText(node, input);
// decide, if the visualization is token based.
if (mappings.containsKey(MAPPING_NODE_KEY)) {
vakyarthaObject.put("t", annotationValue);
} else {
vakyarthaObject.put("t", text);
}
vakyarthaObject.put("annotation", annotationValue);
vakyarthaObject.put("text", text);
vakyarthaObject.put("tooltip", completeAnnotation);
JSONObject govs = new JSONObject();
List<SRelation<SNode, SNode>> sEdges = node.getGraph().getInRelations(node.getId());
for (SRelation<? extends SNode, ? extends SNode> e : sEdges) {
if (e instanceof SPointingRelation) {
SPointingRelation sRelation = (SPointingRelation) e;
boolean includeEdge = true;
// check layer
if (input.getNamespace() != null) {
// must be included in the layer in order to be included
includeEdge = false;
if (sRelation.getLayers() != null) {
for (SLayer layer : sRelation.getLayers()) {
if (input.getNamespace().equals(layer.getName())) {
includeEdge = true;
break;
}
}
}
}
if (includeEdge) {
SNode source = (SNode) sRelation.getSource();
String label = "";
for (SAnnotation anno : sRelation.getAnnotations()) {
label = anno.getValue_STEXT();
break;
}
if (sRelation.getSource() != null && node2Int.containsKey(source)) {
govs.put(String.valueOf(node2Int.get(source)), label);
}
}
}
// end if pointing relation
}
vakyarthaObject.put("govs", govs);
JSONObject attris = new JSONObject();
JSONObject tAttris = new JSONObject();
String tokenColor = "black";
if (input.getMarkedAndCovered().containsKey(node)) {
tokenColor = MatchedNodeColors.getHTMLColorByMatch(input.getMarkedAndCovered().get(node));
}
tAttris.put("fill", tokenColor);
tAttris.put("font", "11px " + StringUtils.join(fontsText, ","));
attris.put("t", tAttris);
JSONObject depAttris = new JSONObject();
depAttris.put("fill", "#999");
depAttris.put("font-style", "italic");
depAttris.put("font", "12px " + StringUtils.join(fontsDep, ","));
attris.put("deptext", depAttris);
vakyarthaObject.put("attris", attris);
writer.append("tokens[").append("" + count++).append("]=");
writer.append(vakyarthaObject.toString().replaceAll("\n", " "));
writer.append(";\n");
}
println("</script>", writer);
println("</head>", writer);
println("<body id=\"holder\">", writer);
// the div to render the javascript to
// println(
// "<div id=\"holder\"> </div>",
// writer);
println("</body>", writer);
println("</html>", writer);
} catch (JSONException ex) {
log.error(null, ex);
} catch (IOException ex) {
log.error(null, ex);
}
}
use of org.corpus_tools.salt.common.SPointingRelation in project ANNIS by korpling.
the class SaltAnnotateExtractor method createNewRelation.
private SRelation createNewRelation(SDocumentGraph graph, SStructuredNode sourceNode, SNode targetNode, String relationName, String type, long componentID, SLayer layer, long pre, FastInverseMap<Long, SNode> nodeByRankID, AtomicInteger numberOfRelations) {
SRelation rel = null;
if (null != type) // create new relation
{
switch(type) {
case "d":
SDominanceRelation domrel = SaltFactory.createSDominanceRelation();
// always set a name by ourself since the SDocumentGraph#basicAddRelation()
// functions otherwise real slow
domrel.setName("sDomRel" + numberOfRelations.incrementAndGet());
rel = domrel;
if (sourceNode != null && !(sourceNode instanceof SStructure)) {
log.debug("Mismatched source type: should be SStructure");
SNode oldNode = sourceNode;
sourceNode = recreateNode(SStructure.class, sourceNode);
updateMapAfterRecreatingNode(oldNode, sourceNode, nodeByRankID);
}
if (relationName == null || relationName.isEmpty()) {
// layer but has a non-empty relation name
if (handleArtificialDominanceRelation(graph, sourceNode, targetNode, rel, layer, componentID, pre)) {
// don't include this relation
rel = null;
}
}
break;
case "c":
SSpanningRelation spanrel = SaltFactory.createSSpanningRelation();
// always set a name by ourself since the SDocumentGraph#basicAddRelation()
// functions is real slow otherwise
spanrel.setName("sSpanRel" + numberOfRelations.incrementAndGet());
rel = spanrel;
sourceNode = testAndFixNonSpan(sourceNode, nodeByRankID);
break;
case "p":
SPointingRelation pointingrel = SaltFactory.createSPointingRelation();
pointingrel.setName("sPointingRel" + numberOfRelations.incrementAndGet());
rel = pointingrel;
break;
default:
throw new IllegalArgumentException("Invalid type " + type + " for new Relation");
}
try {
if (rel != null) {
rel.setType(relationName);
RelannisEdgeFeature featRelation = new RelannisEdgeFeature();
featRelation.setPre(pre);
featRelation.setComponentID(componentID);
SFeature sfeatRelation = SaltFactory.createSFeature();
sfeatRelation.setNamespace(ANNIS_NS);
sfeatRelation.setName(FEAT_RELANNIS_EDGE);
sfeatRelation.setValue(featRelation);
rel.addFeature(sfeatRelation);
rel.setSource(sourceNode);
if ("c".equals(type) && !(targetNode instanceof SToken)) {
log.warn("invalid relation detected: target node ({}) " + "of a coverage relation (from: {}, internal id {}) was not a token", new Object[] { targetNode.getName(), sourceNode == null ? "null" : sourceNode.getName(), "" + pre });
} else {
rel.setTarget(targetNode);
graph.addRelation(rel);
layer.addRelation(rel);
}
}
} catch (SaltException ex) {
log.warn("invalid relation detected", ex);
}
}
return rel;
}
use of org.corpus_tools.salt.common.SPointingRelation in project ANNIS by korpling.
the class CorefVisualizer method setReferent.
/**
* adds a Referent for all Nodes dominated or covered by outgoing Edges of AnnisNode a
* @param n the Node
* @param index index of the Referent
* @param value determines wheather the refered P-Edge is incoming (1) or outgoing (0)
*/
private void setReferent(SNode n, long index, int value) {
if (n instanceof SToken) {
SToken tok = (SToken) n;
if (!referentOfToken.containsKey(tok.getId())) {
HashMap<Long, Integer> newlist = new HashMap<Long, Integer>();
// globalindex?
newlist.put(index, value);
referentOfToken.put(tok.getId(), newlist);
} else {
referentOfToken.get(tok.getId()).put(globalIndex, value);
}
} else {
List<SRelation<SNode, SNode>> outEdges = n.getGraph().getOutRelations(n.getId());
if (outEdges != null) {
for (SRelation<? extends SNode, ? extends SNode> edge : outEdges) {
if (!(edge instanceof SPointingRelation)) {
if (edge.getSource() != null && edge.getTarget() != null) {
setReferent(edge.getTarget(), index, value);
}
}
}
}
}
}
Aggregations