@antv/g6/src/extend/g/group.js

graph visualization frame work

/**
 * @fileOverview extend G.Group
 * @author huangtonger@aliyun.com
 * @ignore
 */

const G = require('@antv/g');
const Util = require('../../util/');
const Mixin = function() {};

Util.augment(Mixin, {
  /**
   * find element by className
   * @param   {String}      className class name
   * @return  {Array}       rst
   */
  findByClass(className) {
    const rst = [];
    this.deepEach(child => {
      if (child.hasClass(className)) {
        rst.push(child);
      }
    });
    return rst;
  },
  /**
   * Check contains the specified class
   * @param   {String}      className class name
   * @return  {Boolean}     boolean
   */
  hasClass(className) {
    const clasees = this.get('class');
    if (clasees && clasees.indexOf(className) !== -1) {
      return true;
    }
    return false;
  },
  /**
   * traverse child node
   * @param  {function} callback callback
   * @param  {boolean} runSelf excute self or not
   */
  deepEach(callback, runSelf) {
    Util.traverseTree(this, callback, parent => {
      return parent.get('children');
    }, runSelf);
  },
  /**
   * radix sort (a stable sort)
   */
  sort() {
    const children = this.get('children');
    this.set('children', Util.radixSort(children, child => {
      return child.get('zIndex');
    }));
  },
  /**
   * sort by callback
   * @param  {function} callback callback
   */
  sortBy(callback) {
    const children = this.get('children');
    this.set('children', Util.radixSort(children, callback));
  },
  /**
   * clear inner elements
   * @param  {boolean} bool if destroy child
   * @return {object}  this
   */
  clear(bool) {
    const children = this.get('children');
    bool = bool !== false;
    while (children.length !== 0) {
      children[children.length - 1].remove(bool);
    }
    return this;
  }
});

Util.mixin(G.canvas.Group, [ Mixin ]);
Util.mixin(G.svg.Group, [ Mixin ]);

module.exports = Mixin;