verletParticleStress.py

import pygame
import random, time
from Lib2D import Vector2D
from VerletIntegration import Integration, Particle, Constraint, Prefabs

# Settings #############################################################

# put any adjustable settings here that would be interesting to tinker with.

FPS = 30
CANVAS_WIDTH = 800
CANVAS_HEIGHT = 500
GRAVITY_DAMPENING = 0.001

##########################################################################

pygame.init()
screen = pygame.display.set_mode((CANVAS_WIDTH, CANVAS_HEIGHT))
done = False
clock = pygame.time.Clock()

verlet = Integration({
    'iterations': 1,
    'stageMinVect': Vector2D(0, 0),
    'stageMaxVect': Vector2D(CANVAS_WIDTH, CANVAS_HEIGHT),
    'gravity': Vector2D(0, 0.05)
})

for i in range(0, 350):
    verlet.addParticle(Particle({
            'vector': Vector2D(random.randint(0, CANVAS_WIDTH), random.randint(0, CANVAS_HEIGHT)),
            'radius': 14,
            'collides': True,
            'data': { 'drawn': True }
        }))

while not done:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            done = True

    # millis = int(round(time.time() * 1000))

    screen.fill((0,0,0))

    verlet.runTimeStep()

    for particle in verlet.particles:
        pygame.draw.circle(screen, (0, 255, 0), (int(particle.vector.x), int(particle.vector.y)), particle.radius, 1)

    pygame.display.flip()

    # print(int(round(time.time() * 1000)) - millis)

    clock.tick(FPS)