Reproduce an early 90s Macintosh screensaver

lines.py

# Reproduce an early 90s Macintosh screensaver
# Nicolas Seriot 2024-11-13 https://seriot.ch
# https://x.com/nst021/status/1856605627264417931

import pygameimport pygame
import random

WIDTH = 800
HEIGHT = 600

NB_GROUPS = 4
NB_LINES_PER_GROUP = 6

vectors = []
lines_to_draw = []

for _ in range(NB_GROUPS):
    vectors.append([random.randint(-10, 10), random.randint(-10, 10), random.randint(-10, 10), random.randint(-10, 10)])
    lines_to_draw.append([random.randrange(WIDTH), random.randrange(HEIGHT), random.randrange(WIDTH), random.randrange(HEIGHT)])
    
pygame.init()

screen = pygame.display.set_mode((WIDTH, HEIGHT))

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

    screen.fill((0,0,0))
        
    for g in range(NB_GROUPS):
        
        next = lines_to_draw[2*g].copy()
    
        v = vectors[g]
        
        for i in range(4):
            if next[i] + v[i] not in range(WIDTH if i % 2 == 0 else HEIGHT):
                v[i] *= -1
            next[i] += v[i]

        lines_to_draw.insert(g, next)
    
    lines_to_draw = lines_to_draw[:NB_LINES_PER_GROUP*NB_GROUPS]
        
    for l in lines_to_draw:
        pygame.draw.line(screen, (255,255,255), l[0:2], l[2:4])
    
    pygame.display.update()
    
    pygame.time.delay(100) # refresh ms