aboutsummaryrefslogtreecommitdiffstats
path: root/subprojects/language/src/main/java/tools/refinery/language/validation/ReferenceCounter.java
blob: 55cbd71d1d2d87117a080f25222f3fd104426773 (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
/*
 * SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package tools.refinery.language.validation;

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

import org.eclipse.emf.ecore.EObject;
import org.eclipse.xtext.util.IResourceScopeCache;
import org.eclipse.xtext.util.Tuples;

import com.google.inject.Inject;
import com.google.inject.Singleton;

import tools.refinery.language.model.problem.Problem;

@Singleton
public class ReferenceCounter {
	@Inject
	private IResourceScopeCache cache = IResourceScopeCache.NullImpl.INSTANCE;

	public int countReferences(Problem problem, EObject eObject) {
		var count = getReferenceCounts(problem).get(eObject);
		if (count == null) {
			return 0;
		}
		return count;
	}

	protected Map<EObject, Integer> getReferenceCounts(Problem problem) {
		var resource = problem.eResource();
		if (resource == null) {
			return doGetReferenceCounts(problem);
		}
		return cache.get(Tuples.create(problem, "referenceCounts"), resource, () -> doGetReferenceCounts(problem));
	}

	protected Map<EObject, Integer> doGetReferenceCounts(Problem problem) {
		var map = new HashMap<EObject, Integer>();
		countCrossReferences(problem, map);
		var iterator = problem.eAllContents();
		while (iterator.hasNext()) {
			var eObject = iterator.next();
			countCrossReferences(eObject, map);
		}
		return map;
	}

	protected void countCrossReferences(EObject eObject, Map<EObject, Integer> map) {
		for (var referencedObject : eObject.eCrossReferences()) {
			map.compute(referencedObject, (key, currentValue) -> currentValue == null ? 1 : currentValue + 1);
		}
	}
}