recharts
Version:
React charts
289 lines (245 loc) • 10.2 kB
JavaScript
import _isFunction from "lodash/isFunction";
import _uniqBy from "lodash/uniqBy";
var _class, _class2, _temp;
function _typeof(obj) { if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i] != null ? arguments[i] : {}; var ownKeys = Object.keys(source); if (typeof Object.getOwnPropertySymbols === 'function') { ownKeys = ownKeys.concat(Object.getOwnPropertySymbols(source).filter(function (sym) { return Object.getOwnPropertyDescriptor(source, sym).enumerable; })); } ownKeys.forEach(function (key) { _defineProperty(target, key, source[key]); }); } return target; }
function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } }
function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); return Constructor; }
function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } return _assertThisInitialized(self); }
function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }
function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function"); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, writable: true, configurable: true } }); if (superClass) _setPrototypeOf(subClass, superClass); }
function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf || function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
/**
* @fileOverview Legend
*/
import React, { Component } from 'react';
import PropTypes from 'prop-types';
import pureRender from '../util/PureRender';
import DefaultLegendContent from './DefaultLegendContent';
import { isNumber } from '../util/DataUtils';
import { LEGEND_TYPES } from '../util/ReactUtils';
var defaultUniqBy = function defaultUniqBy(entry) {
return entry.value;
};
var getUniqPaylod = function getUniqPaylod(option, payload) {
if (option === true) {
return _uniqBy(payload, defaultUniqBy);
}
if (_isFunction(option)) {
return _uniqBy(payload, option);
}
return payload;
};
var renderContent = function renderContent(content, props) {
if (React.isValidElement(content)) {
return React.cloneElement(content, props);
}
if (_isFunction(content)) {
return content(props);
}
return React.createElement(DefaultLegendContent, props);
};
var EPS = 1;
var ICON_TYPES = LEGEND_TYPES.filter(function (type) {
return type !== 'none';
});
var Legend = pureRender(_class = (_temp = _class2 =
/*#__PURE__*/
function (_Component) {
_inherits(Legend, _Component);
function Legend() {
var _getPrototypeOf2;
var _this;
_classCallCheck(this, Legend);
for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
args[_key] = arguments[_key];
}
_this = _possibleConstructorReturn(this, (_getPrototypeOf2 = _getPrototypeOf(Legend)).call.apply(_getPrototypeOf2, [this].concat(args)));
_this.state = {
boxWidth: -1,
boxHeight: -1
};
return _this;
}
_createClass(Legend, [{
key: "componentDidMount",
value: function componentDidMount() {
this.updateBBox();
}
}, {
key: "componentDidUpdate",
value: function componentDidUpdate() {
this.updateBBox();
}
}, {
key: "getBBox",
value: function getBBox() {
var _this$state = this.state,
boxWidth = _this$state.boxWidth,
boxHeight = _this$state.boxHeight;
if (boxWidth >= 0 && boxHeight >= 0) {
return {
width: boxWidth,
height: boxHeight
};
}
return null;
}
}, {
key: "getDefaultPosition",
value: function getDefaultPosition(style) {
var _this$props = this.props,
layout = _this$props.layout,
align = _this$props.align,
verticalAlign = _this$props.verticalAlign,
margin = _this$props.margin,
chartWidth = _this$props.chartWidth,
chartHeight = _this$props.chartHeight;
var hPos, vPos;
if (!style || (style.left === undefined || style.left === null) && (style.right === undefined || style.right === null)) {
if (align === 'center' && layout === 'vertical') {
var box = this.getBBox() || {
width: 0
};
hPos = {
left: ((chartWidth || 0) - box.width) / 2
};
} else {
hPos = align === 'right' ? {
right: margin && margin.right || 0
} : {
left: margin && margin.left || 0
};
}
}
if (!style || (style.top === undefined || style.top === null) && (style.bottom === undefined || style.bottom === null)) {
if (verticalAlign === 'middle') {
var _box = this.getBBox() || {
height: 0
};
vPos = {
top: ((chartHeight || 0) - _box.height) / 2
};
} else {
vPos = verticalAlign === 'bottom' ? {
bottom: margin && margin.bottom || 0
} : {
top: margin && margin.top || 0
};
}
}
return _objectSpread({}, hPos, vPos);
}
}, {
key: "updateBBox",
value: function updateBBox() {
var _this$state2 = this.state,
boxWidth = _this$state2.boxWidth,
boxHeight = _this$state2.boxHeight;
var onBBoxUpdate = this.props.onBBoxUpdate;
if (this.wrapperNode && this.wrapperNode.getBoundingClientRect) {
var box = this.wrapperNode.getBoundingClientRect();
if (Math.abs(box.width - boxWidth) > EPS || Math.abs(box.height - boxHeight) > EPS) {
this.setState({
boxWidth: box.width,
boxHeight: box.height
}, function () {
if (onBBoxUpdate) {
onBBoxUpdate(box);
}
});
}
} else if (boxWidth !== -1 || boxHeight !== -1) {
this.setState({
boxWidth: -1,
boxHeight: -1
}, function () {
if (onBBoxUpdate) {
onBBoxUpdate(null);
}
});
}
}
}, {
key: "render",
value: function render() {
var _this2 = this;
var _this$props2 = this.props,
content = _this$props2.content,
width = _this$props2.width,
height = _this$props2.height,
wrapperStyle = _this$props2.wrapperStyle,
paylodUniqBy = _this$props2.paylodUniqBy,
payload = _this$props2.payload;
var outerStyle = _objectSpread({
position: 'absolute',
width: width || 'auto',
height: height || 'auto'
}, this.getDefaultPosition(wrapperStyle), wrapperStyle);
return React.createElement("div", {
className: "recharts-legend-wrapper",
style: outerStyle,
ref: function ref(node) {
_this2.wrapperNode = node;
}
}, renderContent(content, _objectSpread({}, this.props, {
payload: getUniqPaylod(paylodUniqBy, payload)
})));
}
}], [{
key: "getWithHeight",
value: function getWithHeight(item, chartWidth) {
var layout = item.props.layout;
if (layout === 'vertical' && isNumber(item.props.height)) {
return {
height: item.props.height
};
}
if (layout === 'horizontal') {
return {
width: item.props.width || chartWidth
};
}
return null;
}
}]);
return Legend;
}(Component), _class2.displayName = 'Legend', _class2.propTypes = {
content: PropTypes.oneOfType([PropTypes.element, PropTypes.func]),
wrapperStyle: PropTypes.object,
chartWidth: PropTypes.number,
chartHeight: PropTypes.number,
width: PropTypes.number,
height: PropTypes.number,
iconSize: PropTypes.number,
iconType: PropTypes.oneOf(ICON_TYPES),
layout: PropTypes.oneOf(['horizontal', 'vertical']),
align: PropTypes.oneOf(['center', 'left', 'right']),
verticalAlign: PropTypes.oneOf(['top', 'bottom', 'middle']),
margin: PropTypes.shape({
top: PropTypes.number,
left: PropTypes.number,
bottom: PropTypes.number,
right: PropTypes.number
}),
payload: PropTypes.arrayOf(PropTypes.shape({
value: PropTypes.any,
id: PropTypes.any,
type: PropTypes.oneOf(LEGEND_TYPES)
})),
paylodUniqBy: PropTypes.oneOfType([PropTypes.func, PropTypes.bool]),
formatter: PropTypes.func,
onMouseEnter: PropTypes.func,
onMouseLeave: PropTypes.func,
onClick: PropTypes.func,
onBBoxUpdate: PropTypes.func
}, _class2.defaultProps = {
iconSize: 14,
layout: 'horizontal',
align: 'center',
verticalAlign: 'bottom'
}, _temp)) || _class;
export default Legend;