1 files changed, 59 insertions, 0 deletions

diff --git a/Application/org.eclipse.viatra.solver.language.ui/src/org/eclipse/viatra/solver/language/ui/syntaxcoloring/SolverSemanticHighlightCalculator.xtend b/Application/org.eclipse.viatra.solver.language.ui/src/org/eclipse/viatra/solver/language/ui/syntaxcoloring/SolverSemanticHighlightCalculator.xtend
new file mode 100644
index 00000000..5242a82e
--- /dev/null
+++ b/Application/org.eclipse.viatra.solver.language.ui/src/org/eclipse/viatra/solver/language/ui/syntaxcoloring/SolverSemanticHighlightCalculator.xtend
@@ -0,0 +1,59 @@
+package org.eclipse.viatra.solver.language.ui.syntaxcoloring

+

+import java.util.Random

+import java.util.WeakHashMap

+import org.eclipse.viatra.solver.language.solverLanguage.Reference

+import org.eclipse.xtext.ide.editor.syntaxcoloring.DefaultSemanticHighlightingCalculator

+import org.eclipse.xtext.ide.editor.syntaxcoloring.IHighlightedPositionAcceptor

+import org.eclipse.xtext.nodemodel.util.NodeModelUtils

+import org.eclipse.xtext.resource.XtextResource

+import org.eclipse.xtext.util.CancelIndicator

+

+class SolverSemanticHighlightCalculator extends DefaultSemanticHighlightingCalculator {

+        public static val SYMBOL_CODE = "SymbolCode::"

+

+        val symbolName2ID = new WeakHashMap<String, String>

+

+        override provideHighlightingFor(XtextResource resource, IHighlightedPositionAcceptor acceptor,

+                CancelIndicator cancelIndicator) {

+                if (resource === null || resource.getParseResult() === null) {

+                        return

+                } else {

+                        val modelSymbols = resource.allContents.filter(Reference)

+                        while (modelSymbols.hasNext) {

+                                val symbolElement = modelSymbols.next

+                                val node = NodeModelUtils.getNode(symbolElement)

+                                acceptor.addPosition(

+                                        node.getOffset(),

+                                        node.getLength(),

+                                        getSymbolCode(node.text)

+                                )

+                        }

+                }

+        }

+

+        def getSymbolCode(String name) {

+                if (symbolName2ID.containsKey(name)) {

+                        return symbolName2ID.get(name)

+                } else {

+                        val id = calculateSymbolCode(name)

+                        symbolName2ID.put(name, id)

+                        return id

+                }

+        }

+

+        def String calculateSymbolCode(String name) {

+                val Random random = new Random(name.hashCode)

+                val hue = huePick(random.nextFloat)

+                val saturation = 0.2 + random.nextFloat * 0.3

+                val brightness = random.nextFloat * 0.2 + 0.7f

+                return '''«SYMBOL_CODE» «hue» «saturation» «brightness»'''

+        }

+

+        private def huePick(float random) {

+                val pink = 300

+                val tolerance = 10

+                val degree = random * (360 - 2 * tolerance)

+                if(degree < pink - tolerance) return degree else return degree + 2 * tolerance

+        }

+}