Violin plots

lib/index-cartesian.js

@@ -19,7 +19,8 @@ Plotly.register([
     require('./histogram2dcontour'),
     require('./pie'),
     require('./contour'),
-    require('./scatterternary')
+    require('./scatterternary'),
+    require('./violin')
 ]);
 
 module.exports = Plotly;

lib/index.js

@@ -21,7 +21,7 @@ Plotly.register([
     require('./pie'),
     require('./contour'),
     require('./scatterternary'),
-    require('./sankey'),
+    require('./violin'),
 
     require('./scatter3d'),
     require('./surface'),
@@ -34,10 +34,13 @@ Plotly.register([
     require('./pointcloud'),
     require('./heatmapgl'),
     require('./parcoords'),
-    require('./table'),
 
     require('./scattermapbox'),
 
+    require('./sankey'),
+
+    require('./table'),
+
     require('./carpet'),
     require('./scattercarpet'),
     require('./contourcarpet'),

lib/violin.js

@@ -0,0 +1,11 @@
+/**
+* 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';
+
+module.exports = require('../src/traces/violin');

src/plots/cartesian/constants.js

@@ -57,18 +57,13 @@ module.exports = {
     DFLTRANGEX: [-1, 6],
     DFLTRANGEY: [-1, 4],
 
-    // Layers to keep trace types in the right order.
-    // from back to front:
-    // 1. heatmaps, 2D histos and contour maps
-    // 2. bars / 1D histos
-    // 3. errorbars for bars and scatter
-    // 4. scatter
-    // 5. box plots
+    // Layers to keep trace types in the right order
     traceLayerClasses: [
         'imagelayer',
         'maplayer',
         'barlayer',
         'carpetlayer',
+        'violinlayer',
         'boxlayer',
         'scatterlayer'
     ],

src/plots/plots.js

@@ -1342,7 +1342,6 @@ plots.purge = function(gd) {
     delete gd.firstscatter;
     delete gd._hmlumcount;
     delete gd._hmpixcount;
-    delete gd.numboxes;
     delete gd._transitionData;
     delete gd._transitioning;
     delete gd._initialAutoSize;
@@ -2159,8 +2158,9 @@ plots.doCalcdata = function(gd, traces) {
     // firstscatter: fill-to-next on the first trace goes to zero
     gd.firstscatter = true;
 
-    // how many box plots do we have (in case they're grouped)
-    gd.numboxes = 0;
+    // how many box/violins plots do we have (in case they're grouped)
+    fullLayout._numBoxes = 0;
+    fullLayout._numViolins = 0;
 
     // for calculating avg luminosity of heatmaps
     gd._hmpixcount = 0;

src/traces/box/calc.js

@@ -15,11 +15,14 @@ var Axes = require('../../plots/cartesian/axes');
 
 // outlier definition based on http://www.physics.csbsju.edu/stats/box2.html
 module.exports = function calc(gd, trace) {
+    var fullLayout = gd._fullLayout;
     var xa = Axes.getFromId(gd, trace.xaxis || 'x');
     var ya = Axes.getFromId(gd, trace.yaxis || 'y');
     var orientation = trace.orientation;
     var cd = [];
 
+    var numKey = '_numBoxes';
+
     var i;
     var valAxis, valLetter;
     var posAxis, posLetter;
@@ -37,7 +40,7 @@ module.exports = function calc(gd, trace) {
     }
 
     var val = valAxis.makeCalcdata(trace, valLetter);
-    var pos = getPos(trace, posLetter, posAxis, val, gd.numboxes);
+    var pos = getPos(trace, posLetter, posAxis, val, fullLayout[numKey]);
 
     var dv = Lib.distinctVals(pos);
     var posDistinct = dv.vals;
@@ -115,13 +118,14 @@ module.exports = function calc(gd, trace) {
 
     if(cd.length > 0) {
         cd[0].t = {
-            boxnum: gd.numboxes,
+            num: fullLayout[numKey],
             dPos: dPos
         };
-        gd.numboxes++;
+
+        fullLayout[numKey]++;
         return cd;
     } else {
-        return [{t: {emptybox: true}}];
+        return [{t: {empty: true}}];
     }
 };
 
@@ -130,7 +134,7 @@ module.exports = function calc(gd, trace) {
 // so if you want one box
 // per trace, set x0 (y0) to the x (y) value or category for this trace
 // (or set x (y) to a constant array matching y (x))
-function getPos(trace, posLetter, posAxis, val, numboxes) {
+function getPos(trace, posLetter, posAxis, val, num) {
     if(posLetter in trace) {
         return posAxis.makeCalcdata(trace, posLetter);
     }
@@ -150,7 +154,7 @@ function getPos(trace, posLetter, posAxis, val, numboxes) {
     )) {
         pos0 = trace.name;
     } else {
-        pos0 = numboxes;
+        pos0 = num;
     }
 
     var pos0c = posAxis.d2c(pos0, 0, trace[posLetter + 'calendar']);

src/traces/box/defaults.js

@@ -14,11 +14,25 @@ var Color = require('../../components/color');
 
 var attributes = require('./attributes');
 
-module.exports = function supplyDefaults(traceIn, traceOut, defaultColor, layout) {
+function supplyDefaults(traceIn, traceOut, defaultColor, layout) {
     function coerce(attr, dflt) {
         return Lib.coerce(traceIn, traceOut, attributes, attr, dflt);
     }
 
+    handleSampleDefaults(traceIn, traceOut, coerce, layout);
+    if(traceOut.visible === false) return;
+
+    coerce('line.color', (traceIn.marker || {}).color || defaultColor);
+    coerce('line.width');
+    coerce('fillcolor', Color.addOpacity(traceOut.line.color, 0.5));
+
+    coerce('whiskerwidth');
+    coerce('boxmean');
+
+    handlePointsDefaults(traceIn, traceOut, coerce, {prefix: 'box'});
+}
+
+function handleSampleDefaults(traceIn, traceOut, coerce, layout) {
     var y = coerce('y');
     var x = coerce('x');
 
@@ -39,25 +53,22 @@ module.exports = function supplyDefaults(traceIn, traceOut, defaultColor, layout
     handleCalendarDefaults(traceIn, traceOut, ['x', 'y'], layout);
 
     coerce('orientation', defaultOrientation);
+}
 
-    coerce('line.color', (traceIn.marker || {}).color || defaultColor);
-    coerce('line.width');
-    coerce('fillcolor', Color.addOpacity(traceOut.line.color, 0.5));
-
-    coerce('whiskerwidth');
-    coerce('boxmean');
+function handlePointsDefaults(traceIn, traceOut, coerce, opts) {
+    var prefix = opts.prefix;
 
     var outlierColorDflt = Lib.coerce2(traceIn, traceOut, attributes, 'marker.outliercolor');
     var lineoutliercolor = coerce('marker.line.outliercolor');
 
-    var boxpoints = coerce(
-        'boxpoints',
+    var points = coerce(
+        prefix + 'points',
         (outlierColorDflt || lineoutliercolor) ? 'suspectedoutliers' : undefined
     );
 
-    if(boxpoints) {
-        coerce('jitter', boxpoints === 'all' ? 0.3 : 0);
-        coerce('pointpos', boxpoints === 'all' ? -1.5 : 0);
+    if(points) {
+        coerce('jitter', points === 'all' ? 0.3 : 0);
+        coerce('pointpos', points === 'all' ? -1.5 : 0);
 
         coerce('marker.symbol');
         coerce('marker.opacity');
@@ -66,7 +77,7 @@ module.exports = function supplyDefaults(traceIn, traceOut, defaultColor, layout
         coerce('marker.line.color');
         coerce('marker.line.width');
 
-        if(boxpoints === 'suspectedoutliers') {
+        if(points === 'suspectedoutliers') {
             coerce('marker.line.outliercolor', traceOut.marker.color);
             coerce('marker.line.outlierwidth');
         }
@@ -77,4 +88,10 @@ module.exports = function supplyDefaults(traceIn, traceOut, defaultColor, layout
     }
 
     coerce('hoveron');
+}
+
+module.exports = {
+    supplyDefaults: supplyDefaults,
+    handleSampleDefaults: handleSampleDefaults,
+    handlePointsDefaults: handlePointsDefaults
 };

src/traces/box/index.js

@@ -12,11 +12,11 @@ var Box = {};
 
 Box.attributes = require('./attributes');
 Box.layoutAttributes = require('./layout_attributes');
-Box.supplyDefaults = require('./defaults');
-Box.supplyLayoutDefaults = require('./layout_defaults');
+Box.supplyDefaults = require('./defaults').supplyDefaults;
+Box.supplyLayoutDefaults = require('./layout_defaults').supplyLayoutDefaults;
 Box.calc = require('./calc');
 Box.setPositions = require('./set_positions');
-Box.plot = require('./plot');
+Box.plot = require('./plot').plot;
 Box.style = require('./style');
 Box.hoverPoints = require('./hover');
 Box.selectPoints = require('./select');

src/traces/box/layout_defaults.js

@@ -12,11 +12,7 @@ var Registry = require('../../registry');
 var Lib = require('../../lib');
 var layoutAttributes = require('./layout_attributes');
 
-module.exports = function supplyLayoutDefaults(layoutIn, layoutOut, fullData) {
-    function coerce(attr, dflt) {
-        return Lib.coerce(layoutIn, layoutOut, layoutAttributes, attr, dflt);
-    }
-
+function _supply(layoutIn, layoutOut, fullData, coerce, prefix) {
     var hasBoxes;
     for(var i = 0; i < fullData.length; i++) {
         if(Registry.traceIs(fullData[i], 'box')) {
@@ -26,7 +22,19 @@ module.exports = function supplyLayoutDefaults(layoutIn, layoutOut, fullData) {
     }
     if(!hasBoxes) return;
 
-    coerce('boxmode');
-    coerce('boxgap');
-    coerce('boxgroupgap');
+    coerce(prefix + 'mode');
+    coerce(prefix + 'gap');
+    coerce(prefix + 'groupgap');
+}
+
+function supplyLayoutDefaults(layoutIn, layoutOut, fullData) {
+    function coerce(attr, dflt) {
+        return Lib.coerce(layoutIn, layoutOut, layoutAttributes, attr, dflt);
+    }
+    _supply(layoutIn, layoutOut, fullData, coerce, 'box');
+}
+
+module.exports = {
+    supplyLayoutDefaults: supplyLayoutDefaults,
+    _supply: _supply
 };