Add limit violation custom highlighting style support

Signed-off-by: Ayoub LABIDI <[email protected]>

Author: ayolab

network-area-diagram/src/main/java/com/powsybl/nad/svg/iidm/TopologicalStyleProvider.java

@@ -11,30 +11,55 @@
 import com.powsybl.iidm.network.Network;
 import com.powsybl.iidm.network.Terminal;
 import com.powsybl.nad.model.*;
-
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
+import java.util.Map;
 import java.util.Optional;
 
 /**
  * @author Florian Dupuy {@literal <florian.dupuy at rte-france.com>}
  */
 public class TopologicalStyleProvider extends AbstractVoltageStyleProvider {
 
+    private final Map<String, String> violationStyles;
+
     public TopologicalStyleProvider(Network network) {
-        super(network);
+        this(network, Collections.emptyMap());
     }
 
     public TopologicalStyleProvider(Network network, BaseVoltagesConfig baseVoltageStyle) {
+        this(network, baseVoltageStyle, Collections.emptyMap());
+    }
+
+    public TopologicalStyleProvider(Network network, Map<String, String> violationStyles) {
+        super(network);
+        this.violationStyles = violationStyles != null ? violationStyles : Collections.emptyMap();
+    }
+
+    public TopologicalStyleProvider(Network network, BaseVoltagesConfig baseVoltageStyle, Map<String, String> violationStyles) {
         super(network, baseVoltageStyle);
+        this.violationStyles = violationStyles != null ? violationStyles : Collections.emptyMap();
     }
 
     @Override
     public List<String> getCssFilenames() {
         return Collections.singletonList("topologicalStyle.css");
     }
 
+    @Override
+    public List<String> getBranchEdgeStyleClasses(BranchEdge branchEdge) {
+        // Check custom violations first
+        if (!violationStyles.isEmpty()) {
+            String customStyle = violationStyles.get(branchEdge.getEquipmentId());
+            if (customStyle != null && !customStyle.isBlank()) {
+                return List.of(customStyle);
+            }
+        }
+        // Fallback to default overload detection
+        return super.getBranchEdgeStyleClasses(branchEdge);
+    }
+
     @Override
     public List<String> getBusNodeStyleClasses(BusNode busNode) {
         List<String> styles = new ArrayList<>(super.getBusNodeStyleClasses(busNode));

single-line-diagram/single-line-diagram-core/src/main/java/com/powsybl/sld/svg/styles/iidm/LimitHighlightStyleProvider.java

@@ -22,17 +22,43 @@
  */
 public class LimitHighlightStyleProvider extends EmptyStyleProvider {
     private final Network network;
+    private final Map<String, String> violationStyles;
 
     public LimitHighlightStyleProvider(Network network) {
+        this(network, Collections.emptyMap());
+    }
+
+    public LimitHighlightStyleProvider(Network network, Map<String, String> violationStyles) {
         this.network = network;
+        this.violationStyles = violationStyles != null ? violationStyles : Collections.emptyMap();
     }
 
     @Override
     public List<String> getEdgeStyles(Graph graph, Edge edge) {
+        // Check custom violations first
+        if (!violationStyles.isEmpty()) {
+            Optional<String> customStyle = getCustomViolationStyle(edge);
+            if (customStyle.isPresent()) {
+                return List.of(customStyle.get());
+            }
+        }
+        // Fallback to default overload detection
         Optional<String> overloadStyle = getOverloadStyle(edge);
         return overloadStyle.map(Collections::singletonList).orElse(Collections.emptyList());
     }
 
+    private Optional<String> getCustomViolationStyle(Edge edge) {
+        for (Node node : edge.getNodes()) {
+            if (node instanceof FeederNode feederNode) {
+                String style = violationStyles.get(feederNode.getEquipmentId());
+                if (style != null && !style.isBlank()) {
+                    return Optional.of(style);
+                }
+            }
+        }
+        return Optional.empty();
+    }
+
     private Optional<String> getOverloadStyle(Edge edge) {
         List<Node> nodes = edge.getNodes();
         for (Node node : nodes) {