@alicloud/cloud-charts/lib/Whierarchy/index.js


'use strict';

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");

exports.__esModule = true;
exports["default"] = exports.Hierarchy = void 0;

var _inheritsLoose2 = _interopRequireDefault(require("@babel/runtime/helpers/inheritsLoose"));

var _view = require("@antv/data-set/lib/view");

require("@antv/data-set/lib/api/hierarchy");

require("@antv/data-set/lib/transform/hierarchy/partition");

require("@antv/data-set/lib/connector/hierarchy.js");

var _Base2 = _interopRequireDefault(require("../common/Base"));

var _rectTooltip = _interopRequireDefault(require("../common/rectTooltip"));

var _errorWrap = _interopRequireDefault(require("../common/errorWrap"));

var _index = _interopRequireDefault(require("../themes/index"));

require("./index.css");

function _createForOfIteratorHelperLoose(o, allowArrayLike) { var it; if (typeof Symbol === "undefined" || o[Symbol.iterator] == null) { if (Array.isArray(o) || (it = _unsupportedIterableToArray(o)) || allowArrayLike && o && typeof o.length === "number") { if (it) o = it; var i = 0; return function () { if (i >= o.length) return { done: true }; return { done: false, value: o[i++] }; }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } it = o[Symbol.iterator](); return it.next.bind(it); }

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

var Hierarchy = /*#__PURE__*/function (_Base) {
  (0, _inheritsLoose2["default"])(Hierarchy, _Base);

  function Hierarchy() {
    var _this;

    for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
      args[_key] = arguments[_key];
    }

    _this = _Base.call.apply(_Base, [this].concat(args)) || this;
    _this.chartName = 'G2Hierarchy';
    _this.convertData = false;
    return _this;
  }

  var _proto = Hierarchy.prototype;

  _proto.getDefaultConfig = function getDefaultConfig() {
    return {
      colors: _index["default"].category_12,
      tooltip: {
        titleFormatter: null,
        nameFormatter: null,
        valueFormatter: null
      },
      label: {
        offset: 0 // textStyle: {
        //   fill: '#fff',
        //   shadowBlur: 2,
        //   shadowColor: 'rgba(0,0,0,0.6)',
        // },

      },
      // label 文本展示的策略
      labelRender: function labelRender(depth, name, value, xRange, yRange) {
        // 根据矩形大小判断是否渲染 label
        if (xRange[1] - xRange[0] > 0.03 && yRange[1] - yRange[0] > 0.05) {
          return name;
        }

        return;
      },
      innerRadius: 0,
      polar: false,
      // 区块的 border 样式，包含 lineWidth lineDash stroke 等属性
      borderStyle: {}
    };
  };

  _proto.init = function init(chart, config, data) {
    var dataView = processDataView(data);
    var nodes = parseDataView(dataView);
    chart.data(nodes); // tooltip

    (0, _rectTooltip["default"])(this, chart, config, {}, null, {
      showMarkers: false,
      showCrosshairs: false,
      shared: false
    });
    drawHierarchy(chart, config, config.colors); // chart.render();
  };

  _proto.changeData = function changeData(chart, config, data) {
    var dataView = processDataView(data);
    var nodes = parseDataView(dataView);
    chart && chart.changeData(nodes);
  } // Chart 内容，方法名称与入参和原来一致
  ;

  return Hierarchy;
}(_Base2["default"]);

exports.Hierarchy = Hierarchy;
var Whierarchy = (0, _errorWrap["default"])(Hierarchy);
var _default = Whierarchy; // 将 DataSet 处理后的结果转换为 G2 接受的数据

exports["default"] = _default;

function parseDataView(dv) {
  var nodes = [];

  for (var _iterator = _createForOfIteratorHelperLoose(dv.getAllNodes()), _step; !(_step = _iterator()).done;) {
    var node = _step.value;
    var eachNode = {
      name: node.data.name,
      value: node.value,
      depth: node.depth,
      x: node.x,
      y: node.y,
      path: getNodePath(node).join('/')
    };
    nodes.push(eachNode);
  }

  return nodes;
} // 简单矩形树图


function drawHierarchy(chart, config, colors, field) {
  if (field === void 0) {
    field = 'name';
  }

  // 设置坐标系：极坐标/直角坐标
  if (config.polar) {
    chart.coordinate('polar', {
      innerRadius: config.innerRadius || 0
    });
  } else {
    chart.coordinate();
  }

  chart.axis(false);
  chart.legend(false);
  chart.polygon().position('x*y').color(field, colors).tooltip('name*value*path', function (name, value, path) {
    return {
      name: name,
      value: value,
      title: path
    };
  }).style(config.borderStyle).label('name*value*x*y*depth', function (name, value, xs, ys, depth) {
    return config.labelRender(depth, name, value, [Math.min.apply(Math, xs), Math.max.apply(Math, xs)], [Math.min.apply(Math, ys), Math.max.apply(Math, ys)]);
  } // config.label,
  );
} // 数据分箱


function processDataView(data) {
  return new _view.View().source(data, {
    type: 'hierarchy'
  }).transform({
    type: 'hierarchy.partition',
    as: ['x', 'y']
  });
} // 获取节点的路径


function getNodePath(n) {
  if (!n.parent) {
    return [n.data.name];
  }

  return [].concat(getNodePath(n.parent), [n.data.name]);
}