Screensaver/screensaver.py

#!/usr/bin/python3

import os
import math
import random
import pygame

pygame.init()
screen = pygame.display.set_mode((1000, 600), pygame.RESIZABLE)
pygame.display.set_caption("Screensaver")

gray = (20, 20, 20)
red = (255, 20, 20)

working_directory = os.path.dirname(os.path.realpath(__file__))
file_path = os.path.abspath(__file__)
wobblers = []
max_wobbler_size = 100

images = []
for image in os.listdir(f"{working_directory}/textures"):
    images.append(pygame.image.load(f"{working_directory}/textures/{image}"))
clock = pygame.time.Clock()

running = True


class Wobbler:
    def __init__(self, position: tuple=(100, 100), rotation: int=0):
        self.size = 1

        self.x, self.y = position

        self.color = red

        self.rotation = rotation

        self.speed = 4

        self.age = 0

        self.image_original = random.choice(images)

        self.divider = 1

        self.max_width, self.max_height = self.image_original.get_size()
        while self.max_width > max_wobbler_size * 2 or self.max_height > max_wobbler_size * 2:
            self.divider += 0.01
            self.max_width = self.image_original.get_width() / self.divider
            self.max_height = self.image_original.get_height() / self.divider

        self.width = self.max_width * self.size / 100 / self.divider
        self.height = self.max_height * self.size / 100 / self.divider

        self.image = self.image_original
        self.image = pygame.transform.scale(self.image, (self.width, self.height))
        self.image_no_rot = self.image
        # self.image = pygame.transform.rotate(self.image, self.rotation)
        self.image_rect = self.image.get_rect(center=self.image_no_rot.get_rect(center=(self.x, self.y)).center)

    def draw(self):
        screen.blit(self.image, self.image_rect)

    def update(self):
        self.width = self.max_width * self.size / 100 / self.divider
        self.height = self.max_height * self.size / 100 / self.divider

        self.image = self.image_original
        self.image = pygame.transform.scale(self.image, (self.width, self.height))
        self.image_no_rot = self.image
        # self.image = pygame.transform.rotate(self.image, self.rotation)
        self.image_rect = self.image.get_rect(center=self.image_no_rot.get_rect(center=(self.x, self.y)).center)

    def move(self):
        x_offset, y_offset = deg_to_offset(self.rotation, self.speed)
        self.x += x_offset
        self.y += y_offset


def deg_to_offset(degrees, offset):
    radians = math.radians(degrees)
    x = math.sin(radians) * offset
    y = math.cos(radians) * offset
    return x, y


def update():
    screen.fill(gray)

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            global running

            pygame.quit()

            running = False

        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_w:
                print(wobblers)
                print(len(wobblers))

    if running:
        new_wobbler = random.randrange(1, 61) == 1
        if new_wobbler:
            x = random.randrange(0, screen.get_width() + 1)
            y = random.randrange(0, screen.get_height() + 1)
            rot = random.randrange(0, 360)
            wobblers.append(Wobbler((x, y), rot))

        for wobbler in wobblers:
            if not wobbler.x < 0 or not wobbler.x > screen.get_width() or not wobbler.y < 0 or not wobbler.y > screen.get_height():
                wobbler.move()

            wobbler.update()
            wobbler.draw()

            if wobbler.speed > 1:
                decrease_speed = random.randrange(1, 101 - wobbler.speed) == 1
                if decrease_speed:
                    wobbler.speed -= 1

            change_rot = random.randrange(1, 101 - wobbler.speed) == 1
            if change_rot:
                wobbler.rotation = random.randrange(0, 360)
                wobbler.speed = 4

            if wobbler.age < max_wobbler_size:
                wobbler.size += 1

            wobbler.age += 1

            if wobbler.age >= 600:
                wobbler.size -= 1

            if wobbler.size == 0:
                wobblers.remove(wobbler)

            if wobbler.x < -5 or wobbler.x > screen.get_width() + 5 or wobbler.y < -5 or wobbler.y > screen.get_height() + 5:
                wobbler.rotation -= 180
                wobbler.speed = 4

        pygame.display.update()

        clock.tick(60)


while running:
    update()