subprojects/store-query/src/main/java/tools/refinery/store/query/substitution/Substitution.java


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

/*
 * SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package tools.refinery.store.query.substitution;

import tools.refinery.store.query.term.AnyDataVariable;
import tools.refinery.store.query.term.DataVariable;
import tools.refinery.store.query.term.NodeVariable;
import tools.refinery.store.query.term.Variable;

@FunctionalInterface
public interface Substitution {
	Variable getSubstitute(Variable variable);

	default NodeVariable getTypeSafeSubstitute(NodeVariable variable) {
		var substitute = getSubstitute(variable);
		return substitute.asNodeVariable();
	}

	default AnyDataVariable getTypeSafeSubstitute(AnyDataVariable variable) {
		return getTypeSafeSubstitute((DataVariable<?>) variable);
	}

	default <T> DataVariable<T> getTypeSafeSubstitute(DataVariable<T> variable) {
		var substitute = getSubstitute(variable);
		return substitute.asDataVariable(variable.getType());
	}

	default Substitution andThen(Substitution second) {
		return new CompositeSubstitution(this, second);
	}
}