asteroid.py

import random
import pygame

from constants import *
from circleshape import CircleShape

# asteroid class
class Asteroid(CircleShape):
  def __init__(self, x, y, radius):
    super().__init__(x, y, radius)
    self.radius = radius
  
  def split(self):
    self.kill()

    if self.radius <= ASTEROID_MIN_RADIUS:
      return
    else:
      new_angle = random.uniform(20, 50)
      v1 = self.velocity.rotate(new_angle)
      v2 = self.velocity.rotate(-new_angle)
      r = self.radius - ASTEROID_MIN_RADIUS
      a1 = Asteroid(self.position[0], self.position[1], r)
      a2 = Asteroid(self.position[0], self.position[1], r)
      a1.velocity = v1 * 1.2
      a2.velocity = v2 * 1.2

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