@antv/g6/lib/layouts/tree/base.js

graph visualization frame work

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

/**
 * @fileOverview tree layout base
 * @author huangtonger@aliyun.com
 */

var Base = require('../base');

var CompactBoxTreeLayout = function (_Base) {
  _inherits(CompactBoxTreeLayout, _Base);

  function CompactBoxTreeLayout(options) {
    _classCallCheck(this, CompactBoxTreeLayout);

    var _this = _possibleConstructorReturn(this, _Base.call(this));

    _this.options = options;
    return _this;
  }

  CompactBoxTreeLayout.prototype.execute = function execute() {
    var _this2 = this;

    var options = this.options;
    var roots = this.roots;
    roots.forEach(function (root) {
      var layoutedRoot = _this2.layout(root, options);
      layoutedRoot.eachNode(function (node) {
        node.data.x = node.x + node.data.width / 2 + node.hgap;
        node.data.y = node.y + node.data.height / 2 + node.vgap;
      });
    });
  };

  return CompactBoxTreeLayout;
}(Base);

module.exports = CompactBoxTreeLayout;