Add seed option for random

2 files changed, 15 insertions, 0 deletions

diff --git a/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/dse/DesignSpaceExplorationAdapter.java b/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/dse/DesignSpaceExplorationAdapter.java
index e7ce7b2c..4cfca210 100644
--- a/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/dse/DesignSpaceExplorationAdapter.java
+++ b/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/dse/DesignSpaceExplorationAdapter.java
@@ -10,6 +10,7 @@ import tools.refinery.store.tuple.Tuple1;
 
 import java.util.Collection;
 import java.util.List;
+import java.util.Random;
 
 public interface DesignSpaceExplorationAdapter extends ModelAdapter {
         @Override
@@ -52,4 +53,8 @@ public interface DesignSpaceExplorationAdapter extends ModelAdapter {
         public ObjectiveComparatorHelper getObjectiveComparatorHelper();
 
         public void restoreTrajectory(List<Long>  trajectory);
+
+        public void setRandom(Random random);
+
+        public void setRandom(long seed);
 }
diff --git a/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/dse/internal/DesignSpaceExplorationAdapterImpl.java b/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/dse/internal/DesignSpaceExplorationAdapterImpl.java
index eb2fc063..c4c51e79 100644
--- a/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/dse/internal/DesignSpaceExplorationAdapterImpl.java
+++ b/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/dse/internal/DesignSpaceExplorationAdapterImpl.java
@@ -142,6 +142,16 @@ public class DesignSpaceExplorationAdapterImpl implements DesignSpaceExploration
         }
 
         @Override
+        public void setRandom(Random random) {
+                this.random = random;
+        }
+
+        @Override
+        public void setRandom(long seed) {
+                this.random = new Random(seed);
+        }
+
+        @Override
         public Fitness calculateFitness() {
                 Fitness result = new Fitness();
                 boolean satisfiesHardObjectives = true;