Merge pull request #4595 from plotly/fix-splom-clear-select-via-api

Reset splom selectBatch and unselectBatch on updates

Author: etpinard

src/traces/splom/scene_update.js

@@ -20,7 +20,11 @@ module.exports = function sceneUpdate(gd, trace) {
     var splomScenes = fullLayout._splomScenes;
     if(!splomScenes) splomScenes = fullLayout._splomScenes = {};
 
-    var reset = {dirty: true};
+    var reset = {
+        dirty: true,
+        selectBatch: [],
+        unselectBatch: []
+    };
 
     var first = {
         matrix: false,

test/jasmine/tests/splom_test.js

@@ -1837,4 +1837,45 @@ describe('Test splom select:', function() {
         .catch(failTest)
         .then(done);
     });
+
+    it('should be able to select and then clear using API', function(done) {
+        function _assert(msg, exp) {
+            return function() {
+                var uid = gd._fullData[0].uid;
+                var scene = gd._fullLayout._splomScenes[uid];
+                expect(scene.selectBatch).withContext(msg + ' selectBatch').toEqual(exp.selectBatch);
+                expect(scene.unselectBatch).withContext(msg + ' unselectBatch').toEqual(exp.unselectBatch);
+            };
+        }
+
+        Plotly.plot(gd, [{
+            type: 'splom',
+            dimensions: [{
+                values: [1, 2, 3]
+            }, {
+                values: [2, 3, 0]
+            }]
+        }], {
+            width: 400,
+            height: 400,
+            margin: {l: 0, t: 0, r: 0, b: 0},
+            dragmode: 'select'
+        })
+        .then(_assert('base', {
+            selectBatch: [],
+            unselectBatch: []
+        }))
+        .then(function() { return _select([[50, 50], [195, 195]]); })
+        .then(_assert('after selection', {
+            selectBatch: [1],
+            unselectBatch: [0, 2]
+        }))
+        .then(function() { return Plotly.restyle(gd, 'selectedpoints', null); })
+        .then(_assert('after API clear', {
+            selectBatch: [],
+            unselectBatch: []
+        }))
+        .catch(failTest)
+        .then(done);
+    });
 });