You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.1 KiB
39 lines
1.1 KiB
import CentralGravitySolver from "./CentralGravitySolver"
|
|
|
|
/**
|
|
* @class ForceAtlas2BasedCentralGravitySolver
|
|
* @extends CentralGravitySolver
|
|
*/
|
|
class ForceAtlas2BasedCentralGravitySolver extends CentralGravitySolver {
|
|
/**
|
|
* @param {Object} body
|
|
* @param {{physicsNodeIndices: Array, physicsEdgeIndices: Array, forces: {}, velocities: {}}} physicsBody
|
|
* @param {Object} options
|
|
* @constructor ForceAtlas2BasedCentralGravitySolver
|
|
*/
|
|
constructor(body, physicsBody, options) {
|
|
super(body, physicsBody, options);
|
|
}
|
|
|
|
|
|
/**
|
|
* Calculate the forces based on the distance.
|
|
*
|
|
* @param {number} distance
|
|
* @param {number} dx
|
|
* @param {number} dy
|
|
* @param {Object<vis.Node.id, vis.Node>} forces
|
|
* @param {vis.Node} node
|
|
* @private
|
|
*/
|
|
_calculateForces(distance, dx, dy, forces, node) {
|
|
if (distance > 0) {
|
|
let degree = (node.edges.length + 1);
|
|
let gravityForce = this.options.centralGravity * degree * node.options.mass;
|
|
forces[node.id].x = dx * gravityForce;
|
|
forces[node.id].y = dy * gravityForce;
|
|
}
|
|
}
|
|
}
|
|
|
|
export default ForceAtlas2BasedCentralGravitySolver;
|