asteroid.py

import constants as c
import circleshape
import pygame
from logger import log_state, log_event
import random

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

    def draw(self, screen):
        pygame.draw.circle(screen, (255, 255, 255), self.position, self.radius, c.LINE_WIDTH)

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

    def split(self):
        self.kill()
        if self.radius <= c.ASTEROID_MIN_RADIUS:
            return
        log_event("asteroid_split")
        new_size = self.radius - c.ASTEROID_MIN_RADIUS
        exit_angle = random.uniform(20, 50)
        child1 = Asteroid(self.position.x, self.position.y, new_size)
        child1.velocity = self.velocity.rotate(exit_angle) * 1.2
        child2 = Asteroid(self.position.x, self.position.y, new_size)
        child2.velocity = self.velocity.rotate(-exit_angle) * 1.2