diff options
Diffstat (limited to 'subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CompositeObjective.java')
-rw-r--r-- | subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CompositeObjective.java | 69 |
1 files changed, 69 insertions, 0 deletions
diff --git a/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CompositeObjective.java b/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CompositeObjective.java new file mode 100644 index 00000000..192a824b --- /dev/null +++ b/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CompositeObjective.java | |||
@@ -0,0 +1,69 @@ | |||
1 | /* | ||
2 | * SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/> | ||
3 | * | ||
4 | * SPDX-License-Identifier: EPL-2.0 | ||
5 | */ | ||
6 | package tools.refinery.store.dse.transition.objectives; | ||
7 | |||
8 | import tools.refinery.store.model.Model; | ||
9 | import tools.refinery.store.model.ModelStore; | ||
10 | import tools.refinery.store.model.ModelStoreBuilder; | ||
11 | |||
12 | import java.util.ArrayList; | ||
13 | import java.util.Collection; | ||
14 | import java.util.Collections; | ||
15 | import java.util.List; | ||
16 | |||
17 | public class CompositeObjective implements Objective { | ||
18 | private final List<Objective> objectives; | ||
19 | |||
20 | CompositeObjective(Collection<? extends Objective> objectives) { | ||
21 | var unwrappedObjectives = new ArrayList<Objective>(); | ||
22 | for (var objective : objectives) { | ||
23 | if (objective instanceof CompositeObjective compositeObjective) { | ||
24 | unwrappedObjectives.addAll(compositeObjective.getObjectives()); | ||
25 | } else { | ||
26 | unwrappedObjectives.add(objective); | ||
27 | } | ||
28 | } | ||
29 | this.objectives = Collections.unmodifiableList(unwrappedObjectives); | ||
30 | } | ||
31 | |||
32 | public List<Objective> getObjectives() { | ||
33 | return objectives; | ||
34 | } | ||
35 | |||
36 | @Override | ||
37 | public boolean isAlwaysZero(ModelStore store) { | ||
38 | for (var objective : objectives) { | ||
39 | if (!objective.isAlwaysZero(store)) { | ||
40 | return false; | ||
41 | } | ||
42 | } | ||
43 | return true; | ||
44 | } | ||
45 | |||
46 | @Override | ||
47 | public ObjectiveCalculator createCalculator(Model model) { | ||
48 | var calculators = new ArrayList<ObjectiveCalculator>(); | ||
49 | for (var objective : objectives) { | ||
50 | if (!objective.isAlwaysZero(model.getStore())) { | ||
51 | calculators.add(objective.createCalculator(model)); | ||
52 | } | ||
53 | } | ||
54 | return () -> { | ||
55 | double value = 0; | ||
56 | for (var calculator : calculators) { | ||
57 | value += calculator.getValue(); | ||
58 | } | ||
59 | return value; | ||
60 | }; | ||
61 | } | ||
62 | |||
63 | @Override | ||
64 | public void configure(ModelStoreBuilder storeBuilder) { | ||
65 | for (var objective : objectives) { | ||
66 | objective.configure(storeBuilder); | ||
67 | } | ||
68 | } | ||
69 | } | ||