re-initialize categories internals after calc transforms

Author: etpinard

src/plots/plots.js

@@ -1973,8 +1973,6 @@ plots.doCalcdata = function(gd, traces) {
 
     var trace, _module, i, j;
 
-    var hasCategoryAxis = false;
-
     // XXX: Is this correct? Needs a closer look so that *some* traces can be recomputed without
     // *all* needing doCalcdata:
     var calcdata = new Array(fullData.length);
@@ -1996,20 +1994,6 @@ plots.doCalcdata = function(gd, traces) {
     fullLayout._piecolormap = {};
     fullLayout._piedefaultcolorcount = 0;
 
-    // initialize the category list, if there is one, so we start over
-    // to be filled in later by ax.d2c
-    for(i = 0; i < axList.length; i++) {
-        axList[i]._categories = axList[i]._initialCategories.slice();
-
-        // Build the lookup map for initialized categories
-        axList[i]._categoriesMap = {};
-        for(j = 0; j < axList[i]._categories.length; j++) {
-            axList[i]._categoriesMap[axList[i]._categories[j]] = j;
-        }
-
-        if(axList[i].type === 'category') hasCategoryAxis = true;
-    }
-
     // If traces were specified and this trace was not included,
     // then transfer it over from the old calcdata:
     for(i = 0; i < fullData.length; i++) {
@@ -2019,6 +2003,8 @@ plots.doCalcdata = function(gd, traces) {
         }
     }
 
+    var hasCategoryAxis = plots.initCategories(axList);
+
     var hasCalcTransform = false;
 
     // transform loop
@@ -2051,9 +2037,9 @@ plots.doCalcdata = function(gd, traces) {
             axList[i]._min = [];
             axList[i]._max = [];
             axList[i]._categories = [];
-            // Reset the look up map
             axList[i]._categoriesMap = {};
         }
+        plots.initCategories(axList);
     }
 
     // 'regular' loop
@@ -2099,6 +2085,26 @@ plots.doCalcdata = function(gd, traces) {
     }
 };
 
+plots.initCategories = function(axList) {
+    var hasCategoryAxis = false;
+
+    // initialize the category list, if there is one, so we start over
+    // to be filled in later by ax.d2c
+    for(var i = 0; i < axList.length; i++) {
+        axList[i]._categories = axList[i]._initialCategories.slice();
+
+        // Build the lookup map for initialized categories
+        axList[i]._categoriesMap = {};
+        for(var j = 0; j < axList[i]._categories.length; j++) {
+            axList[i]._categoriesMap[axList[i]._categories[j]] = j;
+        }
+
+        if(axList[i].type === 'category') hasCategoryAxis = true;
+    }
+
+    return hasCategoryAxis;
+};
+
 plots.rehover = function(gd) {
     if(gd._fullLayout._rehover) {
         gd._fullLayout._rehover();

test/jasmine/tests/transform_sort_test.js

@@ -355,4 +355,41 @@ describe('Test sort transform interactions:', function() {
         .catch(fail)
         .then(done);
     });
+
+    it('should honor *categoryarray* when set', function(done) {
+        var gd = createGraphDiv();
+
+        Plotly.plot(gd, [{
+            x: ['C', 'B', 'A'],
+            y: [3, 1, 2],
+            marker: {
+                size: [10, 20, 5]
+            },
+            transforms: [{
+                enabled: false,
+                type: 'sort',
+                target: [0, 2, 1],
+            }]
+        }], {
+            xaxis: {
+                categoryorder: 'trace',
+                categoryarray: ['A', 'B', 'C']
+            }
+        })
+        .then(function() {
+            expect(gd._fullLayout.xaxis._categories).toEqual(['C', 'B', 'A']);
+
+            return Plotly.restyle(gd, 'transforms[0].enabled', true);
+        })
+        .then(function() {
+            expect(gd._fullLayout.xaxis._categories).toEqual(['C', 'A', 'B']);
+
+            return Plotly.relayout(gd, 'xaxis.categoryorder', 'array');
+        })
+        .then(function() {
+            expect(gd._fullLayout.xaxis._categories).toEqual(['A', 'B', 'C']);
+        })
+        .catch(fail)
+        .then(done);
+    });
 });