diff options
author | Kristóf Marussy <marussy@mit.bme.hu> | 2023-09-14 19:29:36 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-09-14 19:29:36 +0200 |
commit | 98ed3b6db5f4e51961a161050cc31c66015116e8 (patch) | |
tree | 8bfd6d9bc8d6ed23b9eb0f889dd40b6c24fe8f92 /subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CountObjective.java | |
parent | Merge pull request #38 from nagilooh/design-space-exploration (diff) | |
parent | Merge remote-tracking branch 'upstream/main' into partial-interpretation (diff) | |
download | refinery-98ed3b6db5f4e51961a161050cc31c66015116e8.tar.gz refinery-98ed3b6db5f4e51961a161050cc31c66015116e8.tar.zst refinery-98ed3b6db5f4e51961a161050cc31c66015116e8.zip |
Merge pull request #39 from kris7t/partial-interpretation
Implement partial interpretation based model generation
Diffstat (limited to 'subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CountObjective.java')
-rw-r--r-- | subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CountObjective.java | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CountObjective.java b/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CountObjective.java new file mode 100644 index 00000000..fbd05ded --- /dev/null +++ b/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/objectives/CountObjective.java | |||
@@ -0,0 +1,47 @@ | |||
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 | import tools.refinery.store.query.ModelQueryAdapter; | ||
12 | import tools.refinery.store.query.ModelQueryBuilder; | ||
13 | import tools.refinery.store.query.ModelQueryStoreAdapter; | ||
14 | import tools.refinery.store.query.dnf.RelationalQuery; | ||
15 | import tools.refinery.store.query.literal.Reduction; | ||
16 | |||
17 | public class CountObjective implements Objective { | ||
18 | private final RelationalQuery query; | ||
19 | private final double weight; | ||
20 | |||
21 | public CountObjective(RelationalQuery query) { | ||
22 | this(query, 1); | ||
23 | } | ||
24 | |||
25 | public CountObjective(RelationalQuery query, double weight) { | ||
26 | this.query = query; | ||
27 | this.weight = weight; | ||
28 | } | ||
29 | |||
30 | @Override | ||
31 | public boolean isAlwaysZero(ModelStore store) { | ||
32 | var queryStore = store.getAdapter(ModelQueryStoreAdapter.class); | ||
33 | var canonicalQuery = queryStore.getCanonicalQuery(query); | ||
34 | return canonicalQuery.getDnf().getReduction() == Reduction.ALWAYS_FALSE; | ||
35 | } | ||
36 | |||
37 | @Override | ||
38 | public ObjectiveCalculator createCalculator(Model model) { | ||
39 | var resultSet = model.getAdapter(ModelQueryAdapter.class).getResultSet(query); | ||
40 | return () -> resultSet.size() * weight; | ||
41 | } | ||
42 | |||
43 | @Override | ||
44 | public void configure(ModelStoreBuilder storeBuilder) { | ||
45 | storeBuilder.getAdapter(ModelQueryBuilder.class).query(query); | ||
46 | } | ||
47 | } | ||