Depth-first search algorithm for BusBarSection IDs

src/main/java/org/gridsuite/network/map/dto/definition/extension/BusbarSectionFinderTraverser.java

@@ -6,42 +6,141 @@
  */
 package org.gridsuite.network.map.dto.definition.extension;
 
-import com.powsybl.iidm.network.IdentifiableType;
-import com.powsybl.iidm.network.Switch;
-import com.powsybl.iidm.network.Terminal;
-import com.powsybl.math.graph.TraverseResult;
+import com.powsybl.iidm.network.*;
+import com.powsybl.iidm.network.extensions.BusbarSectionPosition;
+
+import java.util.*;
 
 /**
- * @author Slimane Amar <slimane.amar at rte-france.com>
+ * @author Ghazwa Rehili <ghazwa.rehili at rte-france.com>
  */
-public class BusbarSectionFinderTraverser implements Terminal.TopologyTraverser {
+// TODO : to remove when this class is available in network-store
+public final class BusbarSectionFinderTraverser {
+
+    private BusbarSectionFinderTraverser() {
+        throw new UnsupportedOperationException();
+    }
 
-    private final boolean onlyConnectedBbs;
+    private record NodePath(int startNode, List<SwitchInfo> traversedSwitches, SwitchInfo lastSwitch) { }
 
-    private String firstTraversedBbsId;
+    public record SwitchInfo(String id, boolean isOpen) { }
 
-    public BusbarSectionFinderTraverser(boolean onlyConnectedBbs) {
-        this.onlyConnectedBbs = onlyConnectedBbs;
+    public record BusbarSectionResult(String busbarSectionId, int depth, SwitchInfo lastSwitch, int busbarIndex, int sectionIndex) { }
+
+    public static String findBusbarSectionId(Terminal terminal) {
+        BusbarSectionResult result = getBusbarSectionResult(terminal);
+        return result != null ? result.busbarSectionId() : terminal.getVoltageLevel().getNodeBreakerView().getBusbarSections().iterator().next().getId();
     }
 
-    @Override
-    public TraverseResult traverse(Terminal terminal, boolean connected) {
+    public static BusbarSectionResult getBusbarSectionResult(Terminal terminal) {
+        VoltageLevel.NodeBreakerView view = terminal.getVoltageLevel().getNodeBreakerView();
+        int startNode = terminal.getNodeBreakerView().getNode();
+        List<BusbarSectionResult> allResults = searchAllBusbars(view, startNode);
+        if (allResults.isEmpty()) {
+            return null;
+        }
+        return selectBestBusbar(allResults);
+    }
+
+    private static BusbarSectionResult selectBestBusbar(List<BusbarSectionResult> results) {
+        // Priority 1: Busbars without switch direct connection
+        List<BusbarSectionResult> withoutSwitch = results.stream().filter(r -> r.lastSwitch() == null).toList();
+        if (!withoutSwitch.isEmpty()) {
+            return withoutSwitch.stream().min(Comparator.comparingInt(BusbarSectionResult::depth)
+                    .thenComparingInt(BusbarSectionResult::busbarIndex)
+                    .thenComparingInt(BusbarSectionResult::sectionIndex)).orElse(null);
+        }
+        // Priority 2: Search for busbar with closed last switch
+        List<BusbarSectionResult> withClosedSwitch = results.stream().filter(r -> r.lastSwitch() != null && !r.lastSwitch().isOpen()).toList();
+        if (!withClosedSwitch.isEmpty()) {
+            return withClosedSwitch.stream().min(Comparator.comparingInt(BusbarSectionResult::depth)
+                    .thenComparingInt(BusbarSectionResult::busbarIndex)
+                    .thenComparingInt(BusbarSectionResult::sectionIndex)).orElse(null);
+        }
+        // Priority 3: Search for busbar with open last switch
+        List<BusbarSectionResult> withOpenSwitch = results.stream().filter(r -> r.lastSwitch() != null && r.lastSwitch().isOpen()).toList();
+        if (!withOpenSwitch.isEmpty()) {
+            return withOpenSwitch.stream().min(Comparator.comparingInt(BusbarSectionResult::depth)
+                    .thenComparingInt(BusbarSectionResult::busbarIndex)
+                    .thenComparingInt(BusbarSectionResult::sectionIndex)).orElse(null);
+        }
+        return results.getFirst();
+    }
+
+    private static List<BusbarSectionResult> searchAllBusbars(VoltageLevel.NodeBreakerView view, int startNode) {
+        List<BusbarSectionResult> results = new ArrayList<>();
+        Set<Integer> visited = new HashSet<>();
+        Queue<NodePath> nodePathsToVisit = new LinkedList<>();
+        nodePathsToVisit.offer(new NodePath(startNode, new ArrayList<>(), null));
+        while (!nodePathsToVisit.isEmpty()) {
+            NodePath currentNodePath = nodePathsToVisit.poll();
+            if (hasBeenVisited(currentNodePath.startNode(), visited)) {
+                continue;
+            }
+            visited.add(currentNodePath.startNode());
+            Optional<BusbarSectionResult> busbarSectionResult = findBusbarSectionAtNode(view, currentNodePath);
+            if (busbarSectionResult.isPresent()) {
+                results.add(busbarSectionResult.get());
+            } else {
+                exploreAdjacentNodes(view, currentNodePath, visited, nodePathsToVisit);
+            }
+        }
+        return results;
+    }
+
+    private static boolean hasBeenVisited(int node, Set<Integer> visited) {
+        return visited.contains(node);
+    }
+
+    private static Optional<BusbarSectionResult> findBusbarSectionAtNode(VoltageLevel.NodeBreakerView view, NodePath currentNodePath) {
+        Optional<Terminal> nodeTerminal = view.getOptionalTerminal(currentNodePath.startNode());
+        if (nodeTerminal.isEmpty()) {
+            return Optional.empty();
+        }
+        Terminal terminal = nodeTerminal.get();
         if (terminal.getConnectable().getType() == IdentifiableType.BUSBAR_SECTION) {
-            firstTraversedBbsId = terminal.getConnectable().getId();
-            return TraverseResult.TERMINATE_TRAVERSER;
+            String busbarSectionId = terminal.getConnectable().getId();
+            int depth = currentNodePath.traversedSwitches().size();
+            SwitchInfo lastSwitch = currentNodePath.lastSwitch();
+            BusbarSection busbarSection = (BusbarSection) terminal.getConnectable();
+            int busbarIndex = 1;
+            int sectionIndex = 1;
+            var busbarSectionPosition = busbarSection.getExtension(BusbarSectionPosition.class);
+            if (busbarSectionPosition != null) {
+                busbarIndex = busbarSectionPosition.getBusbarIndex();
+                sectionIndex = busbarSectionPosition.getSectionIndex();
+            }
+            return Optional.of(new BusbarSectionResult(busbarSectionId, depth, lastSwitch, busbarIndex, sectionIndex));
         }
-        return TraverseResult.CONTINUE;
+        return Optional.empty();
     }
 
-    @Override
-    public TraverseResult traverse(Switch aSwitch) {
-        if (onlyConnectedBbs && aSwitch.isOpen()) {
-            return TraverseResult.TERMINATE_PATH;
+    private static void exploreAdjacentNodes(VoltageLevel.NodeBreakerView view, NodePath currentNodePath, Set<Integer> visited, Queue<NodePath> nodePathsToVisit) {
+        view.getSwitchStream().forEach(sw -> {
+            int node1 = view.getNode1(sw.getId());
+            int node2 = view.getNode2(sw.getId());
+            Optional<Integer> nextNode = getNextNodeIfAdjacent(currentNodePath.startNode(), node1, node2);
+            if (nextNode.isPresent() && !visited.contains(nextNode.get())) {
+                NodePath newNodePath = createNodePath(currentNodePath, sw, nextNode.get());
+                nodePathsToVisit.offer(newNodePath);
+            }
+        });
+    }
+
+    private static Optional<Integer> getNextNodeIfAdjacent(int currentNode, int node1, int node2) {
+        if (node1 == currentNode) {
+            return Optional.of(node2);
+        }
+        if (node2 == currentNode) {
+            return Optional.of(node1);
         }
-        return TraverseResult.CONTINUE;
+        return Optional.empty();
     }
 
-    public String getFirstTraversedBbsId() {
-        return firstTraversedBbsId;
+    private static NodePath createNodePath(NodePath currentNodePath, Switch sw, int nextNode) {
+        List<SwitchInfo> newPathSwitches = new ArrayList<>(currentNodePath.traversedSwitches());
+        SwitchInfo switchInfo = new SwitchInfo(sw.getId(), sw.isOpen());
+        newPathSwitches.add(switchInfo);
+        return new NodePath(nextNode, newPathSwitches, switchInfo);
     }
 }

src/main/java/org/gridsuite/network/map/dto/utils/ElementUtils.java

@@ -8,7 +8,6 @@
 
 import com.powsybl.iidm.network.*;
 import com.powsybl.iidm.network.extensions.ConnectablePosition;
-import com.powsybl.math.graph.TraversalType;
 import org.gridsuite.network.map.dto.common.ReactiveCapabilityCurveMapData;
 import org.gridsuite.network.map.dto.common.TapChangerData;
 import org.gridsuite.network.map.dto.common.TapChangerStepData;
@@ -92,9 +91,8 @@ public static String getBusOrBusbarSection(Terminal terminal) {
                 return terminal.getBusBreakerView().getConnectableBus().getId();
             }
         } else {
-            final BusbarSectionFinderTraverser connectedBusbarSectionFinder = new BusbarSectionFinderTraverser(terminal.isConnected());
-            terminal.traverse(connectedBusbarSectionFinder, TraversalType.BREADTH_FIRST);
-            return connectedBusbarSectionFinder.getFirstTraversedBbsId();
+            // NODE_BREAKER: explore all paths and choose the busbar with the closed disconnector
+            return BusbarSectionFinderTraverser.findBusbarSectionId(terminal);
         }
     }