Adding value supplier method.

Author: staudtMarius

src/main/java/edu/ie3/datamodel/io/source/PowerValueSource.java

@@ -10,6 +10,7 @@
 import edu.ie3.datamodel.models.value.PValue;
 import java.time.ZonedDateTime;
 import java.util.Optional;
+import java.util.function.Supplier;
 import javax.measure.quantity.Energy;
 import javax.measure.quantity.Power;
 import tech.units.indriya.ComparableQuantity;
@@ -30,6 +31,16 @@ public sealed interface PowerValueSource<
    */
   Optional<PValue> getValue(ID data);
 
+  /**
+   * Method to get a supplier for the next power value based on the provided input data. Depending
+   * on the implementation the supplier will either always return the same value or each time a
+   * random value. To return one constant value please use {@link #getValue(InputData)}.
+   *
+   * @param data input data that is used to calculate the next power value.
+   * @return A supplier for an option on the value at the given time step.
+   */
+  Supplier<Optional<PValue>> getValueSupplier(ID data);
+
   /**
    * Method to determine the next timestamp for which data is present.
    *

src/main/java/edu/ie3/datamodel/io/source/csv/CsvLoadProfileSource.java

@@ -20,6 +20,7 @@
 import java.nio.file.Path;
 import java.util.*;
 import java.util.function.Function;
+import java.util.function.Supplier;
 import java.util.stream.Collectors;
 import javax.measure.quantity.Energy;
 import javax.measure.quantity.Power;
@@ -74,6 +75,11 @@ public Optional<PValue> getValue(TimeSeriesInputValue data) {
     return loadProfileTimeSeries.getValue(data.time());
   }
 
+  @Override
+  public Supplier<Optional<PValue>> getValueSupplier(TimeSeriesInputValue data) {
+    return loadProfileTimeSeries.supplyValue(data.time());
+  }
+
   @Override
   public P getProfile() {
     return loadProfileTimeSeries.getLoadProfile();

src/main/java/edu/ie3/datamodel/io/source/sql/SqlLoadProfileSource.java

@@ -24,6 +24,7 @@
 import java.util.Map;
 import java.util.Optional;
 import java.util.Set;
+import java.util.function.Supplier;
 import java.util.stream.Collectors;
 import javax.measure.quantity.Energy;
 import javax.measure.quantity.Power;
@@ -101,12 +102,20 @@ public Set<LoadProfileEntry<V>> getEntries() {
   @Override
   public Optional<PValue> getValue(TimeSeriesInputValue data) {
     ZonedDateTime time = data.time();
+    return queryForValue(time).map(loadValue -> loadValue.getValue(time, profile));
+  }
 
-    Set<LoadProfileEntry<V>> entries =
-        getEntries(queryTime, ps -> ps.setInt(1, TimeSeriesUtils.calculateQuarterHourOfDay(time)));
-    if (entries.isEmpty()) return Optional.empty();
-    if (entries.size() > 1) log.warn("Retrieved more than one result value, using the first");
-    return entries.stream().findFirst().map(entry -> entry.getValue().getValue(time, profile));
+  @Override
+  public Supplier<Optional<PValue>> getValueSupplier(TimeSeriesInputValue data) {
+    ZonedDateTime time = data.time();
+    Optional<LoadValues<P>> loadValueOption = queryForValue(time);
+
+    if (loadValueOption.isPresent()) {
+      LoadValues<P> loadValue = loadValueOption.get();
+      return () -> Optional.of(loadValue.getValue(time, profile));
+    } else {
+      return Optional::empty;
+    }
   }
 
   @Override
@@ -138,6 +147,15 @@ private Set<LoadProfileEntry<V>> getEntries(String query, SqlDataSource.AddParam
         .collect(Collectors.toSet());
   }
 
+  private Optional<LoadValues<P>> queryForValue(ZonedDateTime time) {
+    Set<LoadProfileEntry<V>> entries =
+        getEntries(queryTime, ps -> ps.setInt(1, TimeSeriesUtils.calculateQuarterHourOfDay(time)));
+    if (entries.isEmpty()) return Optional.empty();
+    if (entries.size() > 1) log.warn("Retrieved more than one result value, using the first");
+
+    return entries.stream().findFirst().map(LoadProfileEntry::getValue);
+  }
+
   /**
    * Build a {@link LoadProfileEntry} of type {@code V}, whereas the underlying {@link Value} does
    * not need any additional information.

src/main/java/edu/ie3/datamodel/models/timeseries/repetitive/LoadProfileTimeSeries.java

@@ -11,6 +11,7 @@
 import edu.ie3.datamodel.utils.TimeSeriesUtils;
 import java.time.ZonedDateTime;
 import java.util.*;
+import java.util.function.Supplier;
 import java.util.stream.Collectors;
 import javax.measure.quantity.Energy;
 import javax.measure.quantity.Power;
@@ -21,8 +22,8 @@
  */
 public class LoadProfileTimeSeries<P extends LoadProfile, V extends LoadValues<P>>
     extends RepetitiveTimeSeries<LoadProfileEntry<V>, V, PValue> {
-  private final P loadProfile;
-  private final Map<Integer, V> valueMapping;
+  protected final P loadProfile;
+  protected final Map<Integer, V> valueMapping;
 
   /**
    * The maximum average power consumption per quarter-hour calculated over all seasons and weekday
@@ -76,6 +77,20 @@ public Set<LoadProfileEntry<V>> getEntries() {
     return set;
   }
 
+  /**
+   * Method to get a supplier for the next power value based on the provided input time. Depending
+   * on the implementation the supplier will either always return the same value or each time a
+   * random value. To return one constant value please use {@link #getValue(ZonedDateTime)}.
+   *
+   * @param time Queried time.
+   * @return A supplier for an option on the value at the given time step.
+   */
+  public Supplier<Optional<PValue>> supplyValue(ZonedDateTime time) {
+    int quarterHour = TimeSeriesUtils.calculateQuarterHourOfDay(time);
+    LoadValues<P> loadValue = valueMapping.get(quarterHour);
+    return () -> Optional.ofNullable(loadValue.getValue(time, loadProfile));
+  }
+
   @Override
   public Optional<ZonedDateTime> getPreviousDateTime(ZonedDateTime time) {
     return Optional.of(time.minusMinutes(15));

src/test/groovy/edu/ie3/datamodel/models/timeseries/RandomLoadProfileTimeSeriesTest.groovy

@@ -0,0 +1,31 @@
+/*
+ * © 2025. TU Dortmund University,
+ * Institute of Energy Systems, Energy Efficiency and Energy Economics,
+ * Research group Distribution grid planning and operation
+ */
+package edu.ie3.datamodel.models.timeseries
+
+import edu.ie3.datamodel.io.source.LoadProfileSource
+import edu.ie3.util.TimeUtil
+import spock.lang.Specification
+
+class RandomLoadProfileTimeSeriesTest extends Specification {
+
+
+  def "A RandomLoadProfileTimeSeries should supply a random value correctly"() {
+    given:
+    def lpts = LoadProfileSource.randomLoadProfile.getTimeSeries()
+    def time = TimeUtil.withDefaults.toZonedDateTime("2020-01-01T00:00:00Z")
+
+    def supplier = lpts.supplyValue(time)
+
+    when:
+    def values = [
+      supplier.get(),
+      supplier.get(),
+      supplier.get()
+    ].collect { it.get()}
+    then:
+    values.size() > 1
+  }
+}