From a8e7628e8301248738a2c2bd9e6f67017a1f2102 Mon Sep 17 00:00:00 2001
From: PlutonicOverkill <PlutonicOverkill@users.noreply.github.com>
Date: Fri, 28 Apr 2017 19:28:05 +1200
Subject: [PATCH] Enable highlighting when cursor is placed over a name (script
 editor)

---
 apps/opencs/model/prefs/state.cpp            |  1 +
 apps/opencs/view/world/scriptedit.cpp        | 23 ++++++++++++++++++++
 apps/opencs/view/world/scriptedit.hpp        |  3 +++
 apps/opencs/view/world/scripthighlighter.cpp | 18 ++++++++++++---
 apps/opencs/view/world/scripthighlighter.hpp |  7 +++++-
 5 files changed, 48 insertions(+), 4 deletions(-)

diff --git a/apps/opencs/model/prefs/state.cpp b/apps/opencs/model/prefs/state.cpp
index b8d6102ac..6bef55c44 100644
--- a/apps/opencs/model/prefs/state.cpp
+++ b/apps/opencs/model/prefs/state.cpp
@@ -156,6 +156,7 @@ void CSMPrefs::State::declare()
     declareColour ("colour-int", "Highlight Colour: Integer Literals", QColor ("darkmagenta"));
     declareColour ("colour-float", "Highlight Colour: Float Literals", QColor ("magenta"));
     declareColour ("colour-name", "Highlight Colour: Names", QColor ("grey"));
+    declareColour ("colour-highlight", "Highlight Colour: Highlighting", QColor("palegreen"));
     declareColour ("colour-keyword", "Highlight Colour: Keywords", QColor ("red"));
     declareColour ("colour-special", "Highlight Colour: Special Characters", QColor ("darkorange"));
     declareColour ("colour-comment", "Highlight Colour: Comments", QColor ("green"));
diff --git a/apps/opencs/view/world/scriptedit.cpp b/apps/opencs/view/world/scriptedit.cpp
index 6f27d5656..1581c3e34 100644
--- a/apps/opencs/view/world/scriptedit.cpp
+++ b/apps/opencs/view/world/scriptedit.cpp
@@ -49,6 +49,7 @@ CSVWorld::ScriptEdit::ScriptEdit(
     mDefaultFont(font()),
     mMonoFont(QFont("Monospace")),
     mTabCharCount(4),
+    mMarkOccurrencesRunning(false),
     mDocument(document),
     mWhiteListQoutes("^[a-z|_]{1}[a-z|0-9|_]{0,}$", Qt::CaseInsensitive)
 {
@@ -85,6 +86,8 @@ CSVWorld::ScriptEdit::ScriptEdit(
                   <<CSMWorld::UniversalId::Type_Weapon
                   <<CSMWorld::UniversalId::Type_Script
                   <<CSMWorld::UniversalId::Type_Region;
+    
+    connect(this, SIGNAL(selectionChanged()), this, SLOT(markOccurrences()));
 
     mHighlighter = new ScriptHighlighter (document.getData(), mode, ScriptEdit::document());
 
@@ -284,6 +287,26 @@ void CSVWorld::ScriptEdit::updateLineNumberArea(const QRect &rect, int dy)
         updateLineNumberAreaWidth(0);
 }
 
+void CSVWorld::ScriptEdit::markOccurrences()
+{
+    // prevent infinite recursion with cursor.select(),
+    // which ends up calling this function again
+    // could be fixed with blockSignals, but mDocument is const
+    if (mMarkOccurrencesRunning)
+        return;
+
+    mMarkOccurrencesRunning = true;
+
+    QTextCursor cursor = textCursor();
+    cursor.select(QTextCursor::WordUnderCursor);
+    QString word = cursor.selectedText();
+
+    mHighlighter->setMarkedWord(word.toStdString());
+    mHighlighter->rehighlight();
+
+    mMarkOccurrencesRunning = false;
+}
+
 void CSVWorld::ScriptEdit::resizeEvent(QResizeEvent *e)
 {
     QPlainTextEdit::resizeEvent(e);
diff --git a/apps/opencs/view/world/scriptedit.hpp b/apps/opencs/view/world/scriptedit.hpp
index 4977ed8e0..5c78470b3 100644
--- a/apps/opencs/view/world/scriptedit.hpp
+++ b/apps/opencs/view/world/scriptedit.hpp
@@ -54,6 +54,7 @@ namespace CSVWorld
             QFont mDefaultFont;
             QFont mMonoFont;
             int mTabCharCount;
+            bool mMarkOccurrencesRunning;
 
         protected:
 
@@ -111,6 +112,8 @@ namespace CSVWorld
             void updateLineNumberAreaWidth(int newBlockCount);
 
             void updateLineNumberArea(const QRect &, int);
+
+            void markOccurrences();
     };
 
     class LineNumberArea : public QWidget
diff --git a/apps/opencs/view/world/scripthighlighter.cpp b/apps/opencs/view/world/scripthighlighter.cpp
index 846a61b47..3041eff2d 100644
--- a/apps/opencs/view/world/scripthighlighter.cpp
+++ b/apps/opencs/view/world/scripthighlighter.cpp
@@ -72,7 +72,11 @@ void CSVWorld::ScriptHighlighter::highlight (const Compiler::TokenLoc& loc, Type
     // compensate for bug in Compiler::Scanner (position of token is the character after the token)
     index -= length;
 
-    setFormat (index, length, mScheme[type]);
+    QTextCharFormat scheme = mScheme[type];
+    if (loc.mLiteral == mMarkedWord)
+        scheme.merge(mScheme[Type_Highlight]);
+
+    setFormat (index, length, scheme);
 }
 
 CSVWorld::ScriptHighlighter::ScriptHighlighter (const CSMWorld::Data& data, Mode mode,
@@ -105,6 +109,11 @@ void CSVWorld::ScriptHighlighter::highlightBlock (const QString& text)
     catch (...) {} // ignore syntax errors
 }
 
+void CSVWorld::ScriptHighlighter::setMarkedWord(const std::string& name)
+{
+    mMarkedWord = name;
+}
+
 void CSVWorld::ScriptHighlighter::invalidateIds()
 {
     mContext.invalidateIds();
@@ -117,7 +126,7 @@ bool CSVWorld::ScriptHighlighter::settingChanged (const CSMPrefs::Setting *setti
         static const char *const colours[Type_Id+2] =
         {
             "colour-int", "colour-float", "colour-name", "colour-keyword",
-            "colour-special", "colour-comment", "colour-id",
+            "colour-special", "colour-comment", "colour-highlight", "colour-id",
             0
         };
 
@@ -125,7 +134,10 @@ bool CSVWorld::ScriptHighlighter::settingChanged (const CSMPrefs::Setting *setti
             if (setting->getKey()==colours[i])
             {
                 QTextCharFormat format;
-                format.setForeground (setting->toColor());
+                if (i == Type_Highlight)
+                    format.setBackground (setting->toColor());
+                else
+                    format.setForeground (setting->toColor());
                 mScheme[static_cast<Type> (i)] = format;
                 return true;
             }
diff --git a/apps/opencs/view/world/scripthighlighter.hpp b/apps/opencs/view/world/scripthighlighter.hpp
index 33824da0d..d55cf4839 100644
--- a/apps/opencs/view/world/scripthighlighter.hpp
+++ b/apps/opencs/view/world/scripthighlighter.hpp
@@ -2,6 +2,7 @@
 #define CSV_WORLD_SCRIPTHIGHLIGHTER_H
 
 #include <map>
+#include <string>
 
 #include <QSyntaxHighlighter>
 
@@ -30,7 +31,8 @@ namespace CSVWorld
                 Type_Keyword = 3,
                 Type_Special = 4,
                 Type_Comment = 5,
-                Type_Id = 6
+                Type_Highlight = 6,
+                Type_Id = 7
             };
 
             enum Mode
@@ -47,6 +49,7 @@ namespace CSVWorld
             CSMWorld::ScriptContext mContext;
             std::map<Type, QTextCharFormat> mScheme;
             Mode mMode;
+            std::string mMarkedWord;
 
         private:
 
@@ -91,6 +94,8 @@ namespace CSVWorld
 
             virtual void highlightBlock (const QString& text);
 
+            void setMarkedWord(const std::string& name);
+
             void invalidateIds();
 
             bool settingChanged (const CSMPrefs::Setting *setting);