Implementation of Identifiable::getNetwork()

iidm/iidm-api/src/main/java/com/powsybl/iidm/network/Identifiable.java

@@ -20,6 +20,11 @@
  */
 public interface Identifiable<I extends Identifiable<I>> extends Extendable<I> {
 
+    /**
+     * Get the network associated to the object.
+     */
+    Network getNetwork();
+
     /**
      * Get the unique identifier of the object.
      */

iidm/iidm-impl/src/main/java/com/powsybl/iidm/network/impl/AbstractBus.java

@@ -42,6 +42,11 @@ public boolean isInMainSynchronousComponent() {
         return sc != null && sc.getNum() == ComponentConstants.MAIN_NUM;
     }
 
+    @Override
+    public NetworkImpl getNetwork() {
+        return voltageLevel.getNetwork();
+    }
+
     @Override
     public VoltageLevel getVoltageLevel() {
         return voltageLevel;

iidm/iidm-impl/src/main/java/com/powsybl/iidm/network/impl/AbstractConnectable.java

@@ -42,7 +42,8 @@ public List<TerminalExt> getTerminals() {
         return terminals;
     }
 
-    protected NetworkImpl getNetwork() {
+    @Override
+    public NetworkImpl getNetwork() {
         if (terminals.isEmpty()) {
             throw new PowsyblException(id + " is not attached to a network");
         }

iidm/iidm-impl/src/main/java/com/powsybl/iidm/network/impl/AbstractIdentifiable.java

@@ -38,6 +38,9 @@ public String getName() {
         return name != null ? name : id;
     }
 
+    @Override
+    public abstract NetworkImpl getNetwork();
+
     protected abstract String getTypeDescription();
 
     @Override

iidm/iidm-impl/src/main/java/com/powsybl/iidm/network/impl/MergedBus.java

@@ -87,6 +87,11 @@ public void invalidate() {
         buses.clear();
     }
 
+    @Override
+    public NetworkImpl getNetwork() {
+        return (NetworkImpl) getVoltageLevel().getNetwork();
+    }
+
     @Override
     public VoltageLevel getVoltageLevel() {
         checkValidity();

iidm/iidm-impl/src/main/java/com/powsybl/iidm/network/impl/NetworkImpl.java

@@ -174,6 +174,11 @@ public NetworkIndex getIndex() {
         return index;
     }
 
+    @Override
+    public NetworkImpl getNetwork() {
+        return this;
+    }
+
     @Override
     public VariantManagerImpl getVariantManager() {
         return variantManager;

iidm/iidm-impl/src/main/java/com/powsybl/iidm/network/impl/SwitchImpl.java

@@ -42,6 +42,11 @@ class SwitchImpl extends AbstractIdentifiable<Switch> implements Switch, MultiVa
         }
     }
 
+    @Override
+    public NetworkImpl getNetwork() {
+        return voltageLevel.getNetwork();
+    }
+
     @Override
     public VoltageLevelExt getVoltageLevel() {
         return voltageLevel;
@@ -54,12 +59,12 @@ public SwitchKind getKind() {
 
     @Override
     public boolean isOpen() {
-        return open.get(voltageLevel.getNetwork().getVariantIndex());
+        return open.get(getNetwork().getVariantIndex());
     }
 
     @Override
     public void setOpen(boolean open) {
-        NetworkImpl network = voltageLevel.getNetwork();
+        NetworkImpl network = getNetwork();
         int index = network.getVariantIndex();
         boolean oldValue = this.open.get(index);
         if (oldValue != open) {
@@ -72,15 +77,15 @@ public void setOpen(boolean open) {
 
     @Override
     public boolean isRetained() {
-        return retained.get(voltageLevel.getNetwork().getVariantIndex());
+        return retained.get(getNetwork().getVariantIndex());
     }
 
     @Override
     public void setRetained(boolean retained) {
         if (voltageLevel.getTopologyKind() != TopologyKind.NODE_BREAKER) {
             throw new ValidationException(this, "retain status is not modifiable in a non node/breaker voltage level");
         }
-        NetworkImpl network = voltageLevel.getNetwork();
+        NetworkImpl network = getNetwork();
         int index = network.getVariantIndex();
         boolean oldValue = this.retained.get(index);
         if (oldValue != retained) {
@@ -102,7 +107,7 @@ public void setFictitious(boolean fictitious) {
         if (oldValue != fictitious) {
             this.fictitious = fictitious;
             voltageLevel.invalidateCache();
-            NetworkImpl network = voltageLevel.getNetwork();
+            NetworkImpl network = getNetwork();
             network.getListeners().notifyUpdate(this, "fictitious", oldValue, fictitious);
         }
     }

iidm/iidm-impl/src/test/java/com/powsybl/iidm/network/impl/BusTest.java

@@ -69,6 +69,7 @@ public void testSetterGetter() {
         vscConverterStation.getTerminal().setQ(q2);
 
         assertSame(voltageLevel, bus.getVoltageLevel());
+        assertSame(network, bus.getNetwork());
         try {
             bus.setV(-1.0);
             fail();

iidm/iidm-impl/src/test/java/com/powsybl/iidm/network/impl/CalculatedTopologyTest.java

@@ -11,6 +11,7 @@
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertSame;
 
 /**
  * @author Luma Zamarreño <zamarrenolm at aia.es>
@@ -33,6 +34,7 @@ private void checkSameMergedBus(Network n, String ida, String idb) {
         assertNotNull(ma);
         assertNotNull(mb);
         assertEquals(ma, mb);
+        assertSame(n, ma.getNetwork());
     }
 
     private Network createBusBreaker() {

iidm/iidm-impl/src/test/java/com/powsybl/iidm/network/impl/NetworkTest.java

@@ -41,6 +41,7 @@ public NetworkTest() {
     @Test
     public void testNetwork1() {
         Network network = NetworkTest1Factory.create();
+        assertSame(network, network.getNetwork());
         assertEquals(1, Iterables.size(network.getCountries()));
         assertEquals(1, network.getCountryCount());
         Country country1 = network.getCountries().iterator().next();

iidm/iidm-impl/src/test/java/com/powsybl/iidm/network/impl/VariantManagerImplTest.java

@@ -51,6 +51,11 @@ public String getName() {
             return id;
         }
 
+        @Override
+        public Network getNetwork() {
+            return null;
+        }
+
         @Override
         public boolean hasProperty() {
             return false;