style.js

/**
* Copyright 2012-2017, 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 d3 = require('d3');

var Drawing = require('../../components/drawing');
var heatmapStyle = require('../heatmap/style');

var makeColorMap = require('./make_color_map');


module.exports = function style(gd) {
    var contours = d3.select(gd).selectAll('g.contour');

    contours.style('opacity', function(d) {
        return d.trace.opacity;
    });

    contours.each(function(d) {
        var c = d3.select(this),
            trace = d.trace,
            contours = trace.contours,
            line = trace.line,
            cs = contours.size || 1,
            start = contours.start;

        var colorMap = makeColorMap(trace);

        c.selectAll('g.contourlevel').each(function(d) {
            d3.select(this).selectAll('path')
                .call(Drawing.lineGroupStyle,
                    line.width,
                    contours.coloring === 'lines' ? colorMap(d.level) : line.color,
                    line.dash);
        });

        var firstFill;

        c.selectAll('g.contourfill path')
            .style('fill', function(d) {
                if(firstFill === undefined) firstFill = d.level;
                return colorMap(d.level + 0.5 * cs);
            });

        if(firstFill === undefined) firstFill = start;

        c.selectAll('g.contourbg path')
            .style('fill', colorMap(firstFill - 0.5 * cs));
    });

    heatmapStyle(gd);
};