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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
/*
* SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
*
* SPDX-License-Identifier: EPL-2.0
*/
package tools.refinery.language.resource;
import com.google.inject.Inject;
import com.google.inject.Provider;
import com.google.inject.Singleton;
import com.google.inject.name.Named;
import org.eclipse.emf.common.notify.impl.AdapterImpl;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.util.EcoreUtil;
import org.eclipse.xtext.Constants;
import org.eclipse.xtext.resource.DerivedStateAwareResource;
import org.eclipse.xtext.resource.IDerivedStateComputer;
import org.eclipse.xtext.resource.XtextResource;
import tools.refinery.language.model.problem.*;
import java.util.*;
import java.util.function.Function;
@Singleton
public class ProblemDerivedStateComputer implements IDerivedStateComputer {
public static final String NEW_NODE = "new";
@Inject
@Named(Constants.LANGUAGE_NAME)
private String languageName;
@Inject
private Provider<NodeNameCollector> nodeNameCollectorProvider;
@Inject
private DerivedVariableComputer derivedVariableComputer;
@Override
public void installDerivedState(DerivedStateAwareResource resource, boolean preLinkingPhase) {
var problem = getProblem(resource);
if (problem != null) {
var adapter = getOrInstallAdapter(resource);
installDerivedProblemState(problem, adapter, preLinkingPhase);
}
}
protected Problem getProblem(Resource resource) {
List<EObject> contents = resource.getContents();
if (contents.isEmpty()) {
return null;
}
EObject object = contents.get(0);
if (object instanceof Problem problem) {
return problem;
}
return null;
}
protected void installDerivedProblemState(Problem problem, Adapter adapter, boolean preLinkingPhase) {
installNewNodes(problem, adapter);
if (preLinkingPhase) {
return;
}
Set<String> nodeNames = installDerivedNodes(problem);
derivedVariableComputer.installDerivedVariables(problem, nodeNames);
}
protected void installNewNodes(Problem problem, Adapter adapter) {
for (Statement statement : problem.getStatements()) {
if (statement instanceof ClassDeclaration declaration && !declaration.isAbstract()
&& declaration.getNewNode() == null) {
var newNode = adapter.createNewNodeIfAbsent(declaration, key -> createNode(NEW_NODE));
declaration.setNewNode(newNode);
}
}
}
protected Set<String> installDerivedNodes(Problem problem) {
var collector = nodeNameCollectorProvider.get();
collector.collectNodeNames(problem);
Set<String> nodeNames = collector.getNodeNames();
List<Node> grapNodes = problem.getNodes();
for (String nodeName : nodeNames) {
var graphNode = createNode(nodeName);
grapNodes.add(graphNode);
}
return nodeNames;
}
protected Node createNode(String name) {
var node = ProblemFactory.eINSTANCE.createNode();
node.setName(name);
return node;
}
@Override
public void discardDerivedState(DerivedStateAwareResource resource) {
var problem = getProblem(resource);
if (problem != null) {
var adapter = getOrInstallAdapter(resource);
discardDerivedProblemState(problem, adapter);
}
}
protected void discardDerivedProblemState(Problem problem, Adapter adapter) {
Set<ClassDeclaration> classDeclarations = new HashSet<>();
problem.getNodes().clear();
for (var statement : problem.getStatements()) {
if (statement instanceof ClassDeclaration classDeclaration) {
classDeclaration.setNewNode(null);
classDeclarations.add(classDeclaration);
}
}
adapter.retainAll(classDeclarations);
derivedVariableComputer.discardDerivedVariables(problem);
}
protected Adapter getOrInstallAdapter(Resource resource) {
if (!(resource instanceof XtextResource)) {
return new Adapter();
}
String resourceLanguageName = ((XtextResource) resource).getLanguageName();
if (!languageName.equals(resourceLanguageName)) {
return new Adapter();
}
var adapter = (Adapter) EcoreUtil.getAdapter(resource.eAdapters(), Adapter.class);
if (adapter == null) {
adapter = new Adapter();
resource.eAdapters().add(adapter);
}
return adapter;
}
protected static class Adapter extends AdapterImpl {
private final Map<ClassDeclaration, Node> newNodes = new HashMap<>();
public Node createNewNodeIfAbsent(ClassDeclaration classDeclaration,
Function<ClassDeclaration, Node> createNode) {
return newNodes.computeIfAbsent(classDeclaration, createNode);
}
public void retainAll(Collection<ClassDeclaration> classDeclarations) {
newNodes.keySet().retainAll(classDeclarations);
}
@Override
public boolean isAdapterForType(Object type) {
return Adapter.class == type;
}
}
}
|
