MainGame.py

import pygame
import numpy as np
import math
import os
from car import Car
from Player import Player
from CollisionManager import CollisionManager
from Menu import Menu


pygame.init()

## Set up the screen ##
LARGEUR = 1000
HAUTEUR = 600

## Load the car and the racetrack images ##
CAR1 = pygame.image.load(os.path.join(os.getcwd(), "./images/car.png"))
CAR2 = pygame.image.load(os.path.join(os.getcwd(), "./images/car2.png"))
w = CAR1.get_width()
h = CAR1.get_height()


START_IMAGE = pygame.image.load(os.path.join(os.getcwd(), "./images/start.png"))
START = pygame.transform.scale(START_IMAGE, (LARGEUR, HAUTEUR))

fontlapsP1P2 = pygame.font.Font(None, 20)

offset_initial_position = 30

## Main game class

class MainGame():
    def __init__(self):
        pygame.init()
        # Set up the screen
        self.screen = pygame.display.set_mode((LARGEUR, HAUTEUR))
        
        # Set up the music 
        pygame.mixer.init()
        pygame.mixer.music.load(os.path.join(os.getcwd(), "./sounds/TopGearMusic.mp3"))
        
        # Menu components
        self.menu = Menu(self.screen)
        
        self.selected_circuit = None
        self.laps = None
        
        # Ingame components
        self.initial_position = [0, 0]
        
        self.car1 = Car(600, 244, CAR1)
        self.car2 = Car(650, 244, CAR2)
        
        self.player1 = Player(self.car1, {'left': pygame.K_LEFT, 'right': pygame.K_RIGHT, 'up': pygame.K_UP, 'down': pygame.K_DOWN, 'brake': pygame.K_SPACE})
        self.player2 = Player(self.car2, {'left': pygame.K_q, 'right': pygame.K_d, 'up': pygame.K_z, 'down': pygame.K_s, 'brake': pygame.K_x})
        
        self.collision_manager1 = CollisionManager(self.car1, self.selected_circuit)
        self.collision_manager2 = CollisionManager(self.car2, self.selected_circuit)
        
        self.lapsP1 = 0
        self.lapsP2 = 0
        
        self.x1 = 0
        self.x2 = 0
        
        

    def run_menu(self, get_event=pygame.event.get):
        running = True
        while running:
            
            pygame.mixer.music.play(-1)
            
            self.menu.display_menu()
            
            if self.menu.selected_circuit is None:
                #print("Selected circuit: ", self.selected_circuit)    
                self.selected_circuit = self.menu.racetrack_input()
                self.initial_position = self.menu.initial_position
                self.player1.car.position = [self.initial_position[0], self.initial_position[1] + offset_initial_position]
                self.player2.car.position = [self.initial_position[0], self.initial_position[1]]

                     
            
            if self.menu.laps is None:
                self.laps = self.menu.select_laps()
                #print("Number of laps selected: ", self.laps)


                        
            if self.selected_circuit is not None and self.laps is not None:
                self.screen.blit(START, (0, 0))
                pygame.display.flip()
                pygame.time.wait(2000)
                running = False
                self.run_game()
            
                
            for event in get_event():    
                if event.type == pygame.QUIT:
                    running = False
                    break
                elif event.type == pygame.KEYDOWN:
                    if event.key == pygame.K_ESCAPE:
                        running = False
                        break
            
    
    def run_game(self):
        pygame.init()
        clock = pygame.time.Clock()
        running = True
        while running:
            
            # FPS limit
            clock.tick(60)
            
            ## Update car accordingly to the player 1 inputs ## 
            self.player1.ingame_inputs()
            self.player1.car.update()
            
            """ print("Velocity P1: ", self.player1.car.velocity)
            print()
            print("Position vector P1: ", self.player1.car.position)
            print()
            print("Angle P1: ", self.player1.car.angle) """
            
            ## Update car accordingly to the player 2 inputs ##
            self.player2.ingame_inputs()
            self.player2.car.update()
            
            """ print("Velocity P2: ", self.player2.car.velocity)
            print()
            print("Position vector P2: ", self.player2.car.position)
            print()
            print("Angle P2: ", self.player2.car.angle) """
            
            
            ## Check if a player has won ##
            if self.lapsP1 == self.laps:
                print("Player 1 wins")
                running = False
                break
            elif self.lapsP2 == self.laps:
                print("Player 2 wins")
                running = False
                break
            
            
            # Check for collisions
            if self.collision_manager1.check_boundary_collision(self.player1.car, self.selected_circuit):
                print("Collision P1")
                new_image = pygame.transform.rotate(self.player1.car.image, self.player1.car.angle)
                rect = new_image.get_rect(center=self.player1.car.position)
                self.screen.blit(self.car1.crash, rect)
                pygame.display.flip()
                pygame.time.wait(1000)
                # Raise an exception to stop the games
                raise Exception("Player 1 has collided with the boundary")

   
            if self.collision_manager2.check_boundary_collision(self.player2.car, self.selected_circuit):  
                print("Collision P2")
                new_image = pygame.transform.rotate(self.player2.car.image, self.player2.car.angle)
                rect = new_image.get_rect(center=self.player2.car.position)
                self.screen.blit(self.car2.crash, rect)
                pygame.display.flip()
                pygame.time.wait(1000)
                # Raise an exception to stop the game
                raise Exception("Player 2 has collided with the boundary")
            
            
            # Check for collisions between the two cars
            if self.collision_manager1.check_car_collisions(car1=self.player1.car, car2=self.player2.car):
                print("Collision P1 and P2")
               # Raise an exception to stop the game
                raise Exception("Player 1 and Player 2 have collided")
            
                    
            
            # Display the number of laps on the screen for each player
            # P1 laps
            if self.player1.car.position[0] < self.initial_position[0] - offset_initial_position :
                self.x1 = 1
                
            if self.player1.car.position[0] > self.initial_position[0] and self.x1 == 1:
                self.lapsP1 += 1
                self.x1 = 0
            
            
            # P2 laps
            if self.player2.car.position[0] < self.initial_position[0] - offset_initial_position :
                self.x2 = 1        
            
            if self.player2.car.position[0] > self.initial_position[0] and self.x2 == 1: 
                self.lapsP2 += 1
                self.x2 = 0
                

            # Clear the screen to erase the drag of the car 
            self.screen.fill((0, 0, 0))

            # display race track on the screen with .blit()
            self.screen.blit((self.selected_circuit), (0, 0))
            
            # displays the car on the track
            new_image = pygame.transform.rotate(self.player1.car.image, self.player1.car.angle)
            rect = new_image.get_rect(center=self.player1.car.position)
            self.screen.blit(new_image, rect)
            
            new_image = pygame.transform.rotate(self.player2.car.image, self.player2.car.angle)
            rect = new_image.get_rect(center=self.player2.car.position)
            self.screen.blit(new_image, rect)
            
            
            # Display the number of laps on the screen for each player
            lapsP1txt = fontlapsP1P2.render("Laps P1 : " + str(self.lapsP1), True, (255, 255, 255))                       
            lapsP2txt = fontlapsP1P2.render("Laps P2 : " + str(self.lapsP2), True, (255, 255, 255))
            
            self.screen.blit(lapsP1txt, (5, 5))
            self.screen.blit(lapsP2txt, (5, 25))
            
            # update the display
            pygame.display.flip()
            
                        
            ## Check for quiting events ##
            for event in pygame.event.get():
                    
                if event.type == pygame.QUIT:
                    running = False
                    break
                
                elif event.type == pygame.KEYDOWN:
                    
                    if event.key == pygame.K_ESCAPE:
                        running = False
                        break
            

            
game = MainGame()
game.run_menu()         ## Comment this for unit testing
pygame.mixer.music.stop()
pygame.quit()