aboutsummaryrefslogtreecommitdiffstats
path: root/subprojects/interpreter-rete/src/main/java/tools/refinery/interpreter/rete/itc/alg/representative/StronglyConnectedComponentAlgorithm.java
blob: d2e524f781324fcd4f0bb532aeda452d02c08b04 (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
/*
 * SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/>
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package tools.refinery.interpreter.rete.itc.alg.representative;

import tools.refinery.interpreter.rete.itc.graphimpl.Graph;
import tools.refinery.interpreter.rete.itc.alg.misc.GraphHelper;
import tools.refinery.interpreter.rete.itc.alg.misc.bfs.BFS;
import tools.refinery.interpreter.rete.itc.alg.misc.scc.SCC;

import java.util.Collection;
import java.util.Set;

public class StronglyConnectedComponentAlgorithm<T> extends RepresentativeElectionAlgorithm<T> {
	public StronglyConnectedComponentAlgorithm(Graph<T> graph) {
		super(graph);
	}

	@Override
	protected void initializeComponents() {
		var computedSCCs = SCC.computeSCC(graph).getSccs();
		for (var computedSCC : computedSCCs) {
			initializeSet(computedSCC);
		}
	}

	@Override
	public void edgeInserted(T source, T target) {
		var sourceRoot = getRepresentative(source);
		var targetRoot = getRepresentative(target);
		if (sourceRoot.equals(targetRoot)) {
			// New edge does not change strongly connected components.
			return;
		}
		if (BFS.isReachable(target, source, graph)) {
			var sources = BFS.reachableSources(graph, target);
			var targets = BFS.reachableTargets(graph, source);
			sources.retainAll(targets);
			merge(sources);
		}
	}

	@Override
	public void edgeDeleted(T source, T target) {
		var sourceRoot = getRepresentative(source);
		var targetRoot = getRepresentative(target);
		if (!sourceRoot.equals(targetRoot)) {
			// New edge does not change strongly connected components.
			return;
		}
		var component = GraphHelper.getSubGraph(getComponent(sourceRoot), graph);
		if (!BFS.isReachable(source, target, component)) {
			var newSCCs = SCC.computeSCC(component).getSccs();
			split(sourceRoot, newSCCs);
		}
	}

	private void split(T preservedRepresentative, Collection<? extends Set<T>> sets) {
		for (var set : sets) {
			if (set.contains(preservedRepresentative)) {
				components.put(preservedRepresentative, set);
			} else {
				assignNewRepresentative(preservedRepresentative, set);
			}
		}
	}
}