wui-bigdata/dist-modules/PartitionDiagram.js

Big data visualization library for WUI framework.

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _typeof = typeof Symbol === "function" && typeof Symbol.iterator === "symbol" ? function (obj) { return typeof obj; } : function (obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol ? "symbol" : typeof obj; };

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 _react = require('react');

var _react2 = _interopRequireDefault(_react);

var _d = require('d3');

var _d2 = _interopRequireDefault(_d);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

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; }

var PartitionDiagram = function (_Component) {
  _inherits(PartitionDiagram, _Component);

  function PartitionDiagram() {
    _classCallCheck(this, PartitionDiagram);

    return _possibleConstructorReturn(this, Object.getPrototypeOf(PartitionDiagram).apply(this, arguments));
  }

  _createClass(PartitionDiagram, [{
    key: 'textTransform',
    value: function textTransform(d, i) {
      if (i !== 0) {
        var r = d.x + d.dx / 2;
        var angle = Math.PI / 2;

        r += r < Math.PI ? angle * -1 : angle;
        r *= 180 / Math.PI;

        return 'translate(' + this.arc.centroid(d) + ')rotate(' + r + ')';
      }
    }
  }, {
    key: 'render',
    value: function render() {
      var _this2 = this;

      if (this.props.shape === 'rect') {
        var _ret = function () {
          var width = 800;
          var height = 500;
          var partition = _d2.default.layout.partition().sort(null).size([800, 500]).value(function () {
            return 1;
          });
          var color = _d2.default.scale.category20();
          var nodes = partition.nodes(root);

          return {
            v: _react2.default.createElement(
              'svg',
              { width: width, height: height },
              nodes.map(function (d, index) {
                return _react2.default.createElement(
                  'g',
                  { key: index },
                  _react2.default.createElement('rect', { x: d.x,
                    y: d.y,
                    width: d.dx,
                    height: d.dy,
                    stroke: '#fff',
                    fill: color((d.children ? d : d.parent).name) }),
                  _react2.default.createElement(
                    'text',
                    { className: 'nodeText',
                      x: d.x,
                      y: d.y,
                      dx: 20,
                      dy: 20 },
                    d.name
                  )
                );
              })
            )
          };
        }();

        if ((typeof _ret === 'undefined' ? 'undefined' : _typeof(_ret)) === "object") return _ret.v;
      } else {
        var _ret2 = function () {
          var width = 900;
          var height = 900;
          var radius = 450;
          var partition = _d2.default.layout.partition().sort(null).size([2 * Math.PI, radius * radius]).value(function () {
            return 1;
          });
          var color = _d2.default.scale.category20();
          var nodes = partition.nodes(root);
          _this2.arc = _d2.default.svg.arc().startAngle(function (d) {
            return d.x;
          }).endAngle(function (d) {
            return d.x + d.dx;
          }).innerRadius(function (d) {
            return Math.sqrt(d.y);
          }).outerRadius(function (d) {
            return Math.sqrt(d.y + d.dy);
          });

          return {
            v: _react2.default.createElement(
              'svg',
              { width: width, height: height },
              nodes.map(function (d, index) {
                return _react2.default.createElement(
                  'g',
                  { key: index, transform: 'translate(' + width / 2 + ', ' + height / 2 + ')' },
                  _react2.default.createElement('path', { display: d.depth ? null : 'none',
                    d: _this2.arc(d),
                    stroke: '#fff',
                    fill: color((d.children ? d : d.parent).name) }),
                  _react2.default.createElement(
                    'text',
                    { className: 'nodeText',
                      dy: '.5em',
                      transform: _this2.textTransform(d, index) },
                    d.name
                  )
                );
              })
            )
          };
        }();

        if ((typeof _ret2 === 'undefined' ? 'undefined' : _typeof(_ret2)) === "object") return _ret2.v;
      }
    }
  }]);

  return PartitionDiagram;
}(_react.Component);

exports.default = PartitionDiagram;


var root = {
  'name': '中国',
  'children': [{
    'name': '浙江',
    'children': [{ 'name': '杭州' }, { 'name': '宁波' }, { 'name': '温州' }, { 'name': '绍兴' }]
  }, {
    'name': '广西',
    'children': [{
      'name': '桂林',
      'children': [{ 'name': '秀峰区' }, { 'name': '叠彩区' }, { 'name': '象山区' }, { 'name': '七星区' }]
    }, { 'name': '南宁' }, { 'name': '柳州' }, { 'name': '防城港' }]
  }, {
    'name': '黑龙江',
    'children': [{ 'name': '哈尔滨' }, { 'name': '齐齐哈尔' }, { 'name': '牡丹江' }, { 'name': '大庆' }]
  }, {
    'name': '新疆',
    'children': [{ 'name': '乌鲁木齐' }, { 'name': '克拉玛依' }, { 'name': '吐鲁番' }, { 'name': '哈密' }]
  }]
};