add indexToPoints to groupby transforms

Author: etpinard

src/transforms/groupby.js

@@ -11,6 +11,7 @@
 var Lib = require('../lib');
 var PlotSchema = require('../plot_api/plot_schema');
 var Plots = require('../plots/plots');
+var pointsAccessorFunction = require('./helpers').pointsAccessorFunction;
 
 exports.moduleType = 'transform';
 
@@ -161,6 +162,7 @@ function transformOne(trace, state) {
 
     var opts = state.transform;
     var groups = trace.transforms[state.transformIndex].groups;
+    var originalPointsAccessor = pointsAccessorFunction(trace.transforms, opts);
 
     if(!(Array.isArray(groups)) || groups.length === 0) {
         return [trace];
@@ -184,14 +186,17 @@ function transformOne(trace, state) {
 
     // An index to map group name --> expanded trace index
     var indexLookup = {};
+    var indexCnts = {};
 
     for(i = 0; i < groupNames.length; i++) {
         groupName = groupNames[i];
         indexLookup[groupName] = i;
+        indexCnts[groupName] = 0;
 
         // Start with a deep extend that just copies array references.
         newTrace = newData[i] = Lib.extendDeepNoArrays({}, trace);
         newTrace._group = groupName;
+        newTrace.transforms[state.transformIndex]._indexToPoints = {};
 
         var suppliedName = null;
         if(groupNameObj) {
@@ -246,6 +251,14 @@ function transformOne(trace, state) {
         }
     }
 
+    for(j = 0; j < len; j++) {
+        newTrace = newData[indexLookup[groups[j]]];
+
+        var indexToPoints = newTrace.transforms[state.transformIndex]._indexToPoints;
+        indexToPoints[indexCnts[groups[j]]] = originalPointsAccessor(j);
+        indexCnts[groups[j]]++;
+    }
+
     for(i = 0; i < groupNames.length; i++) {
         groupName = groupNames[i];
         newTrace = newData[i];

test/jasmine/tests/transform_groupby_test.js

@@ -57,8 +57,10 @@ describe('groupby', function() {
                 expect(gd._fullData.length).toEqual(2);
                 expect(gd._fullData[0].x).toEqual([1, -1, 0, 3]);
                 expect(gd._fullData[0].y).toEqual([1, 2, 1, 1]);
+                expect(gd._fullData[0].transforms[0]._indexToPoints).toEqual({0: [0], 1: [1], 2: [3], 3: [6]});
                 expect(gd._fullData[1].x).toEqual([-2, 1, 2]);
                 expect(gd._fullData[1].y).toEqual([3, 2, 3]);
+                expect(gd._fullData[1].transforms[0]._indexToPoints).toEqual({0: [2], 1: [4], 2: [5]});
 
                 assertDims([4, 3]);
 
@@ -460,6 +462,7 @@ describe('groupby', function() {
                     expect(gd._fullData[0].y).toEqual([0, 1, 3, 6]);
                     expect(gd._fullData[0].marker.line.width).toEqual([4, 2, 2, 3]);
 
+
                     expect(gd._fullData[1].ids).toEqual(['r', 'y', 'u']);
                     expect(gd._fullData[1].x).toEqual([-2, 1, 2]);
                     expect(gd._fullData[1].y).toEqual([2, 5, 4]);