1 files changed, 52 insertions, 0 deletions

diff --git a/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/statespace/internal/SolutionStoreImpl.java b/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/statespace/internal/SolutionStoreImpl.java
new file mode 100644
index 00000000..43548eaa
--- /dev/null
+++ b/subprojects/store-dse/src/main/java/tools/refinery/store/dse/transition/statespace/internal/SolutionStoreImpl.java
@@ -0,0 +1,52 @@
+/*
+ * 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.VersionWithObjectiveValue;
+import tools.refinery.store.dse.transition.statespace.SolutionStore;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.PriorityQueue;
+
+
+public class SolutionStoreImpl implements SolutionStore {
+        final int maxNumberSolutions;
+        public static final int UNLIMITED = -1;
+        final PriorityQueue<VersionWithObjectiveValue> solutions;
+
+        public SolutionStoreImpl(int maxNumberSolutions) {
+                this.maxNumberSolutions = maxNumberSolutions;
+                solutions = new PriorityQueue<>(ObjectivePriorityQueueImpl.c1.reversed());
+        }
+
+
+        @Override
+        public synchronized boolean submit(VersionWithObjectiveValue version) {
+                boolean removeLast = hasEnoughSolution();
+                solutions.add(version);
+                if(removeLast) {
+                        var last = solutions.poll();
+                        return last != version;
+                } else {
+                        return true;
+                }
+        }
+
+        @Override
+        public List<VersionWithObjectiveValue> getSolutions() {
+                return new ArrayList<>(solutions);
+        }
+
+        @Override
+        public boolean hasEnoughSolution() {
+                if (maxNumberSolutions == UNLIMITED) {
+                        return false;
+                } else {
+                        return solutions.size() >= maxNumberSolutions;
+                }
+        }
+}