add choropleth lasso/select

- with test support.
- 'simply' use the location's centroid to determine
  if it is selected or not.

Author: etpinard

src/traces/choropleth/index.js

@@ -18,6 +18,7 @@ Choropleth.calc = require('./calc');
 Choropleth.plot = require('./plot');
 Choropleth.hoverPoints = require('./hover');
 Choropleth.eventData = require('./event_data');
+Choropleth.selectPoints = require('./select');
 
 Choropleth.moduleType = 'trace';
 Choropleth.name = 'choropleth';

src/traces/choropleth/select.js

@@ -0,0 +1,54 @@
+/**
+* Copyright 2012-2017, Plotly, Inc.
+* All rights reserved.
+*
+* This source code is licensed under the MIT license found in the
+* LICENSE file in the root directory of this source tree.
+*/
+
+'use strict';
+
+var DESELECTDIM = require('../../constants/interactions').DESELECTDIM;
+
+module.exports = function selectPoints(searchInfo, polygon) {
+    var cd = searchInfo.cd;
+    var xa = searchInfo.xaxis;
+    var ya = searchInfo.yaxis;
+    var selection = [];
+    var node3 = cd[0].node3;
+
+    var i, di, ct, x, y;
+
+    if(polygon === false) {
+        for(i = 0; i < cd.length; i++) {
+            cd[i].dim = 0;
+        }
+    } else {
+        for(i = 0; i < cd.length; i++) {
+            di = cd[i];
+            ct = di.ct;
+
+            if(!ct) continue;
+
+            x = xa.c2p(ct);
+            y = ya.c2p(ct);
+
+            if(polygon.contains([x, y])) {
+                selection.push({
+                    pointNumber: i,
+                    lon: ct[0],
+                    lat: ct[1]
+                });
+                di.dim = 0;
+            } else {
+                di.dim = 1;
+            }
+        }
+    }
+
+    node3.selectAll('path').style('opacity', function(d) {
+        return d.dim ? DESELECTDIM : 1;
+    });
+
+    return selection;
+};

test/jasmine/tests/select_test.js

@@ -703,4 +703,45 @@ describe('Test select box and lasso per trace:', function() {
         .then(done);
     }, LONG_TIMEOUT_INTERVAL);
 
+    it('should work on choropleth traces', function(done) {
+        gd = createGraphDiv();
+
+        var assertPoints = makeAssertPoints(['location', 'z']);
+        var assertRanges = makeAssertRanges('geo', -0.5);
+        var assertLassoPoints = makeAssertLassoPoints('geo', -0.5);
+
+        var fig = Lib.extendDeep({}, require('@mocks/geo_choropleth-text'));
+        fig.layout.width = 870;
+        fig.layout.height = 450;
+        fig.layout.dragmode = 'select';
+        fig.layout.geo.scope = 'europe';
+
+        Plotly.plot(gd, fig)
+        .then(function() {
+            return _run([[350, 200], [400, 250]], function() {
+                assertPoints([['GBR', 26.507354205352502], ['IRL', 86.4125147625692]]);
+                assertRanges([[-19.11, 63.06], [7.31, 53.72]]);
+            }, [280, 190]);
+        })
+        .then(function() {
+            return Plotly.relayout(gd, 'dragmode', 'lasso');
+        })
+        .then(function() {
+            return _run([[350, 200], [400, 200], [400, 250], [350, 250], [350, 200]], function() {
+                assertPoints([['GBR', 26.507354205352502], ['IRL', 86.4125147625692]]);
+                assertLassoPoints([
+                    [-19.11, 63.06], [5.50, 65.25], [7.31, 53.72], [-12.90, 51.70], [-19.11, 63.06]
+                ]);
+            }, [280, 190]);
+        })
+        .then(function() {
+            // make selection handlers don't get called in 'pan' dragmode
+            return Plotly.relayout(gd, 'dragmode', 'pan');
+        })
+        .then(function() {
+            return _runNoSelect([[370, 120], [500, 200]]);
+        })
+        .catch(fail)
+        .then(done);
+    }, LONG_TIMEOUT_INTERVAL);
 });