415 lines
10 KiB
JavaScript
415 lines
10 KiB
JavaScript
/* *
|
|
* Networkgraph series
|
|
*
|
|
* (c) 2010-2019 Paweł Fus
|
|
*
|
|
* License: www.highcharts.com/license
|
|
*/
|
|
|
|
'use strict';
|
|
import H from '../../parts/Globals.js';
|
|
|
|
/**
|
|
* The QuadTree node class. Used in Networkgraph chart as a base for Barnes-Hut
|
|
* approximation.
|
|
*
|
|
* @private
|
|
* @class
|
|
* @name Highcharts.QuadTreeNode
|
|
*
|
|
* @param {Highcharts.RectangleObject} Available space for the node
|
|
*/
|
|
var QuadTreeNode = H.QuadTreeNode = function (box) {
|
|
/**
|
|
* Read only. The available space for node.
|
|
*
|
|
* @name Highcharts.QuadTreeNode#box
|
|
* @type {Highcharts.RectangleObject}
|
|
*/
|
|
this.box = box;
|
|
/**
|
|
* Read only. The minium of width and height values.
|
|
*
|
|
* @name Highcharts.QuadTreeNode#boxSize
|
|
* @type {number}
|
|
*/
|
|
this.boxSize = Math.min(box.width, box.height);
|
|
/**
|
|
* Read only. Array of subnodes. Empty if QuadTreeNode has just one Point.
|
|
* When added another Point to this QuadTreeNode, array is filled with four
|
|
* subnodes.
|
|
*
|
|
* @name Highcharts.QuadTreeNode#nodes
|
|
* @type {Array<Highcharts.QuadTreeNode>}
|
|
*/
|
|
this.nodes = [];
|
|
/**
|
|
* Read only. Flag to determine if QuadTreeNode is internal (and has
|
|
* subnodes with mass and central position) or external (bound to Point).
|
|
*
|
|
* @name Highcharts.QuadTreeNode#isInternal
|
|
* @type {boolean}
|
|
*/
|
|
this.isInternal = false;
|
|
/**
|
|
* Read only. If QuadTreeNode is an external node, Point is stored in
|
|
* `this.body`.
|
|
*
|
|
* @name Highcharts.QuadTreeNode#body
|
|
* @type {boolean|Highcharts.Point}
|
|
*/
|
|
this.body = false;
|
|
/**
|
|
* Read only. Internal nodes when created are empty to reserve the space. If
|
|
* Point is added to this QuadTreeNode, QuadTreeNode is no longer empty.
|
|
*
|
|
* @name Highcharts.QuadTreeNode#isEmpty
|
|
* @type {boolean}
|
|
*/
|
|
this.isEmpty = true;
|
|
};
|
|
|
|
H.extend(
|
|
QuadTreeNode.prototype,
|
|
/** @lends Highcharts.QuadTreeNode.prototype */
|
|
{
|
|
/**
|
|
* Insert recursively point(node) into the QuadTree. If the given
|
|
* quadrant is already occupied, divide it into smaller quadrants.
|
|
*
|
|
* @param {Highcharts.Point} point point/node to be inserted
|
|
* @param {number} depth max depth of the QuadTree
|
|
*/
|
|
insert: function (point, depth) {
|
|
if (this.isInternal) {
|
|
// Internal node:
|
|
this.nodes[this.getBoxPosition(point)].insert(point, depth - 1);
|
|
} else {
|
|
this.isEmpty = false;
|
|
|
|
if (!this.body) {
|
|
// First body in a quadrant:
|
|
this.isInternal = false;
|
|
this.body = point;
|
|
} else {
|
|
if (depth) {
|
|
// Every other body in a quadrant:
|
|
this.isInternal = true;
|
|
this.divideBox();
|
|
// Reinsert main body only once:
|
|
if (this.body !== true) {
|
|
this.nodes[this.getBoxPosition(this.body)]
|
|
.insert(this.body, depth - 1);
|
|
this.body = true;
|
|
}
|
|
// Add second body:
|
|
this.nodes[this.getBoxPosition(point)]
|
|
.insert(point, depth - 1);
|
|
} else {
|
|
this.nodes.push(point);
|
|
}
|
|
|
|
}
|
|
}
|
|
},
|
|
/**
|
|
* Each quad node requires it's mass and center position. That mass and
|
|
* position is used to imitate real node in the layout by approximation.
|
|
*/
|
|
updateMassAndCenter: function () {
|
|
var mass = 0,
|
|
plotX = 0,
|
|
plotY = 0;
|
|
|
|
if (this.isInternal) {
|
|
// Calcualte weightened mass of the quad node:
|
|
this.nodes.forEach(function (pointMass) {
|
|
if (!pointMass.isEmpty) {
|
|
mass += pointMass.mass;
|
|
plotX += pointMass.plotX * pointMass.mass;
|
|
plotY += pointMass.plotY * pointMass.mass;
|
|
}
|
|
});
|
|
plotX /= mass;
|
|
plotY /= mass;
|
|
} else if (this.body) {
|
|
// Just one node, use coordinates directly:
|
|
mass = this.body.mass;
|
|
plotX = this.body.plotX;
|
|
plotY = this.body.plotY;
|
|
}
|
|
|
|
// Store details:
|
|
this.mass = mass;
|
|
this.plotX = plotX;
|
|
this.plotY = plotY;
|
|
},
|
|
/**
|
|
* When inserting another node into the box, that already hove one node,
|
|
* divide the available space into another four quadrants.
|
|
*
|
|
* Indexes of quadrants are:
|
|
*
|
|
* <pre>
|
|
* ------------- -------------
|
|
* | | | | |
|
|
* | | | 0 | 1 |
|
|
* | | divide() | | |
|
|
* | 1 | -----------> -------------
|
|
* | | | | |
|
|
* | | | 3 | 2 |
|
|
* | | | | |
|
|
* ------------- -------------
|
|
* </pre>
|
|
*/
|
|
divideBox: function () {
|
|
var halfWidth = this.box.width / 2,
|
|
halfHeight = this.box.height / 2;
|
|
|
|
// Top left
|
|
this.nodes[0] = new QuadTreeNode({
|
|
left: this.box.left,
|
|
top: this.box.top,
|
|
width: halfWidth,
|
|
height: halfHeight
|
|
});
|
|
|
|
// Top right
|
|
this.nodes[1] = new QuadTreeNode({
|
|
left: this.box.left + halfWidth,
|
|
top: this.box.top,
|
|
width: halfWidth,
|
|
height: halfHeight
|
|
});
|
|
|
|
// Bottom right
|
|
this.nodes[2] = new QuadTreeNode({
|
|
left: this.box.left + halfWidth,
|
|
top: this.box.top + halfHeight,
|
|
width: halfWidth,
|
|
height: halfHeight
|
|
});
|
|
|
|
// Bottom left
|
|
this.nodes[3] = new QuadTreeNode({
|
|
left: this.box.left,
|
|
top: this.box.top + halfHeight,
|
|
width: halfWidth,
|
|
height: halfHeight
|
|
});
|
|
},
|
|
/**
|
|
* Determine which of the quadrants should be used when placing node in
|
|
* the QuadTree. Returned index is always in range `<0, 3>`.
|
|
*
|
|
* @param {Highcharts.Point} node
|
|
* @return {number}
|
|
*/
|
|
getBoxPosition: function (node) {
|
|
var left = node.plotX < this.box.left + this.box.width / 2,
|
|
top = node.plotY < this.box.top + this.box.height / 2,
|
|
index;
|
|
|
|
if (left) {
|
|
if (top) {
|
|
// Top left
|
|
index = 0;
|
|
} else {
|
|
// Bottom left
|
|
index = 3;
|
|
}
|
|
} else {
|
|
if (top) {
|
|
// Top right
|
|
index = 1;
|
|
} else {
|
|
// Bottom right
|
|
index = 2;
|
|
}
|
|
}
|
|
|
|
return index;
|
|
}
|
|
}
|
|
);
|
|
/**
|
|
* The QuadTree class. Used in Networkgraph chart as a base for Barnes-Hut
|
|
* approximation.
|
|
*
|
|
* @private
|
|
* @class
|
|
* @name Highcharts.QuadTree
|
|
*
|
|
* @param {number} x left position of the plotting area
|
|
* @param {number} y top position of the plotting area
|
|
* @param {number} width width of the plotting area
|
|
* @param {number} height height of the plotting area
|
|
*/
|
|
var QuadTree = H.QuadTree = function (x, y, width, height) {
|
|
// Boundary rectangle:
|
|
this.box = {
|
|
left: x,
|
|
top: y,
|
|
width: width,
|
|
height: height
|
|
};
|
|
|
|
this.maxDepth = 25;
|
|
|
|
this.root = new QuadTreeNode(this.box, '0');
|
|
|
|
this.root.isInternal = true;
|
|
this.root.isRoot = true;
|
|
this.root.divideBox();
|
|
};
|
|
|
|
|
|
H.extend(
|
|
QuadTree.prototype,
|
|
/** @lends Highcharts.QuadTree.prototype */
|
|
{
|
|
/**
|
|
* Insert nodes into the QuadTree
|
|
*
|
|
* @param {Array<Highcharts.Point>} points
|
|
*/
|
|
insertNodes: function (nodes) {
|
|
nodes.forEach(function (node) {
|
|
this.root.insert(node, this.maxDepth);
|
|
}, this);
|
|
},
|
|
/**
|
|
* Depfth first treversal (DFS). Using `before` and `after` callbacks,
|
|
* we can get two results: preorder and postorder traversals, reminder:
|
|
*
|
|
* <pre>
|
|
* (a)
|
|
* / \
|
|
* (b) (c)
|
|
* / \
|
|
* (d) (e)
|
|
* </pre>
|
|
*
|
|
* DFS (preorder): `a -> b -> d -> e -> c`
|
|
*
|
|
* DFS (postorder): `d -> e -> b -> c -> a`
|
|
*
|
|
* @param {Highcharts.QuadTreeNode} node
|
|
* @param {function} beforeCallback function to be called before
|
|
* visiting children nodes
|
|
* @param {function} afterCallback function to be called after
|
|
* visiting children nodes
|
|
*/
|
|
visitNodeRecursive: function (
|
|
node,
|
|
beforeCallback,
|
|
afterCallback,
|
|
chart,
|
|
clear
|
|
) {
|
|
var goFurther;
|
|
|
|
if (!node) {
|
|
node = this.root;
|
|
}
|
|
|
|
if (node === this.root && beforeCallback) {
|
|
goFurther = beforeCallback(node);
|
|
}
|
|
|
|
if (goFurther === false) {
|
|
return;
|
|
}
|
|
|
|
node.nodes.forEach(
|
|
function (qtNode) {
|
|
if (chart) {
|
|
// this.renderBox(qtNode, chart, clear);
|
|
}
|
|
if (qtNode.isInternal) {
|
|
if (beforeCallback) {
|
|
goFurther = beforeCallback(qtNode);
|
|
}
|
|
if (goFurther === false) {
|
|
return;
|
|
}
|
|
this.visitNodeRecursive(
|
|
qtNode,
|
|
beforeCallback,
|
|
afterCallback,
|
|
chart,
|
|
clear
|
|
);
|
|
} else if (qtNode.body) {
|
|
if (beforeCallback) {
|
|
beforeCallback(qtNode.body);
|
|
}
|
|
}
|
|
if (afterCallback) {
|
|
afterCallback(qtNode);
|
|
}
|
|
},
|
|
this
|
|
);
|
|
if (node === this.root && afterCallback) {
|
|
afterCallback(node);
|
|
}
|
|
},
|
|
/**
|
|
* Calculate mass of the each QuadNode in the tree.
|
|
*/
|
|
calculateMassAndCenter: function () {
|
|
this.visitNodeRecursive(null, null, function (node) {
|
|
node.updateMassAndCenter();
|
|
});
|
|
},
|
|
render: function (chart, clear) {
|
|
this.visitNodeRecursive(this.root, null, null, chart, clear);
|
|
},
|
|
clear: function (chart) {
|
|
this.render(chart, true);
|
|
},
|
|
renderBox: function (qtNode, chart, clear) {
|
|
if (!qtNode.graphic && !clear) {
|
|
qtNode.graphic = chart.renderer
|
|
.rect(
|
|
qtNode.box.left + chart.plotLeft,
|
|
qtNode.box.top + chart.plotTop,
|
|
qtNode.box.width,
|
|
qtNode.box.height
|
|
)
|
|
.attr({
|
|
stroke: 'rgba(100, 100, 100, 0.5)',
|
|
'stroke-width': 2
|
|
})
|
|
.add();
|
|
|
|
if (!isNaN(qtNode.plotX)) {
|
|
qtNode.graphic2 = chart.renderer
|
|
.circle(
|
|
qtNode.plotX,
|
|
qtNode.plotY,
|
|
qtNode.mass / 10
|
|
)
|
|
.attr({
|
|
fill: 'red',
|
|
translateY: chart.plotTop,
|
|
translateX: chart.plotLeft
|
|
})
|
|
.add();
|
|
}
|
|
} else if (clear) {
|
|
if (qtNode.graphic) {
|
|
qtNode.graphic = qtNode.graphic.destroy();
|
|
}
|
|
if (qtNode.graphic2) {
|
|
qtNode.graphic2 = qtNode.graphic2.destroy();
|
|
}
|
|
if (qtNode.label) {
|
|
qtNode.label = qtNode.label.destroy();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
);
|