From ab3f1f979a72126c0c49c5c5f81c1583d49a43b8 Mon Sep 17 00:00:00 2001
From: Carsten Knoll <carsten.knoll@posteo.de>
Date: Tue, 12 Dec 2023 01:44:15 +0100
Subject: [PATCH] visualization.py: enable lables in resulting svg file

---
 src/pyerk/visualization.py | 38 ++++++++++++++++++++++++++++++++------
 1 file changed, 32 insertions(+), 6 deletions(-)

diff --git a/src/pyerk/visualization.py b/src/pyerk/visualization.py
index 2c3e48e..57fa985 100644
--- a/src/pyerk/visualization.py
+++ b/src/pyerk/visualization.py
@@ -4,6 +4,7 @@
 from typing import Union, List, Tuple, Optional
 import os
 import urllib
+from rdflib import Literal
 
 import networkx as nx
 import nxv  # for graphviz visualization of networkx graphs
@@ -406,7 +407,8 @@ def create_complete_graph(
             pass
         else:
             node = create_node(item, url_template)
-        G.add_node(node, label=item.short_key, color=get_color_for_item(item))
+        label_str = f"{item.short_key}\n{item.R1__has_label}"
+        G.add_node(node, label=label_str, color=get_color_for_item(item))
         added_items_nodes[item_uri] = node
 
         # iterate over relation edges
@@ -477,6 +479,24 @@ def render_graph_to_dot(G: nx.DiGraph) -> str:
     return dot_data
 
 
+def svg_replace(raw_svg_data: str, REPLACEMENTS: dict) -> str:
+    assert isinstance(raw_svg_data, str)
+
+    # prevent some latex stuff to interfere with the handing of the `REPLACEMENTS`
+    # TODO: handle the whole problme more elegantly
+
+    latex_replacements = [(r"\dot{x}", "__LATEX1__")]
+    for orig, subs in latex_replacements:
+        raw_svg_data = raw_svg_data.replace(orig, subs)
+
+    svg_data1: str = raw_svg_data.format(**REPLACEMENTS)
+
+    for orig, subs in latex_replacements:
+        svg_data1 = svg_data1.replace(subs, orig)
+
+    return svg_data1
+
+
 def visualize_entity(uri: str, url_template="", write_tmp_files: bool = False) -> str:
     """
 
@@ -502,8 +522,8 @@ def visualize_entity(uri: str, url_template="", write_tmp_files: bool = False) -
 
     # noinspection PyUnresolvedReferences,PyProtectedMember
     raw_svg_data = nxv._graphviz.run(dot_data, algorithm="dot", format="svg", graphviz_bin=None)
-
-    svg_data1: str = raw_svg_data.decode("utf8").format(**REPLACEMENTS)
+    raw_svg_data = raw_svg_data.decode("utf8")
+    svg_data1 = svg_replace(raw_svg_data, REPLACEMENTS)
 
     if write_tmp_files:
         # for debugging
@@ -521,6 +541,12 @@ def visualize_entity(uri: str, url_template="", write_tmp_files: bool = False) -
     return svg_data1
 
 
+def get_label(entity):
+    res = entity.get("label", "undefined label")
+    if isinstance(res, Literal):
+        return res.value
+    return res
+
 def visualize_all_entities(url_template="", write_tmp_files: bool = False) -> str:
     G = create_complete_graph(url_template)
 
@@ -542,7 +568,7 @@ def visualize_all_entities(url_template="", write_tmp_files: bool = False) -> st
             "width": 1.3,
             "fontsize": 10,
             "color": d.get("color", "black"),
-            "label": d.get("label", "undefined label"),
+            "label": get_label(d),
             "shape": d.get("shape", "circle"),  # see also AbstractNode.shape
         },
         # u: node1, v: node1, d: its attribute dict
@@ -562,7 +588,7 @@ def visualize_all_entities(url_template="", write_tmp_files: bool = False) -> st
             "color": d.get("color", "black"),
             "width": 0.3,
             "fontsize": 2,
-            "label": d.get("label", "undefined label"),
+            "label": get_label(d),
             "fillcolor": "#45454533",
         },
         edge=lambda u, v, d: {
@@ -581,7 +607,7 @@ def visualize_all_entities(url_template="", write_tmp_files: bool = False) -> st
     dot_data = raw_dot_data
     # noinspection PyUnresolvedReferences,PyProtectedMember
     raw_svg_data = nxv._graphviz.run(dot_data, algorithm="dot", format="svg", graphviz_bin=None)
-    svg_data1: str = raw_svg_data.decode("utf8").format(**REPLACEMENTS)
+    svg_data1 = svg_replace(raw_svg_data.decode("utf8"), REPLACEMENTS)
 
     if write_tmp_files:
         # for debugging