Add methods to get measurements by type (#1782)

Signed-off-by: RALAMBOTIANA MIORA <[email protected]>
powsybl · Jul 7, 2021 · 092d0f8 · 092d0f8
1 parent 95fadc4
commit 092d0f8
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/...dm-extensions/src/main/java/com/powsybl/iidm/network/extensions/DiscreteMeasurements.java b/...dm-extensions/src/main/java/com/powsybl/iidm/network/extensions/DiscreteMeasurements.java
@@ -26,6 +26,8 @@ default String getName() {
 
     Collection<DiscreteMeasurement> getDiscreteMeasurements();
 
+    Collection<DiscreteMeasurement> getDiscreteMeasurements(DiscreteMeasurement.Type type);
+
     DiscreteMeasurement getDiscreteMeasurement(String id);
 
     DiscreteMeasurementAdder newDiscreteMeasurement();

diff --git a/iidm/iidm-extensions/src/main/java/com/powsybl/iidm/network/extensions/Measurements.java b/iidm/iidm-extensions/src/main/java/com/powsybl/iidm/network/extensions/Measurements.java
@@ -26,6 +26,8 @@ default String getName() {
 
     Collection<Measurement> getMeasurements();
 
+    Collection<Measurement> getMeasurements(Measurement.Type type);
+
     Measurement getMeasurement(String id);
 
     MeasurementAdder newMeasurement();

diff --git a/...impl/src/main/java/com/powsybl/iidm/network/impl/extensions/DiscreteMeasurementsImpl.java b/...impl/src/main/java/com/powsybl/iidm/network/impl/extensions/DiscreteMeasurementsImpl.java
@@ -16,6 +16,7 @@
 import java.util.Collection;
 import java.util.Collections;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * @author Miora Ralambotiana <miora.ralambotiana at rte-france.com>
@@ -38,6 +39,11 @@ public Collection<DiscreteMeasurement> getDiscreteMeasurements() {
         return Collections.unmodifiableList(discreteMeasurements);
     }
 
+    @Override
+    public Collection<DiscreteMeasurement> getDiscreteMeasurements(DiscreteMeasurement.Type type) {
+        return discreteMeasurements.stream().filter(dm -> dm.getType() == type).collect(Collectors.toList());
+    }
+
     @Override
     public DiscreteMeasurement getDiscreteMeasurement(String id) {
         return discreteMeasurements.stream()

diff --git a/iidm/iidm-impl/src/main/java/com/powsybl/iidm/network/impl/extensions/MeasurementsImpl.java b/iidm/iidm-impl/src/main/java/com/powsybl/iidm/network/impl/extensions/MeasurementsImpl.java
@@ -16,6 +16,7 @@
 import java.util.Collection;
 import java.util.Collections;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * @author Miora Ralambotiana <miora.ralambotiana at rte-france.com>
@@ -38,6 +39,11 @@ public Collection<Measurement> getMeasurements() {
         return Collections.unmodifiableList(measurements);
     }
 
+    @Override
+    public Collection<Measurement> getMeasurements(Measurement.Type type) {
+        return measurements.stream().filter(m -> m.getType() == type).collect(Collectors.toList());
+    }
+
     @Override
     public Measurement getMeasurement(String id) {
         return measurements.stream()