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
|
/*
* SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
*
* SPDX-License-Identifier: EPL-2.0
*/
package tools.refinery.language.resource;
import com.google.common.collect.ImmutableMap;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.xtext.EcoreUtil2;
import org.eclipse.xtext.naming.IQualifiedNameConverter;
import org.eclipse.xtext.naming.QualifiedName;
import org.eclipse.xtext.resource.EObjectDescription;
import org.eclipse.xtext.resource.IEObjectDescription;
import org.eclipse.xtext.resource.impl.DefaultResourceDescriptionStrategy;
import org.eclipse.xtext.util.IAcceptor;
import tools.refinery.language.model.problem.*;
import tools.refinery.language.naming.NamingUtil;
import tools.refinery.language.utils.ProblemUtil;
import java.util.Map;
@Singleton
public class ProblemResourceDescriptionStrategy extends DefaultResourceDescriptionStrategy {
public static final String ERROR_PREDICATE = "tools.refinery.language.resource" +
".ProblemResourceDescriptionStrategy.ERROR_PREDICATE";
public static final String ERROR_PREDICATE_TRUE = "true";
@Inject
private IQualifiedNameConverter qualifiedNameConverter;
@Override
public boolean createEObjectDescriptions(EObject eObject, IAcceptor<IEObjectDescription> acceptor) {
if (!shouldExport(eObject)) {
return false;
}
var qualifiedName = getNameAsQualifiedName(eObject);
if (qualifiedName == null) {
return true;
}
var problem = EcoreUtil2.getContainerOfType(eObject, Problem.class);
var problemQualifiedName = getNameAsQualifiedName(problem);
var userData = getUserData(eObject);
boolean nameExported;
if (shouldExportSimpleName(eObject)) {
acceptEObjectDescription(eObject, problemQualifiedName, qualifiedName, userData, acceptor);
nameExported = true;
} else {
nameExported = false;
}
var parent = eObject.eContainer();
while (parent != null && parent != problem) {
var parentQualifiedName = getNameAsQualifiedName(parent);
if (parentQualifiedName == null) {
parent = parent.eContainer();
continue;
}
qualifiedName = parentQualifiedName.append(qualifiedName);
if (shouldExportSimpleName(parent)) {
acceptEObjectDescription(eObject, problemQualifiedName, qualifiedName, userData, acceptor);
nameExported = true;
} else {
nameExported = false;
}
parent = parent.eContainer();
}
if (!nameExported) {
acceptEObjectDescription(eObject, problemQualifiedName, qualifiedName, userData, acceptor);
}
return true;
}
protected QualifiedName getNameAsQualifiedName(EObject eObject) {
if (!(eObject instanceof NamedElement namedElement)) {
return null;
}
var name = namedElement.getName();
if (NamingUtil.isNullOrEmpty(name)) {
return null;
}
return qualifiedNameConverter.toQualifiedName(name);
}
protected boolean shouldExport(EObject eObject) {
if (eObject instanceof Variable) {
// Variables are always private to the containing predicate definition.
return false;
}
if (eObject instanceof Node node) {
// Only enum literals and new nodes are visible across problem files.
return ProblemUtil.isIndividualNode(node) || ProblemUtil.isNewNode(node);
}
return true;
}
protected Map<String, String> getUserData(EObject eObject) {
var builder = ImmutableMap.<String, String>builder();
if (eObject instanceof PredicateDefinition predicateDefinition && predicateDefinition.isError()) {
builder.put(ERROR_PREDICATE, ERROR_PREDICATE_TRUE);
}
return builder.build();
}
protected boolean shouldExportSimpleName(EObject eObject) {
if (eObject instanceof Node node) {
return !ProblemUtil.isNewNode(node);
}
if (eObject instanceof PredicateDefinition predicateDefinition) {
return !ProblemUtil.isInvalidMultiplicityConstraint(predicateDefinition);
}
return true;
}
private void acceptEObjectDescription(EObject eObject, QualifiedName prefix, QualifiedName qualifiedName,
Map<String, String> userData, IAcceptor<IEObjectDescription> acceptor) {
var qualifiedNameWithPrefix = prefix == null ? qualifiedName : prefix.append(qualifiedName);
var description = EObjectDescription.create(qualifiedNameWithPrefix, eObject, userData);
acceptor.accept(description);
}
}
|