init

pygame_animation/cli/ball.py

@@ -0,0 +1,114 @@
+import random
+import sys
+
+import pygame
+import typer
+
+from pygame_animation.cli.common import verbose_callback
+
+ball_app = typer.Typer()
+
+
+@ball_app.callback()
+def ball(
+    verbose: bool = typer.Option(
+        False,
+        callback=verbose_callback,
+        help="run the ball annimation",
+    ),
+):
+    "ball cli"
+
+
+@ball_app.command()
+def run(
+    verbose: bool = typer.Option(
+        False,
+        callback=verbose_callback,
+        help="show the log messages",
+    ),
+    ball_radius: int = typer.Option(20),
+    restitution: int = typer.Option(-2),
+    gravity: int = typer.Option(5),
+    friction: int = typer.Option(0.9),
+):
+
+    pygame.init()
+
+    # Set up the display
+    screen_width = 640
+    screen_height = 480
+    screen = pygame.display.set_mode((screen_width, screen_height))
+    pygame.display.set_caption("Bouncing Ball Animation")
+
+    # Set up the ball
+    ball_color = (255, 255, 255)
+    ball_x = screen_width // 2
+    ball_y = screen_height // 2
+    ball_speed_x = random.randint(-5, 5)
+    ball_speed_y = random.randint(-5, 5)
+    squishyness = 100
+
+    # Set up the clock
+    clock = pygame.time.Clock()
+
+    # Run the game loop
+    while True:
+        squish = 0
+        # Handle events
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT:
+                pygame.quit()
+                sys.exit()
+
+        # Update the ball position
+        ball_x += ball_speed_x
+        ball_y += ball_speed_y
+
+        # Check for collisions with the walls
+        if ball_x + ball_radius >= screen_width or ball_x - ball_radius <= 0:
+            ball_speed_x = -ball_speed_x
+
+        if ball_y + ball_radius >= screen_height or ball_y - ball_radius <= 0:
+            # restitution
+            ball_speed_y = -ball_speed_y - restitution
+            # friction
+            ball_speed_x *= friction
+
+        if ball_y <= ball_radius:
+            ball_y = ball_radius
+        if ball_y >= screen_height - ball_radius:
+            ball_y = screen_height - ball_radius
+        if ball_x <= ball_radius:
+            ball_x = ball_radius
+        if ball_x >= screen_width - ball_radius:
+            ball_x = screen_width - ball_radius
+
+        if ball_y - ball_radius <= 0:
+            squish = ball_speed_y * squishyness
+
+        else:
+            ball_speed_y += gravity
+
+        # Draw the ball
+        screen.fill((0, 0, 0))
+        if squish != 0:
+            ball_width = ball_radius * (2 + squish)
+            ball_height = ball_radius * (2 - squish)
+            print(ball_width, ball_height)
+            ball_x = ball_x - ball_width // 2
+            pygame.draw.ellipse(
+                screen,
+                ball_color,
+                (ball_x, ball_y, ball_width, ball_height),
+            )
+            # pygame.draw.circle(screen, ball_color, (ball_x, ball_y), ball_radius)
+
+        else:
+            pygame.draw.circle(screen, ball_color, (ball_x, ball_y), ball_radius)
+
+        # Update the display
+        pygame.display.flip()
+
+        # Wait for a little bit to control the frame rate
+        clock.tick(60)