split colorscale methods into seperate files

Author: etpinard

src/components/colorscale/calc.js

@@ -0,0 +1,63 @@
+/**
+* Copyright 2012-2015, 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 Lib = require('../../lib');
+
+var scales = require('./scales');
+var flipScale = require('./flip_scale');
+
+
+module.exports = function calc(trace, vals, containerStr, cLetter) {
+    var container, inputContainer;
+
+    if(containerStr) {
+        container = Lib.nestedProperty(trace, containerStr).get();
+        inputContainer = Lib.nestedProperty(trace._input, containerStr).get();
+    } else {
+        container = trace;
+        inputContainer = trace._input;
+    }
+
+    var auto = container[cLetter + 'auto'],
+        min = container[cLetter + 'min'],
+        max = container[cLetter + 'max'],
+        scl = container.colorscale;
+
+    if(auto!==false || min===undefined) {
+        min = Lib.aggNums(Math.min, null, vals);
+    }
+
+    if(auto!==false || max===undefined) {
+        max = Lib.aggNums(Math.max, null, vals);
+    }
+
+    if(min === max) {
+        min -= 0.5;
+        max += 0.5;
+    }
+
+    container[cLetter + 'min'] = min;
+    container[cLetter + 'max'] = max;
+
+    inputContainer[cLetter + 'min'] = min;
+    inputContainer[cLetter + 'max'] = max;
+
+    if(container.autocolorscale) {
+        if(min * max < 0) scl = scales.RdBu;
+        else if(min >= 0) scl = scales.Reds;
+        else scl = scales.Blues;
+
+        // reversescale is handled at the containerOut level
+        inputContainer.colorscale = scl;
+        if(container.reversescale) scl = flipScale(scl);
+        container.colorscale = scl;
+    }
+};

src/components/colorscale/default_scale.js

@@ -0,0 +1,12 @@
+/**
+* Copyright 2012-2015, 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.
+*/
+
+var scales = require('./scales');
+
+
+module.exports = scales.RdBu;

src/components/colorscale/defaults.js

@@ -0,0 +1,62 @@
+/**
+* Copyright 2012-2015, 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 isNumeric = require('fast-isnumeric');
+
+var Lib = require('../../lib');
+
+var hasColorbar = require('../colorbar/has_colorbar');
+var colorbarDefaults = require('../colorbar/defaults');
+var isValidScale = require('./is_valid_scale');
+var flipScale = require('./flip_scale');
+
+
+module.exports = function colorScaleDefaults(traceIn, traceOut, layout, coerce, opts) {
+    var prefix = opts.prefix,
+        cLetter = opts.cLetter,
+        containerStr = prefix.slice(0, prefix.length-1),
+        containerIn = prefix ?
+            Lib.nestedProperty(traceIn, containerStr).get() || {} :
+            traceIn,
+        containerOut = prefix ?
+            Lib.nestedProperty(traceOut, containerStr).get() || {} :
+            traceOut,
+        minIn = containerIn[cLetter + 'min'],
+        maxIn = containerIn[cLetter + 'max'],
+        sclIn = containerIn.colorscale;
+
+    var validMinMax = isNumeric(minIn) && isNumeric(maxIn) && (minIn < maxIn);
+    coerce(prefix + cLetter + 'auto', !validMinMax);
+    coerce(prefix + cLetter + 'min');
+    coerce(prefix + cLetter + 'max');
+
+    // handles both the trace case (autocolorscale is false by default) and
+    // the marker and marker.line case (autocolorscale is true by default)
+    var autoColorscaleDftl;
+    if(sclIn!==undefined) autoColorscaleDftl = !isValidScale(sclIn);
+    coerce(prefix + 'autocolorscale', autoColorscaleDftl);
+    var sclOut = coerce(prefix + 'colorscale');
+
+    // reversescale is handled at the containerOut level
+    var reverseScale = coerce(prefix + 'reversescale');
+    if(reverseScale) containerOut.colorscale = flipScale(sclOut);
+
+    // ... until Scatter.colorbar can handle marker line colorbars
+    if(prefix === 'marker.line.') return;
+
+    // handle both the trace case where the dflt is listed in attributes and
+    // the marker case where the dflt is determined by hasColorbar
+    var showScaleDftl;
+    if(prefix) showScaleDftl = hasColorbar(containerIn);
+    var showScale = coerce(prefix + 'showscale', showScaleDftl);
+
+    if(showScale) colorbarDefaults(containerIn, containerOut, layout);
+};

src/components/colorscale/flip_scale.js

@@ -0,0 +1,23 @@
+/**
+* Copyright 2012-2015, 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';
+
+module.exports = function flipScale(scl) {
+    var N = scl.length,
+        sclNew = new Array(N),
+        si;
+
+    for(var i = N-1, j = 0; i >= 0; i--, j++) {
+        si = scl[i];
+        sclNew[j] = [1 - si[0], si[1]];
+    }
+
+    return sclNew;
+};

src/components/colorscale/get_scale.js

@@ -0,0 +1,38 @@
+/**
+* Copyright 2012-2015, 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 scales = require('./scales');
+var defaultScale = require('./default_scale');
+var isValidScaleArray = require('./is_valid_scale_array');
+
+
+module.exports = function getScale(scl, dflt) {
+    if(!dflt) dflt = defaultScale;
+    if(!scl) return dflt;
+
+    function parseScale() {
+        try {
+            scl = scales[scl] || JSON.parse(scl);
+        }
+        catch(e) {
+            scl = dflt;
+        }
+    }
+
+    if(typeof scl === 'string') {
+        parseScale();
+        // occasionally scl is double-JSON encoded...
+        if(typeof scl === 'string') parseScale();
+    }
+
+    if(!isValidScaleArray(scl)) return dflt;
+    return scl;
+};

src/components/colorscale/has_colorscale.js

@@ -0,0 +1,44 @@
+/**
+* Copyright 2012-2015, 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 isNumeric = require('fast-isnumeric');
+
+var Lib = require('../../lib');
+
+var isValidScale = require('./is_valid_scale');
+
+
+module.exports = function hasColorscale(trace, containerStr) {
+    var container = containerStr ?
+            Lib.nestedProperty(trace, containerStr).get() || {} :
+            trace,
+        color = container.color,
+        isArrayWithOneNumber = false;
+
+    if(Array.isArray(color)) {
+        for(var i = 0; i < color.length; i++) {
+            if(isNumeric(color[i])) {
+                isArrayWithOneNumber = true;
+                break;
+            }
+        }
+    }
+
+    return (
+        (typeof container==='object' && container!==null) && (
+            isArrayWithOneNumber ||
+            container.showscale===true ||
+            (isNumeric(container.cmin) && isNumeric(container.cmax)) ||
+            isValidScale(container.colorscale) ||
+            (typeof container.colorbar==='object' && container.colorbar!==null)
+        )
+    );
+};