initial commit

Prog_1/Animations/firework.py

@@ -0,0 +1,91 @@
+import random
+
+import pygame
+
+from my_package.Particle_class import Particle
+
+
+class Spark(Particle):
+    def __init__(self, x, y, color, fade_speed=4):
+        super().__init__(x, y, color, fade_speed)
+        self.x = x
+        self.y = y
+        self.rad = CIRCLE_RADIUS
+        self.image = pygame.Surface((self.rad * 2, self.rad * 2), pygame.SRCALPHA)
+        pygame.draw.circle(self.image, self.color, (self.rad, self.rad), self.rad)
+
+    def appear(self, screen):
+        if self.alpha > 0:
+            self.image.set_alpha(self.alpha)
+            screen.blit(self.image, (self.pos[0] - self.rad, self.pos[1] - self.rad))
+
+particles = []
+width = 600
+height = 600
+SIZE = (width, height)
+BLACK = (0, 0, 0)
+CIRCLE_RADIUS = 1
+COLORS = [(255,0,0), (0,0,255), (255,0,255), (0,255,255), (255,0,255)]
+FRICTION = -0.5
+
+
+def main():
+    screen = init_game()
+    game_loop(screen)
+    exit_game()
+
+def init_game():
+    global clock
+    clock = pygame.time.Clock()
+    pygame.init()
+    screen = pygame.display.set_mode(SIZE)
+    return screen
+
+def game_loop(screen):
+    while True:
+        if not event_handling():
+            break
+        if not update_game():
+            break
+        draw_screen(screen)
+
+
+def event_handling():
+    for event in pygame.event.get():
+        if event.type == pygame.QUIT:
+            return False
+
+        if event.type == pygame.MOUSEBUTTONDOWN:
+            x = pygame.mouse.get_pos()[0]
+            y = pygame.mouse.get_pos()[1]
+            for _ in range(3000):
+                particles.append(Spark(x, y, random.choice(COLORS)))
+    return True
+
+def update_game():
+    for particle_object in particles:
+
+        particle_object.apply_force((random.randint(-2,2),random.randint(-2,2)))
+
+        if particle_object.y <= 0 or particle_object.x <= 0 or particle_object.y == height or particle_object.x == width or particle_object.alpha <= 0:
+            particle_object.set_surface(None)
+            print(particle_object)
+            particles.remove(particle_object)
+
+        particle_object.fade()
+        particle_object.update()
+
+    clock.tick(60)
+    return True
+
+def draw_screen(screen):
+    screen.fill(BLACK)
+    for particle_object in particles:
+        particle_object.set_surface(screen)
+        particle_object.appear(screen)
+    pygame.display.flip()
+
+def exit_game():
+    pygame.quit()
+
+main()