UNPKG

recharts

Version:
289 lines (284 loc) 11.6 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.payloadSearcher = exports.addToSunburstNodeIndex = exports.SunburstChart = void 0; var _react = _interopRequireWildcard(require("react")); var React = _react; var _d3Scale = require("victory-vendor/d3-scale"); var _clsx = require("clsx"); var _get = _interopRequireDefault(require("es-toolkit/compat/get")); var _Surface = require("../container/Surface"); var _Layer = require("../container/Layer"); var _Sector = require("../shape/Sector"); var _Text = require("../component/Text"); var _PolarUtils = require("../util/PolarUtils"); var _chartLayoutContext = require("../context/chartLayoutContext"); var _tooltipPortalContext = require("../context/tooltipPortalContext"); var _RechartsWrapper = require("./RechartsWrapper"); var _tooltipSlice = require("../state/tooltipSlice"); var _SetTooltipEntrySettings = require("../state/SetTooltipEntrySettings"); var _RechartsStoreProvider = require("../state/RechartsStoreProvider"); var _hooks = require("../state/hooks"); function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; } function _interopRequireWildcard(e, t) { if ("function" == typeof WeakMap) var r = new WeakMap(), n = new WeakMap(); return (_interopRequireWildcard = function _interopRequireWildcard(e, t) { if (!t && e && e.__esModule) return e; var o, i, f = { __proto__: null, default: e }; if (null === e || "object" != typeof e && "function" != typeof e) return f; if (o = t ? n : r) { if (o.has(e)) return o.get(e); o.set(e, f); } for (var _t in e) "default" !== _t && {}.hasOwnProperty.call(e, _t) && ((i = (o = Object.defineProperty) && Object.getOwnPropertyDescriptor(e, _t)) && (i.get || i.set) ? o(f, _t, i) : f[_t] = e[_t]); return f; })(e, t); } function _extends() { return _extends = Object.assign ? Object.assign.bind() : function (n) { for (var e = 1; e < arguments.length; e++) { var t = arguments[e]; for (var r in t) ({}).hasOwnProperty.call(t, r) && (n[r] = t[r]); } return n; }, _extends.apply(null, arguments); } function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; } function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; } function _defineProperty(e, r, t) { return (r = _toPropertyKey(r)) in e ? Object.defineProperty(e, r, { value: t, enumerable: !0, configurable: !0, writable: !0 }) : e[r] = t, e; } function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == typeof i ? i : i + ""; } function _toPrimitive(t, r) { if ("object" != typeof t || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != typeof i) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); } var defaultTextProps = { fontWeight: 'bold', paintOrder: 'stroke fill', fontSize: '.75rem', stroke: '#FFF', fill: 'black', pointerEvents: 'none' }; function getMaxDepthOf(node) { if (!node.children || node.children.length === 0) return 1; // Calculate depth for each child and find the maximum var childDepths = node.children.map(d => getMaxDepthOf(d)); return 1 + Math.max(...childDepths); } function convertMapToRecord(map) { var record = {}; map.forEach((value, key) => { record[key] = value; }); return record; } function getTooltipEntrySettings(_ref) { var { dataKey, nameKey, data, stroke, fill, positions } = _ref; return { dataDefinedOnItem: data.children, // Redux store will not accept a Map because it's not serializable positions: convertMapToRecord(positions), // Sunburst does not support many of the properties as other charts do so there's plenty of defaults here settings: { stroke, strokeWidth: undefined, fill, nameKey, dataKey, // if there is a nameKey use it, otherwise make the name of the tooltip the dataKey itself name: nameKey ? undefined : dataKey, hide: false, type: undefined, color: fill, unit: '' } }; } // Why is margin not a sunburst prop? No clue. Probably it should be var defaultSunburstMargin = { top: 0, right: 0, bottom: 0, left: 0 }; var payloadSearcher = (data, activeIndex) => { return (0, _get.default)(data, activeIndex); }; exports.payloadSearcher = payloadSearcher; var addToSunburstNodeIndex = exports.addToSunburstNodeIndex = function addToSunburstNodeIndex(indexInChildrenArr) { var activeTooltipIndexSoFar = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : ''; return "".concat(activeTooltipIndexSoFar, "children[").concat(indexInChildrenArr, "]"); }; var preloadedState = { options: { validateTooltipEventTypes: ['item'], defaultTooltipEventType: 'item', chartName: 'Sunburst', tooltipPayloadSearcher: payloadSearcher, eventEmitter: undefined } }; var SunburstChartImpl = _ref2 => { var { className, data, children, width, height, padding = 2, dataKey = 'value', nameKey = 'name', ringPadding = 2, innerRadius = 50, fill = '#333', stroke = '#FFF', textOptions = defaultTextProps, outerRadius = Math.min(width, height) / 2, cx = width / 2, cy = height / 2, startAngle = 0, endAngle = 360, onClick, onMouseEnter, onMouseLeave } = _ref2; var dispatch = (0, _hooks.useAppDispatch)(); var rScale = (0, _d3Scale.scaleLinear)([0, data[dataKey]], [0, endAngle]); var treeDepth = getMaxDepthOf(data); var thickness = (outerRadius - innerRadius) / treeDepth; var sectors = []; var positions = new Map([]); var [tooltipPortal, setTooltipPortal] = (0, _react.useState)(null); // event handlers function handleMouseEnter(node, e) { if (onMouseEnter) onMouseEnter(node, e); dispatch((0, _tooltipSlice.setActiveMouseOverItemIndex)({ activeIndex: node.tooltipIndex, activeDataKey: dataKey, activeCoordinate: positions.get(node.name) })); } function handleMouseLeave(node, e) { if (onMouseLeave) onMouseLeave(node, e); dispatch((0, _tooltipSlice.mouseLeaveItem)()); } function handleClick(node) { if (onClick) onClick(node); dispatch((0, _tooltipSlice.setActiveClickItemIndex)({ activeIndex: node.tooltipIndex, activeDataKey: dataKey, activeCoordinate: positions.get(node.name) })); } // recursively add nodes for each data point and its children function drawArcs(childNodes, options) { var depth = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : 1; var { radius, innerR, initialAngle, childColor, nestedActiveTooltipIndex } = options; var currentAngle = initialAngle; if (!childNodes) return; // base case: no children of this node childNodes.forEach((d, i) => { var _ref3, _d$fill; var currentTooltipIndex = depth === 1 ? "[".concat(i, "]") : addToSunburstNodeIndex(i, nestedActiveTooltipIndex); var nodeWithIndex = _objectSpread(_objectSpread({}, d), {}, { tooltipIndex: currentTooltipIndex }); var arcLength = rScale(d[dataKey]); var start = currentAngle; // color priority - if there's a color on the individual point use that, otherwise use parent color or default var fillColor = (_ref3 = (_d$fill = d === null || d === void 0 ? void 0 : d.fill) !== null && _d$fill !== void 0 ? _d$fill : childColor) !== null && _ref3 !== void 0 ? _ref3 : fill; var { x: textX, y: textY } = (0, _PolarUtils.polarToCartesian)(0, 0, innerR + radius / 2, -(start + arcLength - arcLength / 2)); currentAngle += arcLength; sectors.push( /*#__PURE__*/ // eslint-disable-next-line react/no-array-index-key React.createElement("g", { key: "sunburst-sector-".concat(d.name, "-").concat(i) }, /*#__PURE__*/React.createElement(_Sector.Sector, { onClick: () => handleClick(nodeWithIndex), onMouseEnter: e => handleMouseEnter(nodeWithIndex, e), onMouseLeave: e => handleMouseLeave(nodeWithIndex, e), fill: fillColor, stroke: stroke, strokeWidth: padding, startAngle: start, endAngle: start + arcLength, innerRadius: innerR, outerRadius: innerR + radius, cx: cx, cy: cy }), /*#__PURE__*/React.createElement(_Text.Text, _extends({}, textOptions, { alignmentBaseline: "middle", textAnchor: "middle", x: textX + cx, y: cy - textY }), d[dataKey]))); var { x: tooltipX, y: tooltipY } = (0, _PolarUtils.polarToCartesian)(cx, cy, innerR + radius / 2, start); positions.set(d.name, { x: tooltipX, y: tooltipY }); return drawArcs(d.children, { radius, innerR: innerR + radius + ringPadding, initialAngle: start, childColor: fillColor, nestedActiveTooltipIndex: currentTooltipIndex }, depth + 1); }); } drawArcs(data.children, { radius: thickness, innerR: innerRadius, initialAngle: startAngle }); var layerClass = (0, _clsx.clsx)('recharts-sunburst', className); return /*#__PURE__*/React.createElement(_tooltipPortalContext.TooltipPortalContext.Provider, { value: tooltipPortal }, /*#__PURE__*/React.createElement(_RechartsWrapper.RechartsWrapper, { className: className, width: width // Sunburst doesn't support `style` property, why? , height: height, ref: node => { if (tooltipPortal == null && node != null) { setTooltipPortal(node); } }, onMouseEnter: undefined, onMouseLeave: undefined, onClick: undefined, onMouseMove: undefined, onMouseDown: undefined, onMouseUp: undefined, onContextMenu: undefined, onDoubleClick: undefined, onTouchStart: undefined, onTouchMove: undefined, onTouchEnd: undefined }, /*#__PURE__*/React.createElement(_Surface.Surface, { width: width, height: height }, /*#__PURE__*/React.createElement(_Layer.Layer, { className: layerClass }, sectors), /*#__PURE__*/React.createElement(_SetTooltipEntrySettings.SetTooltipEntrySettings, { fn: getTooltipEntrySettings, args: { dataKey, data, stroke, fill, nameKey, positions } }), children))); }; var SunburstChart = props => { var _props$className; return /*#__PURE__*/React.createElement(_RechartsStoreProvider.RechartsStoreProvider, { preloadedState: preloadedState, reduxStoreName: (_props$className = props.className) !== null && _props$className !== void 0 ? _props$className : 'SunburstChart' }, /*#__PURE__*/React.createElement(_chartLayoutContext.ReportChartSize, { width: props.width, height: props.height }), /*#__PURE__*/React.createElement(_chartLayoutContext.ReportChartMargin, { margin: defaultSunburstMargin }), /*#__PURE__*/React.createElement(SunburstChartImpl, props)); }; exports.SunburstChart = SunburstChart;