aboutsummaryrefslogtreecommitdiffstats
path: root/subprojects/store-query/src/main/java/tools/refinery/store/query/Constraint.java
blob: 916fb35c6e9166869fa97f2f0096670024311925 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
/*
 * SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package tools.refinery.store.query;

import tools.refinery.store.query.equality.LiteralEqualityHelper;
import tools.refinery.store.query.literal.*;
import tools.refinery.store.query.term.*;

import java.util.List;

public interface Constraint {
	String name();

	List<Parameter> getParameters();

	default int arity() {
		return getParameters().size();
	}

	default boolean invalidIndex(int i) {
		return i < 0 || i >= arity();
	}

	default Reduction getReduction() {
		return Reduction.NOT_REDUCIBLE;
	}

	default boolean equals(LiteralEqualityHelper helper, Constraint other) {
		return equals(other);
	}

	default String toReferenceString() {
		return name();
	}

	default CallLiteral call(CallPolarity polarity, List<Variable> arguments) {
		return new CallLiteral(polarity, this, arguments);
	}

	default CallLiteral call(CallPolarity polarity, Variable... arguments) {
		return call(polarity, List.of(arguments));
	}

	default CallLiteral call(Variable... arguments) {
		return call(CallPolarity.POSITIVE, arguments);
	}

	default CallLiteral callTransitive(NodeVariable left, NodeVariable right) {
		return call(CallPolarity.TRANSITIVE, List.of(left, right));
	}

	default AssignedValue<Integer> count(List<Variable> arguments) {
		return targetVariable -> new CountLiteral(targetVariable, this, arguments);
	}

	default AssignedValue<Integer> count(Variable... arguments) {
		return count(List.of(arguments));
	}

	default <R, T> AssignedValue<R> aggregateBy(DataVariable<T> inputVariable, Aggregator<R, T> aggregator,
												List<Variable> arguments) {
		return targetVariable -> new AggregationLiteral<>(targetVariable, aggregator, inputVariable, this, arguments);
	}

	default <R, T> AssignedValue<R> aggregateBy(DataVariable<T> inputVariable, Aggregator<R, T> aggregator,
												Variable... arguments) {
		return aggregateBy(inputVariable, aggregator, List.of(arguments));
	}
}