add scattergeo lasso/select

- with test support.

Author: etpinard

src/traces/scattergeo/index.js

@@ -18,11 +18,12 @@ ScatterGeo.calc = require('./calc');
 ScatterGeo.plot = require('./plot');
 ScatterGeo.hoverPoints = require('./hover');
 ScatterGeo.eventData = require('./event_data');
+ScatterGeo.selectPoints = require('./select');
 
 ScatterGeo.moduleType = 'trace';
 ScatterGeo.name = 'scattergeo';
 ScatterGeo.basePlotModule = require('../../plots/geo');
-ScatterGeo.categories = ['geo', 'symbols', 'markerColorscale', 'showLegend'];
+ScatterGeo.categories = ['geo', 'symbols', 'markerColorscale', 'showLegend', 'scatter-like'];
 ScatterGeo.meta = {
     hrName: 'scatter_geo',
     description: [

src/traces/scattergeo/select.js

@@ -0,0 +1,63 @@
+/**
+* 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 subtypes = require('../scatter/subtypes');
+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 trace = cd[0].trace;
+    var node3 = cd[0].node3;
+
+    var di, lonlat, x, y, i;
+
+    var hasOnlyLines = (!subtypes.hasMarkers(trace) && !subtypes.hasText(trace));
+    if(trace.visible !== true || hasOnlyLines) return;
+
+    var marker = trace.marker;
+    var opacity = Array.isArray(marker.opacity) ? 1 : marker.opacity;
+
+    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];
+            lonlat = di.lonlat;
+            x = xa.c2p(lonlat);
+            y = ya.c2p(lonlat);
+
+            if(polygon.contains([x, y])) {
+                selection.push({
+                    pointNumber: i,
+                    lon: lonlat[0],
+                    lat: lonlat[1]
+                });
+                di.dim = 0;
+            } else {
+                di.dim = 1;
+            }
+        }
+    }
+
+    node3.selectAll('path.point').style('opacity', function(d) {
+        return ((d.mo + 1 || opacity + 1) - 1) * (d.dim ? DESELECTDIM : 1);
+    });
+
+    node3.selectAll('text').style('opacity', function(d) {
+        return d.dim ? DESELECTDIM : 1;
+    });
+
+    return selection;
+};

test/jasmine/tests/select_test.js

@@ -653,4 +653,54 @@ describe('Test select box and lasso per trace:', function() {
         .catch(fail)
         .then(done);
     }, LONG_TIMEOUT_INTERVAL);
+
+    it('should work on scattergeo traces', function(done) {
+        gd = createGraphDiv();
+
+        var assertPoints = makeAssertPoints(['lon', 'lat']);
+        var assertRanges = makeAssertRanges('geo');
+        var assertLassoPoints = makeAssertLassoPoints('geo');
+
+        Plotly.plot(gd, [{
+            type: 'scattergeo',
+            lon: [10, 20, 30],
+            lat: [10, 20, 30]
+        }, {
+            type: 'scattergeo',
+            lon: [-10, -20, -30],
+            lat: [10, 20, 30]
+        }], {
+            showlegend: false,
+            dragmode: 'select',
+            width: 800,
+            height: 600
+        })
+        .then(function() {
+            return _run([[350, 200], [450, 400]], function() {
+                assertPoints([[10, 10], [20, 20], [-10, 10], [-20, 20]]);
+                assertRanges([[-28.13, 61.88], [28.13, -50.64]]);
+            });
+        })
+        .then(function() {
+            return Plotly.relayout(gd, 'dragmode', 'lasso');
+        })
+        .then(function() {
+            return _run([[300, 200], [300, 300], [400, 300], [400, 200], [300, 200]], function() {
+                assertPoints([[-10, 10], [-20, 20], [-30, 30]]);
+                assertLassoPoints([
+                    [-56.25, 61.88], [-56.24, 5.63], [0, 5.63], [0, 61.88], [-56.25, 61.88]
+                ]);
+            });
+        })
+        .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);
+
 });