get number of voltage levels in nad diagram (#486)

* Get nb voltage levels for a given voltageLevelFilter * Add unit test Signed-off-by: Franck LECUYER <franck.lecuyer@rte-france.com>
powsybl · Jan 31, 2023 · 897ac08 · 897ac08
1 parent e61ed8e
commit 897ac08
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/network-area-diagram/src/main/java/com/powsybl/nad/build/iidm/VoltageLevelFilter.java b/network-area-diagram/src/main/java/com/powsybl/nad/build/iidm/VoltageLevelFilter.java
@@ -27,6 +27,10 @@ public VoltageLevelFilter(Set<VoltageLevel> voltageLevels) {
         this.voltageLevels = voltageLevels;
     }
 
+    public int getNbVoltageLevels() {
+        return voltageLevels.size();
+    }
+
     private Set<VoltageLevel> getVoltageLevels() {
         return voltageLevels;
     }

diff --git a/...ork-area-diagram/src/test/java/com/powsybl/nad/layout/LayoutWithInitialPositionsTest.java b/...ork-area-diagram/src/test/java/com/powsybl/nad/layout/LayoutWithInitialPositionsTest.java
@@ -59,6 +59,16 @@ public void testDiamond() {
         checkLayoutWithInitialPositions(LayoutNetworkFactory.createDiamond());
     }
 
+    @Test
+    public void testNbVoltageLevels() {
+        Network network = LayoutNetworkFactory.createDiamond();
+
+        assertEquals(1, VoltageLevelFilter.createVoltageLevelDepthFilter(network, "A 400", 0).getNbVoltageLevels());
+        assertEquals(4, VoltageLevelFilter.createVoltageLevelDepthFilter(network, "A 400", 3).getNbVoltageLevels());
+        assertEquals(7, VoltageLevelFilter.createVoltageLevelDepthFilter(network, "A 400", 5).getNbVoltageLevels());
+        assertEquals(1, VoltageLevelFilter.createVoltageLevelDepthFilter(network, "A 400", 0).getNbVoltageLevels());
+    }
+
     private void checkLayoutWithInitialPositions(Network network) {
         Predicate<VoltageLevel> filter = vl -> vl.getNominalV() >= 100;