lib.py

import pygame
import random
import math
from constants import *
pygame.font.init()

# We create the asteroids by append it to the ASTEROIDS list 
# and its coordinates to the ASTEROID_X and ASTEROID_Y list
def generate_asteroid():
    for i in range(NUM_OF_ASTEROIDS):
        ASTEROIDS.append(ASTEROID)
        ASTEROID_X.append(WIDTH*random.random())
        ASTEROID_Y.append(0)

# Shoot the asteroids, simply by increasing the y-coordinates of each asteroids in the list
def shooting_asteroid():
    for i in range(NUM_OF_ASTEROIDS):
        ASTEROID_Y[i] += random.randint(5, 10)

# Checking whether any asteroid has collided with the spaceships or not, 
# it it does, return yes, false for the other
def is_hit(SPACESHIP, i):
    # Draw a rectangle arounf the asteroid
    rect = pygame.Rect(ASTEROID_X[i], ASTEROID_Y[i], ASTEROID_WIDTH, ASTEROID_HEIGHT) 
    #python provides us the tool
    if SPACESHIP.colliderect(rect):
        return True
    else:
        return False

#Draw the winner
def draw_winner(text, COLOR):
    #The winner text
    draw_text = WINNER_FONT.render(text, 1, COLOR)
    #Display the text
    WIN.blit(draw_text,(
    WIDTH/2 - draw_text.get_width() /2, HEIGHT/2 - draw_text.get_height()/2))
    #Update the text
    pygame.display.update()
    # Delay the time for 5000 milliseconds
    pygame.time.delay(5000)

#Draw the window
def draw_window(red_health, yellow_health):
    # Draw the background of the window
    WIN.blit(SPACE, (0,0))
    # Draw the border 
    pygame.draw.rect(WIN, BLACK, BORDER)
    # Display 2 spaceships
    WIN.blit(YELLOW_SPACESHIP,(yellow.x,yellow.y))
    WIN.blit(RED_SPACESHIP,(red.x,red.y))

    # Display the asteroids
    for i in range(NUM_OF_ASTEROIDS):
        WIN.blit(ASTEROIDS[i], (ASTEROID_X[i], ASTEROID_Y[i]))

    # Display the health data
    red_health_text = HEALTH.render(
    "Health :" + str(red_health), 1, WHITE)
    yellow_health_text = HEALTH.render(
    "Health :" + str(yellow_health), 1, WHITE)
    
    WIN.blit(red_health_text, (10,10))
    WIN.blit(yellow_health_text, (
    (WIDTH - 10 - yellow_health_text.get_width()),10))

    # Display the bullets
    for bullet in red_bullets:
        pygame.draw.rect(WIN, RED, bullet)
    for bullet in yellow_bullets:
        pygame.draw.rect(WIN, YELLOW, bullet)
    # Update the window
    pygame.display.update() 

#This class deals with bullets, shoot it by increasing its y-coordinates 
def handle_bullet_function(red_bullets, yellow_bullets):
    for bullet in yellow_bullets:
        bullet.x -= bullet_velocity
        if red.colliderect(bullet): 
# If it collides the other spaceship, 
# post the RED_HIT event and remove the bullet from the list
            pygame.event.post(pygame.event.Event(RED_HIT))
            yellow_bullets.remove(bullet)
        elif bullet.x < 0:
            yellow_bullets.remove(bullet)

    for bullet in red_bullets:
        bullet.x += bullet_velocity
        if yellow.colliderect(bullet):
            pygame.event.post(pygame.event.Event(YELLOW_HIT))
            red_bullets.remove(bullet)
        elif bullet.x > WIDTH:
            red_bullets.remove(bullet)

# The controller of the red_spaceship, 
# increasing its coordinates to move around but it cannot get pass the border
def red_spaceship_controller(key_pressed):
    if key_pressed[pygame.K_a] and red.x - VEL > 0:
            red.x -= VEL   
    if key_pressed[pygame.K_d] and red.x + VEL + SPACESHIP_WIDTH < BORDER.x:
            red.x += VEL  
    if key_pressed[pygame.K_w] and red.y - VEL > 0:
            red.y -= VEL
    if key_pressed[pygame.K_s] and red.y + VEL + SPACESHIP_HEIGHT < HEIGHT:
            red.y += VEL    

# The controller of the yellow
""" Speed controller of yellow spaceship """  
def yellow_spaceship_controller(key_pressed):
    if key_pressed[pygame.K_LEFT] and yellow.x - VEL > BORDER.x + BORDER.width:
            yellow.x -= VEL   
    if key_pressed[pygame.K_RIGHT] and yellow.x + VEL + SPACESHIP_WIDTH < WIDTH :
            yellow.x += VEL  
    if key_pressed[pygame.K_UP] and yellow.y - VEL > 0:
            yellow.y -= VEL
    if key_pressed[pygame.K_DOWN] and yellow.y + VEL + SPACESHIP_HEIGHT < HEIGHT:
            yellow.y += VEL