Improved JavaScript implementation of Verlet

Author: Butt4cak3

contents/verlet_integration/code/javascript/verlet.js

@@ -1,55 +1,56 @@
-function verlet(pos, acc, dt){
-
-    var prev_pos, temp_pos, time;
-    prev_pos = pos;
-    time = 0;
-
-    while (pos > 0){
-        time += dt;
-        temp_pos = pos;
-        pos = pos*2 - prev_pos + acc * dt * dt;
-        prev_pos = temp_pos;
-    }
-
-   return time;
-
+function verlet(pos, acc, dt) {
+  let prevPos = pos;
+  let time = 0;
+  let tempPos;
+
+  while (pos > 0) {
+    time += dt;
+    tempPos = pos;
+    pos = pos * 2 - prevPos + acc * dt * dt;
+    prevPos = tempPos;
+  }
+
+  return time;
 }
 
-function stormer_verlet(pos, acc, dt){
+function stormerVerlet(pos, acc, dt) {
+  let prevPos = pos;
+  let time = 0;
+  let vel = 0;
+  let tempPos;
 
-    var prev_pos, temp_pos, time, vel;
-    prev_pos = pos;
-    vel = 0;
-    time = 0;
-    while (pos > 0){
-        time += dt;
-        temp_pos = pos;
-        pos = pos*2 - prev_pos + acc * dt * dt;
-        prev_pos = temp_pos;
+  while (pos > 0) {
+    time += dt;
+    tempPos = pos;
+    pos = pos * 2 - prevPos + acc * dt * dt;
+    prevPos = tempPos;
 
-        vel += acc*dt;
-    }
-
-   return time;
+    vel += acc * dt;
+  }
 
+  return { time, vel };
 }
 
-function velocity_verlet(pos, acc, dt){
-
-    var time, vel;
-    vel = 0;
-    time = 0;
-    while (pos > 0){
-        time += dt;
-        pos += vel*dt + 0.5*acc * dt * dt;
-        vel += acc*dt;
-    }
+function velocityVerlet(pos, acc, dt) {
+  let time = 0;
+  let vel = 0;
 
-   return time;
+  while (pos > 0) {
+    time += dt;
+    pos += vel * dt + 0.5 * acc * dt * dt;
+    vel += acc * dt;
+  }
 
+  return { time, vel };
 }
 
-console.log(verlet(5.0, -10, 0.01));
-console.log(stormer_verlet(5.0, -10, 0.01));
-console.log(velocity_verlet(5.0, -10, 0.01));
+const time = verlet(5, -10, 0.01);
+console.log(`Time for Verlet integration is: ${time}\n`);
+
+const stormer = stormerVerlet(5, -10, 0.01);
+console.log(`Time for Stormer Verlet integration is: ${stormer.time}`);
+console.log(`Velocity for Stormer Verlet integration is: ${stormer.vel}\n`);
 
+const velocity = velocityVerlet(5, -10, 0.01);
+console.log(`Time for Velocity Verlet integration is: ${velocity.time}`);
+console.log(`Velocity for Velocity Verlet integration is: ${velocity.vel}\n`);

contents/verlet_integration/verlet_integration.md

@@ -52,7 +52,7 @@ Unfortunately, this has not yet been implemented in matlab, so here's Julia code
 Unfortunately, this has not yet been implemented in LabVIEW, so here's Julia code:
 [import:1-13, lang:"julia"](code/julia/verlet.jl)
 {% sample lang="javascript" %}
-[import:1-16, lang:"javascript"](code/javascript/verlet.js)
+[import:1-14, lang:"javascript"](code/javascript/verlet.js)
 {% sample lang="rs" %}
 [import:1-13, lang:"rust"](code/rust/verlet.rs)
 {% sample lang="swift" %}
@@ -98,7 +98,7 @@ Unfortunately, this has not yet been implemented in matlab, so here's Julia code
 Unfortunately, this has not yet been implemented in LabVIEW, so here's Julia code:
 [import:15-31, lang:"julia"](code/julia/verlet.jl)
 {% sample lang="javascript" %}
-[import:18-35, lang:"javascript"](code/javascript/verlet.js)
+[import:16-32, lang:"javascript"](code/javascript/verlet.js)
 {% sample lang="rs" %}
 [import:15-32, lang:"rust"](code/rust/verlet.rs)
 {% sample lang="swift" %}
@@ -158,7 +158,7 @@ Unfortunately, this has not yet been implemented in matlab, so here's Julia code
 Unfortunately, this has not yet been implemented in LabVIEW, so here's Julia code:
 [import:33-45, lang:"julia"](code/julia/verlet.jl)
 {% sample lang="javascript" %}
-[import:37-50, lang:"javascript"](code/javascript/verlet.js)
+[import:34-45, lang:"javascript"](code/javascript/verlet.js)
 {% sample lang="rs" %}
 [import:34-45, lang:"rust"](code/rust/verlet.rs)
 {% sample lang="swift" %}