From b6563387515aa1b05bd2ae21785194e571485734 Mon Sep 17 00:00:00 2001
From: Kota Yamaguchi <KotaYamaguchi1984@gmail.com>
Date: Tue, 21 Jul 2015 17:37:18 +0900
Subject: [PATCH] Added onhighlight and onmousemove callbacks

---
 css/main.css                   |  3 +++
 js/app/edit.js                 | 14 +++++++++++++-
 js/helper/segment-annotator.js | 18 ++++++++++++++----
 js/main.js                     |  2 +-
 4 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/css/main.css b/css/main.css
index c572803..7f8f067 100644
--- a/css/main.css
+++ b/css/main.css
@@ -82,6 +82,9 @@
 .edit-sidebar-button-enabled {
   background-color: #ccc;
 }
+.edit-sidebar-button-highlight {
+  background-color: #aaa;
+}
 .edit-sidebar-legend-colorbox {
   background-color: white;
   border: 1px solid gray;
diff --git a/js/app/edit.js b/js/app/edit.js
index 537fee9..2111edc 100644
--- a/js/app/edit.js
+++ b/js/app/edit.js
@@ -293,6 +293,17 @@ function(Layer, Annotator, util) {
     return pickButton;
   }
 
+  // Hightlight legend labels.
+  function highlightLabel(label) {
+    var highlightClass = "edit-sidebar-button-highlight",
+        elements = document.getElementsByClassName(highlightClass);
+    for (var i = 0; i < elements.length; ++i)
+      elements[i].classList.remove(highlightClass);
+    var pickButton = document.getElementById("label-" + label + "-button");
+    if (pickButton)
+      pickButton.classList.add(highlightClass);
+  }
+
   // Create the label picker button.
   function createLabelPicker(params, data, annotator) {
     var container = document.createElement("div");
@@ -398,7 +409,8 @@ function(Layer, Annotator, util) {
           },
           onrightclick: function (label) {
             document.getElementById("label-" + label + "-button").click();
-          }
+          },
+          onmousemove: highlightLabel
         }),
         imageLayer = new Layer(data.imageURLs[id], {
           width: params.width,
diff --git a/js/helper/segment-annotator.js b/js/helper/segment-annotator.js
index e5d9b0c..e4b7508 100644
--- a/js/helper/segment-annotator.js
+++ b/js/helper/segment-annotator.js
@@ -32,6 +32,8 @@ function(Layer, segmentation, morph) {
     this.onchange = options.onchange || null;
     this.onrightclick = options.onrightclick || null;
     this.onleftclick = options.onleftclick || null;
+    this.onhighlight = options.onhighlight || null;
+    this.onmousemove = options.onmousemove || null;
     this._createLayers(options);
     this._initializeHistory(options);
     var annotator = this;
@@ -311,15 +313,18 @@ function(Layer, segmentation, morph) {
     function updateIfActive(event) {
       var offset = annotator._getClickOffset(event),
           superpixelData = annotator.layers.superpixel.imageData.data,
+          annotationData = annotator.layers.annotation.imageData.data,
           superpixelIndex = _getEncodedLabel(superpixelData, offset),
-          pixels = annotator.pixelIndex[superpixelIndex];
+          pixels = annotator.pixelIndex[superpixelIndex],
+          existingLabel = _getEncodedLabel(annotationData, offset);
       annotator._updateHighlight(pixels);
+      if (typeof annotator.onmousemove === "function") {
+        annotator.onmousemove.call(annotator, existingLabel);
+      }
       if (mousestate.down) {
         if (mousestate.button == 2 &&
             typeof annotator.onrightclick === "function") {
-          var annotationData = annotator.layers.annotation.imageData.data;
-          annotator.onrightclick.call(annotator,
-                                      _getEncodedLabel(annotationData, offset));
+          annotator.onrightclick.call(annotator, existingLabel);
         }
         else {
           annotator._updateAnnotation(pixels, annotator.currentLabel);
@@ -332,6 +337,9 @@ function(Layer, segmentation, morph) {
     canvas.addEventListener('mouseup', updateIfActive);
     canvas.addEventListener('mouseleave', function (event) {
       annotator._updateHighlight(null);
+      if (typeof annotator.onmousemove === "function") {
+        annotator.onmousemove.call(annotator, null);
+      }
     });
     canvas.addEventListener('mousedown', function (event) {
       mousestate.down = true;
@@ -436,6 +444,8 @@ function(Layer, segmentation, morph) {
     }
     this.layers.visualization.render();
     this.layers.boundary.render();
+    if (typeof this.onhighlight === "function")
+      this.onhighlight.call(this);
   };
 
   Annotator.prototype._fillPixels = function (pixels, labels) {
diff --git a/js/main.js b/js/main.js
index b0f2a76..e77ad4f 100644
--- a/js/main.js
+++ b/js/main.js
@@ -17,7 +17,7 @@ function(indexPage, editPage, colormap, util) {
       }) :
       [[255, 255, 255],
        [226, 196, 196],
-       [64, 32, 32]].concat(colormap.create("hhsv", {
+       [64, 32, 32]].concat(colormap.create("hsv", {
         size: labels.length - 3
       }));
   }