1 files changed, 55 insertions, 0 deletions

diff --git a/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/statespace/internal/ActivationStoreWorker.java b/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/statespace/internal/ActivationStoreWorker.java
new file mode 100644
index 00000000..1d7c5ce5
--- /dev/null
+++ b/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/statespace/internal/ActivationStoreWorker.java
@@ -0,0 +1,55 @@
+/*
+ * SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/>
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ */
+package tools.refinery.store.dse.transition.statespace.internal;
+
+import tools.refinery.store.dse.transition.Transformation;
+import tools.refinery.store.dse.transition.VersionWithObjectiveValue;
+import tools.refinery.store.dse.transition.statespace.ActivationStore;
+
+import java.util.List;
+import java.util.Random;
+
+public class ActivationStoreWorker {
+        final ActivationStore store;
+        final List<Transformation> transformations;
+
+        public ActivationStoreWorker(ActivationStore store, List<Transformation> transformations) {
+                this.store = store;
+                this.transformations = transformations;
+        }
+
+        public int[] calculateEmptyActivationSize() {
+                int[] result = new int[transformations.size()];
+                for (int i = 0; i < result.length; i++) {
+                        result[i] = transformations.get(i).getAllActivationsAsResultSet().size();
+                }
+                return result;
+        }
+
+
+        public ActivationStore.VisitResult fireRandomActivation(VersionWithObjectiveValue thisVersion, Random random) {
+                var result = store.getRandomAndMarkAsVisited(thisVersion, random);
+                if (result.successfulVisit()) {
+                        int selectedTransformation = result.transformation();
+                        int selectedActivation = result.activation();
+
+                        Transformation transformation = transformations.get(selectedTransformation);
+                        var tuple = transformation.getActivation(selectedActivation);
+
+                        boolean success = transformation.fireActivation(tuple);
+                        if (success) {
+                                return result;
+                        } else {
+                                return new ActivationStore.VisitResult(
+                                                false,
+                                                result.mayHaveMore(),
+                                                selectedTransformation,
+                                                selectedActivation);
+                        }
+                }
+                return result;
+        }
+}