|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234 |
- var _createClass = 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); } } return function (Constructor, protoProps, staticProps) { if (protoProps) defineProperties(Constructor.prototype, protoProps); if (staticProps) defineProperties(Constructor, staticProps); return Constructor; }; }();
-
- 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; };
-
- function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
-
- function _possibleConstructorReturn(self, call) { if (!self) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return call && (typeof call === "object" || typeof call === "function") ? call : self; }
-
- function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function, not " + typeof superClass); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, enumerable: false, writable: true, configurable: true } }); if (superClass) Object.setPrototypeOf ? Object.setPrototypeOf(subClass, superClass) : subClass.__proto__ = superClass; }
-
- // 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, pack, partition, treemapSquarify, treemapResquarify, treemapSlice, treemapDice, treemapSliceDice, treemapBinary, treemap } from 'd3-hierarchy';
-
- import { CONTINUOUS_COLOR_RANGE, DEFAULT_COLOR, DEFAULT_OPACITY, OPACITY_TYPE } from '../theme';
- import { AnimationPropType } from '../animation';
- import { getAttributeFunctor, getMissingScaleProps } from '../utils/scales-utils';
- import { MarginPropType, getInnerDimensions } from '../utils/chart-utils';
-
- import TreemapDOM from './treemap-dom';
- import TreemapSVG from './treemap-svg';
-
- var TREEMAP_TILE_MODES = {
- squarify: treemapSquarify,
- resquarify: treemapResquarify,
- slice: treemapSlice,
- dice: treemapDice,
- slicedice: treemapSliceDice,
- binary: treemapBinary
- };
-
- var TREEMAP_LAYOUT_MODES = ['circlePack', 'partition', 'partition-pivot'];
-
- var NOOP = function NOOP(d) {
- return d;
- };
-
- var ATTRIBUTES = ['opacity', 'color'];
-
- var DEFAULT_MARGINS = {
- left: 40,
- right: 10,
- top: 10,
- bottom: 40
- };
-
- /**
- * Get the map of scale functions from the given props.
- * @param {Object} props Props for the component.
- * @returns {Object} Map of scale functions.
- * @private
- */
- function _getScaleFns(props) {
- var data = props.data;
-
- var allData = data.children || [];
-
- // Adding _allData property to the object to reuse the existing
- // getAttributeFunctor function.
- var compatibleProps = _extends({}, props, getMissingScaleProps(props, allData, ATTRIBUTES), {
- _allData: allData
- });
- return {
- opacity: getAttributeFunctor(compatibleProps, 'opacity'),
- color: getAttributeFunctor(compatibleProps, 'color')
- };
- }
-
- var Treemap = function (_React$Component) {
- _inherits(Treemap, _React$Component);
-
- function Treemap(props) {
- _classCallCheck(this, Treemap);
-
- var _this = _possibleConstructorReturn(this, (Treemap.__proto__ || Object.getPrototypeOf(Treemap)).call(this, props));
-
- _this.state = _extends({
- scales: _getScaleFns(props)
- }, getInnerDimensions(props, props.margin));
- return _this;
- }
-
- _createClass(Treemap, [{
- key: 'componentWillReceiveProps',
- value: function componentWillReceiveProps(props) {
- this.setState(_extends({
- scales: _getScaleFns(props)
- }, getInnerDimensions(props, props.margin)));
- }
-
- /**
- * Create the list of nodes to render.
- * @returns {Array} Array of nodes.
- * @private
- */
-
- }, {
- key: '_getNodesToRender',
- value: function _getNodesToRender() {
- var _state = this.state,
- innerWidth = _state.innerWidth,
- innerHeight = _state.innerHeight;
- var _props = this.props,
- data = _props.data,
- mode = _props.mode,
- padding = _props.padding,
- sortFunction = _props.sortFunction,
- getSize = _props.getSize;
-
- if (!data) {
- return [];
- }
-
- if (mode === 'partition' || mode === 'partition-pivot') {
- var partitionFunction = partition().size(mode === 'partition-pivot' ? [innerHeight, innerWidth] : [innerWidth, innerHeight]).padding(padding);
- var _structuredInput = hierarchy(data).sum(getSize).sort(function (a, b) {
- return sortFunction(a, b, getSize);
- });
- var mappedNodes = partitionFunction(_structuredInput).descendants();
- if (mode === 'partition-pivot') {
- return mappedNodes.map(function (node) {
- return _extends({}, node, {
- x0: node.y0,
- x1: node.y1,
- y0: node.x0,
- y1: node.x1
- });
- });
- }
- return mappedNodes;
- }
- if (mode === 'circlePack') {
- var packingFunction = pack().size([innerWidth, innerHeight]).padding(padding);
- var _structuredInput2 = hierarchy(data).sum(getSize).sort(function (a, b) {
- return sortFunction(a, b, getSize);
- });
- return packingFunction(_structuredInput2).descendants();
- }
-
- var tileFn = TREEMAP_TILE_MODES[mode];
- var treemapingFunction = treemap(tileFn).tile(tileFn).size([innerWidth, innerHeight]).padding(padding);
- var structuredInput = hierarchy(data).sum(getSize).sort(function (a, b) {
- return sortFunction(a, b, getSize);
- });
- return treemapingFunction(structuredInput).descendants();
- }
- }, {
- key: 'render',
- value: function render() {
- var renderMode = this.props.renderMode;
- var scales = this.state.scales;
-
- var nodes = this._getNodesToRender();
- var TreemapElement = renderMode === 'SVG' ? TreemapSVG : TreemapDOM;
- return React.createElement(TreemapElement, _extends({}, this.props, { nodes: nodes, scales: scales }));
- }
- }]);
-
- return Treemap;
- }(React.Component);
-
- Treemap.displayName = 'Treemap';
- Treemap.propTypes = {
- animation: AnimationPropType,
- className: PropTypes.string,
- data: PropTypes.object.isRequired,
- height: PropTypes.number.isRequired,
- hideRootNode: PropTypes.bool,
- margin: MarginPropType,
- mode: PropTypes.oneOf(Object.keys(TREEMAP_TILE_MODES).concat(TREEMAP_LAYOUT_MODES)),
- onLeafClick: PropTypes.func,
- onLeafMouseOver: PropTypes.func,
- onLeafMouseOut: PropTypes.func,
- useCirclePacking: PropTypes.bool,
- padding: PropTypes.number.isRequired,
- sortFunction: PropTypes.func,
- width: PropTypes.number.isRequired,
- getSize: PropTypes.func,
- getColor: PropTypes.func
- };
-
- Treemap.defaultProps = {
- className: '',
- colorRange: CONTINUOUS_COLOR_RANGE,
- _colorValue: DEFAULT_COLOR,
- data: {
- children: []
- },
- hideRootNode: false,
- margin: DEFAULT_MARGINS,
- mode: 'squarify',
- onLeafClick: NOOP,
- onLeafMouseOver: NOOP,
- onLeafMouseOut: NOOP,
- opacityType: OPACITY_TYPE,
- _opacityValue: DEFAULT_OPACITY,
- padding: 1,
- sortFunction: function sortFunction(a, b, accessor) {
- if (!accessor) {
- return 0;
- }
- return accessor(a) - accessor(b);
- },
- getSize: function getSize(d) {
- return d.size;
- },
- getColor: function getColor(d) {
- return d.color;
- },
- getLabel: function getLabel(d) {
- return d.title;
- }
- };
- export default Treemap;
|