fix(tree-view): Fix direction of Generalization edges

capellambse_context_diagrams/collectors/tree_view.py

@@ -80,8 +80,8 @@ def process_class(self, cls, params):
                 self.data["edges"].append(
                     {
                         "id": edge.uuid,
-                        "sources": [cls.generalizes.uuid],
-                        "targets": [cls.source.uuid],
+                        "sources": [cls.source.uuid],
+                        "targets": [cls.generalizes.uuid],
                     }
                 )
 
@@ -122,9 +122,9 @@ def collector(
     """Return the class tree data for ELK."""
     assert isinstance(diagram.target, information.Class)
     data = generic.collector(diagram, no_symbol=True)
-    all_associations: cabc.Iterable[information.Association] = (
-        diagram._model.search("Association")
-    )
+    all_associations: cabc.Iterable[
+        information.Association
+    ] = diagram._model.search("Association")
     _set_layout_options(data, params)
     processor = ClassProcessor(data, all_associations)
     processor._set_data_types_and_labels(data["children"][0], diagram.target)

capellambse_context_diagrams/serializers.py

@@ -9,6 +9,7 @@
 """
 from __future__ import annotations
 
+import collections.abc as cabc
 import logging
 import typing as t
 
@@ -149,6 +150,8 @@ class type that stores all previously named classes.
         elif child["type"] == "edge":
             styleclass = child.get("styleclass", styleclass)  # type: ignore[assignment]
             styleclass = REMAP_STYLECLASS.get(styleclass, styleclass)  # type: ignore[arg-type]
+            EDGE_HANDLER.get(styleclass, lambda c: c)(child)
+
             if child["routingPoints"]:
                 refpoints = [
                     ref + (point["x"], point["y"])
@@ -315,3 +318,16 @@ def route_shortest_connection(
         line_end, source=line_start, style=diagram.RoutingStyle.OBLIQUE
     )
     return [source_intersection, target_intersection]
+
+
+def reverse_edge_refpoints(child: _elkjs.ELKOutputEdge) -> None:
+    source = child["sourceId"]
+    target = child["targetId"]
+    child["targetId"] = source
+    child["sourceId"] = target
+    child["routingPoints"] = child["routingPoints"][::-1]
+
+
+EDGE_HANDLER: dict[str, cabc.Callable[[_elkjs.ELKOutputEdge], None]] = {
+    "Generalization": reverse_edge_refpoints
+}