subprojects/store-query/src/main/java/tools/refinery/store/query/rewriter/InputParameterResolver.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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

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

import tools.refinery.store.query.dnf.Dnf;
import tools.refinery.store.query.dnf.DnfBuilder;
import tools.refinery.store.query.literal.Literal;
import tools.refinery.store.query.term.ParameterDirection;
import tools.refinery.store.query.term.Variable;

import java.util.HashSet;
import java.util.List;

public class InputParameterResolver extends AbstractRecursiveRewriter {
	@Override
	protected Dnf doRewrite(Dnf dnf) {
		return rewriteWithContext(List.of(), dnf);
	}

	Dnf rewriteWithContext(List<Literal> context, Dnf dnf) {
		var dnfName = dnf.name();
		var builder = Dnf.builder(dnfName);
		createSymbolicParameters(context, dnf, builder);
		builder.functionalDependencies(dnf.getFunctionalDependencies());
		var clauses = dnf.getClauses();
		int clauseCount = clauses.size();
		for (int i = 0; i < clauseCount; i++) {
			var clause = clauses.get(i);
			var clauseRewriter = new ClauseInputParameterResolver(this, context, clause, dnfName, i);
			builder.clause(clauseRewriter.rewriteClause());
		}
		return builder.build();
	}

	private static void createSymbolicParameters(List<Literal> context, Dnf dnf, DnfBuilder builder) {
		var positiveInContext = new HashSet<Variable>();
		for (var literal : context) {
			positiveInContext.addAll(literal.getOutputVariables());
		}
		for (var symbolicParameter : dnf.getSymbolicParameters()) {
			var variable = symbolicParameter.getVariable();
			var isOutput = symbolicParameter.getDirection() == ParameterDirection.OUT ||
					positiveInContext.contains(variable);
			var direction = isOutput ? ParameterDirection.OUT : ParameterDirection.IN;
			builder.parameter(variable, direction);
		}
	}
}