1 files changed, 62 insertions, 0 deletions

diff --git a/subprojects/store-dse/src/main/java/tools/refinery/store/dse/internal/DesignSpaceExplorationBuilderImpl.java b/subprojects/store-dse/src/main/java/tools/refinery/store/dse/internal/DesignSpaceExplorationBuilderImpl.java
new file mode 100644
index 00000000..03508adc
--- /dev/null
+++ b/subprojects/store-dse/src/main/java/tools/refinery/store/dse/internal/DesignSpaceExplorationBuilderImpl.java
@@ -0,0 +1,62 @@
+package tools.refinery.store.dse.internal;
+
+import tools.refinery.store.adapter.AbstractModelAdapterBuilder;
+import tools.refinery.store.model.ModelStore;
+import tools.refinery.store.model.ModelStoreBuilder;
+import tools.refinery.store.query.dnf.RelationalQuery;
+import tools.refinery.store.dse.DesignSpaceExplorationBuilder;
+import tools.refinery.store.dse.Strategy;
+import tools.refinery.store.dse.objectives.Objective;
+
+import java.util.LinkedHashSet;
+import java.util.LinkedList;
+import java.util.List;
+
+public class DesignSpaceExplorationBuilderImpl
+                extends AbstractModelAdapterBuilder<DesignSpaceExplorationStoreAdapterImpl>
+                implements DesignSpaceExplorationBuilder {
+        private final LinkedHashSet<TransformationRule> transformationSpecifications = new LinkedHashSet<>();
+        private final LinkedHashSet<RelationalQuery> globalConstraints = new LinkedHashSet<>();
+        private final List<Objective> objectives = new LinkedList<>();
+        private Strategy strategy;
+
+        @Override
+        protected DesignSpaceExplorationStoreAdapterImpl doBuild(ModelStore store) {
+                return new DesignSpaceExplorationStoreAdapterImpl(store, transformationSpecifications, globalConstraints,
+                                objectives, strategy);
+        }
+
+        @Override
+        public DesignSpaceExplorationBuilder transformation(TransformationRule transformationRule) {
+                checkNotConfigured();
+                transformationSpecifications.add(transformationRule);
+                return this;
+        }
+
+        @Override
+        public DesignSpaceExplorationBuilder globalConstraint(RelationalQuery globalConstraint) {
+                checkNotConfigured();
+                globalConstraints.add(globalConstraint);
+                return this;
+        }
+
+        @Override
+        public DesignSpaceExplorationBuilder objective(Objective objective) {
+                checkNotConfigured();
+                objectives.add(objective);
+                return this;
+        }
+
+        @Override
+        public DesignSpaceExplorationBuilder strategy(Strategy strategy) {
+                checkNotConfigured();
+                this.strategy = strategy;
+                return this;
+        }
+
+        @Override
+        protected void doConfigure(ModelStoreBuilder storeBuilder) {
+                storeBuilder.symbols(DesignSpaceExplorationAdapterImpl.NODE_COUNT_SYMBOL);
+                super.doConfigure(storeBuilder);
+        }
+}