const isFunction = function(o){ return typeof o === 'function'; }; const defaults = require('./defaults'); const assign = require('./assign'); const dagre = require('dagre'); // constructor // options : object containing layout options function DagreLayout( options ){ this.options = assign( {}, defaults, options ); } // runs the layout DagreLayout.prototype.run = function(){ let options = this.options; let layout = this; let cy = options.cy; // cy is automatically populated for us in the constructor let eles = options.eles; let getVal = function( ele, val ){ return isFunction(val) ? val.apply( ele, [ ele ] ) : val; }; let bb = options.boundingBox || { x1: 0, y1: 0, w: cy.width(), h: cy.height() }; if( bb.x2 === undefined ){ bb.x2 = bb.x1 + bb.w; } if( bb.w === undefined ){ bb.w = bb.x2 - bb.x1; } if( bb.y2 === undefined ){ bb.y2 = bb.y1 + bb.h; } if( bb.h === undefined ){ bb.h = bb.y2 - bb.y1; } let g = new dagre.graphlib.Graph({ multigraph: true, compound: true }); let gObj = {}; let setGObj = function( name, val ){ if( val != null ){ gObj[ name ] = val; } }; setGObj( 'nodesep', options.nodeSep ); setGObj( 'edgesep', options.edgeSep ); setGObj( 'ranksep', options.rankSep ); setGObj( 'rankdir', options.rankDir ); setGObj( 'align', options.align); setGObj( 'ranker', options.ranker ); setGObj( 'acyclicer', options.acyclicer); g.setGraph( gObj ); g.setDefaultEdgeLabel(function() { return {}; }); g.setDefaultNodeLabel(function() { return {}; }); // add nodes to dagre let nodes = eles.nodes(); if ( isFunction(options.sort) ) { nodes = nodes.sort( options.sort ); } for( let i = 0; i < nodes.length; i++ ){ let node = nodes[i]; let nbb = node.layoutDimensions( options ); g.setNode( node.id(), { width: nbb.w, height: nbb.h, name: node.id() } ); // console.log( g.node(node.id()) ); } // set compound parents for( let i = 0; i < nodes.length; i++ ){ let node = nodes[i]; if( node.isChild() ){ g.setParent( node.id(), node.parent().id() ); } } // add edges to dagre let edges = eles.edges().stdFilter(function( edge ){ return !edge.source().isParent() && !edge.target().isParent(); // dagre can't handle edges on compound nodes }); if ( isFunction(options.sort) ) { edges = edges.sort( options.sort ); } for( let i = 0; i < edges.length; i++ ){ let edge = edges[i]; g.setEdge( edge.source().id(), edge.target().id(), { minlen: getVal( edge, options.minLen ), weight: getVal( edge, options.edgeWeight ), name: edge.id() }, edge.id() ); // console.log( g.edge(edge.source().id(), edge.target().id(), edge.id()) ); } dagre.layout( g ); let gNodeIds = g.nodes(); for( let i = 0; i < gNodeIds.length; i++ ){ let id = gNodeIds[i]; let n = g.node( id ); cy.getElementById(id).scratch().dagre = n; } let dagreBB; if( options.boundingBox ){ dagreBB = { x1: Infinity, x2: -Infinity, y1: Infinity, y2: -Infinity }; nodes.forEach(function( node ){ let dModel = node.scratch().dagre; dagreBB.x1 = Math.min( dagreBB.x1, dModel.x ); dagreBB.x2 = Math.max( dagreBB.x2, dModel.x ); dagreBB.y1 = Math.min( dagreBB.y1, dModel.y ); dagreBB.y2 = Math.max( dagreBB.y2, dModel.y ); }); dagreBB.w = dagreBB.x2 - dagreBB.x1; dagreBB.h = dagreBB.y2 - dagreBB.y1; } else { dagreBB = bb; } let constrainPos = function( p ){ if( options.boundingBox ){ let xPct = dagreBB.w === 0 ? 0 : (p.x - dagreBB.x1) / dagreBB.w; let yPct = dagreBB.h === 0 ? 0 : (p.y - dagreBB.y1) / dagreBB.h; return { x: bb.x1 + xPct * bb.w, y: bb.y1 + yPct * bb.h }; } else { return p; } }; nodes.layoutPositions(layout, options, function( ele ){ ele = typeof ele === "object" ? ele : this; let dModel = ele.scratch().dagre; return constrainPos({ x: dModel.x, y: dModel.y }); }); return this; // chaining }; module.exports = DagreLayout;