refactor simulation boundary handling to use half the simulation area

Author: NiniEdo

src/renderer.ts

@@ -42,7 +42,7 @@ function animate(now: number) {
     if (deltaTime > 1 / fps) {
         if (runSimulation === true) {
             let solver = Solver.getInstance()
-            solver.solve(spheres, simulationArea, e);
+            solver.solve(spheres, simulationArea/2, e);
         }
         then = now;
         renderer.render(scene, camera);
@@ -55,7 +55,7 @@ export function addBody(mass: number = 10, radius: number = 0.2) {
     const material = new THREE.MeshStandardMaterial({ color: color, flatShading: true });
     const sphere = new THREE.Mesh(geometry, material);
 
-    sphere.position.set(Utils.getRandomNumber(-simulationArea, simulationArea), Utils.getRandomNumber(-simulationArea, simulationArea), Utils.getRandomNumber(-simulationArea, simulationArea));
+    sphere.position.set(Utils.getRandomNumber(-simulationArea/2, simulationArea/2), Utils.getRandomNumber(-simulationArea/2, simulationArea/2), Utils.getRandomNumber(-simulationArea/2, simulationArea/2));
     sphere.userData = {
         velocity: new THREE.Vector3(Utils.getRandomNumber(-0.05, 0.05), Utils.getRandomNumber(-0.05, 0.05), Utils.getRandomNumber(-0.05, 0.05)),
         mass: mass,

src/solver.ts

@@ -14,7 +14,7 @@ export default class Solver {
         return Solver.instance;
     }
 
-    public solve(spheres: THREE.Mesh[], simulationArea: number, e: number): void {
+    public solve(spheres: THREE.Mesh[], bouns: number, e: number): void {
         spheres.forEach((sphere) => {
             let objectAcceleration = new THREE.Vector3(0, 0, 0);
             spheres.forEach((other) => {
@@ -36,28 +36,29 @@ export default class Solver {
 
             sphere.position.add(sphere.userData.velocity.clone().multiplyScalar(this.deltaT));
 
-            if (sphere.position.x + sphere.userData.radius > simulationArea) {
-                sphere.position.x = simulationArea - sphere.userData.radius;
-                sphere.userData.velocity.x *= 0.8;
-            } else if (sphere.position.x - sphere.userData.radius < -simulationArea) {
-                sphere.position.x = -simulationArea + sphere.userData.radius;
-                sphere.userData.velocity.x *= 0.8;
+            let wallBounceE : number = 0.8; 
+            if (sphere.position.x + sphere.userData.radius > bouns) {
+                sphere.position.x = bouns - sphere.userData.radius;
+                sphere.userData.velocity.x *= wallBounceE;
+            } else if (sphere.position.x - sphere.userData.radius < -bouns) {
+                sphere.position.x = -bouns + sphere.userData.radius;
+                sphere.userData.velocity.x *= wallBounceE;
             }
 
-            if (sphere.position.y + sphere.userData.radius > simulationArea) {
-                sphere.position.y = simulationArea - sphere.userData.radius;
-                sphere.userData.velocity.y *= 0.8;
-            } else if (sphere.position.y - sphere.userData.radius < -simulationArea) {
-                sphere.position.y = -simulationArea + sphere.userData.radius;
-                sphere.userData.velocity.y *= 0.8;
+            if (sphere.position.y + sphere.userData.radius > bouns) {
+                sphere.position.y = bouns - sphere.userData.radius;
+                sphere.userData.velocity.y *= wallBounceE;
+            } else if (sphere.position.y - sphere.userData.radius < -bouns) {
+                sphere.position.y = -bouns + sphere.userData.radius;
+                sphere.userData.velocity.y *= wallBounceE;
             }
 
-            if (sphere.position.z + sphere.userData.radius > simulationArea) {
-                sphere.position.z = simulationArea - sphere.userData.radius;
-                sphere.userData.velocity.z *= 0.8;
-            } else if (sphere.position.z - sphere.userData.radius < -simulationArea) {
-                sphere.position.z = -simulationArea + sphere.userData.radius;
-                sphere.userData.velocity.z *= 0.8;
+            if (sphere.position.z + sphere.userData.radius > bouns) {
+                sphere.position.z = bouns - sphere.userData.radius;
+                sphere.userData.velocity.z *= wallBounceE;
+            } else if (sphere.position.z - sphere.userData.radius < -bouns) {
+                sphere.position.z = -bouns + sphere.userData.radius;
+                sphere.userData.velocity.z *= wallBounceE;
             }
 
             let maxVel = 2.5;