main.py

import pygame
from random import randint
from boid import Boid
from pygame.locals import (
    K_ESCAPE,
    K_SPACE,
    KEYDOWN,
    MOUSEBUTTONUP,
    QUIT,
)

# Initialize pygame modules
pygame.init()

# Create the screen object
pygame.display.set_caption("Flocking Simulation")
screen = pygame.display.set_mode((1000, 1000))
background = pygame.Surface(screen.get_size())
background.fill((0, 0, 0))

# Make groups of sprites
boids = pygame.sprite.Group()

# Populate with boids
numBoids = 30
for i in range(numBoids):
    position = (randint(0, 1000), randint(0, 1000))
    boids.add(Boid(position))

# Setup the clock to limit fps
clock = pygame.time.Clock()

# Running control var
running = True

while running:
    for event in pygame.event.get():
        if event.type == QUIT:
            running = False
        elif event.type == MOUSEBUTTONUP:
            pos = pygame.mouse.get_pos()
            boids.add(Boid(pos))
        elif event.type == KEYDOWN:
            if event.key == K_ESCAPE:
                running = False
            if event.key == K_SPACE:
                print("Space")
                # Pause the sim

    # Paint the background
    screen.blit(background, (0, 0))

    # Update then paint the boids
    boids.update(boids)
    for boid in boids:
        screen.blit(boid.surf, boid.rect)

    pygame.display.flip()
    clock.tick(60)