asteroid.py

from circleshape import *
from constants import *
import random

class Asteroid(CircleShape):
    def __init__(self, x, y, radius):
        super().__init__(x, y, radius)

    def draw(self, screen):
        pygame.draw.circle(screen, (0, 0, 255), self.position, self.radius, 2)
    
    def update(self, dt):
        self.position += self.velocity * dt

    def split(self, asteroids):
        asteroids_groups = asteroids
        self.kill()
        if self.radius <= ASTEROID_MIN_RADIUS:
            return
        angle = random.uniform(20, 50)
        new_asteroid1 = Asteroid(self.position[0], self.position[1], self.radius - ASTEROID_MIN_RADIUS)
        new_asteroid2 = Asteroid(self.position[0], self.position[1], self.radius - ASTEROID_MIN_RADIUS)
        new_asteroid1.velocity = pygame.math.Vector2.rotate(self.velocity, angle) * 1.2
        new_asteroid2.velocity = pygame.math.Vector2.rotate(self.velocity, -angle) * 1.2
        asteroids_groups.add(new_asteroid1)
        asteroids_groups.add(new_asteroid2)