Adding order and direction to SwitchNode serialization (#346)

Signed-off-by: Marin Pinchinat-Loth <marin.pinchinat@rte-france.com>
powsybl · Feb 11, 2022 · 4c9d03e · 4c9d03e
1 parent 8790ed1
commit 4c9d03e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/single-line-diagram-core/src/main/java/com/powsybl/sld/model/SwitchNode.java b/single-line-diagram-core/src/main/java/com/powsybl/sld/model/SwitchNode.java
@@ -11,6 +11,7 @@
 
 import java.io.IOException;
 import java.util.Objects;
+import java.util.Optional;
 
 /**
  * @author Benoit Jeanson <benoit.jeanson at rte-france.com>
@@ -50,5 +51,12 @@ public Node getOtherAdjNode(Node adj) {
     protected void writeJsonContent(JsonGenerator generator, boolean includeCoordinates) throws IOException {
         super.writeJsonContent(generator, includeCoordinates);
         generator.writeStringField("kind", kind.name());
+        Optional<Integer> optOrder = getOrder();
+        if (optOrder.isPresent()) {
+            generator.writeNumberField("order", optOrder.get());
+        }
+        if (getDirection() != null && getDirection() != BusCell.Direction.UNDEFINED) {
+            generator.writeStringField("direction", getDirection().name());
+        }
     }
 }
diff --git a/single-line-diagram-core/src/test/resources/TestInternCellExplicitPosition.json b/single-line-diagram-core/src/test/resources/TestInternCellExplicitPosition.json
@@ -123,7 +123,9 @@
     "y" : 345.0,
     "rotationAngle" : 90.0,
     "open" : false,
-    "kind" : "BREAKER"
+    "kind" : "BREAKER",
+    "order" : 2,
+    "direction" : "BOTTOM"
   }, {
     "type" : "SWITCH",
     "id" : "bc2",
@@ -135,7 +137,8 @@
     "y" : 240.0,
     "rotationAngle" : 90.0,
     "open" : false,
-    "kind" : "BREAKER"
+    "kind" : "BREAKER",
+    "order" : 4
   }, {
     "type" : "SWITCH",
     "id" : "bl1",