Add substation creation with voltage level creation (#17)

Signed-off-by: Rehili Ghazwa <[email protected]>
gridsuite · Jan 16, 2025 · a7554a2 · a7554a2
1 parent 9ba9250
commit a7554a2
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/main/java/org/gridsuite/modification/utils/ModificationUtils.java b/src/main/java/org/gridsuite/modification/utils/ModificationUtils.java
@@ -361,14 +361,19 @@ public void createSubstation(SubstationCreationInfos substationCreationInfos,
     public void createVoltageLevel(VoltageLevelCreationInfos voltageLevelCreationInfos,
                                    ReportNode subReportNode, Network network) {
         String substationId = voltageLevelCreationInfos.getSubstationId();
-        if (voltageLevelCreationInfos.getSubstationCreation() != null) {
-            createSubstation(voltageLevelCreationInfos.getSubstationCreation(), subReportNode, network);
+        SubstationCreationInfos substationCreation = voltageLevelCreationInfos.getSubstationCreation();
+        Substation substation;
+        if (substationCreation != null) {
+            substationId = substationCreation.getEquipmentId();
+            createSubstation(substationCreation, subReportNode, network);
+            substation = network.getSubstation(substationId);
+            PropertiesUtils.applyProperties(substation, subReportNode, substationCreation.getProperties(), "SubstationProperties");
+        } else {
+            substation = network.getSubstation(substationId);
         }
-        Substation substation = network.getSubstation(substationId);
         if (substation == null) {
             throw new NetworkModificationException(SUBSTATION_NOT_FOUND, substationId);
         }
-
         VoltageLevel voltageLevel = substation.newVoltageLevel()
             .setId(voltageLevelCreationInfos.getEquipmentId())
             .setName(voltageLevelCreationInfos.getEquipmentName())