From 9308c84cdf5b52509ac6281d47bf015802e57668 Mon Sep 17 00:00:00 2001
From: Manindra Moharana <mkmoharana29@gmail.com>
Date: Sat, 5 Apr 2014 01:23:08 +0530
Subject: [PATCH] Further work on highlighting issue

---
 .../mode/experimental/ASTGenerator.java       |  7 +-
 .../mode/experimental/ASTNodeWrapper.java     | 76 +++++++++++++++++++
 2 files changed, 82 insertions(+), 1 deletion(-)

diff --git a/pdex/src/processing/mode/experimental/ASTGenerator.java b/pdex/src/processing/mode/experimental/ASTGenerator.java
index 325a98b81..8a58ae2a2 100644
--- a/pdex/src/processing/mode/experimental/ASTGenerator.java
+++ b/pdex/src/processing/mode/experimental/ASTGenerator.java
@@ -1678,7 +1678,12 @@ public class ASTGenerator {
       ASTNode simpName2 = getNodeName(decl,nameOfNode);
       logE("FINAL String decl: " + getNodeAsString(decl));
       logE("FINAL String label: " + getNodeAsString(simpName2));
-      errorCheckerService.highlightNode(simpName2);
+      //errorCheckerService.highlightNode(simpName2);
+      ASTNodeWrapper declWrap = new ASTNodeWrapper(simpName2,nodeLabel);
+      errorCheckerService.highlightNode(declWrap);
+//      if (!declWrap.highlightNode(this)) {
+//        logE("Highlighting failed.");
+//      }
     } 
 
     return new ASTNodeWrapper(decl,nodeLabel);
diff --git a/pdex/src/processing/mode/experimental/ASTNodeWrapper.java b/pdex/src/processing/mode/experimental/ASTNodeWrapper.java
index eb35e2215..3ad8bb16c 100644
--- a/pdex/src/processing/mode/experimental/ASTNodeWrapper.java
+++ b/pdex/src/processing/mode/experimental/ASTNodeWrapper.java
@@ -18,6 +18,7 @@
 
 package processing.mode.experimental;
 import static processing.mode.experimental.ExperimentalMode.log;
+import static processing.mode.experimental.ExperimentalMode.logE;
 import static processing.mode.experimental.ExperimentalMode.log2;
 import java.util.Iterator;
 import java.util.List;
@@ -25,6 +26,10 @@ import java.util.TreeMap;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
+import javax.swing.text.BadLocationException;
+import javax.swing.text.Element;
+import javax.swing.text.PlainDocument;
+
 import org.eclipse.jdt.core.dom.ASTNode;
 import org.eclipse.jdt.core.dom.CompilationUnit;
 import org.eclipse.jdt.core.dom.ExpressionStatement;
@@ -516,6 +521,77 @@ public class ASTNodeWrapper {
     return new int[][]{javaCodeMap,pdeCodeMap};
   }
   
+  public boolean highlightNode(ASTGenerator astGenerator){
+    if(!(Node instanceof SimpleName)){
+      return false; 
+    }
+    SimpleName nodeName = (SimpleName) Node;
+    try {
+      int javaLineNumber = getLineNumber(nodeName);
+      int pdeOffs[] = astGenerator.errorCheckerService
+          .calculateTabIndexAndLineNumber(javaLineNumber);
+      PlainDocument javaSource = new PlainDocument();
+      javaSource.insertString(0, astGenerator.errorCheckerService.sourceCode, null);
+      Element lineElement = javaSource.getDefaultRootElement()
+          .getElement(javaLineNumber-1);
+      if(lineElement == null) {
+        log(lineNumber + " line element null while highlighting " + nodeName);
+        return false;
+      }
+      
+      String javaLine = javaSource.getText(lineElement.getStartOffset(),
+                                           lineElement.getEndOffset()
+                                               - lineElement.getStartOffset());
+      astGenerator.editor.getSketch().setCurrentCode(pdeOffs[0]);
+      String pdeLine = astGenerator.editor.getLineText(pdeOffs[1]);
+      String lookingFor = nodeName.toString();
+      log(lookingFor + ", " + nodeName.getStartPosition());
+      log("JL " + javaLine + " LSO " + lineElement.getStartOffset() + ","
+          + lineElement.getEndOffset());
+      log("PL " + pdeLine);
+      if (!javaLine.contains(lookingFor) || !pdeLine.contains(lookingFor)) {
+        logE("Logical error in highLightNode(). Please file a bug report.");
+        return false;
+      }
+      Pattern toFind = Pattern.compile("\\b" + nodeName.toString() + "\\b");
+      Matcher matcher = toFind.matcher(javaLine);
+      int lsto = lineElement.getStartOffset();
+      while(matcher.find()){
+        System.out.println(matcher.start() + lsto);
+      }
+      // find the count of the name in the java code
+      int count = 0, index = 0;
+      do {
+        index = javaLine.indexOf(lookingFor, index);
+        if (index != -1) {
+          count++;
+          index += lookingFor.length();
+        } else
+          break;
+      } while (true);
+      log("count=" + count);
+      // find the offset of the name of that index in the pde code
+      index = 0;
+      while (count > 0) {
+        index = pdeLine.indexOf(lookingFor, index);
+        if (index != -1) {
+          count--;
+          index += lookingFor.length();
+        }
+      }
+      log("pde lso " + (index - lookingFor.length()));
+      
+      int lso = astGenerator.editor.ta.getLineStartOffset(pdeOffs[1]);
+      astGenerator.editor.setSelection(lso + index - lookingFor.length(), lso
+          + index);
+      return true;
+    } catch (BadLocationException e) {
+      logE("BLE in highLightNode() for " + nodeName);
+      e.printStackTrace();
+    }
+    return false;
+  }
+  
   /**
    * Gets offset mapping between java and pde code
    * int[0][x] stores the java code offset and