group all Axes.makeLabelFns functions in one object

etpinard · etpinard · commit 7e55b517fd78 · 2019-02-12T15:44:07.000-05:00
diff --git a/src/components/colorbar/draw.js b/src/components/colorbar/draw.js
@@ -434,7 +434,6 @@ module.exports = function draw(gd, id) {
 
                     var vals = Axes.calcTicks(cbAxisOut);
                     var transFn = Axes.makeTransFn(cbAxisOut);
-                    var labelFns = Axes.makeLabelFns(cbAxisOut, shift);
                     var tickSign = Axes.getTickSigns(cbAxisOut)[2];
 
                     Axes.drawTicks(gd, cbAxisOut, {
@@ -448,9 +447,7 @@ module.exports = function draw(gd, id) {
                         vals: vals,
                         layer: axisLayer,
                         transFn: transFn,
-                        labelXFn: labelFns.labelXFn,
-                        labelYFn: labelFns.labelYFn,
-                        labelAnchorFn: labelFns.labelAnchorFn
+                        labelFns: Axes.makeLabelFns(cbAxisOut, shift)
                     });
                 },
                 function() {
diff --git a/src/plots/cartesian/axes.js b/src/plots/cartesian/axes.js
@@ -1802,14 +1802,11 @@ axes.drawOne = function(gd, ax, opts) {
     // TODO: mirror labels, esp for subplots
 
     seq.push(function() {
-        var labelFns = axes.makeLabelFns(ax, mainLinePosition);
         return axes.drawLabels(gd, ax, {
             vals: vals,
             layer: mainAxLayer,
             transFn: transFn,
-            labelXFn: labelFns.labelXFn,
-            labelYFn: labelFns.labelYFn,
-            labelAnchorFn: labelFns.labelAnchorFn,
+            labelFns: axes.makeLabelFns(ax, mainLinePosition)
         });
     });
 
@@ -1821,8 +1818,6 @@ axes.drawOne = function(gd, ax, opts) {
         seq.push(function() {
             labelLength += getLabelLevelSpan(ax, axId + 'tick') + pad;
             labelLength += ax._tickAngles[axId + 'tick'] ? ax.tickfont.size * LINE_SPACING : 0;
-            var secondaryPosition = mainLinePosition + labelLength * sgn;
-            var secondaryLabelFns = axes.makeLabelFns(ax, secondaryPosition);
 
             return axes.drawLabels(gd, ax, {
                 vals: getSecondaryLabelVals(ax, vals),
@@ -1831,9 +1826,7 @@ axes.drawOne = function(gd, ax, opts) {
                 repositionOnUpdate: true,
                 secondary: true,
                 transFn: transFn,
-                labelXFn: secondaryLabelFns.labelXFn,
-                labelYFn: secondaryLabelFns.labelYFn,
-                labelAnchorFn: secondaryLabelFns.labelAnchorFn,
+                labelFns: axes.makeLabelFns(ax, mainLinePosition + labelLength * sgn)
             });
         });
 
@@ -2178,9 +2171,9 @@ axes.makeTickPath = function(ax, shift, sgn, len) {
  * @param {number} shift
  * @param {number} angle [in degrees] ...
  * @return {object}
- *  - {fn} labelXFn
- *  - {fn} labelYFn
- *  - {fn} labelAnchorFn
+ *  - {fn} xFn
+ *  - {fn} yFn
+ *  - {fn} anchorFn
  *  - {number} labelStandoff (gap parallel to ticks)
  *  - {number} labelShift (gap perpendicular to ticks)
  */
@@ -2210,15 +2203,16 @@ axes.makeLabelFns = function(ax, shift, angle) {
     };
 
     var x0, y0, ff, flipIt;
+
     if(axLetter === 'x') {
         flipIt = ax.side === 'bottom' ? 1 : -1;
         x0 = labelShift * flipIt;
         y0 = shift + labelStandoff * flipIt;
         ff = ax.side === 'bottom' ? 1 : -0.2;
 
-        out.labelXFn = function(d) { return d.dx + x0; };
-        out.labelYFn = function(d) { return d.dy + y0 + d.fontSize * ff; };
-        out.labelAnchorFn = function(a) {
+        out.xFn = function(d) { return d.dx + x0; };
+        out.yFn = function(d) { return d.dy + y0 + d.fontSize * ff; };
+        out.anchorFn = function(d, a) {
             if(!isNumeric(a) || a === 0 || a === 180) {
                 return 'middle';
             }
@@ -2230,9 +2224,9 @@ axes.makeLabelFns = function(ax, shift, angle) {
         y0 = -labelShift * flipIt;
         ff = Math.abs(ax.tickangle) === 90 ? 0.5 : 0;
 
-        out.labelXFn = function(d) { return d.dx + shift + (x0 + d.fontSize * ff) * flipIt; };
-        out.labelYFn = function(d) { return d.dy + y0 + d.fontSize * MID_SHIFT; };
-        out.labelAnchorFn = function(a) {
+        out.xFn = function(d) { return d.dx + shift + (x0 + d.fontSize * ff) * flipIt; };
+        out.yFn = function(d) { return d.dy + y0 + d.fontSize * MID_SHIFT; };
+        out.anchorFn = function(d, a) {
             if(isNumeric(a) && Math.abs(a) === 90) {
                 return 'middle';
             }
@@ -2411,10 +2405,11 @@ axes.drawZeroLine = function(gd, ax, opts) {
  * - {string (optional)} cls (node className)
  * - {boolean} repositionOnUpdate (set to true to reposition update selection)
  * - {boolean} secondary
- * - {fn} transFn
- * - {fn} labelXFn
- * - {fn} labelYFn
- * - {fn} labelAnchorFn
+ * - {object} labelFns
+ *   + {fn} transFn
+ *   + {fn} labelXFn
+ *   + {fn} labelYFn
+ *   + {fn} labelAnchorFn
  */
 axes.drawLabels = function(gd, ax, opts) {
     opts = opts || {};
@@ -2423,9 +2418,7 @@ axes.drawLabels = function(gd, ax, opts) {
     var axLetter = axId.charAt(0);
     var cls = opts.cls || axId + 'tick';
     var vals = opts.vals;
-    var labelXFn = opts.labelXFn;
-    var labelYFn = opts.labelYFn;
-    var labelAnchorFn = opts.labelAnchorFn;
+    var labelFns = opts.labelFns;
     var tickAngle = opts.secondary ? 0 : ax.tickangle;
     var lastAngle = (ax._tickAngles || {})[cls];
 
@@ -2445,7 +2438,7 @@ axes.drawLabels = function(gd, ax, opts) {
                 var newPromise = gd._promises.length;
 
                 thisLabel
-                    .call(svgTextUtils.positionText, labelXFn(d), labelYFn(d))
+                    .call(svgTextUtils.positionText, labelFns.xFn(d), labelFns.yFn(d))
                     .call(Drawing.font, d.font, d.fontSize, d.fontColor)
                     .text(d.text)
                     .call(svgTextUtils.convertToTspans, gd);
@@ -2469,7 +2462,7 @@ axes.drawLabels = function(gd, ax, opts) {
     if(opts.repositionOnUpdate) {
         tickLabels.each(function(d) {
             d3.select(this).select('text')
-                .call(svgTextUtils.positionText, labelXFn(d), labelYFn(d));
+                .call(svgTextUtils.positionText, labelFns.xFn(d), labelFns.yFn(d));
         });
     }
 
@@ -2497,12 +2490,12 @@ axes.drawLabels = function(gd, ax, opts) {
         s.each(function(d) {
             var thisLabel = d3.select(this);
             var mathjaxGroup = thisLabel.select('.text-math-group');
-            var anchor = labelAnchorFn(angle, d);
+            var anchor = labelFns.anchorFn(d, angle);
 
             var transform = opts.transFn.call(thisLabel.node(), d) +
                 ((isNumeric(angle) && +angle !== 0) ?
-                (' rotate(' + angle + ',' + labelXFn(d) + ',' +
-                    (labelYFn(d) - d.fontSize / 2) + ')') :
+                (' rotate(' + angle + ',' + labelFns.xFn(d) + ',' +
+                    (labelFns.yFn(d) - d.fontSize / 2) + ')') :
                 '');
 
             var anchorHeight = getAnchorHeight(
diff --git a/src/plots/polar/polar.js b/src/plots/polar/polar.js
@@ -404,7 +404,6 @@ proto.updateRadialAxis = function(fullLayout, polarLayout) {
 
         var vals = Axes.calcTicks(ax);
         var valsClipped = Axes.clipEnds(ax, vals);
-        var labelFns = Axes.makeLabelFns(ax, 0);
         var tickSign = Axes.getTickSigns(ax)[2];
 
         Axes.drawTicks(gd, ax, {
@@ -427,9 +426,7 @@ proto.updateRadialAxis = function(fullLayout, polarLayout) {
             vals: vals,
             layer: layers['radial-axis'],
             transFn: transFn,
-            labelXFn: labelFns.labelXFn,
-            labelYFn: labelFns.labelYFn,
-            labelAnchorFn: labelFns.labelAnchorFn
+            labelFns: Axes.makeLabelFns(ax, 0)
         });
     }
 
@@ -556,20 +553,21 @@ proto.updateAngularAxis = function(fullLayout, polarLayout) {
 
     var out = Axes.makeLabelFns(ax, 0);
     var labelStandoff = out.labelStandoff;
+    var labelFns = {};
 
-    var labelXFn = function(d) {
+    labelFns.xFn = function(d) {
         var rad = t2g(d);
         return Math.cos(rad) * labelStandoff;
     };
 
-    var labelYFn = function(d) {
+    labelFns.yFn = function(d) {
         var rad = t2g(d);
         var ff = Math.sin(rad) > 0 ? 0.2 : 1;
         return -Math.sin(rad) * (labelStandoff + d.fontSize * ff) +
             Math.abs(Math.cos(rad)) * (d.fontSize * MID_SHIFT);
     };
 
-    var labelAnchorFn = function(angle, d) {
+    labelFns.anchorFn = function(d) {
         var rad = t2g(d);
         var cos = Math.cos(rad);
         return Math.abs(cos) < 0.1 ?
@@ -635,9 +633,7 @@ proto.updateAngularAxis = function(fullLayout, polarLayout) {
             layer: layers['angular-axis'],
             repositionOnUpdate: true,
             transFn: transFn,
-            labelXFn: labelXFn,
-            labelYFn: labelYFn,
-            labelAnchorFn: labelAnchorFn
+            labelFns: labelFns
         });
     }
 
diff --git a/src/plots/ternary/ternary.js b/src/plots/ternary/ternary.js
@@ -459,15 +459,11 @@ proto.drawAx = function(ax) {
         crisp: false
     });
 
-    var labelFns = Axes.makeLabelFns(ax, 0, counterAngle);
-
     Axes.drawLabels(gd, ax, {
         vals: vals,
         layer: axLayer,
         transFn: transFn,
-        labelXFn: labelFns.labelXFn,
-        labelYFn: labelFns.labelYFn,
-        labelAnchorFn: labelFns.labelAnchorFn
+        labelFns: Axes.makeLabelFns(ax, 0, counterAngle)
     });
 };
 
