NatureOfCode2/grav2.3.html at master · brlebtag/NatureOfCode2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
<html>
    <head>
        <title>Gravitational Attraction 2.3 - n body problem</title>
        <script src="https://cdn.jsdelivr.net/npm/p5@1.5.0/lib/p5.min.js"></script>
        <script>
            const G = 25;

            class Mover {
                constructor(x, y, vx, vy, m = 1) {
                    this.pos = createVector(x, y);
                    this.velocity = p5.Vector.random2D().mult(0.1); /*p5.Vector.random2D().mult(5)*/
                    this.acceleration = createVector(0, 0);
                    this.mass = m;
                    this.radius = sqrt(this.mass) * 2;
                }

                applyForce(force) {
                    this.acceleration.add(p5.Vector.div(force, this.mass)); // F = m * a => a = F / m (1) => a = F;
                }

                update() {
                    this.velocity.add(this.acceleration);
                    this.pos.add(this.velocity);
                    this.acceleration.set(0, 0);
                }

                show() {
                    stroke(255);
                    strokeWeight(3);
                    fill(255, 100);
                    ellipse(this.pos.x, this.pos.y, this.radius * 2);
                }

                attract(mover) {
                    let force = p5.Vector.sub(this.pos, mover.pos);
                    let distanceSq = constrain(force.magSq(), 100, 1000);
                    let strength = G * (this.mass * mover.mass) / distanceSq;
                    force.setMag(strength);
                    mover.applyForce(force);
                }
            }

            var movers = [];
            const TotalMovers = 2;

            function setup() {
                createCanvas(800, 600);

                /*for (let i = 0; i < TotalMovers; i++) {
                    movers.push(new Mover(random(width), random(height), 50));
                }*/
                movers[0] = new Mover(300, 200, 0, 5, 10);
                movers[1] = new Mover(100, 200, 0, -5, 10);
                movers[2] = new Mover(200, 300, 5, 0, 10);
                movers[3] = new Mover(200, 100, -5, 0, 10);
                background(0);
            }

            function draw() {
                background(0, 10);

                for (const mover of movers) {
                    for (const other of movers) {
                        if (mover !== other) {
                            mover.attract(other);
                            stroke(255);
                            line(mover.pos.x, mover.pos.y, other.pos.x, other.pos.y);
                        }
                    }
                }

                for (const mover of movers) {
                    mover.update();
                    mover.show();
                }
            }
        </script>
    </head>
</html>