[WIP] Run marmot using fast_rao (POC)

ra-optimisation/search-tree-rao/src/main/java/com/powsybl/openrao/searchtreerao/commons/FlowCnecSorting.java

@@ -31,4 +31,12 @@ public static List<FlowCnec> sortByMargin(Set<FlowCnec> flowCnecs, Unit unit, Ma
             .sorted(Comparator.comparingDouble(flowCnec -> marginEvaluator.getMargin(flowResult, flowCnec, unit)))
             .toList();
     }
+
+    public static List<FlowCnec> sortByNegativeMargin(Set<FlowCnec> flowCnecs, Unit unit, MarginEvaluator marginEvaluator, FlowResult flowResult) {
+        return flowCnecs.stream()
+            .filter(Cnec::isOptimized)
+            .filter(flowCnec -> marginEvaluator.getMargin(flowResult, flowCnec, unit) < 0)
+            .sorted(Comparator.comparingDouble(flowCnec -> marginEvaluator.getMargin(flowResult, flowCnec, unit)))
+            .toList();
+    }
 }

ra-optimisation/search-tree-rao/src/main/java/com/powsybl/openrao/searchtreerao/commons/RaoLogger.java

@@ -71,10 +71,35 @@ public static void logSensitivityAnalysisResults(String prefix,
             numberOfLoggedLimitingElements);
     }
 
+    public static void logObjectiveFunctionResult(String prefix,
+                                                 ObjectiveFunctionResult objectiveFunctionResult,
+                                                 PrePerimeterResult sensitivityAnalysisResult,
+                                                 RaoParameters raoParameters,
+                                                 int numberOfLoggedLimitingElements) {
+
+        if (!BUSINESS_LOGS.isInfoEnabled()) {
+            return;
+        }
+
+        Map<String, Double> virtualCostDetailed = getVirtualCostDetailed(objectiveFunctionResult);
+
+        BUSINESS_LOGS.info(prefix + "cost = {} (functional: {}, virtual: {}{})",
+            formatDoubleBasedOnMargin(objectiveFunctionResult.getCost(), -objectiveFunctionResult.getCost()),
+            formatDoubleBasedOnMargin(objectiveFunctionResult.getFunctionalCost(), -objectiveFunctionResult.getCost()),
+            formatDoubleBasedOnMargin(objectiveFunctionResult.getVirtualCost(), -objectiveFunctionResult.getCost()),
+            virtualCostDetailed.isEmpty() ? "" : " " + virtualCostDetailed);
+
+        RaoLogger.logMostLimitingElementsResults(BUSINESS_LOGS,
+            sensitivityAnalysisResult,
+            raoParameters.getObjectiveFunctionParameters().getType(),
+            raoParameters.getObjectiveFunctionParameters().getUnit(),
+            numberOfLoggedLimitingElements);
+    }
+
     public static void logRangeActions(OpenRaoLogger logger,
                                        Leaf leaf,
-                                       OptimizationPerimeter
-                                           optimizationContext, String prefix) {
+                                       OptimizationPerimeter optimizationContext,
+                                       String prefix) {
 
         boolean globalPstOptimization = optimizationContext instanceof GlobalOptimizationPerimeter;
 

ra-optimisation/search-tree-rao/src/main/java/com/powsybl/openrao/searchtreerao/commons/marginevaluator/MarginEvaluatorWithMarginDecreaseUnoptimizedCnecs.java

@@ -50,7 +50,7 @@ public double getMargin(FlowResult flowResult, FlowCnec flowCnec, TwoSides side,
     }
 
     private double computeMargin(FlowCnec flowCnec, double newMargin, double prePerimeterMargin) {
-        if (countriesNotToOptimize.contains(flowCnec.getOperator()) && newMargin > prePerimeterMargin - .0001 * Math.abs(prePerimeterMargin)) {
+        if (countriesNotToOptimize.contains(flowCnec.getOperator()) && newMargin > prePerimeterMargin - .0001 * Math.abs(prePerimeterMargin) && flowCnec.getState().getInstant().isCurative()) {
             return Double.MAX_VALUE;
         }
         return newMargin;

ra-optimisation/search-tree-rao/src/main/java/com/powsybl/openrao/searchtreerao/commons/objectivefunctionevaluator/MinMarginViolationEvaluator.java

@@ -8,11 +8,20 @@
 package com.powsybl.openrao.searchtreerao.commons.objectivefunctionevaluator;
 
 import com.powsybl.openrao.commons.Unit;
+import com.powsybl.openrao.data.crac.api.State;
 import com.powsybl.openrao.data.crac.api.cnec.FlowCnec;
+import com.powsybl.openrao.searchtreerao.commons.FlowCnecSorting;
+import com.powsybl.openrao.searchtreerao.commons.costevaluatorresult.CostEvaluatorResult;
+import com.powsybl.openrao.searchtreerao.commons.costevaluatorresult.MaxCostEvaluatorResult;
 import com.powsybl.openrao.searchtreerao.commons.marginevaluator.MarginEvaluator;
 import com.powsybl.openrao.searchtreerao.result.api.FlowResult;
+import com.powsybl.openrao.searchtreerao.result.api.RemedialActionActivationResult;
 
+import java.util.Map;
 import java.util.Set;
+import java.util.stream.Collectors;
+
+import static com.powsybl.openrao.searchtreerao.commons.objectivefunctionevaluator.CostEvaluatorUtils.groupFlowCnecsPerState;
 
 /**
  * @author Thomas Bouquet {@literal <thomas.bouquet at rte-france.com>}
@@ -33,4 +42,11 @@ public String getName() {
     protected double computeCostForState(FlowResult flowResult, Set<FlowCnec> flowCnecsOfState) {
         return Math.max(0, super.computeCostForState(flowResult, flowCnecsOfState)) * OVERLOAD_PENALTY;
     }
+
+    @Override
+    public CostEvaluatorResult evaluate(FlowResult flowResult, RemedialActionActivationResult remedialActionActivationResult) {
+        Map<State, Set<FlowCnec>> flowCnecsPerState = groupFlowCnecsPerState(flowCnecs);
+        Map<State, Double> costPerState = flowCnecsPerState.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, entry -> computeCostForState(flowResult, entry.getValue())));
+        return new MaxCostEvaluatorResult(costPerState, FlowCnecSorting.sortByNegativeMargin(flowCnecs, unit, marginEvaluator, flowResult));
+    }
 }