blob: 3795daa421bc850dd028fe56c3f53d0df6b70657 (
plain) (
blame)
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
59
60
61
62
63
64
65
66
67
68
69
70
71
|
/*
* SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/>
*
* SPDX-License-Identifier: EPL-2.0
*/
package tools.refinery.store.dse.strategy;
import org.eclipse.collections.api.block.procedure.Procedure;
import tools.refinery.store.dse.transition.DesignSpaceExplorationStoreAdapter;
import tools.refinery.store.dse.transition.VersionWithObjectiveValue;
import tools.refinery.store.dse.transition.statespace.ActivationStore;
import tools.refinery.store.dse.transition.statespace.EquivalenceClassStore;
import tools.refinery.store.dse.transition.statespace.ObjectivePriorityQueue;
import tools.refinery.store.dse.transition.statespace.SolutionStore;
import tools.refinery.store.dse.transition.statespace.internal.ActivationStoreImpl;
import tools.refinery.store.dse.transition.statespace.internal.FastEquivalenceClassStore;
import tools.refinery.store.dse.transition.statespace.internal.ObjectivePriorityQueueImpl;
import tools.refinery.store.dse.transition.statespace.internal.SolutionStoreImpl;
import tools.refinery.store.map.Version;
import tools.refinery.store.model.ModelStore;
import tools.refinery.store.statecoding.StateCoderStoreAdapter;
public class BestFirstStoreManager {
ModelStore modelStore;
ObjectivePriorityQueue objectiveStore;
ActivationStore activationStore;
SolutionStore solutionStore;
EquivalenceClassStore equivalenceClassStore;
public BestFirstStoreManager(ModelStore modelStore) {
this.modelStore = modelStore;
DesignSpaceExplorationStoreAdapter storeAdapter =
modelStore.getAdapter(DesignSpaceExplorationStoreAdapter.class);
objectiveStore = new ObjectivePriorityQueueImpl(storeAdapter.getObjectives());
Procedure<VersionWithObjectiveValue> whenAllActivationsVisited = x -> objectiveStore.remove(x);
activationStore = new ActivationStoreImpl(storeAdapter.getTransformations().size(), whenAllActivationsVisited);
solutionStore = new SolutionStoreImpl(10);
equivalenceClassStore = new FastEquivalenceClassStore(modelStore.getAdapter(StateCoderStoreAdapter.class)) {
@Override
protected void delegate(VersionWithObjectiveValue version, int[] emptyActivations, boolean accept) {
objectiveStore.submit(version);
activationStore.markNewAsVisited(version, emptyActivations);
if(accept) {
solutionStore.submit(version);
}
}
};
}
ObjectivePriorityQueue getObjectiveStore() {
return objectiveStore;
}
ActivationStore getActivationStore() {
return activationStore;
}
public SolutionStore getSolutionStore() {
return solutionStore;
}
EquivalenceClassStore getEquivalenceClassStore() {
return equivalenceClassStore;
}
public void startExploration(Version initial) {
BestFirstExplorer bestFirstExplorer = new BestFirstExplorer(this, modelStore.createModelForState(initial), 1);
bestFirstExplorer.explore();
}
}
|
