replace selectAll with for(k in _plots) loop

Author: etpinard

src/components/shapes/draw.js

@@ -42,7 +42,11 @@ function draw(gd) {
     // Remove previous shapes before drawing new in shapes in fullLayout.shapes
     fullLayout._shapeUpperLayer.selectAll('path').remove();
     fullLayout._shapeLowerLayer.selectAll('path').remove();
-    fullLayout._shapeSubplotLayers.selectAll('path').remove();
+
+    for(var k in fullLayout._plots) {
+        var shapelayer = fullLayout._plots[k].shapelayer;
+        if(shapelayer) shapelayer.selectAll('path').remove();
+    }
 
     for(var i = 0; i < fullLayout.shapes.length; i++) {
         if(fullLayout.shapes[i].visible) {

src/plot_api/subroutines.js

@@ -529,10 +529,6 @@ exports.drawData = function(gd) {
         basePlotModules[i].plot(gd);
     }
 
-    // keep reference to shape layers in subplots
-    var layerSubplot = fullLayout._paper.selectAll('.layer-subplot');
-    fullLayout._shapeSubplotLayers = layerSubplot.selectAll('.shapelayer');
-
     // styling separate from drawing
     Plots.style(gd);