@antv/g6/lib/extend/g/shape.js

graph visualization frame work

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

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

Util.augment(Mixin, {
  /**
   * get shape init attrs
   * @return {object} rst
   */
  getAttrs: function getAttrs() {
    var _this = this;

    var attrs = this.get('attrs');
    var rst = {};
    Util.each(attrs, function (v, k) {
      rst[k] = _this.attr(k);
    });
    return rst;
  },

  /**
   * Check contains the specified class
   * @param   {String}      className class name
   * @return  {Boolean}     boolean
   */
  hasClass: function hasClass(className) {
    var clasees = this.get('class');
    if (clasees && clasees.indexOf(className) !== -1) {
      return true;
    }
    return false;
  }
});

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

module.exports = Mixin;