select colorbar group from infolayer

etpinard · etpinard · commit b38a1cf041f9 · 2016-01-27T14:04:27.000-05:00
diff --git a/src/plot_api/plot_api.js b/src/plot_api/plot_api.js
@@ -297,8 +297,14 @@ Plotly.plot = function(gd, data, layout, config) {
             trace = cd[0].trace;
             if (trace.visible !== true || !trace._module.colorbar) {
                 uid = trace.uid;
-                fullLayout._paper.selectAll('.hm'+uid+',.contour'+uid+',.cb'+uid+',#clip'+uid)
-                    .remove();
+
+                fullLayout._paper.selectAll(
+                    '.hm' + uid +
+                    ',.contour' + uid +
+                    ',#clip' + uid
+                ).remove();
+
+                fullLayout._infolayer.selectAll('.cb' + uid).remove();
             }
         }
 
diff --git a/src/traces/contour/plot.js b/src/traces/contour/plot.js
@@ -64,7 +64,8 @@ function plotOne(gd, plotinfo, cd) {
         pathinfo = emptyPathinfo(contours, plotinfo, cd[0]);
 
     if(trace.visible !== true) {
-        fullLayout._paper.selectAll('.'+id+',.cb'+uid+',.hm'+uid).remove();
+        fullLayout._paper.selectAll('.' + id + ',.hm' + id).remove();
+        fullLayout._infolayer.selectAll('.cb' + uid).remove();
         return;
     }
 
diff --git a/src/traces/heatmap/plot.js b/src/traces/heatmap/plot.js
@@ -33,14 +33,14 @@ function plotOne(gd, plotinfo, cd) {
         xa = plotinfo.x(),
         ya = plotinfo.y(),
         fullLayout = gd._fullLayout,
-        id = 'hm' + uid,
-        cbId = 'cb' + uid;
+        id = 'hm' + uid;
 
-    fullLayout._paper.selectAll('.contour' + uid).remove(); // in case this used to be a contour map
+    // in case this used to be a contour map
+    fullLayout._paper.selectAll('.contour' + uid).remove();
 
     if(trace.visible !== true) {
         fullLayout._paper.selectAll('.' + id).remove();
-        fullLayout._paper.selectAll('.' + cbId).remove();
+        fullLayout._infolayer.selectAll('.cb' + uid).remove();
         return;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,8 @@ function plotOne(gd, plotinfo, cd) {`
`64`	`64`	`pathinfo = emptyPathinfo(contours, plotinfo, cd[0]);`
`65`	`65`
`66`	`66`	`if(trace.visible !== true) {`
`67`		`- fullLayout._paper.selectAll('.'+id+',.cb'+uid+',.hm'+uid).remove();`
	`67`	`+ fullLayout._paper.selectAll('.' + id + ',.hm' + id).remove();`
	`68`	`+ fullLayout._infolayer.selectAll('.cb' + uid).remove();`
`68`	`69`	`return;`
`69`	`70`	`}`
`70`	`71`