DEV Community

Martin Heinz
Martin Heinz

Posted on

Implementing 2D Physics in Javascript

Note: This was originally posted at martinheinz.dev

Physics and implementations of real looking animations might seem very complex and difficult, but it's actually not the case. These algorithms can be very simple and can produce realistic simulations of various physics concepts, including velocity, acceleration or gravity.

Alt Text

So, let's see how those algorithms work while implementing 2D physics simulation in JavaScript!

You can checkout the animations and examples here: https://martinheinz.github.io/physics-visual/

TL;DR: Source code is available in my repository here: https://github.com/MartinHeinz/physics-visual

Uniform and Accelerated Movement

Let's start with the most basic thing - moving stuff around.

If we want just uniform movement, then we can use code like this:

function move(dt) {
    x += vx * dt;
    y += vy * dt;
}
Enter fullscreen mode Exit fullscreen mode

In the code above x and y are coordinates of an object, e.g ellipse, next vx and vy are velocities in horizontal and vertical axis respectively and dt (time delta) is a time between 2 ticks of timer, which in case of JavaScript are 2 calls to requestAnimationFrame.

As an example - if we wanted to move object sitting at (150, 50) and moving southwest, then we would have the following (movement after single tick):

x = 150 += -1 * 0.1 -> 149.9
y =  50 +=  1 * 0.1 -> 50.1
Enter fullscreen mode Exit fullscreen mode

Moving uniformly is pretty boring though, so let's accelerate movement of our objects:

function move(dt) {
    vx += ax * dt;
    vy += ay * dt;
    x += vx * dt;
    y += vy * dt;
}
Enter fullscreen mode Exit fullscreen mode

In this piece of code we added ax and ay which represent acceleration on x and y axis respectively. We use the acceleration to calculate change in velocity or speed (vx/vy), which we then use to move objects like before. Now, if we copy previous example and add acceleration only on x axis (going west), we get:

vx =  -1 += -1   * 0.1 ->  -1.1  // vx += ax * dt;
vy =   1 +=  0   * 0.1 ->   1    // vy += ay * dt;
 x = 150 += -1.1 * 0.1 -> 149.89 //  x += vx * dt;  Moved further (-0.01) than in previous example!
 y =  50 +=  1   * 0.1 ->  50.1  //  y += vy * dt;
Enter fullscreen mode Exit fullscreen mode

Gravity

Now that we can move things around, how about moving objects towards other objects? Well, that's just called gravity. What do we need to add, to implement that?

Just so you know what we are trying to get to:

Alt Text

First things first, let's recall a few equations from high school:

Equation of force:

F = m * a    ... Force is Mass times Acceleration
a = F / m    ... From that we can derive that force acting on some object (mass) accelerates
Enter fullscreen mode Exit fullscreen mode

If we now want to extend this to force of 2 object acting on each other, we get:

Alt Text

It's getting little complicated (for me at least), so let's break it down. In this equation |F| is magnitude of force, which is same for both objects, just in opposite direction. These objects are represented by their mass - m_1 and m_2. k here is a gravitational constant and r is distance of centers of gravity of these object. If it still doesn't make much sense, then here's a picture:

Alt Text

If we want to create some visualization we will end up with more than 2 object, right? So, what happens when we have more objects acting on each other?

Alt Text

Looking at the picture above, we can see 2 orange objects pulling black one with forces F_1 and F_2, what we are interested in though is final force F, which we can calculate like this:

  • We first calculate forces F_1 and F_2 using equations from above
  • We then break it down into vectors:

Alt Text

  • Finally we get F:

Alt Text

Alright, we have all the math we need, now how will the code look? I will spare you all the steps and just show you final code with comments, if you need more information, feel free to reach out to me. ๐Ÿ™‚

function moveWithGravity(dt, o) {  // "o" refers to Array of objects we are moving
    for (let o1 of o) {  // Zero-out accumulator of forces for each object
        o1.fx = 0;
        o1.fy = 0;
    }
    for (let [i, o1] of o.entries()) {  // For each pair of objects...
        for (let [j, o2] of o.entries()) {
            if (i < j) {  // To not do same pair twice
                let dx = o2.x - o1.x;  // Compute distance between centers of objects
                let dy = o2.y - o1.y;
                let r = Math.sqrt(Math.pow(dx, 2) + Math.pow(dy, 2));
                if (r < 1) {  // To avoid division by 0
                    r = 1;
                }
                // Compute force for this pair; k = 1000
                let f = (1000 * o1.m * o2.m) / Math.pow(r, 2);  
                let fx = f * dx / r;  // Break it down into components
                let fy = f * dy / r;
                o1.fx += fx;  // Accumulate for first object
                o1.fy += fy;
                o2.fx -= fx;  // And for second object in opposite direction
                o2.fy -= fy;
            }
        }
    }
    for (let o1 of o) {  // for each object update...
        let ax = o1.fx / o1.m;  // ...acceleration 
        let ay = o1.fy / o1.m;

        o1.vx += ax * dt;  // ...speed
        o1.vy += ay * dt;

        o1.x += o1.vx * dt;  // ...position
        o1.y += o1.vy * dt;
    }
}
Enter fullscreen mode Exit fullscreen mode

Collisions

When things move around they will also collide at some point. We have two options for solving collisions - push objects out of collision or bounce away, let's look at the pushing solution first:

Alt Text

Before we can resolve collision we need to first check whether 2 objects are actually colliding:

class Collision {
    constructor(o1, o2, dx, dy, d) {
        this.o1 = o1;
        this.o2 = o2;

        this.dx = dx;
        this.dy = dy;
        this.d = d;
    }
}

function checkCollision(o1, o2) {
    let dx = o2.x - o1.x;
    let dy = o2.y - o1.y;
    let d = Math.sqrt(Math.pow(dx, 2) + Math.pow(dy, 2));
    if (d < o1.r + o2.r) {
        return  {
            collisionInfo: new Collision(o1, o2, dx, dy, d),
            collided: true
        }
    }
    return  {
        collisionInfo: null,
        collided: false
    }
}
Enter fullscreen mode Exit fullscreen mode

We first declare Collision class which represents 2 colliding objects. In the checkCollision function we first compute x and y components of distances of objects and then compute their actual distance d. If sum of their radii is lower than their distance d, then they must be in collision so we return new Collision object.

Alt Text

Now, to resolve their collision, we need to know direction of displacement and it's magnitude:

n_x = d_x / d        ... this is eigenvector
n_y = d_y / d

s = r_1 + r_2 - d  ... s is size of collision (see picture)  
Enter fullscreen mode Exit fullscreen mode

Alt Text

So, in JavaScript code that would be:

function resolveCollision(info) {  // "info" is a Collision object from above
    let nx = info.dx /info.d;  // Compute eigen vectors
    let ny = info.dy /info.d;
    let s = info.o1.r + info.o2.r - info.d; // Compute penetration depth
    info.o1.x -= nx * s/2;  // Move first object by half of collision size
    info.o1.y -= ny * s/2;
    info.o2.x += nx * s/2;  // Move other object by half of collision size in opposite direction
    info.o2.y += ny * s/2;
}
Enter fullscreen mode Exit fullscreen mode

You can view interactive example of this collision resolution at https://martinheinz.github.io/physics-visual/ (Click on Pushing Through Objects)

Solving Collisions with Force

Aaaaand final piece of puzzle - resolving collisions by bouncing objects. In this case, it's better to omit all the math as it would make the article twice as long, so all I'm gonna tell you is that we need to account for law of momentum conservation and law of energy conservation which helps us build and solve following magical equation:

k = -2 * ((o2.vx - o1.vx) * nx + (o2.vy - o1.vy) * ny) / (1/o1.m + 1/o2.m) ... *Magic*
Enter fullscreen mode Exit fullscreen mode

Well, how does this magical k help us? We know direction in which the objects will move (we can compute that using eigenvectors like before with n_x and n_y), but we don't know by how much and that's the k. So, this is how we compute vector (z), that tells us where to move those objects:

Alt Text

Alt Text

And now the final code:

function resolveCollisionWithBounce(info) {
    let nx = info.dx /info.d;
    let ny = info.dy /info.d;
    let s = info.o1.r + info.o2.r - info.d;
    info.o1.x -= nx * s/2;
    info.o1.y -= ny * s/2;
    info.o2.x += nx * s/2;
    info.o2.y += ny * s/2;

    // Magic...
    let k = -2 * ((info.o2.vx - info.o1.vx) * nx + (info.o2.vy - info.o1.vy) * ny) / (1/info.o1.m + 1/info.o2.m);
    info.o1.vx -= k * nx / info.o1.m;  // Same as before, just added "k" and switched to "m" instead of "s/2"
    info.o1.vy -= k * ny / info.o1.m;
    info.o2.vx += k * nx / info.o2.m;
    info.o2.vy += k * ny / info.o2.m;
}
Enter fullscreen mode Exit fullscreen mode

Conclusion

This post includes lots of math, but most of it is pretty simple, so I hope this helped you understand and familiarize yourself with these physical concepts. If you want to see more details, then you can check out code in my repository here and interactive demo here.

Top comments (7)

Collapse
 
anuraghazra profile image
Anurag Hazra

Cool man. I also love physics stuff on web. I built this Verlyjs physics engine too.:D

Collapse
 
jdforsythe profile image
Jeremy Forsythe

Great article! My team will have fun with this.

Collapse
 
sheygs profile image
Olusegun Ekoh

Cool stuff man!

Collapse
 
marksparkyryan profile image
Mark Ryan

๐Ÿ˜ฏ Awesome.

Collapse
 
anrodriguez profile image
An Rodriguez

hehe as complicated as I thought it might get : ) good job!

Some comments may only be visible to logged-in visitors. Sign in to view all comments.