Exemplo de movimento e colisão que vem no Processing Python Mode

exemplo_bouncy_processing.pyde

"""
Para roder este exemplo:
https://abav.lugaralgum.com/como-instalar-o-processing-modo-python/
---
Bouncy Bubbles
based on code from Keith Peters.

Multiple-object collision.
The "bounding-box" rendering is imprecise because we've implemented springy
bodies which don't show their deformation.
"""

balls = []
NUM_BALLS = 12


def setup():
    size(640, 360, P3D)
    for i in range(NUM_BALLS):
        balls.append(Ball(random(width), random(height),
                          random(15, 45), i, balls))
    noStroke()
    ellipseMode(RADIUS)
    fill(255, 204)


def draw():
    background(0)
    for ball in balls:
        ball.collide()
        ball.move()
        ball.display()
        
class Ball(object):
    # Class variables. These are shared amongst all instances of Ball.
    Spring = 0.05
    Gravity = 0.03
    Friction = -0.9

    def __init__(self, x, y, radius, index, others):
        self.x = x
        self.y = y
        self.radius = radius
        self.index = index
        self.others = others
        self.vx = 0
        self.vy = 0

    def collide(self):
        for other in self.others[self.index:]:
            dx = other.x - self.x
            dy = other.y - self.y
            minDist = other.radius + self.radius
            if dist(other.x, other.y, self.x, self.y) < minDist:
                angle = atan2(dy, dx)
                targetX = self.x + cos(angle) * minDist
                targetY = self.y + sin(angle) * minDist
                ax = (targetX - other.x) * Ball.Spring
                ay = (targetY - other.y) * Ball.Spring
                self.vx -= ax
                self.vy -= ay
                other.vx += ax
                other.vy += ay

    def move(self):
        self.vy += Ball.Gravity
        self.x += self.vx
        self.y += self.vy

        if self.x + self.radius > width:
            self.x = width - self.radius
            self.vx *= Ball.Friction
        elif self.x - self.radius < 0:
            self.x = self.radius
            self.vx *= Ball.Friction
        if self.y + self.radius > height:
            self.y = height - self.radius
            self.vy *= Ball.Friction
        elif self.y - self.radius < 0:
            self.y = self.radius
            self.vy *= Ball.Friction

    def display(self):
        ellipse(self.x, self.y, self.radius, self.radius)