asteroid.py

import random
import pygame
from circleshape import CircleShape
from constants import ASTEROID_MIN_RADIUS


class Asteroid(CircleShape):
    def __init__(self, x, y, radius: int | float, velocity=pygame.Vector2(0, 0)):
        super().__init__(x, y, radius, velocity=velocity)

    def draw(self, screen):
        pygame.draw.circle(
            surface=screen,
            color="white",
            center=self.position,
            radius=self.radius,
            width=2,
        )

    def update(self, dt):
        self.position += self.velocity * dt

    def split(self):
        if self.radius > ASTEROID_MIN_RADIUS:
            angle = random.uniform(20, 50)
            split1 = self.velocity.rotate(angle)
            split2 = self.velocity.rotate(-angle)
            new_radius = self.radius - ASTEROID_MIN_RADIUS
            asteroid_chunk1 = Asteroid(
                self.position[0], self.position[1], new_radius, velocity=split1 * 1.2
            )
            asteroid_chunk2 = Asteroid(
                self.position[0], self.position[1], new_radius, velocity=split2 * 1.2
            )
        self.kill()