player.py

from circleshape import CircleShape
from constants import *
import pygame
from shot import Shot


# inherited class from CircleShape
class Player(CircleShape):
    def __init__(self, x, y):
        # call the parent constructor
        super().__init__(x, y, PLAYER_RADIUS)
        self.rotation = 0
        self.cooldown_timer = 0
    
    
    def triangle(self):
        forward = pygame.Vector2(0, 1).rotate(self.rotation)
        right = pygame.Vector2(0, 1).rotate(self.rotation + 90) * self.radius / 1.5
        a = self.position + forward * self.radius
        b = self.position - forward * self.radius - right
        c = self.position - forward * self.radius + right
        return [a, b, c]
    
    def draw(self, screen):
        # draw the triangle
        pygame.draw.polygon(screen, (255, 255, 255), self.triangle(), width=2)
    
    def rotate(self, dt):
        # rotate the player
        return self.rotation + dt * PLAYER_TURN_SPEED
    
    def update(self, dt):
        keys = pygame.key.get_pressed()
        # check for key presses
        
        if keys[pygame.K_a]:
            # rotate left
            self.rotation = self.rotate(-dt)

        if keys[pygame.K_d]:
            # rotate right
            self.rotation = self.rotate(dt)

        if keys[pygame.K_w]:
            # move forward
            self.move(dt)

        if keys[pygame.K_s]:
            # move backward
            self.move(-dt)

        if keys[pygame.K_SPACE]:
            # shoot
            if self.cooldown_timer > 0:
                self.cooldown_timer -= dt
                return
            self.shoot()
            self.cooldown_timer = PLAYER_SHOOT_COOLDOWN

    def move(self, dt):
            # move the player
            forward = pygame.Vector2(0, 1).rotate(self.rotation)
            self.position += forward * PLAYER_SPEED * dt

    def shoot(self):
        # shoot a bullet
        forward = pygame.Vector2(0, 1).rotate(self.rotation)
        shot = Shot(self.position.x, self.position.y, SHOT_RADIUS)
        shot.velocity = forward * PLAYER_SHOT_SPEED  
        # The containers will handle adding it to the right groups