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.resource;
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);
}
}
}
|