|
|
- 'use strict';
- /**
- * There's a mix-up with terms in the code. Following are the formal definitions:
- *
- * tree - a strict hierarchical network, i.e. every node has at most one parent
- * forest - a collection of trees. These distinct trees are thus not connected.
- *
- * So:
- * - in a network that is not a tree, there exist nodes with multiple parents.
- * - a network consisting of unconnected sub-networks, of which at least one
- * is not a tree, is not a forest.
- *
- * In the code, the definitions are:
- *
- * tree - any disconnected sub-network, strict hierarchical or not.
- * forest - a bunch of these sub-networks
- *
- * The difference between tree and not-tree is important in the code, notably within
- * to the block-shifting algorithm. The algorithm assumes formal trees and fails
- * for not-trees, often in a spectacular manner (search for 'exploding network' in the issues).
- *
- * In order to distinguish the definitions in the following code, the adjective 'formal' is
- * used. If 'formal' is absent, you must assume the non-formal definition.
- *
- * ----------------------------------------------------------------------------------
- * NOTES
- * =====
- *
- * A hierarchical layout is a different thing from a hierarchical network.
- * The layout is a way to arrange the nodes in the view; this can be done
- * on non-hierarchical networks as well. The converse is also possible.
- */
- let util = require('../../util');
- var NetworkUtil = require('../NetworkUtil').default;
-
-
- /**
- * Container for derived data on current network, relating to hierarchy.
- *
- * Local, private class.
- */
- class HierarchicalStatus {
-
- constructor() {
- this.childrenReference = {}; // child id's per node id
- this.parentReference = {}; // parent id's per node id
- this.trees = {}; // tree id per node id; i.e. to which tree does given node id belong
-
- this.distributionOrdering = {}; // The nodes per level, in the display order
- this.levels = {}; // hierarchy level per node id
- this.distributionIndex = {}; // The position of the node in the level sorting order, per node id.
-
- this.isTree = false; // True if current network is a formal tree
- this.treeIndex = -1; // Highest tree id in current network.
- }
-
-
- /**
- * Add the relation between given nodes to the current state.
- */
- addRelation(parentNodeId, childNodeId) {
- if (this.childrenReference[parentNodeId] === undefined) {
- this.childrenReference[parentNodeId] = [];
- }
- this.childrenReference[parentNodeId].push(childNodeId);
-
- if (this.parentReference[childNodeId] === undefined) {
- this.parentReference[childNodeId] = [];
- }
- this.parentReference[childNodeId].push(parentNodeId);
- }
-
-
- /**
- * Check if the current state is for a formal tree or formal forest.
- *
- * This is the case if every node has at most one parent.
- *
- * Pre: parentReference init'ed properly for current network
- */
- checkIfTree() {
- for (let i in this.parentReference) {
- if (this.parentReference[i].length > 1) {
- this.isTree = false;
- return;
- }
- }
-
- this.isTree = true;
- }
-
-
- /**
- * Return the number of separate trees in the current network.
- */
- numTrees() {
- return (this.treeIndex + 1); // This assumes the indexes are assigned consecitively
- }
-
-
- /**
- * Assign a tree id to a node
- */
- setTreeIndex(node, treeId) {
- if (this.trees[node.id] === undefined) {
- this.trees[node.id] = treeId;
- this.treeIndex = Math.max(treeId, this.treeIndex);
- }
- }
-
-
- /**
- * Ensure level for given id is defined.
- *
- * Sets level to zero for given node id if not already present
- */
- ensureLevel(nodeId) {
- if (this.levels[nodeId] === undefined) {
- this.levels[nodeId] = 0;
- }
- }
-
-
- /**
- * get the maximum level of a branch.
- *
- * TODO: Never entered; find a test case to test this!
- */
- getMaxLevel(nodeId) {
- let accumulator = {};
-
- let _getMaxLevel = (nodeId) => {
- if (accumulator[nodeId] !== undefined) {
- return accumulator[nodeId];
- }
- let level = this.levels[nodeId];
- if (this.childrenReference[nodeId]) {
- let children = this.childrenReference[nodeId];
- if (children.length > 0) {
- for (let i = 0; i < children.length; i++) {
- level = Math.max(level,_getMaxLevel(children[i]));
- }
- }
- }
- accumulator[nodeId] = level;
- return level;
- };
-
- return _getMaxLevel(nodeId);
- }
-
-
- levelDownstream(nodeA, nodeB) {
- if (this.levels[nodeB.id] === undefined) {
- // set initial level
- if (this.levels[nodeA.id] === undefined) {
- this.levels[nodeA.id] = 0;
- }
- // set level
- this.levels[nodeB.id] = this.levels[nodeA.id] + 1;
- }
- }
-
-
- /**
- * Small util method to set the minimum levels of the nodes to zero.
- */
- setMinLevelToZero(nodes) {
- let minLevel = 1e9;
- // get the minimum level
- for (let nodeId in nodes) {
- if (nodes.hasOwnProperty(nodeId)) {
- if (this.levels[nodeId] !== undefined) {
- minLevel = Math.min(this.levels[nodeId], minLevel);
- }
- }
- }
-
- // subtract the minimum from the set so we have a range starting from 0
- for (let nodeId in nodes) {
- if (nodes.hasOwnProperty(nodeId)) {
- if (this.levels[nodeId] !== undefined) {
- this.levels[nodeId] -= minLevel;
- }
- }
- }
- }
-
-
- /**
- * Get the min and max xy-coordinates of a given tree
- */
- getTreeSize(nodes, index) {
- let min_x = 1e9;
- let max_x = -1e9;
- let min_y = 1e9;
- let max_y = -1e9;
-
- for (let nodeId in this.trees) {
- if (this.trees.hasOwnProperty(nodeId)) {
- if (this.trees[nodeId] === index) {
- let node = nodes[nodeId];
- min_x = Math.min(node.x, min_x);
- max_x = Math.max(node.x, max_x);
- min_y = Math.min(node.y, min_y);
- max_y = Math.max(node.y, max_y);
- }
- }
- }
-
- return {
- min_x: min_x,
- max_x: max_x,
- min_y: min_y,
- max_y: max_y
- };
- }
-
-
- /**
- * Check if two nodes have the same parent(s)
- *
- * @return true if the two nodes have a same ancestor node, false otherwise
- */
- hasSameParent(node1, node2) {
- let parents1 = this.parentReference[node1.id];
- let parents2 = this.parentReference[node2.id];
- if (parents1 === undefined || parents2 === undefined) {
- return false;
- }
-
- for (let i = 0; i < parents1.length; i++) {
- for (let j = 0; j < parents2.length; j++) {
- if (parents1[i] == parents2[j]) {
- return true;
- }
- }
- }
- return false;
- }
-
-
- /**
- * Check if two nodes are in the same tree.
- *
- * @return true if this is so, false otherwise
- */
- inSameSubNetwork(node1, node2) {
- return (this.trees[node1.id] === this.trees[node2.id]);
- }
-
-
- /**
- * Get a list of the distinct levels in the current network
- */
- getLevels() {
- return Object.keys(this.distributionOrdering);
- }
-
-
- /**
- * Add a node to the ordering per level
- */
- addToOrdering(node, level) {
- if (this.distributionOrdering[level] === undefined) {
- this.distributionOrdering[level] = [];
- }
-
- var isPresent = false;
- var curLevel = this.distributionOrdering[level];
- for (var n in curLevel) {
- //if (curLevel[n].id === node.id) {
- if (curLevel[n] === node) {
- isPresent = true;
- break;
- }
- }
-
- if (!isPresent) {
- this.distributionOrdering[level].push(node);
- this.distributionIndex[node.id] = this.distributionOrdering[level].length - 1;
- }
- }
- }
-
-
- class LayoutEngine {
- constructor(body) {
- this.body = body;
-
- this.initialRandomSeed = Math.round(Math.random() * 1000000);
- this.randomSeed = this.initialRandomSeed;
- this.setPhysics = false;
- this.options = {};
- this.optionsBackup = {physics:{}};
-
- this.defaultOptions = {
- randomSeed: undefined,
- improvedLayout: true,
- hierarchical: {
- enabled:false,
- levelSeparation: 150,
- nodeSpacing: 100,
- treeSpacing: 200,
- blockShifting: true,
- edgeMinimization: true,
- parentCentralization: true,
- direction: 'UD', // UD, DU, LR, RL
- sortMethod: 'hubsize' // hubsize, directed
- }
- };
- util.extend(this.options, this.defaultOptions);
- this.bindEventListeners();
- }
-
- bindEventListeners() {
- this.body.emitter.on('_dataChanged', () => {
- this.setupHierarchicalLayout();
- });
- this.body.emitter.on('_dataLoaded', () => {
- this.layoutNetwork();
- });
- this.body.emitter.on('_resetHierarchicalLayout', () => {
- this.setupHierarchicalLayout();
- });
- this.body.emitter.on('_adjustEdgesForHierarchicalLayout', () => {
- if (this.options.hierarchical.enabled !== true) {
- return;
- }
- // get the type of static smooth curve in case it is required
- let type = this.getStaticType();
-
- // force all edges into static smooth curves.
- this.body.emitter.emit('_forceDisableDynamicCurves', type, false);
- });
- }
-
- setOptions(options, allOptions) {
- if (options !== undefined) {
- let hierarchical = this.options.hierarchical;
- let prevHierarchicalState = hierarchical.enabled;
- util.selectiveDeepExtend(["randomSeed", "improvedLayout"],this.options, options);
- util.mergeOptions(this.options, options, 'hierarchical');
- if (options.randomSeed !== undefined) {this.initialRandomSeed = options.randomSeed;}
-
- if (hierarchical.enabled === true) {
- if (prevHierarchicalState === true) {
- // refresh the overridden options for nodes and edges.
- this.body.emitter.emit('refresh', true);
- }
-
- // make sure the level separation is the right way up
- if (hierarchical.direction === 'RL' || hierarchical.direction === 'DU') {
- if (hierarchical.levelSeparation > 0) {
- hierarchical.levelSeparation *= -1;
- }
- }
- else {
- if (hierarchical.levelSeparation < 0) {
- hierarchical.levelSeparation *= -1;
- }
- }
-
- this.body.emitter.emit('_resetHierarchicalLayout');
- // because the hierarchical system needs it's own physics and smooth curve settings,
- // we adapt the other options if needed.
- return this.adaptAllOptionsForHierarchicalLayout(allOptions);
- }
- else {
- if (prevHierarchicalState === true) {
- // refresh the overridden options for nodes and edges.
- this.body.emitter.emit('refresh');
- return util.deepExtend(allOptions,this.optionsBackup);
- }
- }
- }
- return allOptions;
- }
-
- adaptAllOptionsForHierarchicalLayout(allOptions) {
- if (this.options.hierarchical.enabled === true) {
- let backupPhysics = this.optionsBackup.physics;
-
- // set the physics
- if (allOptions.physics === undefined || allOptions.physics === true) {
- allOptions.physics = {
- enabled: backupPhysics.enabled === undefined ? true : backupPhysics.enabled,
- solver :'hierarchicalRepulsion'
- };
- backupPhysics.enabled = backupPhysics.enabled === undefined ? true : backupPhysics.enabled;
- backupPhysics.solver = backupPhysics.solver || 'barnesHut';
- }
- else if (typeof allOptions.physics === 'object') {
- backupPhysics.enabled = allOptions.physics.enabled === undefined ? true : allOptions.physics.enabled;
- backupPhysics.solver = allOptions.physics.solver || 'barnesHut';
- allOptions.physics.solver = 'hierarchicalRepulsion';
- }
- else if (allOptions.physics !== false) {
- backupPhysics.solver ='barnesHut';
- allOptions.physics = {solver:'hierarchicalRepulsion'};
- }
-
- // get the type of static smooth curve in case it is required
- let type = this.getStaticType();
-
- // disable smooth curves if nothing is defined. If smooth curves have been turned on,
- // turn them into static smooth curves.
- if (allOptions.edges === undefined) {
- this.optionsBackup.edges = {smooth:{enabled:true, type:'dynamic'}};
- allOptions.edges = {smooth: false};
- }
- else if (allOptions.edges.smooth === undefined) {
- this.optionsBackup.edges = {smooth:{enabled:true, type:'dynamic'}};
- allOptions.edges.smooth = false;
- }
- else {
- if (typeof allOptions.edges.smooth === 'boolean') {
- this.optionsBackup.edges = {smooth:allOptions.edges.smooth};
- allOptions.edges.smooth = {enabled: allOptions.edges.smooth, type:type}
- }
- else {
- let smooth = allOptions.edges.smooth;
-
- // allow custom types except for dynamic
- if (smooth.type !== undefined && smooth.type !== 'dynamic') {
- type = smooth.type;
- }
-
- // TODO: this is options merging; see if the standard routines can be used here.
- this.optionsBackup.edges = {
- smooth : smooth.enabled === undefined ? true : smooth.enabled,
- type : smooth.type === undefined ? 'dynamic': smooth.type,
- roundness : smooth.roundness === undefined ? 0.5 : smooth.roundness,
- forceDirection: smooth.forceDirection === undefined ? false : smooth.forceDirection
- };
-
-
- // NOTE: Copying an object to self; this is basically setting defaults for undefined variables
- allOptions.edges.smooth = {
- enabled : smooth.enabled === undefined ? true : smooth.enabled,
- type : type,
- roundness : smooth.roundness === undefined ? 0.5 : smooth.roundness,
- forceDirection: smooth.forceDirection === undefined ? false: smooth.forceDirection
- }
- }
- }
-
- // Force all edges into static smooth curves.
- // Only applies to edges that do not use the global options for smooth.
- this.body.emitter.emit('_forceDisableDynamicCurves', type);
- }
-
- return allOptions;
- }
-
- seededRandom() {
- let x = Math.sin(this.randomSeed++) * 10000;
- return x - Math.floor(x);
- }
-
- positionInitially(nodesArray) {
- if (this.options.hierarchical.enabled !== true) {
- this.randomSeed = this.initialRandomSeed;
- let radius = nodesArray.length + 50;
- for (let i = 0; i < nodesArray.length; i++) {
- let node = nodesArray[i];
- let angle = 2 * Math.PI * this.seededRandom();
- if (node.x === undefined) {
- node.x = radius * Math.cos(angle);
- }
- if (node.y === undefined) {
- node.y = radius * Math.sin(angle);
- }
- }
- }
- }
-
-
- /**
- * Use Kamada Kawai to position nodes. This is quite a heavy algorithm so if there are a lot of nodes we
- * cluster them first to reduce the amount.
- */
- layoutNetwork() {
- if (this.options.hierarchical.enabled !== true && this.options.improvedLayout === true) {
- let indices = this.body.nodeIndices;
-
- // first check if we should Kamada Kawai to layout. The threshold is if less than half of the visible
- // nodes have predefined positions we use this.
- let positionDefined = 0;
- for (let i = 0; i < indices.length; i++) {
- let node = this.body.nodes[indices[i]];
- if (node.predefinedPosition === true) {
- positionDefined += 1;
- }
- }
-
- // if less than half of the nodes have a predefined position we continue
- if (positionDefined < 0.5 * indices.length) {
- let MAX_LEVELS = 10;
- let level = 0;
- let clusterThreshold = 150;
- // Performance enhancement, during clustering edges need only be simple straight lines.
- // These options don't propagate outside the clustering phase.
- let clusterOptions = {
- clusterEdgeProperties:{
- smooth: {
- enabled: false
- }
- }
- };
-
- // if there are a lot of nodes, we cluster before we run the algorithm.
- // NOTE: this part fails to find clusters for large scale-free networks, which should
- // be easily clusterable.
- // TODO: examine why this is so
- if (indices.length > clusterThreshold) {
- let startLength = indices.length;
- while (indices.length > clusterThreshold && level <= MAX_LEVELS) {
- //console.time("clustering")
- level += 1;
- let before = indices.length;
- // if there are many nodes we do a hubsize cluster
- if (level % 3 === 0) {
- this.body.modules.clustering.clusterBridges(clusterOptions);
- }
- else {
- this.body.modules.clustering.clusterOutliers(clusterOptions);
- }
- let after = indices.length;
- if (before == after && level % 3 !== 0) {
- this._declusterAll();
- this.body.emitter.emit("_layoutFailed");
- console.info("This network could not be positioned by this version of the improved layout algorithm."
- + " Please disable improvedLayout for better performance.");
- return;
- }
- //console.timeEnd("clustering")
- //console.log(before,level,after);
- }
- // increase the size of the edges
- this.body.modules.kamadaKawai.setOptions({springLength: Math.max(150, 2 * startLength)})
- }
- if (level > MAX_LEVELS){
- console.info("The clustering didn't succeed within the amount of interations allowed,"
- + " progressing with partial result.");
- }
-
- // position the system for these nodes and edges
- this.body.modules.kamadaKawai.solve(indices, this.body.edgeIndices, true);
-
- // shift to center point
- this._shiftToCenter();
-
- // perturb the nodes a little bit to force the physics to kick in
- let offset = 70;
- for (let i = 0; i < indices.length; i++) {
- // Only perturb the nodes that aren't fixed
- let node = this.body.nodes[indices[i]];
- if (node.predefinedPosition === false) {
- node.x += (0.5 - this.seededRandom())*offset;
- node.y += (0.5 - this.seededRandom())*offset;
- }
- }
-
- // uncluster all clusters
- this._declusterAll();
-
- // reposition all bezier nodes.
- this.body.emitter.emit("_repositionBezierNodes");
- }
- }
- }
-
- /**
- * Move all the nodes towards to the center so gravitational pull wil not move the nodes away from view
- * @private
- */
- _shiftToCenter() {
- let range = NetworkUtil.getRangeCore(this.body.nodes, this.body.nodeIndices);
- let center = NetworkUtil.findCenter(range);
- for (let i = 0; i < this.body.nodeIndices.length; i++) {
- let node = this.body.nodes[this.body.nodeIndices[i]];
- node.x -= center.x;
- node.y -= center.y;
- }
- }
-
- _declusterAll() {
- let clustersPresent = true;
- while (clustersPresent === true) {
- clustersPresent = false;
- for (let i = 0; i < this.body.nodeIndices.length; i++) {
- if (this.body.nodes[this.body.nodeIndices[i]].isCluster === true) {
- clustersPresent = true;
- this.body.modules.clustering.openCluster(this.body.nodeIndices[i], {}, false);
- }
- }
- if (clustersPresent === true) {
- this.body.emitter.emit('_dataChanged');
- }
- }
- }
-
- getSeed() {
- return this.initialRandomSeed;
- }
-
- /**
- * This is the main function to layout the nodes in a hierarchical way.
- * It checks if the node details are supplied correctly
- *
- * @private
- */
- setupHierarchicalLayout() {
- if (this.options.hierarchical.enabled === true && this.body.nodeIndices.length > 0) {
- // get the size of the largest hubs and check if the user has defined a level for a node.
- let node, nodeId;
- let definedLevel = false;
- let definedPositions = true;
- let undefinedLevel = false;
- this.lastNodeOnLevel = {};
- this.hierarchical = new HierarchicalStatus();
-
- for (nodeId in this.body.nodes) {
- if (this.body.nodes.hasOwnProperty(nodeId)) {
- node = this.body.nodes[nodeId];
- if (node.options.x === undefined && node.options.y === undefined) {
- definedPositions = false;
- }
- if (node.options.level !== undefined) {
- definedLevel = true;
- this.hierarchical.levels[nodeId] = node.options.level;
- }
- else {
- undefinedLevel = true;
- }
- }
- }
-
- // if the user defined some levels but not all, alert and run without hierarchical layout
- if (undefinedLevel === true && definedLevel === true) {
- throw new Error('To use the hierarchical layout, nodes require either no predefined levels'
- + ' or levels have to be defined for all nodes.');
- }
- else {
- // define levels if undefined by the users. Based on hubsize.
- if (undefinedLevel === true) {
- let sortMethod = this.options.hierarchical.sortMethod;
- if (sortMethod === 'hubsize') {
- this._determineLevelsByHubsize();
- }
- else if (sortMethod === 'directed') {
- this._determineLevelsDirected();
- }
- else if (sortMethod === 'custom') {
- this._determineLevelsCustomCallback();
- }
- }
-
-
- // fallback for cases where there are nodes but no edges
- for (let nodeId in this.body.nodes) {
- if (this.body.nodes.hasOwnProperty(nodeId)) {
- this.hierarchical.ensureLevel(nodeId);
- }
- }
- // check the distribution of the nodes per level.
- let distribution = this._getDistribution();
-
- // get the parent children relations.
- this._generateMap();
-
- // place the nodes on the canvas.
- this._placeNodesByHierarchy(distribution);
-
- // condense the whitespace.
- this._condenseHierarchy();
-
- // shift to center so gravity does not have to do much
- this._shiftToCenter();
- }
- }
- }
-
- /**
- * @private
- */
- _condenseHierarchy() {
- // Global var in this scope to define when the movement has stopped.
- let stillShifting = false;
- let branches = {};
- // first we have some methods to help shifting trees around.
- // the main method to shift the trees
- let shiftTrees = () => {
- let treeSizes = getTreeSizes();
- let shiftBy = 0;
- for (let i = 0; i < treeSizes.length - 1; i++) {
- let diff = treeSizes[i].max - treeSizes[i+1].min;
- shiftBy += diff + this.options.hierarchical.treeSpacing;
- shiftTree(i + 1, shiftBy);
- }
- };
-
- // shift a single tree by an offset
- let shiftTree = (index, offset) => {
- let trees = this.hierarchical.trees;
-
- for (let nodeId in trees) {
- if (trees.hasOwnProperty(nodeId)) {
- if (trees[nodeId] === index) {
- let node = this.body.nodes[nodeId];
- let pos = this._getPositionForHierarchy(node);
- this._setPositionForHierarchy(node, pos + offset, undefined, true);
- }
- }
- }
- };
-
- // get the width of a tree
- let getTreeSize = (index) => {
- let res = this.hierarchical.getTreeSize(this.body.nodes, index);
- if (this._isVertical()) {
- return {min: res.min_x, max: res.max_x};
- } else {
- return {min: res.min_y, max: res.max_y};
- }
- };
-
- // get the width of all trees
- let getTreeSizes = () => {
- let treeWidths = [];
- for (let i = 0; i <= this.hierarchical.numTrees(); i++) {
- treeWidths.push(getTreeSize(i));
- }
- return treeWidths;
- };
-
-
- // get a map of all nodes in this branch
- let getBranchNodes = (source, map) => {
- if (map[source.id]) {
- return;
- }
- map[source.id] = true;
- if (this.hierarchical.childrenReference[source.id]) {
- let children = this.hierarchical.childrenReference[source.id];
- if (children.length > 0) {
- for (let i = 0; i < children.length; i++) {
- getBranchNodes(this.body.nodes[children[i]], map);
- }
- }
- }
- };
-
- // get a min max width as well as the maximum movement space it has on either sides
- // we use min max terminology because width and height can interchange depending on the direction of the layout
- let getBranchBoundary = (branchMap, maxLevel = 1e9) => {
- let minSpace = 1e9;
- let maxSpace = 1e9;
- let min = 1e9;
- let max = -1e9;
- for (let branchNode in branchMap) {
- if (branchMap.hasOwnProperty(branchNode)) {
- let node = this.body.nodes[branchNode];
- let level = this.hierarchical.levels[node.id];
- let position = this._getPositionForHierarchy(node);
-
- // get the space around the node.
- let [minSpaceNode, maxSpaceNode] = this._getSpaceAroundNode(node,branchMap);
- minSpace = Math.min(minSpaceNode, minSpace);
- maxSpace = Math.min(maxSpaceNode, maxSpace);
-
- // the width is only relevant for the levels two nodes have in common. This is why we filter on this.
- if (level <= maxLevel) {
- min = Math.min(position, min);
- max = Math.max(position, max);
- }
- }
- }
-
- return [min, max, minSpace, maxSpace];
- };
-
-
- // check what the maximum level is these nodes have in common.
- let getCollisionLevel = (node1, node2) => {
- let maxLevel1 = this.hierarchical.getMaxLevel(node1.id);
- let maxLevel2 = this.hierarchical.getMaxLevel(node2.id);
- return Math.min(maxLevel1, maxLevel2);
- };
-
-
- /**
- * Condense elements. These can be nodes or branches depending on the callback.
- */
- let shiftElementsCloser = (callback, levels, centerParents) => {
- let hier = this.hierarchical;
-
- for (let i = 0; i < levels.length; i++) {
- let level = levels[i];
- let levelNodes = hier.distributionOrdering[level];
- if (levelNodes.length > 1) {
- for (let j = 0; j < levelNodes.length - 1; j++) {
- let node1 = levelNodes[j];
- let node2 = levelNodes[j+1];
-
- // NOTE: logic maintained as it was; if nodes have same ancestor,
- // then of course they are in the same sub-network.
- if (hier.hasSameParent(node1, node2) && hier.inSameSubNetwork(node1, node2) ) {
- callback(node1, node2, centerParents);
- }
- }
- }
- }
- };
-
-
- // callback for shifting branches
- let branchShiftCallback = (node1, node2, centerParent = false) => {
- //window.CALLBACKS.push(() => {
- let pos1 = this._getPositionForHierarchy(node1);
- let pos2 = this._getPositionForHierarchy(node2);
- let diffAbs = Math.abs(pos2 - pos1);
- let nodeSpacing = this.options.hierarchical.nodeSpacing;
- //console.log("NOW CHECKING:", node1.id, node2.id, diffAbs);
- if (diffAbs > nodeSpacing) {
- let branchNodes1 = {};
- let branchNodes2 = {};
-
- getBranchNodes(node1, branchNodes1);
- getBranchNodes(node2, branchNodes2);
-
- // check the largest distance between the branches
- let maxLevel = getCollisionLevel(node1, node2);
- let [min1,max1, minSpace1, maxSpace1] = getBranchBoundary(branchNodes1, maxLevel);
- let [min2,max2, minSpace2, maxSpace2] = getBranchBoundary(branchNodes2, maxLevel);
-
- //console.log(node1.id, getBranchBoundary(branchNodes1, maxLevel), node2.id,
- // getBranchBoundary(branchNodes2, maxLevel), maxLevel);
- let diffBranch = Math.abs(max1 - min2);
- if (diffBranch > nodeSpacing) {
- let offset = max1 - min2 + nodeSpacing;
- if (offset < -minSpace2 + nodeSpacing) {
- offset = -minSpace2 + nodeSpacing;
- //console.log("RESETTING OFFSET", max1 - min2 + this.options.hierarchical.nodeSpacing, -minSpace2, offset);
- }
- if (offset < 0) {
- //console.log("SHIFTING", node2.id, offset);
- this._shiftBlock(node2.id, offset);
- stillShifting = true;
-
- if (centerParent === true)
- this._centerParent(node2);
- }
- }
-
- }
- //this.body.emitter.emit("_redraw");})
- };
-
- let minimizeEdgeLength = (iterations, node) => {
- //window.CALLBACKS.push(() => {
- // console.log("ts",node.id);
- let nodeId = node.id;
- let allEdges = node.edges;
- let nodeLevel = this.hierarchical.levels[node.id];
-
- // gather constants
- let C2 = this.options.hierarchical.levelSeparation * this.options.hierarchical.levelSeparation;
- let referenceNodes = {};
- let aboveEdges = [];
- for (let i = 0; i < allEdges.length; i++) {
- let edge = allEdges[i];
- if (edge.toId != edge.fromId) {
- let otherNode = edge.toId == nodeId ? edge.from : edge.to;
- referenceNodes[allEdges[i].id] = otherNode;
- if (this.hierarchical.levels[otherNode.id] < nodeLevel) {
- aboveEdges.push(edge);
- }
- }
- }
-
- // differentiated sum of lengths based on only moving one node over one axis
- let getFx = (point, edges) => {
- let sum = 0;
- for (let i = 0; i < edges.length; i++) {
- if (referenceNodes[edges[i].id] !== undefined) {
- let a = this._getPositionForHierarchy(referenceNodes[edges[i].id]) - point;
- sum += a / Math.sqrt(a * a + C2);
- }
- }
- return sum;
- };
-
- // doubly differentiated sum of lengths based on only moving one node over one axis
- let getDFx = (point, edges) => {
- let sum = 0;
- for (let i = 0; i < edges.length; i++) {
- if (referenceNodes[edges[i].id] !== undefined) {
- let a = this._getPositionForHierarchy(referenceNodes[edges[i].id]) - point;
- sum -= (C2 * Math.pow(a * a + C2, -1.5));
- }
- }
- return sum;
- };
-
- let getGuess = (iterations, edges) => {
- let guess = this._getPositionForHierarchy(node);
- // Newton's method for optimization
- let guessMap = {};
- for (let i = 0; i < iterations; i++) {
- let fx = getFx(guess, edges);
- let dfx = getDFx(guess, edges);
-
- // we limit the movement to avoid instability.
- let limit = 40;
- let ratio = Math.max(-limit, Math.min(limit, Math.round(fx/dfx)));
- guess = guess - ratio;
- // reduce duplicates
- if (guessMap[guess] !== undefined) {
- break;
- }
- guessMap[guess] = i;
- }
- return guess;
- };
-
- let moveBranch = (guess) => {
- // position node if there is space
- let nodePosition = this._getPositionForHierarchy(node);
-
- // check movable area of the branch
- if (branches[node.id] === undefined) {
- let branchNodes = {};
- getBranchNodes(node, branchNodes);
- branches[node.id] = branchNodes;
- }
- let [minBranch, maxBranch, minSpaceBranch, maxSpaceBranch] = getBranchBoundary(branches[node.id]);
-
- let diff = guess - nodePosition;
-
- // check if we are allowed to move the node:
- let branchOffset = 0;
- if (diff > 0) {
- branchOffset = Math.min(diff, maxSpaceBranch - this.options.hierarchical.nodeSpacing);
- }
- else if (diff < 0) {
- branchOffset = -Math.min(-diff, minSpaceBranch - this.options.hierarchical.nodeSpacing);
- }
-
- if (branchOffset != 0) {
- //console.log("moving branch:",branchOffset, maxSpaceBranch, minSpaceBranch)
- this._shiftBlock(node.id, branchOffset);
- //this.body.emitter.emit("_redraw");
- stillShifting = true;
- }
- };
-
- let moveNode = (guess) => {
- let nodePosition = this._getPositionForHierarchy(node);
-
- // position node if there is space
- let [minSpace, maxSpace] = this._getSpaceAroundNode(node);
- let diff = guess - nodePosition;
- // check if we are allowed to move the node:
- let newPosition = nodePosition;
- if (diff > 0) {
- newPosition = Math.min(nodePosition + (maxSpace - this.options.hierarchical.nodeSpacing), guess);
- }
- else if (diff < 0) {
- newPosition = Math.max(nodePosition - (minSpace - this.options.hierarchical.nodeSpacing), guess);
- }
-
- if (newPosition !== nodePosition) {
- //console.log("moving Node:",diff, minSpace, maxSpace);
- this._setPositionForHierarchy(node, newPosition, undefined, true);
- //this.body.emitter.emit("_redraw");
- stillShifting = true;
- }
- };
-
- let guess = getGuess(iterations, aboveEdges);
- moveBranch(guess);
- guess = getGuess(iterations, allEdges);
- moveNode(guess);
- //})
- };
-
- // method to remove whitespace between branches. Because we do bottom up, we can center the parents.
- let minimizeEdgeLengthBottomUp = (iterations) => {
- let levels = this.hierarchical.getLevels();
- levels = levels.reverse();
- for (let i = 0; i < iterations; i++) {
- stillShifting = false;
- for (let j = 0; j < levels.length; j++) {
- let level = levels[j];
- let levelNodes = this.hierarchical.distributionOrdering[level];
- for (let k = 0; k < levelNodes.length; k++) {
- minimizeEdgeLength(1000, levelNodes[k]);
- }
- }
- if (stillShifting !== true) {
- //console.log("FINISHED minimizeEdgeLengthBottomUp IN " + i);
- break;
- }
- }
- };
-
- // method to remove whitespace between branches. Because we do bottom up, we can center the parents.
- let shiftBranchesCloserBottomUp = (iterations) => {
- let levels = this.hierarchical.getLevels();
- levels = levels.reverse();
- for (let i = 0; i < iterations; i++) {
- stillShifting = false;
- shiftElementsCloser(branchShiftCallback, levels, true);
- if (stillShifting !== true) {
- //console.log("FINISHED shiftBranchesCloserBottomUp IN " + (i+1));
- break;
- }
- }
- };
-
- // center all parents
- let centerAllParents = () => {
- for (let nodeId in this.body.nodes) {
- if (this.body.nodes.hasOwnProperty(nodeId))
- this._centerParent(this.body.nodes[nodeId]);
- }
- };
-
- // center all parents
- let centerAllParentsBottomUp = () => {
- let levels = this.hierarchical.getLevels();
- levels = levels.reverse();
- for (let i = 0; i < levels.length; i++) {
- let level = levels[i];
- let levelNodes = this.hierarchical.distributionOrdering[level];
- for (let j = 0; j < levelNodes.length; j++) {
- this._centerParent(levelNodes[j]);
- }
- }
- };
-
- // the actual work is done here.
- if (this.options.hierarchical.blockShifting === true) {
- shiftBranchesCloserBottomUp(5);
- centerAllParents();
- }
-
- // minimize edge length
- if (this.options.hierarchical.edgeMinimization === true) {
- minimizeEdgeLengthBottomUp(20);
- }
-
- if (this.options.hierarchical.parentCentralization === true) {
- centerAllParentsBottomUp()
- }
-
- shiftTrees();
- }
-
- /**
- * This gives the space around the node. IF a map is supplied, it will only check against nodes NOT in the map.
- * This is used to only get the distances to nodes outside of a branch.
- * @param node
- * @param map
- * @returns {*[]}
- * @private
- */
- _getSpaceAroundNode(node, map) {
- let useMap = true;
- if (map === undefined) {
- useMap = false;
- }
- let level = this.hierarchical.levels[node.id];
- if (level !== undefined) {
- let index = this.hierarchical.distributionIndex[node.id];
- let position = this._getPositionForHierarchy(node);
- let ordering = this.hierarchical.distributionOrdering[level];
- let minSpace = 1e9;
- let maxSpace = 1e9;
- if (index !== 0) {
- let prevNode = ordering[index - 1];
- if ((useMap === true && map[prevNode.id] === undefined) || useMap === false) {
- let prevPos = this._getPositionForHierarchy(prevNode);
- minSpace = position - prevPos;
- }
- }
-
- if (index != ordering.length - 1) {
- let nextNode = ordering[index + 1];
- if ((useMap === true && map[nextNode.id] === undefined) || useMap === false) {
- let nextPos = this._getPositionForHierarchy(nextNode);
- maxSpace = Math.min(maxSpace, nextPos - position);
- }
- }
-
- return [minSpace, maxSpace];
- }
- else {
- return [0, 0];
- }
- }
-
-
- /**
- * We use this method to center a parent node and check if it does not cross other nodes when it does.
- * @param node
- * @private
- */
- _centerParent(node) {
- if (this.hierarchical.parentReference[node.id]) {
- let parents = this.hierarchical.parentReference[node.id];
- for (var i = 0; i < parents.length; i++) {
- let parentId = parents[i];
- let parentNode = this.body.nodes[parentId];
- let children = this.hierarchical.childrenReference[parentId];
-
- if (children !== undefined) {
- // get the range of the children
- let newPosition = this._getCenterPosition(children);
-
- let position = this._getPositionForHierarchy(parentNode);
- let [minSpace, maxSpace] = this._getSpaceAroundNode(parentNode);
- let diff = position - newPosition;
- if ((diff < 0 && Math.abs(diff) < maxSpace - this.options.hierarchical.nodeSpacing) ||
- (diff > 0 && Math.abs(diff) < minSpace - this.options.hierarchical.nodeSpacing)) {
- this._setPositionForHierarchy(parentNode, newPosition, undefined, true);
- }
- }
- }
- }
- }
-
-
- /**
- * This function places the nodes on the canvas based on the hierarchial distribution.
- *
- * @param {Object} distribution | obtained by the function this._getDistribution()
- * @private
- */
- _placeNodesByHierarchy(distribution) {
- this.positionedNodes = {};
- // start placing all the level 0 nodes first. Then recursively position their branches.
- for (let level in distribution) {
- if (distribution.hasOwnProperty(level)) {
- // sort nodes in level by position:
- let nodeArray = Object.keys(distribution[level]);
- nodeArray = this._indexArrayToNodes(nodeArray);
- this._sortNodeArray(nodeArray);
- let handledNodeCount = 0;
-
- for (let i = 0; i < nodeArray.length; i++) {
- let node = nodeArray[i];
- if (this.positionedNodes[node.id] === undefined) {
- let spacing = this.options.hierarchical.nodeSpacing;
- let pos = spacing * handledNodeCount;
- // We get the X or Y values we need and store them in pos and previousPos.
- // The get and set make sure we get X or Y
- if (handledNodeCount > 0) {
- pos = this._getPositionForHierarchy(nodeArray[i-1]) + spacing;
- }
- this._setPositionForHierarchy(node, pos, level);
- this._validatePositionAndContinue(node, level, pos);
-
- handledNodeCount++;
- }
- }
- }
- }
- }
-
-
- /**
- * This is a recursively called function to enumerate the branches from the largest hubs and place the nodes
- * on a X position that ensures there will be no overlap.
- *
- * @param parentId
- * @param parentLevel
- * @private
- */
- _placeBranchNodes(parentId, parentLevel) {
- let childRef = this.hierarchical.childrenReference[parentId];
-
- // if this is not a parent, cancel the placing. This can happen with multiple parents to one child.
- if (childRef === undefined) {
- return;
- }
-
- // get a list of childNodes
- let childNodes = [];
- for (let i = 0; i < childRef.length; i++) {
- childNodes.push(this.body.nodes[childRef[i]]);
- }
-
- // use the positions to order the nodes.
- this._sortNodeArray(childNodes);
-
- // position the childNodes
- for (let i = 0; i < childNodes.length; i++) {
- let childNode = childNodes[i];
- let childNodeLevel = this.hierarchical.levels[childNode.id];
- // check if the child node is below the parent node and if it has already been positioned.
- if (childNodeLevel > parentLevel && this.positionedNodes[childNode.id] === undefined) {
- // get the amount of space required for this node. If parent the width is based on the amount of children.
- let spacing = this.options.hierarchical.nodeSpacing;
- let pos;
-
- // we get the X or Y values we need and store them in pos and previousPos.
- // The get and set make sure we get X or Y
- if (i === 0) {pos = this._getPositionForHierarchy(this.body.nodes[parentId]);}
- else {pos = this._getPositionForHierarchy(childNodes[i-1]) + spacing;}
- this._setPositionForHierarchy(childNode, pos, childNodeLevel);
- this._validatePositionAndContinue(childNode, childNodeLevel, pos);
- }
- else {
- return;
- }
- }
-
- // center the parent nodes.
- let center = this._getCenterPosition(childNodes);
- this._setPositionForHierarchy(this.body.nodes[parentId], center, parentLevel);
- }
-
-
- /**
- * This method checks for overlap and if required shifts the branch. It also keeps records of positioned nodes.
- * Finally it will call _placeBranchNodes to place the branch nodes.
- * @param node
- * @param level
- * @param pos
- * @private
- */
- _validatePositionAndContinue(node, level, pos) {
- // This method only works for formal trees and formal forests
- // Early exit if this is not the case
- if (!this.hierarchical.isTree) return;
-
- // if overlap has been detected, we shift the branch
- if (this.lastNodeOnLevel[level] !== undefined) {
- let previousPos = this._getPositionForHierarchy(this.body.nodes[this.lastNodeOnLevel[level]]);
- if (pos - previousPos < this.options.hierarchical.nodeSpacing) {
- let diff = (previousPos + this.options.hierarchical.nodeSpacing) - pos;
- let sharedParent = this._findCommonParent(this.lastNodeOnLevel[level], node.id);
- this._shiftBlock(sharedParent.withChild, diff);
- }
- }
-
- this.lastNodeOnLevel[level] = node.id; // store change in position.
- this.positionedNodes[node.id] = true;
- this._placeBranchNodes(node.id, level);
- }
-
- /**
- * Receives an array with node indices and returns an array with the actual node references.
- * Used for sorting based on node properties.
- * @param idArray
- */
- _indexArrayToNodes(idArray) {
- let array = [];
- for (let i = 0; i < idArray.length; i++) {
- array.push(this.body.nodes[idArray[i]])
- }
- return array;
- }
-
- /**
- * This function get the distribution of levels based on hubsize
- *
- * @returns {Object}
- * @private
- */
- _getDistribution() {
- let distribution = {};
- let nodeId, node;
-
- // we fix Y because the hierarchy is vertical,
- // we fix X so we do not give a node an x position for a second time.
- // the fix of X is removed after the x value has been set.
- for (nodeId in this.body.nodes) {
- if (this.body.nodes.hasOwnProperty(nodeId)) {
- node = this.body.nodes[nodeId];
- let level = this.hierarchical.levels[nodeId] === undefined ? 0 : this.hierarchical.levels[nodeId];
- if(this._isVertical()) {
- node.y = this.options.hierarchical.levelSeparation * level;
- node.options.fixed.y = true;
- }
- else {
- node.x = this.options.hierarchical.levelSeparation * level;
- node.options.fixed.x = true;
- }
- if (distribution[level] === undefined) {
- distribution[level] = {};
- }
- distribution[level][nodeId] = node;
- }
- }
- return distribution;
- }
-
-
- /**
- * Return the active (i.e. visible) edges for this node
- *
- * @returns {array} Array of edge instances
- * @private
- */
- _getActiveEdges(node) {
- let result = [];
-
- for (let j in node.edges) {
- let edge = node.edges[j];
- if (this.body.edgeIndices.indexOf(edge.id) !== -1) {
- result.push(edge);
- }
- }
-
- return result;
- }
-
-
- /**
- * Get the hubsizes for all active nodes.
- *
- * @returns {number}
- * @private
- */
- _getHubSizes() {
- let hubSizes = {};
- let nodeIds = this.body.nodeIndices;
-
- for (let i in nodeIds) {
- let nodeId = nodeIds[i];
- let node = this.body.nodes[nodeId];
- let hubSize = this._getActiveEdges(node).length;
- hubSizes[hubSize] = true;
- }
-
- // Make an array of the size sorted descending
- let result = [];
- for (let size in hubSizes) {
- result.push(Number(size));
- }
- result.sort(function(a, b) {
- return b - a;
- });
-
- return result;
- }
-
-
- /**
- * this function allocates nodes in levels based on the recursive branching from the largest hubs.
- *
- * @private
- */
- _determineLevelsByHubsize() {
- let levelDownstream = (nodeA, nodeB) => {
- this.hierarchical.levelDownstream(nodeA, nodeB);
- }
-
- let hubSizes = this._getHubSizes();
-
- for (let i = 0; i < hubSizes.length; ++i ) {
- let hubSize = hubSizes[i];
- if (hubSize === 0) break;
-
- let nodeIds = this.body.nodeIndices;
- for (let j in nodeIds) {
- let nodeId = nodeIds[j];
- let node = this.body.nodes[nodeId];
-
- if (hubSize === this._getActiveEdges(node).length) {
- this._crawlNetwork(levelDownstream, nodeId);
- }
- }
- }
- }
-
-
- /**
- * TODO: release feature
- * TODO: Determine if this feature is needed at all
- *
- * @private
- */
- _determineLevelsCustomCallback() {
- let minLevel = 100000;
-
- // TODO: this should come from options.
- let customCallback = function(nodeA, nodeB, edge) {
-
- };
-
- // TODO: perhaps move to HierarchicalStatus.
- // But I currently don't see the point, this method is not used.
- let levelByDirection = (nodeA, nodeB, edge) => {
- let levelA = this.hierarchical.levels[nodeA.id];
- // set initial level
- if (levelA === undefined) { levelA = this.hierarchical.levels[nodeA.id] = minLevel;}
-
- let diff = customCallback(
- NetworkUtil.cloneOptions(nodeA,'node'),
- NetworkUtil.cloneOptions(nodeB,'node'),
- NetworkUtil.cloneOptions(edge,'edge')
- );
-
- this.hierarchical.levels[nodeB.id] = levelA + diff;
- };
-
- this._crawlNetwork(levelByDirection);
- this.hierarchical.setMinLevelToZero(this.body.nodes);
- }
-
- /**
- * Allocate nodes in levels based on the direction of the edges.
- *
- * @param hubsize
- * @private
- */
- _determineLevelsDirected() {
- let minLevel = 10000;
-
- /**
- * Check if there is an edge going the opposite direction for given edge
- */
- let self = this;
- let isBidirectional = (edge) => {
- for (let key in self.body.edges) {
- let otherEdge = self.body.edges[key];
- if (otherEdge.toId === edge.fromId && otherEdge.fromId === edge.toId) {
- return true;
- }
- }
-
- return false;
- };
-
- let levelByDirection = (nodeA, nodeB, edge) => {
- let levelA = this.hierarchical.levels[nodeA.id];
- let levelB = this.hierarchical.levels[nodeB.id];
-
- if (isBidirectional(edge) && levelA !== undefined && levelB !== undefined) {
- // Don't redo the level determination if already done in this case.
- return;
- }
-
- // set initial level
- if (levelA === undefined) { levelA = this.hierarchical.levels[nodeA.id] = minLevel;}
- if (edge.toId == nodeB.id) {
- this.hierarchical.levels[nodeB.id] = levelA + 1;
- }
- else {
- this.hierarchical.levels[nodeB.id] = levelA - 1;
- }
- };
-
- this._crawlNetwork(levelByDirection);
- this.hierarchical.setMinLevelToZero(this.body.nodes);
- }
-
-
- /**
- * Update the bookkeeping of parent and child.
- * @private
- */
- _generateMap() {
- let fillInRelations = (parentNode, childNode) => {
- if (this.hierarchical.levels[childNode.id] > this.hierarchical.levels[parentNode.id]) {
- this.hierarchical.addRelation(parentNode.id, childNode.id);
- }
- };
-
- this._crawlNetwork(fillInRelations);
- this.hierarchical.checkIfTree();
- }
-
-
- /**
- * Crawl over the entire network and use a callback on each node couple that is connected to each other.
- * @param callback | will receive nodeA, nodeB and the connecting edge. A and B are distinct.
- * @param startingNodeId
- * @private
- */
- _crawlNetwork(callback = function() {}, startingNodeId) {
- let progress = {};
-
- let crawler = (node, tree) => {
- if (progress[node.id] === undefined) {
- this.hierarchical.setTreeIndex(node, tree);
-
- progress[node.id] = true;
- let childNode;
- let edges = this._getActiveEdges(node);
- for (let i = 0; i < edges.length; i++) {
- let edge = edges[i];
- if (edge.connected === true) {
- if (edge.toId == node.id) { // Not '===' because id's can be string and numeric
- childNode = edge.from;
- }
- else {
- childNode = edge.to;
- }
-
- if (node.id != childNode.id) { // Not '!==' because id's can be string and numeric
- callback(node, childNode, edge);
- crawler(childNode, tree);
- }
- }
- }
- }
- };
-
-
- if (startingNodeId === undefined) {
- // Crawl over all nodes
- let treeIndex = 0; // Serves to pass a unique id for the current distinct tree
-
- for (let i = 0; i < this.body.nodeIndices.length; i++) {
- let nodeId = this.body.nodeIndices[i];
-
- if (progress[nodeId] === undefined) {
- let node = this.body.nodes[nodeId];
- crawler(node, treeIndex);
- treeIndex += 1;
- }
- }
- }
- else {
- // Crawl from the given starting node
- let node = this.body.nodes[startingNodeId];
- if (node === undefined) {
- console.error("Node not found:", startingNodeId);
- return;
- }
- crawler(node);
- }
- }
-
-
- /**
- * Shift a branch a certain distance
- * @param parentId
- * @param diff
- * @private
- */
- _shiftBlock(parentId, diff) {
- let progress = {};
- let shifter = (parentId) => {
- if (progress[parentId]) {
- return;
- }
- progress[parentId] = true;
- if(this._isVertical()) {
- this.body.nodes[parentId].x += diff;
- }
- else {
- this.body.nodes[parentId].y += diff;
- }
-
- let childRef = this.hierarchical.childrenReference[parentId];
- if (childRef !== undefined) {
- for (let i = 0; i < childRef.length; i++) {
- shifter(childRef[i]);
- }
- }
- };
- shifter(parentId);
- }
-
-
- /**
- * Find a common parent between branches.
- * @param childA
- * @param childB
- * @returns {{foundParent, withChild}}
- * @private
- */
- _findCommonParent(childA,childB) {
- let parents = {};
- let iterateParents = (parents,child) => {
- let parentRef = this.hierarchical.parentReference[child];
- if (parentRef !== undefined) {
- for (let i = 0; i < parentRef.length; i++) {
- let parent = parentRef[i];
- parents[parent] = true;
- iterateParents(parents, parent)
- }
- }
- };
- let findParent = (parents, child) => {
- let parentRef = this.hierarchical.parentReference[child];
- if (parentRef !== undefined) {
- for (let i = 0; i < parentRef.length; i++) {
- let parent = parentRef[i];
- if (parents[parent] !== undefined) {
- return {foundParent:parent, withChild:child};
- }
- let branch = findParent(parents, parent);
- if (branch.foundParent !== null) {
- return branch;
- }
- }
- }
- return {foundParent:null, withChild:child};
- };
-
- iterateParents(parents, childA);
- return findParent(parents, childB);
- }
-
- /**
- * Abstract the getting of the position so we won't have to repeat the check for direction all the time
- * @param node
- * @param position
- * @param level
- * @private
- */
- _setPositionForHierarchy(node, position, level, doNotUpdate = false) {
- //console.log('_setPositionForHierarchy',node.id, position)
- if (doNotUpdate !== true) {
- this.hierarchical.addToOrdering(node, level);
- }
-
- if(this._isVertical()) {
- node.x = position;
- }
- else {
- node.y = position;
- }
- }
-
-
- /**
- * Utility function to cut down on typing this all the time.
- *
- * TODO: use this in all applicable situations in this class.
- *
- * @private
- */
- _isVertical() {
- return (this.options.hierarchical.direction === 'UD' || this.options.hierarchical.direction === 'DU');
- }
-
- /**
- * Abstract the getting of the position of a node so we do not have to repeat the direction check all the time.
- * @param node
- * @returns {number|*}
- * @private
- */
- _getPositionForHierarchy(node) {
- if(this._isVertical()) {
- return node.x;
- }
- else {
- return node.y;
- }
- }
-
- /**
- * Use the x or y value to sort the array, allowing users to specify order.
- * @param nodeArray
- * @private
- */
- _sortNodeArray(nodeArray) {
- if (nodeArray.length > 1) {
- if(this._isVertical()) {
- nodeArray.sort(function (a, b) {
- return a.x - b.x;
- })
- }
- else {
- nodeArray.sort(function (a, b) {
- return a.y - b.y;
- })
- }
- }
- }
-
-
- /**
- * Get the type of static smooth curve in case it is required.
- *
- * The return value is the type to use to translate dynamic curves to
- * another type, in the case of hierarchical layout. Dynamic curves do
- * not work for that layout type.
- */
- getStaticType() {
- // Node that 'type' is the edge type, and therefore 'orthogonal' to the layout type.
- let type = 'horizontal';
- if (!this._isVertical()) {
- type = 'vertical';
- }
-
- return type;
- }
-
-
- /**
- * Determine the center position of a branch from the passed list of child nodes
- *
- * This takes into account the positions of all the child nodes.
- * @param childNodes {array} Array of either child nodes or node id's
- * @return {number}
- * @private
- */
- _getCenterPosition(childNodes) {
- let minPos = 1e9;
- let maxPos = -1e9;
-
- for (let i = 0; i < childNodes.length; i++) {
- let childNode;
- if (childNodes[i].id !== undefined) {
- childNode = childNodes[i];
- } else {
- let childNodeId = childNodes[i];
- childNode = this.body.nodes[childNodeId];
- }
-
- let position = this._getPositionForHierarchy(childNode);
- minPos = Math.min(minPos, position);
- maxPos = Math.max(maxPos, position);
- }
-
- return 0.5 * (minPos + maxPos);
- }
- }
-
- export default LayoutEngine;
|