feat: partial load on bus network components

Signed-off-by: Joris Mancini <[email protected]>

Author: TheMaskedTurtle

src/main/java/org/gridsuite/network/map/dto/InfoTypeParameters.java

@@ -16,6 +16,7 @@ public class InfoTypeParameters {
     public static final String QUERY_PARAM_DC_POWERFACTOR = "dcPowerFactor";
     public static final String QUERY_PARAM_LOAD_OPERATIONAL_LIMIT_GROUPS = "loadOperationalLimitGroups";
     public static final String QUERY_PARAM_LOAD_REGULATING_TERMINALS = "loadRegulatingTerminals";
+    public static final String QUERY_PARAM_LOAD_NETWORK_COMPONENTS = "loadNetworkComponents";
 
     public static final InfoTypeParameters TAB = new InfoTypeParameters(ElementInfos.InfoType.TAB, null);
 

src/main/java/org/gridsuite/network/map/dto/mapper/BusInfosMapper.java

@@ -13,6 +13,9 @@
 import org.gridsuite.network.map.dto.definition.bus.BusTabInfos;
 import org.gridsuite.network.map.dto.utils.ElementUtils;
 
+import java.util.Optional;
+
+import static org.gridsuite.network.map.dto.InfoTypeParameters.QUERY_PARAM_LOAD_NETWORK_COMPONENTS;
 import static org.gridsuite.network.map.dto.utils.ElementUtils.getProperties;
 import static org.gridsuite.network.map.dto.utils.ElementUtils.mapCountry;
 
@@ -25,27 +28,34 @@ private BusInfosMapper() {
     }
 
     public static ElementInfos toData(Identifiable<?> identifiable, InfoTypeParameters infoTypeParameters) {
+        boolean shouldLoadNetworkComponents = Optional.ofNullable(infoTypeParameters.getOptionalParameters().get(QUERY_PARAM_LOAD_NETWORK_COMPONENTS))
+            .map(Boolean::valueOf)
+            .orElse(false);
         return switch (infoTypeParameters.getInfoType()) {
             case LIST -> ElementInfosMapper.toListInfos(identifiable);
-            case TAB -> toTabInfos(identifiable);
+            case TAB -> toTabInfos(identifiable, shouldLoadNetworkComponents);
             default -> throw new UnsupportedOperationException("TODO");
         };
     }
 
-    private static BusTabInfos toTabInfos(Identifiable<?> identifiable) {
+    private static BusTabInfos toTabInfos(Identifiable<?> identifiable, boolean shouldLoadNetworkComponents) {
         Bus bus = (Bus) identifiable;
         BusTabInfos.BusTabInfosBuilder<?, ?> builder = BusTabInfos.builder().id(bus.getId())
                 .angle(bus.getAngle())
                 .v(bus.getV())
                 .voltageLevelId(bus.getVoltageLevel().getId())
                 .nominalVoltage(bus.getVoltageLevel().getNominalV())
                 .country(mapCountry(bus.getVoltageLevel().getSubstation().orElse(null)))
-                .synchronousComponentNum(bus.getSynchronousComponent().getNum())
                 .properties(getProperties(bus))
-                .connectedComponentNum(bus.getConnectedComponent().getNum())
                 .substationProperties(bus.getVoltageLevel().getSubstation().map(ElementUtils::getProperties).orElse(null))
                 .voltageLevelProperties(getProperties(bus.getVoltageLevel()));
 
+        if (shouldLoadNetworkComponents) {
+            builder
+                .synchronousComponentNum(bus.getSynchronousComponent().getNum())
+                .connectedComponentNum(bus.getConnectedComponent().getNum());
+        }
+
         return builder.build();
     }
 }