class CentralGravitySolver { constructor(body, physicsBody, options) { this.body = body; this.physicsBody = physicsBody; this.setOptions(options); } setOptions(options) { this.options = options; } solve() { let dx, dy, distance, node; let nodes = this.body.nodes; let nodeIndices = this.physicsBody.physicsNodeIndices; let forces = this.physicsBody.forces; for (let i = 0; i < nodeIndices.length; i++) { let nodeId = nodeIndices[i]; node = nodes[nodeId]; dx = -node.x; dy = -node.y; distance = Math.sqrt(dx * dx + dy * dy); this._calculateForces(distance, dx, dy, forces, node); } } /** * Calculate the forces based on the distance. * @private */ _calculateForces(distance, dx, dy, forces, node) { let gravityForce = (distance === 0) ? 0 : (this.options.centralGravity / distance); forces[node.id].x = dx * gravityForce; forces[node.id].y = dy * gravityForce; } } export default CentralGravitySolver;