feat: command line model generator

4 files changed, 42 insertions, 15 deletions

diff --git a/subprojects/generator/src/main/java/tools/refinery/generator/ModelGenerator.java b/subprojects/generator/src/main/java/tools/refinery/generator/ModelGenerator.java
index 5b44c10a..1515dceb 100644
--- a/subprojects/generator/src/main/java/tools/refinery/generator/ModelGenerator.java
+++ b/subprojects/generator/src/main/java/tools/refinery/generator/ModelGenerator.java
@@ -5,7 +5,10 @@
  */
 package tools.refinery.generator;
 
+import com.google.inject.Provider;
+import tools.refinery.language.model.problem.Problem;
 import tools.refinery.language.semantics.ProblemTrace;
+import tools.refinery.language.semantics.SolutionSerializer;
 import tools.refinery.store.dse.strategy.BestFirstStoreManager;
 import tools.refinery.store.map.Version;
 import tools.refinery.store.model.ModelStore;
@@ -16,21 +19,22 @@ import tools.refinery.store.reasoning.seed.ModelSeed;
 
 public class ModelGenerator extends ModelFacade {
         private final Version initialVersion;
-
-        private int randomSeed = 0;
-
+        private final Provider<SolutionSerializer> solutionSerializerProvider;
+        private long randomSeed = 1;
         private boolean lastGenerationSuccessful;
 
-        public ModelGenerator(ProblemTrace problemTrace, ModelStore store, ModelSeed modelSeed) {
+        ModelGenerator(ProblemTrace problemTrace, ModelStore store, ModelSeed modelSeed,
+                          Provider<SolutionSerializer> solutionSerializerProvider) {
                 super(problemTrace, store, modelSeed, Concreteness.CANDIDATE);
+                this.solutionSerializerProvider = solutionSerializerProvider;
                 initialVersion = getModel().commit();
         }
 
-        public int getRandomSeed() {
+        public long getRandomSeed() {
                 return randomSeed;
         }
 
-        public void setRandomSeed(int randomSeed) {
+        public void setRandomSeed(long randomSeed) {
                 this.randomSeed = randomSeed;
                 this.lastGenerationSuccessful = false;
         }
@@ -50,7 +54,7 @@ public class ModelGenerator extends ModelFacade {
                 if (solutions.isEmpty()) {
                         return false;
                 }
-                getModel().restore(solutions.get(0).version());
+                getModel().restore(solutions.getFirst().version());
                 lastGenerationSuccessful = true;
                 return true;
         }
@@ -63,9 +67,19 @@ public class ModelGenerator extends ModelFacade {
 
         @Override
         public <A, C> PartialInterpretation<A, C> getPartialInterpretation(PartialSymbol<A, C> partialSymbol) {
+                checkSuccessfulGeneration();
+                return super.getPartialInterpretation(partialSymbol);
+        }
+
+        public Problem serializeSolution() {
+                checkSuccessfulGeneration();
+                var serializer = solutionSerializerProvider.get();
+                return serializer.serializeSolution(getProblemTrace(), getModel());
+        }
+
+        private void checkSuccessfulGeneration() {
                 if (!lastGenerationSuccessful) {
                         throw new IllegalStateException("No generated model is available");
                 }
-                return super.getPartialInterpretation(partialSymbol);
         }
 }
diff --git a/subprojects/generator/src/main/java/tools/refinery/generator/ModelGeneratorFactory.java b/subprojects/generator/src/main/java/tools/refinery/generator/ModelGeneratorFactory.java
index 6642d591..587601f2 100644
--- a/subprojects/generator/src/main/java/tools/refinery/generator/ModelGeneratorFactory.java
+++ b/subprojects/generator/src/main/java/tools/refinery/generator/ModelGeneratorFactory.java
@@ -9,6 +9,7 @@ import com.google.inject.Inject;
 import com.google.inject.Provider;
 import tools.refinery.language.model.problem.Problem;
 import tools.refinery.language.semantics.ModelInitializer;
+import tools.refinery.language.semantics.SolutionSerializer;
 import tools.refinery.store.dse.propagation.PropagationAdapter;
 import tools.refinery.store.dse.transition.DesignSpaceExplorationAdapter;
 import tools.refinery.store.model.ModelStore;
@@ -25,6 +26,9 @@ public final class ModelGeneratorFactory {
         @Inject
         private Provider<ModelInitializer> initializerProvider;
 
+        @Inject
+        private Provider<SolutionSerializer> solutionSerializerProvider;
+
         private CancellationToken cancellationToken = CancellationToken.NONE;
 
         private boolean debugPartialInterpretations;
@@ -53,7 +57,8 @@ public final class ModelGeneratorFactory {
                                                 .requiredInterpretations(getRequiredInterpretations()));
                 initializer.configureStoreBuilder(storeBuilder);
                 var store = storeBuilder.build();
-                return new ModelGenerator(initializer.getProblemTrace(), store, initializer.getModelSeed());
+                return new ModelGenerator(initializer.getProblemTrace(), store, initializer.getModelSeed(),
+                solutionSerializerProvider);
         }
 
         private Collection<Concreteness> getRequiredInterpretations() {
diff --git a/subprojects/generator/src/main/java/tools/refinery/generator/ModelSemantics.java b/subprojects/generator/src/main/java/tools/refinery/generator/ModelSemantics.java
index bc02c887..7207a509 100644
--- a/subprojects/generator/src/main/java/tools/refinery/generator/ModelSemantics.java
+++ b/subprojects/generator/src/main/java/tools/refinery/generator/ModelSemantics.java
@@ -11,7 +11,7 @@ import tools.refinery.store.reasoning.literal.Concreteness;
 import tools.refinery.store.reasoning.seed.ModelSeed;
 
 public class ModelSemantics extends ModelFacade {
-        public ModelSemantics(ProblemTrace problemTrace, ModelStore store, ModelSeed modelSeed) {
+        ModelSemantics(ProblemTrace problemTrace, ModelStore store, ModelSeed modelSeed) {
                 super(problemTrace, store, modelSeed, Concreteness.PARTIAL);
         }
 }
diff --git a/subprojects/generator/src/main/java/tools/refinery/generator/ProblemLoader.java b/subprojects/generator/src/main/java/tools/refinery/generator/ProblemLoader.java
index 2d93a5ad..abfbe7e6 100644
--- a/subprojects/generator/src/main/java/tools/refinery/generator/ProblemLoader.java
+++ b/subprojects/generator/src/main/java/tools/refinery/generator/ProblemLoader.java
@@ -48,21 +48,29 @@ public class ProblemLoader {
                 return this;
         }
 
-        public Problem loadString(String problemString) throws IOException {
+        public Problem loadString(String problemString, URI uri) throws IOException {
                 try (var stream = new LazyStringInputStream(problemString)) {
-                        return loadStream(stream);
+                        return loadStream(stream, uri);
                 }
         }
 
-        public Problem loadStream(InputStream inputStream) throws IOException {
+        public Problem loadString(String problemString) throws IOException {
+                return loadString(problemString, null);
+        }
+
+        public Problem loadStream(InputStream inputStream, URI uri) throws IOException {
                 var resourceSet = resourceSetProvider.get();
-                var uri = URI.createFileURI("__synthetic." + fileExtension);
-                var resource = resourceFactory.createResource(uri);
+                var resourceUri = uri == null ? URI.createFileURI("__synthetic." + fileExtension) : uri;
+                var resource = resourceFactory.createResource(resourceUri);
                 resourceSet.getResources().add(resource);
                 resource.load(inputStream, Map.of());
                 return loadResource(resource);
         }
 
+        public Problem loadStream(InputStream inputStream) throws IOException {
+                return loadStream(inputStream, null);
+        }
+
         public Problem loadFile(File file) throws IOException {
                 return loadFile(file.getAbsolutePath());
         }