aboutsummaryrefslogtreecommitdiffstats
path: root/subprojects/store-reasoning/src/main/java/tools/refinery/store/reasoning/lifting/DnfLifter.java
blob: 1eeb5de15cf77e3551a42444758f3f569349a9cc (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.reasoning.lifting;

import tools.refinery.logic.dnf.*;
import tools.refinery.logic.equality.DnfEqualityChecker;
import tools.refinery.logic.literal.Literal;
import tools.refinery.store.reasoning.literal.Concreteness;
import tools.refinery.store.reasoning.literal.Modality;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class DnfLifter {
	private final Map<ModalDnf, Dnf> cache = new HashMap<>();

	public <T> Query<T> lift(Modality modality, Concreteness concreteness, Query<T> query) {
		var liftedDnf = lift(modality, concreteness, query.getDnf());
		return query.withDnf(liftedDnf);
	}

	public RelationalQuery lift(Modality modality, Concreteness concreteness, RelationalQuery query) {
		var liftedDnf = lift(modality, concreteness, query.getDnf());
		return query.withDnf(liftedDnf);
	}

	public <T> FunctionalQuery<T> lift(Modality modality, Concreteness concreteness, FunctionalQuery<T> query) {
		var liftedDnf = lift(modality, concreteness, query.getDnf());
		return query.withDnf(liftedDnf);
	}

	public Dnf lift(Modality modality, Concreteness concreteness, Dnf dnf) {
		return cache.computeIfAbsent(new ModalDnf(modality, concreteness, dnf), this::doLift);
	}

	private Dnf doLift(ModalDnf modalDnf) {
		var modality = modalDnf.modality();
		var concreteness = modalDnf.concreteness();
		var dnf = modalDnf.dnf();
		var builder = Dnf.builder(decorateName(dnf.name(), modality, concreteness));
		builder.symbolicParameters(dnf.getSymbolicParameters());
		builder.functionalDependencies(dnf.getFunctionalDependencies());
		for (var clause : dnf.getClauses()) {
			builder.clause(liftClause(modality, concreteness, dnf, clause));
		}
		var liftedDnf = builder.build();
		if (dnf.equalsWithSubstitution(DnfEqualityChecker.DEFAULT, liftedDnf)) {
			return dnf;
		}
		return liftedDnf;
	}

	private List<Literal> liftClause(Modality modality, Concreteness concreteness, Dnf dnf, DnfClause clause) {
		var lifter = new ClauseLifter(modality, concreteness, dnf, clause);
		return lifter.liftClause();
	}

	private record ModalDnf(Modality modality, Concreteness concreteness, Dnf dnf) {
		@Override
		public String toString() {
			return "%s %s %s".formatted(modality, concreteness, dnf.name());
		}
	}

	public static String decorateName(String name, Modality modality, Concreteness concreteness) {
		return "%s#%s#%s".formatted(name, modality, concreteness);
	}
}