highcharts
Version:
JavaScript charting framework
144 lines (143 loc) • 3.48 kB
JavaScript
/* *
*
* Networkgraph series
*
* (c) 2010-2025 Paweł Fus
*
* License: www.highcharts.com/license
*
* !!!!!!! SOURCE GETS TRANSPILED BY TYPESCRIPT. EDIT TS FILE ONLY. !!!!!!!
*
* */
'use strict';
import QuadTreeNode from './QuadTreeNode.js';
/* *
*
* Class
*
* */
/**
* 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
*/
class QuadTree {
/* *
*
* Constructor
*
* */
constructor(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);
this.root.isInternal = true;
this.root.isRoot = true;
this.root.divideBox();
}
/* *
*
* Functions
*
* */
/**
* Calculate mass of the each QuadNode in the tree.
*/
calculateMassAndCenter() {
this.visitNodeRecursive(null, null, function (node) {
node.updateMassAndCenter();
});
}
/**
* Insert nodes into the QuadTree
*
* @param {Array<Highcharts.Point>} points
* Points as nodes
*/
insertNodes(points) {
for (const point of points) {
this.root.insert(point, this.maxDepth);
}
}
/**
* Depth first treversal (DFS). Using `before` and `after` callbacks,
* we can get two results: preorder and postorder traversals, reminder:
*
* ```
* (a)
* / \
* (b) (c)
* / \
* (d) (e)
* ```
*
* DFS (preorder): `a -> b -> d -> e -> c`
*
* DFS (postorder): `d -> e -> b -> c -> a`
*
* @param {Highcharts.QuadTreeNode|null} node
* QuadTree node
* @param {Function} [beforeCallback]
* Function to be called before visiting children nodes.
* @param {Function} [afterCallback]
* Function to be called after visiting children nodes.
*/
visitNodeRecursive(node, beforeCallback, afterCallback) {
let goFurther;
if (!node) {
node = this.root;
}
if (node === this.root && beforeCallback) {
goFurther = beforeCallback(node);
}
if (goFurther === false) {
return;
}
for (const qtNode of node.nodes) {
if (qtNode.isInternal) {
if (beforeCallback) {
goFurther = beforeCallback(qtNode);
}
if (goFurther === false) {
continue;
}
this.visitNodeRecursive(qtNode, beforeCallback, afterCallback);
}
else if (qtNode.body) {
if (beforeCallback) {
beforeCallback(qtNode.body);
}
}
if (afterCallback) {
afterCallback(qtNode);
}
}
if (node === this.root && afterCallback) {
afterCallback(node);
}
}
}
/* *
*
* Default Export
*
* */
export default QuadTree;