g6/plugins/layout.dagre/index.js

/**
 * @fileOverview DAG 分层布局
 * @author huangtonger@aliyun.com
 */
const G6 = require('@antv/g6');
const Layout = require('./layout');

G6.Layouts.Dagre = Layout;

class Plugin {
  constructor(options) {
    this.options = options;
  }
  init() {
    const graph = this.graph;
    graph.on('beforeinit', () => {
      const layout = new Layout(this.options);
      graph.set('layout', layout);
    });
  }
}

G6.Plugins['layout.dagre'] = Plugin;

module.exports = Plugin;