From 7e57102595a3adb2bd8499b6490cf97744d300e9 Mon Sep 17 00:00:00 2001
From: EKNr1 <emil@i21k.de>
Date: Wed, 19 Jul 2023 14:41:48 +0200
Subject: [PATCH] Got it working.

---
 screensaver.py | 132 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 132 insertions(+)
 create mode 100644 screensaver.py

diff --git a/screensaver.py b/screensaver.py
new file mode 100644
index 0000000..043f838
--- /dev/null
+++ b/screensaver.py
@@ -0,0 +1,132 @@
+#!/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)
+
+wobblers = []
+max_wobbler_size = 50
+
+clock = pygame.time.Clock()
+
+running = True
+
+
+class Wobbler:
+    def __init__(self, position: tuple=(100, 100), rotation: int=0):
+        self.size = 1
+        self.width = self.size
+        self.height = self.size
+
+        self.x, self.y = position
+
+        self.color = red
+
+        self.rotation = rotation
+
+        self.speed = 4
+
+        self.age = 0
+
+        self.cube = pygame.Surface((self.height, self.width))
+        self.cube.fill(self.color)
+        self.cube_no_rot = self.cube
+        self.cube.set_colorkey(gray)
+        self.cube = pygame.transform.rotate(self.cube, self.rotation)
+        self.cube_rect = self.cube.get_rect(center=self.cube_no_rot.get_rect(center=(self.x, self.y)).center)
+
+    def draw(self):
+        screen.blit(self.cube, self.cube_rect)
+
+    def update(self):
+        self.width = self.size
+        self.height = self.size
+
+        self.cube = pygame.Surface((self.height, self.width))
+        self.cube.fill(self.color)
+        self.cube_no_rot = self.cube
+        self.cube.set_colorkey(gray)
+        self.cube = pygame.transform.rotate(self.cube, self.rotation)
+        self.cube_rect = self.cube.get_rect(center=self.cube_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, 121) == 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)
+
+            if wobbler.age < max_wobbler_size:
+                wobbler.size += 1
+
+            wobbler.age += 1
+
+            if wobbler.age >= 300:
+                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
+
+        pygame.display.update()
+
+        clock.tick(60)
+
+
+while running:
+    update()