add isAllBusbars (#91)

Signed-off-by: Rehili Ghazwa <[email protected]>

Author: ghazwarhili

src/main/java/org/gridsuite/modification/modifications/CreateVoltageLevelSection.java

@@ -74,29 +74,47 @@ public void apply(Network network, ReportNode subReportNode) {
         CreateVoltageLevelSections modification = new CreateVoltageLevelSectionsBuilder()
                 .withReferenceBusbarSectionId(busbarSection.getId())
                 .withCreateTheBusbarSectionsAfterTheReferenceBusbarSection(modificationInfos.isAfterBusbarSectionId())
-                .withAllBusbars(false)
+                .withAllBusbars(modificationInfos.isAllBusbars())
                 .withLeftSwitchKind(modificationInfos.getLeftSwitchKind() != null ? SwitchKind.valueOf(modificationInfos.getLeftSwitchKind()) : SwitchKind.DISCONNECTOR)
                 .withRightSwitchKind(modificationInfos.getRightSwitchKind() != null ? SwitchKind.valueOf(modificationInfos.getRightSwitchKind()) : SwitchKind.DISCONNECTOR)
                 .withSwitchPrefixId(voltageLevel.getId())
                 .withBusbarSectionPrefixId(voltageLevel.getId())
                 .build();
         modification.apply(network, true, subReportNode);
 
-        BusbarSection newBusbarSection = null;
-        for (BusbarSection bbs : voltageLevel.getNodeBreakerView().getBusbarSections()) {
-            if (!busBarIds.contains(bbs.getId())) {
-                newBusbarSection = bbs;
-                break;
+        if (modificationInfos.isAllBusbars()) {
+            List<BusbarSection> newBusbarSections = new ArrayList<>();
+            for (BusbarSection bbs : voltageLevel.getNodeBreakerView().getBusbarSections()) {
+                if (!busBarIds.contains(bbs.getId())) {
+                    newBusbarSections.add(bbs);
+                }
+            }
+            for (BusbarSection section : newBusbarSections) {
+                subReportNode.newReportNode()
+                        .withMessageTemplate("network.modification.voltageLevel.sectionCreation")
+                        .withUntypedValue("sectionId", section.getId())
+                        .withUntypedValue("busbarIndex", section.getExtension(BusbarSectionPosition.class).getBusbarIndex())
+                        .withUntypedValue("sectionIndex", section.getExtension(BusbarSectionPosition.class).getSectionIndex())
+                        .withSeverity(TypedValue.INFO_SEVERITY)
+                        .add();
+            }
+        } else {
+            BusbarSection newBusbarSection = null;
+            for (BusbarSection bbs : voltageLevel.getNodeBreakerView().getBusbarSections()) {
+                if (!busBarIds.contains(bbs.getId())) {
+                    newBusbarSection = bbs;
+                    break;
+                }
+            }
+            if (newBusbarSection != null) {
+                subReportNode.newReportNode()
+                        .withMessageTemplate("network.modification.voltageLevel.sectionCreation")
+                        .withUntypedValue("sectionId", newBusbarSection.getId())
+                        .withUntypedValue("busbarIndex", newBusbarSection.getExtension(BusbarSectionPosition.class).getBusbarIndex())
+                        .withUntypedValue("sectionIndex", newBusbarSection.getExtension(BusbarSectionPosition.class).getSectionIndex())
+                        .withSeverity(TypedValue.INFO_SEVERITY)
+                        .add();
             }
-        }
-        if (newBusbarSection != null) {
-            subReportNode.newReportNode()
-                    .withMessageTemplate("network.modification.voltageLevel.sectionCreation")
-                    .withUntypedValue("sectionId", newBusbarSection.getId())
-                    .withUntypedValue("busbarIndex", newBusbarSection.getExtension(BusbarSectionPosition.class).getBusbarIndex())
-                    .withUntypedValue("sectionIndex", newBusbarSection.getExtension(BusbarSectionPosition.class).getSectionIndex())
-                    .withSeverity(TypedValue.INFO_SEVERITY)
-                    .add();
         }
     }
 

src/test/java/org/gridsuite/modification/modifications/CreateVoltageLevelSectionTest.java

@@ -10,6 +10,8 @@
 import com.powsybl.commons.report.ReportNode;
 import com.powsybl.iidm.network.Network;
 import com.powsybl.iidm.network.Switch;
+import com.powsybl.iidm.network.VoltageLevel;
+import com.powsybl.iidm.network.extensions.BusbarSectionPositionAdder;
 import org.gridsuite.modification.NetworkModificationException;
 import org.gridsuite.modification.dto.CreateVoltageLevelSectionInfos;
 import org.gridsuite.modification.dto.ModificationInfos;
@@ -105,4 +107,30 @@ void testCreateSubReportNode() {
         modification.createSubReportNode(reportNode);
         assertLogMessage("Adding busbar section on v1", "network.modification.voltageLevel.section.created", reportNode);
     }
+
+    @Test
+    void testCreateModificationWithAllBusbars() {
+        Network network = getNetwork();
+        VoltageLevel voltageLevel = network.getVoltageLevel("v1");
+        var bbs = voltageLevel.getNodeBreakerView().newBusbarSection()
+                .setId("bbs1_2")
+                .setName("bbs1_2")
+                .setNode(1)
+                .add();
+        bbs.newExtension(BusbarSectionPositionAdder.class).withBusbarIndex(1).withSectionIndex(0).add();
+        CreateVoltageLevelSectionInfos.builder()
+                .stashed(false)
+                .voltageLevelId("v1")
+                .busbarSectionId("bbs1_2")
+                .busbarIndex(2)
+                .isAfterBusbarSectionId(true)
+                .leftSwitchKind("BREAKER")
+                .rightSwitchKind("DISCONNECTOR")
+                .isAllBusbars(true)
+                .build().toModification().apply(network);
+        List<String> busBarIds = new ArrayList<>();
+        getNetwork().getBusbarSections().forEach(busbarSection -> busBarIds.add(busbarSection.getId()));
+        assertEquals(7, busBarIds.size());
+        assertTrue(busBarIds.containsAll(List.of("bbs1", "bbs2", "bbs3", "bbs4", "bbs1_2", "v1_0_1", "v1_1_1")));
+    }
 }