-
Notifications
You must be signed in to change notification settings - Fork 5
/
NodeResultFactory.java
58 lines (50 loc) · 2.14 KB
/
NodeResultFactory.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
/*
* © 2021. TU Dortmund University,
* Institute of Energy Systems, Energy Efficiency and Energy Economics,
* Research group Distribution grid planning and operation
*/
package edu.ie3.datamodel.io.factory.result;
import edu.ie3.datamodel.io.factory.EntityData;
import edu.ie3.datamodel.models.StandardUnits;
import edu.ie3.datamodel.models.result.NodeResult;
import java.time.ZonedDateTime;
import java.time.format.DateTimeFormatter;
import java.util.*;
import javax.measure.quantity.Angle;
import javax.measure.quantity.Dimensionless;
import tech.units.indriya.ComparableQuantity;
public class NodeResultFactory extends ResultEntityFactory<NodeResult> {
private static final String VMAG = "vMag";
private static final String VANG = "vAng";
public NodeResultFactory() {
super(NodeResult.class);
}
/**
* Create a new factory to build {@link NodeResult}s and utilize the given date time formatter
* pattern to parse date time strings
*
* @param dateTimeFormatter to parse date time strings
*/
public NodeResultFactory(DateTimeFormatter dateTimeFormatter) {
super(dateTimeFormatter, NodeResult.class);
}
@Override
protected List<Set<String>> getFields(Class<?> entityClass) {
Set<String> minConstructorParams = newSet(TIME, INPUT_MODEL, VMAG, VANG);
Set<String> optionalFields = expandSet(minConstructorParams, ENTITY_UUID);
return Arrays.asList(minConstructorParams, optionalFields);
}
@Override
protected NodeResult buildModel(EntityData data) {
ZonedDateTime zdtTime = timeUtil.toZonedDateTime(data.getField(TIME));
UUID inputModelUuid = data.getUUID(INPUT_MODEL);
ComparableQuantity<Dimensionless> vMagValue =
data.getQuantity(VMAG, StandardUnits.VOLTAGE_MAGNITUDE);
ComparableQuantity<Angle> vAngValue = data.getQuantity(VANG, StandardUnits.VOLTAGE_ANGLE);
Optional<UUID> uuidOpt =
data.containsKey(ENTITY_UUID) ? Optional.of(data.getUUID(ENTITY_UUID)) : Optional.empty();
return uuidOpt
.map(uuid -> new NodeResult(uuid, zdtTime, inputModelUuid, vMagValue, vAngValue))
.orElseGet(() -> new NodeResult(zdtTime, inputModelUuid, vMagValue, vAngValue));
}
}