asteroids.py

from proj import *
from bullet import *
from ship import *
from asteroid import *
from vec2 import *
from mat2 import *
from drawn_object import *


def main():

    # Initialize turtle parameters
    tracer(0)
    bgcolor('BLACK')

    # Initial asteroid count
    asteroid_count = 4
    # Create our player, bullet list, and asteroid list
    player, bullets, asteroids, fragments, hud = initialize(asteroid_count)

    # Game loop
    while True:

        # Create more asteroids if the player killed all previous asteroids
        if len(asteroids) == 0:
            asteroid_count += 1
            asteroids = create_asteroids(asteroid_count, player.center)

        # Get the time at the start of this frame
        start_time = perf_counter()
        # Must clear entire previous frame if using single turtle
        Drawn_Object.drawer.clear()

        # Get input from the player
        get_input(player, bullets)

        # Move all relevant objects
        move_fragments(fragments)
        move_asteroids(asteroids)
        move_bullets(player, bullets, asteroids, fragments, hud)
        move_player(player, asteroids, fragments, hud)

        hud.draw_object()
        # Draw the new frame to the screen
        update()

        # Keep the frames
        elapsed_time = perf_counter()
        if (frame - (elapsed_time - start_time)) > 0:
            sleep(frame - (elapsed_time - start_time))
        else:
            print(frame - (elapsed_time - start_time))


# Moves all bullets and checks if they have collided with any asteroids
def move_bullets(player, bullets, asteroids, fragments, hud):

        # Check for bullet collision with asteroids then move and draw bullets
        for i in reversed(range(len(bullets))):
            # See if the bullet has traveled max distance
            bullets[i].decrement_timer()
            if bullets[i].timer <= 0:
                del bullets[i]
                continue
            # See if bullet collided with an asteroid
            collided, asteroid_index = bullets[i].continuous_collision_check(asteroids)
            # Check for a collision
            if collided:
                # Increase player score
                player.increase_score(asteroids[asteroid_index].points, hud)
                # Split the asteroid
                asteroids[asteroid_index].split(asteroids, fragments)
                # Clean up our asteroid and bullet
                del asteroids[asteroid_index]
                del bullets[i]
                continue
            bullets[i].move_object()
            bullets[i].draw_object()


# Move asteroids and check if any asteroids collided with the player
def move_asteroids(asteroids):

        # Move and draw all asteroids
        for i in reversed(range(len(asteroids))):
            # Move the asteroids
            asteroids[i].move_object()
            # Draw the asteroid if it wasn't hit
            asteroids[i].draw_object()


# Moves all fragments and despawns them if needed
def move_fragments(fragments):

        # Move and draw all fragmentsS
        for i in reversed(range(len(fragments))):
            fragments[i].decrement_timer()
            # Kill fragments if they run out of time
            if fragments[i].timer <= 0:
                del fragments[i]
                continue
            fragments[i].move_object()
            fragments[i].draw_object()


def move_player(player, asteroids, fragments, hud):

    # Move the player
    if player.velocity.get_magnitude() > 0:
        player.move_object()
    # If the player isn't respawning check collision
    if not player.respawning:
        # See if bullet collided with an asteroid
        collided, asteroid_index = player.continuous_collision_check(asteroids)
        # Check for a collision
        if collided:
            # Increase player score
            player.increase_score(asteroids[asteroid_index].points, hud)
            # Split the asteroid
            asteroids[asteroid_index].split(asteroids, fragments)
            # Clean up our asteroid and respawn player
            del asteroids[asteroid_index]
            player.respawn(hud)
    player.draw_object()


# Get user input
def get_input(player, bullets):

    # Move the ship forward
    if is_pressed('w'):
        player.accelerate_object()
    elif player.velocity.get_magnitude() > 0:
        player.decelerate_object()
    # Rotate the ship right
    if is_pressed('d'):
        player.rotate_object('d')
    # Rotate the ship left
    elif is_pressed('a'):
        player.rotate_object('a')
    # Fire a bullet
    if player.queued_shots == 0:
        if is_pressed(' '):
            player.shoot(bullets)
            player.shot_last_frame = True
        else:
            player.shot_last_frame = False
    else:
        player.shoot(bullets)


# Create necessary intial objects
def initialize(asteroid_count):

    # Create the player's ship
    player = Ship()
    # Create the initial hud
    hud = HUD(player)
    # Create a list to contain the bullets
    bullets = []
    # Creates a list to contain asteroid fragments
    fragments = []
    # Create the initial four asteroids
    asteroids = create_asteroids(asteroid_count, player.center)

    # Return the player, the bullet list, the asteroid list, the fragment list, and the hud
    return player, bullets, asteroids, fragments, hud


# Fill the asteroid list
def create_asteroids(asteroid_count, player_center):
    return [Asteroid(player_center=player_center) for i in range(asteroid_count)]


main()