add styleOnSelect for geo traces

Author: etpinard

src/traces/choropleth/index.js

@@ -16,7 +16,8 @@ Choropleth.supplyDefaults = require('./defaults');
 Choropleth.colorbar = require('../heatmap/colorbar');
 Choropleth.calc = require('./calc');
 Choropleth.plot = require('./plot');
-Choropleth.style = require('./style');
+Choropleth.style = require('./style').style;
+Choropleth.styleOnSelect = require('./style').styleOnSelect;
 Choropleth.hoverPoints = require('./hover');
 Choropleth.eventData = require('./event_data');
 Choropleth.selectPoints = require('./select');

src/traces/choropleth/plot.js

@@ -15,7 +15,7 @@ var polygon = require('../../lib/polygon');
 
 var getTopojsonFeatures = require('../../lib/topojson_utils').getTopojsonFeatures;
 var locationToFeature = require('../../lib/geo_location_utils').locationToFeature;
-var style = require('./style');
+var style = require('./style').style;
 
 module.exports = function plot(gd, geo, calcData) {
     for(var i = 0; i < calcData.length; i++) {

src/traces/choropleth/style.js

@@ -13,9 +13,9 @@ var Color = require('../../components/color');
 var Drawing = require('../../components/drawing');
 var Colorscale = require('../../components/colorscale');
 
-module.exports = function style(gd, calcTrace) {
+function style(gd, calcTrace) {
     if(calcTrace) styleTrace(gd, calcTrace);
-};
+}
 
 function styleTrace(gd, calcTrace) {
     var trace = calcTrace[0].trace;
@@ -40,5 +40,21 @@ function styleTrace(gd, calcTrace) {
             .style('opacity', marker.opacity);
     });
 
-    Drawing.selectedPointStyle(locs, trace);
+    Drawing.selectedPointStyle(locs, trace, gd);
 }
+
+function styleOnSelect(gd, calcTrace) {
+    var s = calcTrace[0].node3;
+    var trace = calcTrace[0].trace;
+
+    if(trace.selectedpoints) {
+        Drawing.selectedPointStyle(s.selectAll('.choroplethlocation'), trace, gd);
+    } else {
+        styleTrace(gd, calcTrace);
+    }
+}
+
+module.exports = {
+    style: style,
+    styleOnSelect: styleOnSelect
+};

src/traces/scattergeo/index.js

@@ -17,6 +17,7 @@ ScatterGeo.colorbar = require('../scatter/colorbar');
 ScatterGeo.calc = require('./calc');
 ScatterGeo.plot = require('./plot');
 ScatterGeo.style = require('./style');
+ScatterGeo.styleOnSelect = require('../scatter/style').styleOnSelect;
 ScatterGeo.hoverPoints = require('./hover');
 ScatterGeo.eventData = require('./event_data');
 ScatterGeo.selectPoints = require('./select');