1 files changed, 39 insertions, 0 deletions

diff --git a/org.eclipse.viatra.solver.language.parent/org.eclipse.viatra.solver.language/src/main/java/org/eclipse/viatra/solver/language/resource/ProblemResourceDescriptionStrategy.java b/org.eclipse.viatra.solver.language.parent/org.eclipse.viatra.solver.language/src/main/java/org/eclipse/viatra/solver/language/resource/ProblemResourceDescriptionStrategy.java
new file mode 100644
index 00000000..7b37ffd6
--- /dev/null
+++ b/org.eclipse.viatra.solver.language.parent/org.eclipse.viatra.solver.language/src/main/java/org/eclipse/viatra/solver/language/resource/ProblemResourceDescriptionStrategy.java
@@ -0,0 +1,39 @@
+package org.eclipse.viatra.solver.language.resource;
+
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.viatra.solver.language.model.problem.ReferenceDeclaration;
+import org.eclipse.viatra.solver.language.model.problem.Variable;
+import org.eclipse.xtext.naming.QualifiedName;
+import org.eclipse.xtext.resource.IEObjectDescription;
+import org.eclipse.xtext.resource.impl.AliasedEObjectDescription;
+import org.eclipse.xtext.resource.impl.DefaultResourceDescriptionStrategy;
+import org.eclipse.xtext.util.IAcceptor;
+
+import com.google.inject.Singleton;
+
+@Singleton
+public class ProblemResourceDescriptionStrategy extends DefaultResourceDescriptionStrategy {
+        @Override
+        public boolean createEObjectDescriptions(EObject eObject, IAcceptor<IEObjectDescription> acceptor) {
+                IAcceptor<IEObjectDescription> wrappedAcceptor;
+                if (eObject instanceof Variable) {
+                        return false;
+                }
+                if (eObject instanceof ReferenceDeclaration) {
+                        wrappedAcceptor = new IAcceptor<IEObjectDescription>() {
+                                public void accept(IEObjectDescription description) {
+                                        acceptor.accept(description);
+                                        QualifiedName qualifiedName = description.getQualifiedName();
+                                        if (qualifiedName.getSegmentCount() >= 2) {
+                                                QualifiedName simpleName = QualifiedName.create(qualifiedName.getLastSegment());
+                                                IEObjectDescription aliasedDescription = new AliasedEObjectDescription(simpleName, description);
+                                                acceptor.accept(aliasedDescription);
+                                        }
+                                };
+                        };
+                } else {
+                        wrappedAcceptor = acceptor;
+                }
+                return super.createEObjectDescriptions(eObject, wrappedAcceptor);
+        }
+}