recharts
Version:
React charts
399 lines (398 loc) • 22.3 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.Funnel = void 0;
var _isEqual2 = _interopRequireDefault(require("lodash/isEqual"));
var _isPlainObject2 = _interopRequireDefault(require("lodash/isPlainObject"));
var _isFunction2 = _interopRequireDefault(require("lodash/isFunction"));
var _omit2 = _interopRequireDefault(require("lodash/omit"));
var _isString2 = _interopRequireDefault(require("lodash/isString"));
var _isNumber2 = _interopRequireDefault(require("lodash/isNumber"));
var _react = _interopRequireWildcard(require("react"));
var _reactSmooth = _interopRequireDefault(require("react-smooth"));
var _classnames = _interopRequireDefault(require("classnames"));
var _Layer = require("../container/Layer");
var _Trapezoid = require("../shape/Trapezoid");
var _LabelList = require("../component/LabelList");
var _Cell = require("../component/Cell");
var _ReactUtils = require("../util/ReactUtils");
var _Global = require("../util/Global");
var _DataUtils = require("../util/DataUtils");
var _ChartUtils = require("../util/ChartUtils");
var _types = require("../util/types");
function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { "default": obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj["default"] = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
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(arr, i) { var _i = null == arr ? null : "undefined" != typeof Symbol && arr[Symbol.iterator] || arr["@@iterator"]; if (null != _i) { var _s, _e, _x, _r, _arr = [], _n = !0, _d = !1; try { if (_x = (_i = _i.call(arr)).next, 0 === i) { if (Object(_i) !== _i) return; _n = !1; } else for (; !(_n = (_s = _x.call(_i)).done) && (_arr.push(_s.value), _arr.length !== i); _n = !0); } catch (err) { _d = !0, _e = err; } finally { try { if (!_n && null != _i["return"] && (_r = _i["return"](), Object(_r) !== _r)) return; } finally { if (_d) throw _e; } } return _arr; } }
function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
function _typeof(obj) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (obj) { return typeof obj; } : function (obj) { return obj && "function" == typeof Symbol && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }, _typeof(obj); }
function _extends() { _extends = Object.assign ? Object.assign.bind() : 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; }; return _extends.apply(this, arguments); }
function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { _defineProperty(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
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, _toPropertyKey(descriptor.key), descriptor); } }
function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); Object.defineProperty(Constructor, "prototype", { writable: false }); return Constructor; }
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 } }); Object.defineProperty(subClass, "prototype", { writable: false }); if (superClass) _setPrototypeOf(subClass, superClass); }
function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf ? Object.setPrototypeOf.bind() : function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = _getPrototypeOf(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = _getPrototypeOf(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return _possibleConstructorReturn(this, result); }; }
function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } else if (call !== void 0) { throw new TypeError("Derived constructors may only return object or undefined"); } 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 _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }
function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf.bind() : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
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(arg) { var key = _toPrimitive(arg, "string"); return _typeof(key) === "symbol" ? key : String(key); }
function _toPrimitive(input, hint) { if (_typeof(input) !== "object" || input === null) return input; var prim = input[Symbol.toPrimitive]; if (prim !== undefined) { var res = prim.call(input, hint || "default"); if (_typeof(res) !== "object") return res; throw new TypeError("@@toPrimitive must return a primitive value."); } return (hint === "string" ? String : Number)(input); }
var Funnel = /*#__PURE__*/function (_PureComponent) {
_inherits(Funnel, _PureComponent);
var _super = _createSuper(Funnel);
function Funnel() {
var _this;
_classCallCheck(this, Funnel);
for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
args[_key] = arguments[_key];
}
_this = _super.call.apply(_super, [this].concat(args));
_defineProperty(_assertThisInitialized(_this), "state", {
isAnimationFinished: false
});
_defineProperty(_assertThisInitialized(_this), "handleAnimationEnd", function () {
var onAnimationEnd = _this.props.onAnimationEnd;
_this.setState({
isAnimationFinished: true
});
if ((0, _isFunction2["default"])(onAnimationEnd)) {
onAnimationEnd();
}
});
_defineProperty(_assertThisInitialized(_this), "handleAnimationStart", function () {
var onAnimationStart = _this.props.onAnimationStart;
_this.setState({
isAnimationFinished: false
});
if ((0, _isFunction2["default"])(onAnimationStart)) {
onAnimationStart();
}
});
return _this;
}
_createClass(Funnel, [{
key: "isActiveIndex",
value: function isActiveIndex(i) {
var activeIndex = this.props.activeIndex;
if (Array.isArray(activeIndex)) {
return activeIndex.indexOf(i) !== -1;
}
return i === activeIndex;
}
}, {
key: "renderTrapezoidsStatically",
value: function renderTrapezoidsStatically(trapezoids) {
var _this2 = this;
var activeShape = this.props.activeShape;
return trapezoids.map(function (entry, i) {
var trapezoidOptions = _this2.isActiveIndex(i) ? activeShape : null;
var trapezoidProps = _objectSpread(_objectSpread({}, entry), {}, {
stroke: entry.stroke
});
return /*#__PURE__*/_react["default"].createElement(_Layer.Layer, _extends({
className: "recharts-funnel-trapezoid"
}, (0, _types.adaptEventsOfChild)(_this2.props, entry, i), {
key: "trapezoid-".concat(i) // eslint-disable-line react/no-array-index-key
,
role: "img"
}), Funnel.renderTrapezoidItem(trapezoidOptions, trapezoidProps));
});
}
}, {
key: "renderTrapezoidsWithAnimation",
value: function renderTrapezoidsWithAnimation() {
var _this3 = this;
var _this$props = this.props,
trapezoids = _this$props.trapezoids,
isAnimationActive = _this$props.isAnimationActive,
animationBegin = _this$props.animationBegin,
animationDuration = _this$props.animationDuration,
animationEasing = _this$props.animationEasing,
animationId = _this$props.animationId;
var prevTrapezoids = this.state.prevTrapezoids;
return /*#__PURE__*/_react["default"].createElement(_reactSmooth["default"], {
begin: animationBegin,
duration: animationDuration,
isActive: isAnimationActive,
easing: animationEasing,
from: {
t: 0
},
to: {
t: 1
},
key: "funnel-".concat(animationId),
onAnimationStart: this.handleAnimationStart,
onAnimationEnd: this.handleAnimationEnd
}, function (_ref) {
var t = _ref.t;
var stepData = trapezoids.map(function (entry, index) {
var prev = prevTrapezoids && prevTrapezoids[index];
if (prev) {
var _interpolatorX = (0, _DataUtils.interpolateNumber)(prev.x, entry.x);
var _interpolatorY = (0, _DataUtils.interpolateNumber)(prev.y, entry.y);
var _interpolatorUpperWidth = (0, _DataUtils.interpolateNumber)(prev.upperWidth, entry.upperWidth);
var _interpolatorLowerWidth = (0, _DataUtils.interpolateNumber)(prev.lowerWidth, entry.lowerWidth);
var _interpolatorHeight = (0, _DataUtils.interpolateNumber)(prev.height, entry.height);
return _objectSpread(_objectSpread({}, entry), {}, {
x: _interpolatorX(t),
y: _interpolatorY(t),
upperWidth: _interpolatorUpperWidth(t),
lowerWidth: _interpolatorLowerWidth(t),
height: _interpolatorHeight(t)
});
}
var interpolatorX = (0, _DataUtils.interpolateNumber)(entry.x + entry.upperWidth / 2, entry.x);
var interpolatorY = (0, _DataUtils.interpolateNumber)(entry.y + entry.height / 2, entry.y);
var interpolatorUpperWidth = (0, _DataUtils.interpolateNumber)(0, entry.upperWidth);
var interpolatorLowerWidth = (0, _DataUtils.interpolateNumber)(0, entry.lowerWidth);
var interpolatorHeight = (0, _DataUtils.interpolateNumber)(0, entry.height);
return _objectSpread(_objectSpread({}, entry), {}, {
x: interpolatorX(t),
y: interpolatorY(t),
upperWidth: interpolatorUpperWidth(t),
lowerWidth: interpolatorLowerWidth(t),
height: interpolatorHeight(t)
});
});
return /*#__PURE__*/_react["default"].createElement(_Layer.Layer, null, _this3.renderTrapezoidsStatically(stepData));
});
}
}, {
key: "renderTrapezoids",
value: function renderTrapezoids() {
var _this$props2 = this.props,
trapezoids = _this$props2.trapezoids,
isAnimationActive = _this$props2.isAnimationActive;
var prevTrapezoids = this.state.prevTrapezoids;
if (isAnimationActive && trapezoids && trapezoids.length && (!prevTrapezoids || !(0, _isEqual2["default"])(prevTrapezoids, trapezoids))) {
return this.renderTrapezoidsWithAnimation();
}
return this.renderTrapezoidsStatically(trapezoids);
}
}, {
key: "render",
value: function render() {
var _this$props3 = this.props,
hide = _this$props3.hide,
trapezoids = _this$props3.trapezoids,
className = _this$props3.className,
isAnimationActive = _this$props3.isAnimationActive;
var isAnimationFinished = this.state.isAnimationFinished;
if (hide || !trapezoids || !trapezoids.length) {
return null;
}
var layerClass = (0, _classnames["default"])('recharts-trapezoids', className);
return /*#__PURE__*/_react["default"].createElement(_Layer.Layer, {
className: layerClass
}, this.renderTrapezoids(), (!isAnimationActive || isAnimationFinished) && _LabelList.LabelList.renderCallByParent(this.props, trapezoids));
}
}], [{
key: "getDerivedStateFromProps",
value: function getDerivedStateFromProps(nextProps, prevState) {
if (nextProps.animationId !== prevState.prevAnimationId) {
return {
prevAnimationId: nextProps.animationId,
curTrapezoids: nextProps.trapezoids,
prevTrapezoids: prevState.curTrapezoids
};
}
if (nextProps.trapezoids !== prevState.curTrapezoids) {
return {
curTrapezoids: nextProps.trapezoids
};
}
return null;
}
}, {
key: "renderTrapezoidItem",
value: function renderTrapezoidItem(option, props) {
if ( /*#__PURE__*/_react["default"].isValidElement(option)) {
return /*#__PURE__*/_react["default"].cloneElement(option, props);
}
if ((0, _isFunction2["default"])(option)) {
return option(props);
}
if ((0, _isPlainObject2["default"])(option)) {
return /*#__PURE__*/_react["default"].createElement(_Trapezoid.Trapezoid, _extends({}, props, option));
}
return /*#__PURE__*/_react["default"].createElement(_Trapezoid.Trapezoid, props);
}
}]);
return Funnel;
}(_react.PureComponent);
exports.Funnel = Funnel;
_defineProperty(Funnel, "displayName", 'Funnel');
_defineProperty(Funnel, "defaultProps", {
stroke: '#fff',
fill: '#808080',
legendType: 'rect',
labelLine: true,
hide: false,
isAnimationActive: !_Global.Global.isSsr,
animationBegin: 400,
animationDuration: 1500,
animationEasing: 'ease',
nameKey: 'name',
lastShapeType: 'triangle'
});
_defineProperty(Funnel, "getRealFunnelData", function (item) {
var _item$props = item.props,
data = _item$props.data,
children = _item$props.children;
var presentationProps = (0, _ReactUtils.filterProps)(item.props);
var cells = (0, _ReactUtils.findAllByType)(children, _Cell.Cell);
if (data && data.length) {
return data.map(function (entry, index) {
return _objectSpread(_objectSpread(_objectSpread({
payload: entry
}, presentationProps), entry), cells && cells[index] && cells[index].props);
});
}
if (cells && cells.length) {
return cells.map(function (cell) {
return _objectSpread(_objectSpread({}, presentationProps), cell.props);
});
}
return [];
});
_defineProperty(Funnel, "getRealWidthHeight", function (item, offset) {
var customWidth = item.props.width;
var width = offset.width,
height = offset.height,
left = offset.left,
right = offset.right,
top = offset.top,
bottom = offset.bottom;
var realHeight = height;
var realWidth = width;
if ((0, _isNumber2["default"])(customWidth)) {
realWidth = customWidth;
} else if ((0, _isString2["default"])(customWidth)) {
realWidth = realWidth * parseFloat(customWidth) / 100;
}
return {
realWidth: realWidth - left - right - 50,
realHeight: realHeight - bottom - top,
offsetX: (width - realWidth) / 2,
offsetY: (height - realHeight) / 2
};
});
_defineProperty(Funnel, "getComposedData", function (_ref2) {
var item = _ref2.item,
offset = _ref2.offset;
var funnelData = Funnel.getRealFunnelData(item);
var _item$props2 = item.props,
dataKey = _item$props2.dataKey,
nameKey = _item$props2.nameKey,
tooltipType = _item$props2.tooltipType,
lastShapeType = _item$props2.lastShapeType,
reversed = _item$props2.reversed;
var left = offset.left,
top = offset.top;
var _Funnel$getRealWidthH = Funnel.getRealWidthHeight(item, offset),
realHeight = _Funnel$getRealWidthH.realHeight,
realWidth = _Funnel$getRealWidthH.realWidth,
offsetX = _Funnel$getRealWidthH.offsetX,
offsetY = _Funnel$getRealWidthH.offsetY;
var maxValue = Math.max.apply(null, funnelData.map(function (entry) {
return (0, _ChartUtils.getValueByDataKey)(entry, dataKey, 0);
}));
var len = funnelData.length;
var rowHeight = realHeight / len;
var parentViewBox = {
x: offset.left,
y: offset.top,
width: offset.width,
height: offset.height
};
var trapezoids = funnelData.map(function (entry, i) {
var rawVal = (0, _ChartUtils.getValueByDataKey)(entry, dataKey, 0);
var name = (0, _ChartUtils.getValueByDataKey)(entry, nameKey, i);
var val = rawVal;
var nextVal;
if (i !== len - 1) {
nextVal = (0, _ChartUtils.getValueByDataKey)(funnelData[i + 1], dataKey, 0);
if (nextVal instanceof Array) {
var _nextVal = nextVal;
var _nextVal2 = _slicedToArray(_nextVal, 1);
nextVal = _nextVal2[0];
}
} else if (rawVal instanceof Array && rawVal.length === 2) {
var _rawVal = _slicedToArray(rawVal, 2);
val = _rawVal[0];
nextVal = _rawVal[1];
} else if (lastShapeType === 'rectangle') {
nextVal = val;
} else {
nextVal = 0;
}
var x = (maxValue - val) * realWidth / (2 * maxValue) + top + 25 + offsetX;
var y = rowHeight * i + left + offsetY;
var upperWidth = val / maxValue * realWidth;
var lowerWidth = nextVal / maxValue * realWidth;
var tooltipPayload = [{
name: name,
value: val,
payload: entry,
dataKey: dataKey,
type: tooltipType
}];
var tooltipPosition = {
x: x + upperWidth / 2,
y: y + rowHeight / 2
};
return _objectSpread(_objectSpread({
x: x,
y: y,
width: Math.max(upperWidth, lowerWidth),
upperWidth: upperWidth,
lowerWidth: lowerWidth,
height: rowHeight,
name: name,
val: val,
tooltipPayload: tooltipPayload,
tooltipPosition: tooltipPosition
}, (0, _omit2["default"])(entry, 'width')), {}, {
payload: entry,
parentViewBox: parentViewBox,
labelViewBox: {
x: x + (upperWidth - lowerWidth) / 4,
y: y,
width: Math.abs(upperWidth - lowerWidth) / 2 + Math.min(upperWidth, lowerWidth),
height: rowHeight
}
});
});
if (reversed) {
trapezoids = trapezoids.map(function (entry, index) {
var newY = entry.y - index * rowHeight + (len - 1 - index) * rowHeight;
return _objectSpread(_objectSpread({}, entry), {}, {
upperWidth: entry.lowerWidth,
lowerWidth: entry.upperWidth,
x: entry.x - (entry.lowerWidth - entry.upperWidth) / 2,
y: entry.y - index * rowHeight + (len - 1 - index) * rowHeight,
tooltipPosition: _objectSpread(_objectSpread({}, entry.tooltipPosition), {}, {
y: newY + rowHeight / 2
}),
labelViewBox: _objectSpread(_objectSpread({}, entry.labelViewBox), {}, {
y: newY
})
});
});
}
return {
trapezoids: trapezoids,
data: funnelData
};
});