react-vis
Version:
Data visualization library based on React and d3.
230 lines (209 loc) • 7.88 kB
JavaScript
var _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; };
// Copyright (c) 2016 - 2017 Uber Technologies, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
import React from 'react';
import PropTypes from 'prop-types';
import { hierarchy, partition } from 'd3-hierarchy';
import { scaleLinear, scaleSqrt } from 'd3-scale';
import { AnimationPropType } from '../animation';
import LabelSeries from '../plot/series/label-series';
import ArcSeries from '../plot/series/arc-series';
import XYPlot from '../plot/xy-plot';
import { getRadialDomain } from '../utils/series-utils';
import { getRadialLayoutMargin } from '../utils/chart-utils';
var predefinedClassName = 'rv-sunburst';
var LISTENERS_TO_OVERWRITE = ['onValueMouseOver', 'onValueMouseOut', 'onValueClick', 'onValueRightClick', 'onSeriesMouseOver', 'onSeriesMouseOut', 'onSeriesClick', 'onSeriesRightClick'];
/**
* Create the list of nodes to render.
* @param {Object} props
props.data {Object} - tree structured data (each node has a name anc an array of children)
props.height {number} - the height of the graphic to be rendered
props.hideRootNode {boolean} - whether or not to hide the root node
props.width {number} - the width of the graphic to be rendered
props.getSize {function} - accessor for the size
* @returns {Array} Array of nodes.
*/
function getNodesToRender(_ref) {
var data = _ref.data,
height = _ref.height,
hideRootNode = _ref.hideRootNode,
width = _ref.width,
getSize = _ref.getSize;
var partitionFunction = partition();
var structuredInput = hierarchy(data).sum(getSize);
var radius = Math.min(width, height) / 2 - 10;
var x = scaleLinear().range([0, 2 * Math.PI]);
var y = scaleSqrt().range([0, radius]);
return partitionFunction(structuredInput).descendants().reduce(function (res, cell, index) {
if (hideRootNode && index === 0) {
return res;
}
return res.concat([_extends({
angle0: Math.max(0, Math.min(2 * Math.PI, x(cell.x0))),
angle: Math.max(0, Math.min(2 * Math.PI, x(cell.x1))),
radius0: Math.max(0, y(cell.y0)),
radius: Math.max(0, y(cell.y1)),
depth: cell.depth,
parent: cell.parent
}, cell.data)]);
}, []);
}
/**
* Convert arc nodes into label rows.
* Important to use mappedData rather than regular data, bc it is already unrolled
* @param {Array} mappedData - Array of nodes.
* @param {Object} accessors - object of accessors
* @returns {Array} array of node for rendering as labels
*/
function buildLabels(mappedData, accessors) {
var getAngle = accessors.getAngle,
getAngle0 = accessors.getAngle0,
getLabel = accessors.getLabel,
getRadius0 = accessors.getRadius0;
return mappedData.filter(getLabel).map(function (row) {
var truedAngle = -1 * getAngle(row) + Math.PI / 2;
var truedAngle0 = -1 * getAngle0(row) + Math.PI / 2;
var angle = (truedAngle0 + truedAngle) / 2;
var rotateLabels = !row.dontRotateLabel;
var rotAngle = -angle / (2 * Math.PI) * 360;
return _extends({}, row, {
children: null,
angle: null,
radius: null,
x: getRadius0(row) * Math.cos(angle),
y: getRadius0(row) * Math.sin(angle),
style: _extends({
textAnchor: rotAngle > 90 ? 'end' : 'start'
}, row.labelStyle),
rotation: rotateLabels ? rotAngle > 90 ? rotAngle + 180 : rotAngle === 90 ? 90 : rotAngle : null
});
});
}
var NOOP = function NOOP() {};
function Sunburst(props) {
var getAngle = props.getAngle,
getAngle0 = props.getAngle0,
animation = props.animation,
className = props.className,
children = props.children,
data = props.data,
height = props.height,
hideRootNode = props.hideRootNode,
getLabel = props.getLabel,
width = props.width,
getSize = props.getSize,
colorType = props.colorType;
var mappedData = getNodesToRender({
data: data,
height: height,
hideRootNode: hideRootNode,
width: width,
getSize: getSize
});
var radialDomain = getRadialDomain(mappedData);
var margin = getRadialLayoutMargin(width, height, radialDomain);
var labelData = buildLabels(mappedData, {
getAngle: getAngle,
getAngle0: getAngle0,
getLabel: getLabel,
getRadius0: function getRadius0(d) {
return d.radius0;
}
});
var hofBuilder = function hofBuilder(f) {
return function (e, i) {
return f ? f(mappedData[e.index], i) : NOOP;
};
};
return React.createElement(
XYPlot,
{
height: height,
hasTreeStructure: true,
width: width,
className: predefinedClassName + ' ' + className,
margin: margin,
xDomain: [-radialDomain, radialDomain],
yDomain: [-radialDomain, radialDomain]
},
React.createElement(ArcSeries, _extends({
colorType: colorType
}, props, {
animation: animation,
radiusDomain: [0, radialDomain],
// need to present a stripped down version for interpolation
data: animation ? mappedData.map(function (row, index) {
return _extends({}, row, {
parent: null,
children: null,
index: index
});
}) : mappedData,
_data: animation ? mappedData : null,
arcClassName: predefinedClassName + '__series--radial__arc'
}, LISTENERS_TO_OVERWRITE.reduce(function (acc, propName) {
var prop = props[propName];
acc[propName] = animation ? hofBuilder(prop) : prop;
return acc;
}, {}))),
labelData.length > 0 && React.createElement(LabelSeries, { data: labelData, getLabel: getLabel }),
children
);
}
Sunburst.displayName = 'Sunburst';
Sunburst.propTypes = {
animation: AnimationPropType,
getAngle: PropTypes.func,
getAngle0: PropTypes.func,
className: PropTypes.string,
colorType: PropTypes.string,
data: PropTypes.object.isRequired,
height: PropTypes.number.isRequired,
hideRootNode: PropTypes.bool,
getLabel: PropTypes.func,
onValueClick: PropTypes.func,
onValueMouseOver: PropTypes.func,
onValueMouseOut: PropTypes.func,
getSize: PropTypes.func,
width: PropTypes.number.isRequired,
padAngle: PropTypes.oneOfType([PropTypes.func, PropTypes.number])
};
Sunburst.defaultProps = {
getAngle: function getAngle(d) {
return d.angle;
},
getAngle0: function getAngle0(d) {
return d.angle0;
},
className: '',
colorType: 'literal',
getColor: function getColor(d) {
return d.color;
},
hideRootNode: false,
getLabel: function getLabel(d) {
return d.label;
},
getSize: function getSize(d) {
return d.size;
},
padAngle: 0
};
export default Sunburst;