1 files changed, 32 insertions, 0 deletions

diff --git a/language/src/main/java/org/eclipse/viatra/solver/language/resource/ProblemLocationInFileProvider.java b/language/src/main/java/org/eclipse/viatra/solver/language/resource/ProblemLocationInFileProvider.java
new file mode 100644
index 00000000..94dbdfee
--- /dev/null
+++ b/language/src/main/java/org/eclipse/viatra/solver/language/resource/ProblemLocationInFileProvider.java
@@ -0,0 +1,32 @@
+package org.eclipse.viatra.solver.language.resource;
+
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.viatra.solver.language.ProblemUtil;
+import org.eclipse.viatra.solver.language.model.problem.ImplicitVariable;
+import org.eclipse.viatra.solver.language.model.problem.Node;
+import org.eclipse.xtext.resource.DefaultLocationInFileProvider;
+import org.eclipse.xtext.util.ITextRegion;
+
+public class ProblemLocationInFileProvider extends DefaultLocationInFileProvider {
+        @Override
+        protected ITextRegion doGetTextRegion(EObject obj, RegionDescription query) {
+                if (obj instanceof Node) {
+                        return getNodeTextRegion((Node) obj, query);
+                }
+                if (obj instanceof ImplicitVariable) {
+                        return ITextRegion.EMPTY_REGION;
+                }
+                return super.doGetTextRegion(obj, query);
+        }
+
+        protected ITextRegion getNodeTextRegion(Node node, RegionDescription query) {
+                if (ProblemUtil.isEnumLiteral(node)) {
+                        return super.doGetTextRegion(node, query);
+                }
+                if (ProblemUtil.isNewNode(node)) {
+                        EObject container = node.eContainer();
+                        return doGetTextRegion(container, query);
+                }
+                return ITextRegion.EMPTY_REGION;
+        }
+}