Support of LoadAction in AC and DC security analysis

src/main/java/com/powsybl/openloadflow/network/BusDcState.java

@@ -18,13 +18,15 @@ public class BusDcState extends ElementState<LfBus> {
     private final Map<String, Double> generatorsTargetP;
     private final Map<String, Boolean> participatingGenerators;
     private final double absVariableLoadTargetP;
+    private final Map<String, Boolean> loadsStatus;
 
     public BusDcState(LfBus bus) {
         super(bus);
         this.loadTargetP = bus.getLoadTargetP();
         this.generatorsTargetP = bus.getGenerators().stream().collect(Collectors.toMap(LfGenerator::getId, LfGenerator::getTargetP));
         this.participatingGenerators = bus.getGenerators().stream().collect(Collectors.toMap(LfGenerator::getId, LfGenerator::isParticipating));
         this.absVariableLoadTargetP = bus.getAggregatedLoads().getAbsVariableLoadTargetP();
+        this.loadsStatus = bus.getAggregatedLoads().getLoadsStatus();
     }
 
     @Override
@@ -34,6 +36,7 @@ public void restore() {
         element.getGenerators().forEach(g -> g.setTargetP(generatorsTargetP.get(g.getId())));
         element.getGenerators().forEach(g -> g.setParticipating(participatingGenerators.get(g.getId())));
         element.getAggregatedLoads().setAbsVariableLoadTargetP(absVariableLoadTargetP);
+        element.getAggregatedLoads().setLoadsStatus(loadsStatus);
     }
 
     public static BusDcState save(LfBus bus) {

src/main/java/com/powsybl/openloadflow/network/LfAction.java

@@ -6,11 +6,15 @@
  */
 package com.powsybl.openloadflow.network;
 
+import com.powsybl.iidm.network.Bus;
+import com.powsybl.iidm.network.Load;
+import com.powsybl.iidm.network.Network;
+import com.powsybl.iidm.network.Terminal;
+import com.powsybl.loadflow.LoadFlowParameters;
 import com.powsybl.openloadflow.graph.GraphConnectivity;
-import com.powsybl.security.action.Action;
-import com.powsybl.security.action.LineConnectionAction;
-import com.powsybl.security.action.PhaseTapChangerTapPositionAction;
-import com.powsybl.security.action.SwitchAction;
+import com.powsybl.openloadflow.util.PerUnit;
+import com.powsybl.security.action.*;
+import org.apache.commons.lang3.tuple.Pair;
 
 import java.util.*;
 
@@ -54,58 +58,93 @@ public boolean isRelative() {
 
     private final TapPositionChange tapPositionChange;
 
-    private LfAction(String id, LfBranch disabledBranch, LfBranch enabledBranch, TapPositionChange tapPositionChange) {
+    private final Map<Pair<LfBus, String>, Pair<Double, PowerShift>> busesLoadShift; // key: bus and loadId.
+
+    private LfAction(String id, LfBranch disabledBranch, LfBranch enabledBranch, TapPositionChange tapPositionChange, Map<Pair<LfBus, String>, Pair<Double, PowerShift>> busesLoadShift) {
         this.id = Objects.requireNonNull(id);
         this.disabledBranch = disabledBranch;
         this.enabledBranch = enabledBranch;
         this.tapPositionChange = tapPositionChange;
+        this.busesLoadShift = busesLoadShift;
     }
 
-    public static Optional<LfAction> create(Action action, LfNetwork network) {
+    public static Optional<LfAction> create(Action action, LfNetwork lfNetwork, Network network, boolean breakers) {
         Objects.requireNonNull(action);
         Objects.requireNonNull(network);
         switch (action.getType()) {
             case SwitchAction.NAME:
-                return create((SwitchAction) action, network);
+                return create((SwitchAction) action, lfNetwork);
 
             case LineConnectionAction.NAME:
-                return create((LineConnectionAction) action, network);
+                return create((LineConnectionAction) action, lfNetwork);
 
             case PhaseTapChangerTapPositionAction.NAME:
-                return create((PhaseTapChangerTapPositionAction) action, network);
+                return create((PhaseTapChangerTapPositionAction) action, lfNetwork);
+
+            case LoadAction.NAME:
+                return create((LoadAction) action, lfNetwork, network, breakers);
 
             default:
                 throw new UnsupportedOperationException("Unsupported action type: " + action.getType());
         }
     }
 
-    private static Optional<LfAction> create(PhaseTapChangerTapPositionAction action, LfNetwork network) {
-        LfBranch branch = network.getBranchById(action.getTransformerId()); // only two windings transformer for the moment.
+    private static Optional<LfAction> create(LoadAction action, LfNetwork lfNetwork, Network network, boolean breakers) {
+        Load load = network.getLoad(action.getLoadId());
+        Terminal terminal = load.getTerminal();
+        Bus bus = breakers ? terminal.getBusBreakerView().getBus() : terminal.getBusView().getBus();
+        if (bus != null) {
+            LfBus lfBus = lfNetwork.getBusById(bus.getId());
+            if (!lfBus.getAggregatedLoads().isDisabled(action.getLoadId())) {
+                double activePowerShift = 0;
+                double reactivePowerShift = 0;
+                Optional<Double> activePowerValue = action.getActivePowerValue();
+                Optional<Double> reactivePowerValue = action.getReactivePowerValue();
+                if (activePowerValue.isPresent()) {
+                    activePowerShift = action.isRelativeValue() ? activePowerValue.get() : activePowerValue.get() - load.getP0();
+                }
+                if (reactivePowerValue.isPresent()) {
+                    reactivePowerShift = action.isRelativeValue() ? reactivePowerValue.get() : reactivePowerValue.get() - load.getQ0();
+                }
+                // In case of a power shift, we suppose that the shift on a load P0 is exactly the same on the variable active power
+                // of P0 that could be described in a LoadDetail extension.
+                PowerShift powerShift = new PowerShift(activePowerShift / PerUnit.SB, activePowerShift / PerUnit.SB, reactivePowerShift / PerUnit.SB);
+                Map<Pair<LfBus, String>, Pair<Double, PowerShift>> busesLoadShift = new HashMap<>();
+                busesLoadShift.put(Pair.of(lfBus, load.getId()), Pair.of(load.getP0(), powerShift));
+                return Optional.of(new LfAction(action.getId(), null, null, null, busesLoadShift));
+            }
+        }
+
+        return Optional.empty(); // could be in another component or in contingency.
+    }
+
+    private static Optional<LfAction> create(PhaseTapChangerTapPositionAction action, LfNetwork lfNetwork) {
+        LfBranch branch = lfNetwork.getBranchById(action.getTransformerId()); // only two windings transformer for the moment.
         if (branch != null) {
             if (branch.getPiModel() instanceof SimplePiModel) {
                 throw new UnsupportedOperationException("Phase tap changer tap connection action: only one tap in the branch {" + action.getTransformerId() + "}");
             } else {
                 var tapPositionChange = new TapPositionChange(branch, action.getValue(), action.isRelativeValue());
-                return Optional.of(new LfAction(action.getId(), null, null, tapPositionChange));
+                return Optional.of(new LfAction(action.getId(), null, null, tapPositionChange, Collections.emptyMap()));
             }
         }
         return Optional.empty(); // could be in another component
     }
 
-    private static Optional<LfAction> create(LineConnectionAction action, LfNetwork network) {
-        LfBranch branch = network.getBranchById(action.getLineId());
+    private static Optional<LfAction> create(LineConnectionAction action, LfNetwork lfNetwork) {
+        LfBranch branch = lfNetwork.getBranchById(action.getLineId());
         if (branch != null) {
             if (action.isOpenSide1() && action.isOpenSide2()) {
-                return Optional.of(new LfAction(action.getId(), branch, null, null));
+                return Optional.of(new LfAction(action.getId(), branch, null, null, Collections.emptyMap()));
             } else {
                 throw new UnsupportedOperationException("Line connection action: only open line at both sides is supported yet.");
             }
         }
         return Optional.empty(); // could be in another component
     }
 
-    private static Optional<LfAction> create(SwitchAction action, LfNetwork network) {
-        LfBranch branch = network.getBranchById(action.getSwitchId());
+    private static Optional<LfAction> create(SwitchAction action, LfNetwork lfNetwork) {
+        LfBranch branch = lfNetwork.getBranchById(action.getSwitchId());
         if (branch != null) {
             LfBranch disabledBranch = null;
             LfBranch enabledBranch = null;
@@ -114,7 +153,7 @@ private static Optional<LfAction> create(SwitchAction action, LfNetwork network)
             } else {
                 enabledBranch = branch;
             }
-            return Optional.of(new LfAction(action.getId(), disabledBranch, enabledBranch, null));
+            return Optional.of(new LfAction(action.getId(), disabledBranch, enabledBranch, null, Collections.emptyMap()));
         }
         return Optional.empty(); // could be in another component
     }
@@ -131,7 +170,7 @@ public LfBranch getEnabledBranch() {
         return enabledBranch;
     }
 
-    public static void apply(List<LfAction> actions, LfNetwork network, LfContingency contingency) {
+    public static void apply(List<LfAction> actions, LfNetwork network, LfContingency contingency, LoadFlowParameters.BalanceType balanceType) {
         Objects.requireNonNull(actions);
         Objects.requireNonNull(network);
 
@@ -140,7 +179,7 @@ public static void apply(List<LfAction> actions, LfNetwork network, LfContingenc
 
         // then process remaining changes of actions
         for (LfAction action : actions) {
-            action.apply();
+            action.apply(balanceType);
         }
     }
 
@@ -193,13 +232,31 @@ public void updateConnectivity(GraphConnectivity<LfBus, LfBranch> connectivity)
         }
     }
 
-    public void apply() {
+    public void apply(LoadFlowParameters.BalanceType balanceType) {
         if (tapPositionChange != null) {
             LfBranch branch = tapPositionChange.getBranch();
             int tapPosition = branch.getPiModel().getTapPosition();
             int value = tapPositionChange.getValue();
             int newTapPosition = tapPositionChange.isRelative() ? tapPosition + value : value;
             branch.getPiModel().setTapPosition(newTapPosition);
         }
+
+        if (!busesLoadShift.isEmpty()) {
+            for (var e : busesLoadShift.entrySet()) {
+                LfBus bus = e.getKey().getLeft();
+                String loadId = e.getKey().getRight();
+                if (!bus.getAggregatedLoads().isDisabled(loadId)) {
+                    Double loadP0 = e.getValue().getLeft(); // initial load P0.
+                    PowerShift shift = e.getValue().getRight();
+                    double newP0 = loadP0 / PerUnit.SB + shift.getActive();
+                    double oldUpdatedP0 = LfContingency.getUpdatedLoadP0(bus, balanceType, loadP0 / PerUnit.SB, loadP0 / PerUnit.SB);
+                    double newUpdatedP0 = LfContingency.getUpdatedLoadP0(bus, balanceType, newP0, newP0);
+                    bus.setLoadTargetP(bus.getLoadTargetP() + newUpdatedP0 - oldUpdatedP0);
+                    bus.setLoadTargetQ(bus.getLoadTargetQ() + shift.getReactive());
+                    bus.getAggregatedLoads().setAbsVariableLoadTargetP(bus.getAggregatedLoads().getAbsVariableLoadTargetP()
+                            + Math.signum(shift.getActive()) * Math.abs(shift.getVariableActive()) * PerUnit.SB);
+                }
+            }
+        }
     }
 }

src/main/java/com/powsybl/openloadflow/network/LfAggregatedLoads.java

@@ -7,6 +7,7 @@
 package com.powsybl.openloadflow.network;
 
 import java.util.List;
+import java.util.Map;
 
 /**
  * @author Anne Tilloy <anne.tilloy at rte-france.com>
@@ -22,4 +23,12 @@ public interface LfAggregatedLoads extends PropertyBag {
     double getLoadCount();
 
     double getLoadTargetQ(double diffLoadTargetP);
+
+    boolean isDisabled(String originalId);
+
+    void setDisabled(String originalId, boolean disabled);
+
+    Map<String, Boolean> getLoadsStatus();
+
+    void setLoadsStatus(Map<String, Boolean> loadsStatus);
 }

src/main/java/com/powsybl/openloadflow/network/LfContingency.java

@@ -18,6 +18,7 @@
 import java.util.Map;
 import java.util.Objects;
 import java.util.Set;
+import java.util.stream.Collectors;
 
 /**
  * @author Geoffroy Jamgotchian <geoffroy.jamgotchian at rte-france.com>
@@ -38,12 +39,14 @@ public class LfContingency {
 
     private final Map<LfBus, PowerShift> busesLoadShift;
 
+    private final Set<String> originalPowerShiftIds;
+
     private final Set<LfGenerator> lostGenerators;
 
     private double activePowerLoss = 0;
 
     public LfContingency(String id, int index, Set<LfBus> disabledBuses, Set<LfBranch> disabledBranches, Map<LfShunt, AdmittanceShift> shuntsShift,
-                         Map<LfBus, PowerShift> busesLoadShift, Set<LfGenerator> lostGenerators, Set<LfHvdc> disabledHvdcs) {
+                         Map<LfBus, PowerShift> busesLoadShift, Set<LfGenerator> lostGenerators, Set<LfHvdc> disabledHvdcs, Set<String> originalPowerShiftIds) {
         this.id = Objects.requireNonNull(id);
         this.index = index;
         this.disabledBuses = Objects.requireNonNull(disabledBuses);
@@ -52,6 +55,7 @@ public LfContingency(String id, int index, Set<LfBus> disabledBuses, Set<LfBranc
         this.shuntsShift = Objects.requireNonNull(shuntsShift);
         this.busesLoadShift = Objects.requireNonNull(busesLoadShift);
         this.lostGenerators = Objects.requireNonNull(lostGenerators);
+        this.originalPowerShiftIds = Objects.requireNonNull(originalPowerShiftIds);
         for (LfBus bus : disabledBuses) {
             activePowerLoss += bus.getGenerationTargetP() - bus.getLoadTargetP();
         }
@@ -116,6 +120,11 @@ public void apply(LoadFlowParameters.BalanceType balanceType) {
             bus.setLoadTargetP(bus.getLoadTargetP() - getUpdatedLoadP0(bus, balanceType, shift.getActive(), shift.getVariableActive()));
             bus.setLoadTargetQ(bus.getLoadTargetQ() - shift.getReactive());
             bus.getAggregatedLoads().setAbsVariableLoadTargetP(bus.getAggregatedLoads().getAbsVariableLoadTargetP() - Math.abs(shift.getVariableActive()) * PerUnit.SB);
+            Set<String> loadsIdsInContingency = originalPowerShiftIds.stream()
+                    .distinct()
+                    .filter(bus.getAggregatedLoads().getOriginalIds()::contains) // maybe not optimized.
+                    .collect(Collectors.toSet());
+            loadsIdsInContingency.stream().forEach(loadId -> bus.getAggregatedLoads().setDisabled(loadId, true));
         }
         Set<LfBus> generatorBuses = new HashSet<>();
         for (LfGenerator generator : lostGenerators) {

src/main/java/com/powsybl/openloadflow/network/impl/LfAggregatedLoadsImpl.java

@@ -12,8 +12,7 @@
 import com.powsybl.openloadflow.network.LfAggregatedLoads;
 import com.powsybl.openloadflow.util.PerUnit;
 
-import java.util.ArrayList;
-import java.util.List;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -31,6 +30,8 @@ class LfAggregatedLoadsImpl extends AbstractPropertyBag implements LfAggregatedL
 
     private boolean initialized;
 
+    private Map<String, Boolean> loadsStatus = new LinkedHashMap<>();
+
     LfAggregatedLoadsImpl(boolean distributedOnConformLoad) {
         this.distributedOnConformLoad = distributedOnConformLoad;
     }
@@ -42,6 +43,7 @@ public List<String> getOriginalIds() {
 
     void add(Load load) {
         loadsRefs.add(new Ref<>(load));
+        loadsStatus.put(load.getId(), false);
         initialized = false;
     }
 
@@ -112,6 +114,26 @@ public double getLoadTargetQ(double diffLoadTargetP) {
         return newLoadTargetQ;
     }
 
+    @Override
+    public boolean isDisabled(String originalId) {
+        return loadsStatus.get(originalId);
+    }
+
+    @Override
+    public void setDisabled(String originalId, boolean disabled) {
+        loadsStatus.put(originalId, disabled);
+    }
+
+    @Override
+    public Map<String, Boolean> getLoadsStatus() {
+        return loadsStatus;
+    }
+
+    @Override
+    public void setLoadsStatus(Map<String, Boolean> loadsStatus) {
+        this.loadsStatus = loadsStatus;
+    }
+
     private static double getPowerFactor(Load load) {
         return load.getP0() != 0 ? load.getQ0() / load.getP0() : 1;
     }

src/main/java/com/powsybl/openloadflow/network/impl/PropagatedContingency.java

@@ -108,7 +108,7 @@ private static PowerShift getLoadPowerShift(Load load, boolean slackDistribution
         }
         return new PowerShift(load.getP0() / PerUnit.SB,
                               variableActivePower / PerUnit.SB,
-                              load.getQ0() / PerUnit.SB);
+                              load.getQ0() / PerUnit.SB); // ensurePowerFactorConstant is not supported.
     }
 
     public static List<PropagatedContingency> createList(Network network, List<Contingency> contingencies,
@@ -352,6 +352,6 @@ public Optional<LfContingency> toLfContingency(LfNetwork network) {
             return Optional.empty();
         }
 
-        return Optional.of(new LfContingency(contingency.getId(), index, buses, branches, shunts, busesLoadShift, generators, hvdcs));
+        return Optional.of(new LfContingency(contingency.getId(), index, buses, branches, shunts, busesLoadShift, generators, hvdcs, originalPowerShiftIds));
     }
 }