plasma.py

import time
import math

import pygame


WIDTH = 320     # width of screen
HEIGHT = 240    # height of screen
FPS = 30        # frames per second setting


def draw_frame(pixels):
    zoom = 1 / len(pixels)
    _time = time.time()

    for y, row in enumerate(pixels):
        yy = y * zoom
        for x, pixel in enumerate(row):
            xx = x * zoom
            # Making and summing components
            # https://web.archive.org/web/20210119091116/http://www.bidouille.org/prog/plasma
            v = 0
            # Vertical wave
            v += math.sin((xx * 10 + _time))
            # Wave from left to right
            v += math.sin((yy * 10 + _time) / 2.0)
            # Diagonal component
            v += math.sin((xx * 10 + yy * 10 + _time) / 2.0)

            # Circular component
            cx = xx + .5 * math.sin(_time / 5.0)
            cy = yy + .5 * math.cos(_time / 3.0)
            v += math.sin(math.sqrt(100 * (cx * cx + cy * cy) + 1) + _time)

            pixel[:] = (
                255 * (.5 + .5 * math.sin(math.pi * v)),
                255 * (.5 + .5 * math.cos(math.pi * v)),
                0
            )


def main():
    pygame.display.init()

    fpsClock = pygame.time.Clock()

    screen = pygame.display.set_mode((WIDTH, HEIGHT), pygame.DOUBLEBUF, 32)
    pixels = pygame.surfarray.pixels3d(screen)

    done = False
    while not done:
        draw_frame(pixels)
        pygame.display.update()
        fpsClock.tick(FPS)

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


if __name__ == "__main__":
    main()