From 0bf4858c2b08d795bfa0e66a1f6a5cd820a8f0da Mon Sep 17 00:00:00 2001
From: Geoffroy Jamgotchian <geoffroy.jamgotchian@rte-france.com>
Date: Thu, 21 Sep 2023 11:12:36 +0200
Subject: [PATCH] Merge loads optimization for network store IIDM impl

Signed-off-by: Geoffroy Jamgotchian <geoffroy.jamgotchian@rte-france.com>
---
 .../java/com/powsybl/dynawo/commons/NetworkResultsUpdater.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/commons/src/main/java/com/powsybl/dynawo/commons/NetworkResultsUpdater.java b/commons/src/main/java/com/powsybl/dynawo/commons/NetworkResultsUpdater.java
index d344f3436..1957632ee 100644
--- a/commons/src/main/java/com/powsybl/dynawo/commons/NetworkResultsUpdater.java
+++ b/commons/src/main/java/com/powsybl/dynawo/commons/NetworkResultsUpdater.java
@@ -149,8 +149,9 @@ private static void updateLoads(Network targetNetwork, Network sourceNetwork, bo
                 update(targetNetwork.getLoad(sourceLoad.getId()).getTerminal(), sourceLoad.getTerminal());
             }
         } else {
+            Map<String, Bus> sourceBusesById = sourceNetwork.getBusBreakerView().getBusStream().collect(Collectors.toMap(Identifiable::getId, Function.identity()));
             for (Bus busTarget : targetNetwork.getBusBreakerView().getBuses()) {
-                updateLoads(busTarget, sourceNetwork.getBusBreakerView().getBus(busTarget.getId()));
+                updateLoads(busTarget, sourceBusesById.get(busTarget.getId()));
             }
         }
     }