@lobehub/charts
Version:
React modern charts components built on recharts
350 lines (348 loc) • 19.2 kB
JavaScript
'use client';
function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
var _templateObject, _templateObject2;
var _excluded = ["data", "x", "y", "size", "category", "colors", "showOpacity", "yAxisAlign", "sizeRange", "valueFormatter", "startEndOnly", "showXAxis", "showYAxis", "yAxisWidth", "intervalType", "animationDuration", "showAnimation", "showTooltip", "showLegend", "loading", "showGridLines", "autoMinXValue", "minXValue", "maxXValue", "autoMinYValue", "minYValue", "maxYValue", "allowDecimals", "noDataText", "onValueChange", "customTooltip", "rotateLabelX", "className", "enableLegendSlider", "tickGap", "xAxisLabel", "yAxisLabel", "width", "height", "style", "customCategories"];
function _taggedTemplateLiteral(strings, raw) { if (!raw) { raw = strings.slice(0); } return Object.freeze(Object.defineProperties(strings, { raw: { value: Object.freeze(raw) } })); }
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(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(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); }
function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
function _objectWithoutProperties(source, excluded) { if (source == null) return {}; var target = _objectWithoutPropertiesLoose(source, excluded); var key, i; if (Object.getOwnPropertySymbols) { var sourceSymbolKeys = Object.getOwnPropertySymbols(source); for (i = 0; i < sourceSymbolKeys.length; i++) { key = sourceSymbolKeys[i]; if (excluded.indexOf(key) >= 0) continue; if (!Object.prototype.propertyIsEnumerable.call(source, key)) continue; target[key] = source[key]; } } return target; }
function _objectWithoutPropertiesLoose(source, excluded) { if (source == null) return {}; var target = {}; var sourceKeys = Object.keys(source); var key, i; for (i = 0; i < sourceKeys.length; i++) { key = sourceKeys[i]; if (excluded.indexOf(key) >= 0) continue; target[key] = source[key]; } return target; }
import { Skeleton } from 'antd';
import { css } from 'antd-style';
import { forwardRef, useMemo, useState } from 'react';
import { Flexbox } from 'react-layout-kit';
import { CartesianGrid, Label, Legend, ScatterChart as ReChartsScatterChart, ResponsiveContainer, Scatter, Tooltip, XAxis, YAxis, ZAxis } from 'recharts';
import { renderShape } from "./renderShape";
import ChartLegend from "../common/ChartLegend";
import CustomYAxisTick from "../common/CustomYAxisTick";
import { constructCategories, constructCategoryColors, deepEqual, getYAxisDomain } from "../common/utils";
import { useThemeColorRange } from "../hooks/useThemeColorRange";
import { defaultValueFormatter } from "../utils";
import { getMaxLabelLength } from "../utils/getMaxLabelLength";
import NoData from "../common/NoData";
import ScatterChartTooltip from "./ScatterChartTooltip";
import { useStyles } from "./styles";
import { jsx as _jsx } from "react/jsx-runtime";
import { jsxs as _jsxs } from "react/jsx-runtime";
var ScatterChart = /*#__PURE__*/forwardRef(function (props, ref) {
var _useStyles = useStyles(),
cx = _useStyles.cx,
theme = _useStyles.theme,
styles = _useStyles.styles;
var themeColorRange = useThemeColorRange();
var _props$data = props.data,
data = _props$data === void 0 ? [] : _props$data,
x = props.x,
y = props.y,
size = props.size,
category = props.category,
_props$colors = props.colors,
colors = _props$colors === void 0 ? themeColorRange : _props$colors,
_props$showOpacity = props.showOpacity,
showOpacity = _props$showOpacity === void 0 ? false : _props$showOpacity,
_props$yAxisAlign = props.yAxisAlign,
yAxisAlign = _props$yAxisAlign === void 0 ? 'left' : _props$yAxisAlign,
_props$sizeRange = props.sizeRange,
sizeRange = _props$sizeRange === void 0 ? [1, 1000] : _props$sizeRange,
_props$valueFormatter = props.valueFormatter,
valueFormatter = _props$valueFormatter === void 0 ? {
size: defaultValueFormatter,
x: defaultValueFormatter,
y: defaultValueFormatter
} : _props$valueFormatter,
_props$startEndOnly = props.startEndOnly,
startEndOnly = _props$startEndOnly === void 0 ? false : _props$startEndOnly,
_props$showXAxis = props.showXAxis,
showXAxis = _props$showXAxis === void 0 ? true : _props$showXAxis,
_props$showYAxis = props.showYAxis,
showYAxis = _props$showYAxis === void 0 ? true : _props$showYAxis,
yAxisWidth = props.yAxisWidth,
_props$intervalType = props.intervalType,
intervalType = _props$intervalType === void 0 ? 'equidistantPreserveStart' : _props$intervalType,
_props$animationDurat = props.animationDuration,
animationDuration = _props$animationDurat === void 0 ? 900 : _props$animationDurat,
_props$showAnimation = props.showAnimation,
showAnimation = _props$showAnimation === void 0 ? false : _props$showAnimation,
_props$showTooltip = props.showTooltip,
showTooltip = _props$showTooltip === void 0 ? true : _props$showTooltip,
_props$showLegend = props.showLegend,
showLegend = _props$showLegend === void 0 ? true : _props$showLegend,
loading = props.loading,
_props$showGridLines = props.showGridLines,
showGridLines = _props$showGridLines === void 0 ? true : _props$showGridLines,
_props$autoMinXValue = props.autoMinXValue,
autoMinXValue = _props$autoMinXValue === void 0 ? false : _props$autoMinXValue,
minXValue = props.minXValue,
maxXValue = props.maxXValue,
_props$autoMinYValue = props.autoMinYValue,
autoMinYValue = _props$autoMinYValue === void 0 ? false : _props$autoMinYValue,
minYValue = props.minYValue,
maxYValue = props.maxYValue,
_props$allowDecimals = props.allowDecimals,
allowDecimals = _props$allowDecimals === void 0 ? true : _props$allowDecimals,
noDataText = props.noDataText,
onValueChange = props.onValueChange,
customTooltip = props.customTooltip,
rotateLabelX = props.rotateLabelX,
className = props.className,
_props$enableLegendSl = props.enableLegendSlider,
enableLegendSlider = _props$enableLegendSl === void 0 ? false : _props$enableLegendSl,
_props$tickGap = props.tickGap,
tickGap = _props$tickGap === void 0 ? 5 : _props$tickGap,
xAxisLabel = props.xAxisLabel,
yAxisLabel = props.yAxisLabel,
_props$width = props.width,
width = _props$width === void 0 ? '100%' : _props$width,
_props$height = props.height,
height = _props$height === void 0 ? 280 : _props$height,
style = props.style,
customCategories = props.customCategories,
rest = _objectWithoutProperties(props, _excluded);
var _useState = useState(60),
_useState2 = _slicedToArray(_useState, 2),
legendHeight = _useState2[0],
setLegendHeight = _useState2[1];
var _useState3 = useState(),
_useState4 = _slicedToArray(_useState3, 2),
activeNode = _useState4[0],
setActiveNode = _useState4[1];
var _useState5 = useState(),
_useState6 = _slicedToArray(_useState5, 2),
activeLegend = _useState6[0],
setActiveLegend = _useState6[1];
var calculatedYAxisWidth = useMemo(function () {
if (yAxisWidth) return yAxisWidth;
return getMaxLabelLength({
data: data,
index: y,
isScatterChart: true,
margin: 0,
valueFormatter: valueFormatter === null || valueFormatter === void 0 ? void 0 : valueFormatter.y
});
}, [yAxisWidth, data, valueFormatter, y]);
if (loading || !data) return /*#__PURE__*/_jsx(Skeleton.Button, {
active: true,
block: true,
style: {
height: height,
width: width
}
});
var CustomTooltip = customTooltip;
var hasOnValueChange = !!onValueChange;
var onNodeClick = function onNodeClick(data, index, event) {
event.stopPropagation();
if (!hasOnValueChange) return;
if (deepEqual(activeNode, data.node)) {
setActiveLegend(undefined);
setActiveNode(undefined);
onValueChange === null || onValueChange === void 0 || onValueChange(null);
} else {
setActiveNode(data.node);
setActiveLegend(data.payload[category]);
onValueChange === null || onValueChange === void 0 || onValueChange(_objectSpread({
categoryClicked: data.payload[category],
eventType: 'bubble'
}, data.payload));
}
};
var onCategoryClick = function onCategoryClick(dataKey) {
if (!hasOnValueChange) return;
if (dataKey === activeLegend && !activeNode) {
setActiveLegend(undefined);
onValueChange === null || onValueChange === void 0 || onValueChange(null);
} else {
setActiveLegend(dataKey);
onValueChange === null || onValueChange === void 0 || onValueChange({
categoryClicked: dataKey,
eventType: 'category'
});
}
setActiveNode(undefined);
};
var categories = constructCategories(data, category);
var categoryColors = constructCategoryColors(categories, colors);
//maybe rename getYAxisDomain to getAxisDomain
var xAxisDomain = getYAxisDomain(autoMinXValue, minXValue, maxXValue);
var yAxisDomain = getYAxisDomain(autoMinYValue, minYValue, maxYValue);
return /*#__PURE__*/_jsx(Flexbox, _objectSpread(_objectSpread({
className: className,
height: height,
ref: ref,
style: _objectSpread({
position: 'relative'
}, style),
width: width
}, rest), {}, {
children: /*#__PURE__*/_jsx(ResponsiveContainer, {
children: data !== null && data !== void 0 && data.length ? /*#__PURE__*/_jsxs(ReChartsScatterChart, {
margin: {
bottom: xAxisLabel ? 20 : undefined,
left: 20,
right: 20,
top: 5
},
onClick: hasOnValueChange && (activeLegend || activeNode) ? function () {
setActiveNode(undefined);
setActiveLegend(undefined);
onValueChange === null || onValueChange === void 0 || onValueChange(null);
} : undefined,
children: [showGridLines ? /*#__PURE__*/_jsx(CartesianGrid, {
className: styles.gridLines,
horizontal: true,
vertical: true
}) : null, x ? /*#__PURE__*/_jsx(XAxis, {
allowDataOverflow: true,
angle: rotateLabelX === null || rotateLabelX === void 0 ? void 0 : rotateLabelX.angle,
axisLine: false,
className: styles.label,
dataKey: x,
domain: xAxisDomain,
dy: rotateLabelX === null || rotateLabelX === void 0 ? void 0 : rotateLabelX.verticalShift,
fill: "",
hide: !showXAxis,
interval: startEndOnly ? 'preserveStartEnd' : intervalType,
minTickGap: tickGap,
name: x,
stroke: "",
tick: {
transform: 'translate(0, 6)'
},
tickFormatter: valueFormatter.x,
tickLine: false,
ticks: startEndOnly ? [data[0][x], data.at(-1)[x]] : undefined,
type: "number",
children: xAxisLabel && /*#__PURE__*/_jsx(Label, {
className: cx(styles.strongLabel, styles.emphasis),
offset: -20,
position: "insideBottom",
children: xAxisLabel
})
}) : null, y ? /*#__PURE__*/_jsx(YAxis, {
allowDataOverflow: true,
allowDecimals: allowDecimals,
axisLine: false,
className: styles.label,
dataKey: y,
domain: yAxisDomain,
fill: "",
hide: !showYAxis,
name: y,
stroke: "",
tick: function tick(props) {
return /*#__PURE__*/_jsx(CustomYAxisTick, _objectSpread(_objectSpread({}, props), {}, {
align: yAxisAlign,
formatter: valueFormatter.y,
textAnchor: yAxisAlign === 'left' ? 'start' : 'end',
yAxisLabel: Boolean(yAxisLabel)
}));
},
tickFormatter: valueFormatter.y,
tickLine: false,
type: "number",
width: calculatedYAxisWidth,
children: yAxisLabel && /*#__PURE__*/_jsx(Label, {
angle: -90,
className: cx(styles.strongLabel, styles.emphasis),
offset: -15,
position: "insideLeft",
style: {
textAnchor: 'middle'
},
children: yAxisLabel
})
}) : null, /*#__PURE__*/_jsx(Tooltip, {
content: showTooltip ? function (_ref) {
var _payload$;
var active = _ref.active,
payload = _ref.payload,
label = _ref.label;
var color = category ? payload === null || payload === void 0 || (_payload$ = payload[0]) === null || _payload$ === void 0 || (_payload$ = _payload$.payload) === null || _payload$ === void 0 ? void 0 : _payload$[category] : label;
return CustomTooltip ? /*#__PURE__*/_jsx(CustomTooltip, {
active: active,
customCategories: customCategories,
label: color,
payload: payload === null || payload === void 0 ? void 0 : payload.map(function (payloadItem) {
var _categoryColors$get;
return _objectSpread(_objectSpread({}, payloadItem), {}, {
color: (_categoryColors$get = categoryColors.get(color)) !== null && _categoryColors$get !== void 0 ? _categoryColors$get : theme.colorPrimary
});
}),
valueFormatter: valueFormatter
}) : /*#__PURE__*/_jsx(ScatterChartTooltip, {
active: active,
axis: {
size: size,
x: x,
y: y
},
category: category,
categoryColors: categoryColors,
customCategories: customCategories,
label: color,
payload: payload,
valueFormatter: valueFormatter
});
} : undefined,
cursor: {
stroke: theme.colorTextSecondary,
strokeWidth: 1
},
isAnimationActive: false,
wrapperStyle: {
outline: 'none'
}
}), size ? /*#__PURE__*/_jsx(ZAxis, {
dataKey: size,
name: size,
range: sizeRange,
type: "number"
}) : null, categories.map(function (cat) {
var _categoryColors$get2, _categoryColors$get3;
return /*#__PURE__*/_jsx(Scatter, {
animationDuration: animationDuration,
className: cx(css(_templateObject || (_templateObject = _taggedTemplateLiteral(["\n fill: ", ";\n "])), (_categoryColors$get2 = categoryColors.get(cat)) !== null && _categoryColors$get2 !== void 0 ? _categoryColors$get2 : theme.colorPrimary), showOpacity && css(_templateObject2 || (_templateObject2 = _taggedTemplateLiteral(["\n stroke: ", ";\n "])), (_categoryColors$get3 = categoryColors.get(cat)) !== null && _categoryColors$get3 !== void 0 ? _categoryColors$get3 : theme.colorPrimary)),
data: category ? data.filter(function (d) {
return d[category] === cat;
}) : data,
fill: "url(#".concat(categoryColors.get(cat), ")"),
fillOpacity: showOpacity ? 0.7 : 1,
isAnimationActive: showAnimation,
name: cat,
onClick: onNodeClick,
shape: function shape(props) {
return renderShape(props, activeNode, activeLegend);
},
style: {
cursor: onValueChange ? 'pointer' : undefined
}
}, cat);
}), showLegend ? /*#__PURE__*/_jsx(Legend, {
content: function content(_ref2) {
var payload = _ref2.payload;
return ChartLegend({
payload: payload
}, categoryColors, setLegendHeight, activeLegend, hasOnValueChange ? function (clickedLegendItem) {
return onCategoryClick(clickedLegendItem);
} : undefined, enableLegendSlider, customCategories);
},
height: legendHeight,
verticalAlign: "top"
}) : undefined]
}) : /*#__PURE__*/_jsx(NoData, {
noDataText: noDataText
})
})
}));
});
ScatterChart.displayName = 'ScatterChart';
export default ScatterChart;