1 files changed, 46 insertions, 0 deletions

diff --git a/subprojects/store-query/src/main/java/tools/refinery/store/query/dnf/FunctionalQueryBuilder.java b/subprojects/store-query/src/main/java/tools/refinery/store/query/dnf/FunctionalQueryBuilder.java
new file mode 100644
index 00000000..ca2bc006
--- /dev/null
+++ b/subprojects/store-query/src/main/java/tools/refinery/store/query/dnf/FunctionalQueryBuilder.java
@@ -0,0 +1,46 @@
+package tools.refinery.store.query.dnf;
+
+import tools.refinery.store.query.literal.Literal;
+import tools.refinery.store.query.term.Variable;
+
+import java.util.Collection;
+import java.util.Set;
+
+public final class FunctionalQueryBuilder<T> {
+        private final DnfBuilder dnfBuilder;
+        private final Class<T> type;
+
+        FunctionalQueryBuilder(DnfBuilder dnfBuilder, Class<T> type) {
+                this.dnfBuilder = dnfBuilder;
+                this.type = type;
+        }
+
+        public FunctionalQueryBuilder<T> functionalDependencies(Collection<FunctionalDependency<Variable>> functionalDependencies) {
+                dnfBuilder.functionalDependencies(functionalDependencies);
+                return this;
+        }
+
+        public FunctionalQueryBuilder<T> functionalDependency(FunctionalDependency<Variable> functionalDependency) {
+                dnfBuilder.functionalDependency(functionalDependency);
+                return this;
+        }
+
+        public FunctionalQueryBuilder<T> functionalDependency(Set<? extends Variable> forEach, Set<? extends Variable> unique) {
+                dnfBuilder.functionalDependency(forEach, unique);
+                return this;
+        }
+
+        public FunctionalQueryBuilder<T> clause(Literal... literals) {
+                dnfBuilder.clause(literals);
+                return this;
+        }
+
+        public FunctionalQueryBuilder<T> clause(Collection<? extends Literal> literals) {
+                dnfBuilder.clause(literals);
+                return this;
+        }
+
+        public FunctionalQuery<T> build() {
+                return dnfBuilder.build().asFunction(type);
+        }
+}