Drakora/Player.py

"""
Player entity class
"""


import pygame
import math
import os
from CollisionBox import CollisionBox


class Player(pygame.sprite.Sprite):
    imgDir = os.path.join(os.path.dirname(__file__), 'data')
    playerImage = pygame.image.load(os.path.join(imgDir, 'player.png'))#.convert()
    walkImages = (
        pygame.transform.scale(playerImage.subsurface((0, 0, 16, 24)), (64, 98)),
        pygame.transform.scale(playerImage.subsurface((16, 0, 16, 24)), (64, 98)),
        pygame.transform.scale(playerImage.subsurface((32, 0, 16, 24)), (64, 98)),
        pygame.transform.scale(playerImage.subsurface((48, 0, 16, 24)), (64, 98)),
    )

    upImages = (
        pygame.transform.scale(playerImage.subsurface((0, 24, 16, 24)), (64, 98)),
        pygame.transform.scale(playerImage.subsurface((16, 24, 16, 24)), (64, 98)),
        pygame.transform.scale(playerImage.subsurface((32, 24, 16, 24)), (64, 98)),
        pygame.transform.scale(playerImage.subsurface((48, 24, 16, 24)), (64, 98)),
    )

    downImages = (
        pygame.transform.scale(playerImage.subsurface((0, 48, 16, 24)), (64, 98)),
        pygame.transform.scale(playerImage.subsurface((16, 48, 16, 24)), (64, 98)),
        pygame.transform.scale(playerImage.subsurface((32, 48, 16, 24)), (64, 98)),
        pygame.transform.scale(playerImage.subsurface((48, 48, 16, 24)), (64, 98)),
    )

    crouchImages = (
        pygame.transform.scale(playerImage.subsurface((0, 72, 16, 16)), (64, 64)),
        pygame.transform.scale(playerImage.subsurface((16, 72, 16, 16)), (64, 64)),
        pygame.transform.scale(playerImage.subsurface((32, 72, 16, 16)), (64, 64)),
        pygame.transform.scale(playerImage.subsurface((48, 72, 16, 16)), (64, 64)),
    )

    for array in (walkImages, upImages, downImages, crouchImages):
        for image in array:
            image.set_colorkey((255,0,255))

    def getCollisionBoxes(self):
        return self.collisionBoxes

    def __init__(self):
        pygame.sprite.Sprite.__init__(self)

        self.currentWalkImage = 0
        self.currentUpImage = 0
        self.currentDownImage = 0
        self.currentCrouchImage = 0

        self.image = Player.downImages[self.currentDownImage]

        self.rect = self.image.get_rect()
        self.rect.center = (100, 400)
        self.speed = 0.0
        self.isJumping = False
        self.isCrouching = False
        self.hoverCount = 0
        self.isOnFloor = False
        self.isDownJump = False
        self.isDownCrouch = False
        self.buttonsJump = (pygame.K_UP, pygame.K_SPACE,)
        self.buttonsCrouch = (pygame.K_DOWN,)
        self.gameSpeed = 1
        self.updateCount = 0

        self.collisionBoxes = pygame.sprite.Group()

        collision = CollisionBox(0, 20, 60, 20, self.rect.center)
        self.collisionBoxes.add(collision)
        collision = CollisionBox(-10, 5, 30, 20, self.rect.center)
        self.collisionBoxes.add(collision)
        collision = CollisionBox(0, 35, 25, 40, self.rect.center)
        self.collisionBoxes.add(collision)


    def crouch(self):
        if not self.isCrouching:
            self.isCrouching = True
            self.rect = self.rect.inflate(0, -32)

            for i in self.collisionBoxes:
                i.rect.y -= 32


    def standup(self):
        if self.isCrouching:
            self.isCrouching = False
            self.rect = self.rect.inflate(0, 32)

            for i in self.collisionBoxes:
                i.rect.y += 32

    def updateSpeed(self, newGameSpeed):
        self.gameSpeed = newGameSpeed

    def control(self, event):
        if event.type == pygame.KEYDOWN:
            if event.key in self.buttonsCrouch:
                self.isDownCrouch = True
            elif event.key in self.buttonsJump:
                self.isDownJump = True

        elif event.type == pygame.KEYUP:
            if event.key in self.buttonsCrouch:
                self.isDownCrouch = False
            elif event.key in self.buttonsJump:
                self.isDownJump = False


    def update(self):
        self.updateCount += 1

        if not self.isDownJump:
            self.hoverCount = 0

        if self.isOnFloor:
            self.speed = 0

            if self.isDownJump:
                self.isJumping = True

                if self.isCrouching:
                    self.standup()

            elif self.isDownCrouch:
                if not self.isCrouching:
                    self.crouch()

            elif self.isCrouching:
                    self.standup()

        if self.isJumping:
            if self.gameSpeed <= 2:     maxHoverCount = 30
            elif self.gameSpeed <= 4:   maxHoverCount = 23
            elif self.gameSpeed <= 8:   maxHoverCount = 16
            elif self.gameSpeed <= 16:  maxHoverCount = 9
            elif self.gameSpeed <= 32:  maxHoverCount = 5
            elif self.gameSpeed <= 64:  maxHoverCount = 2
            else:                       maxHoverCount = 1

            if self.isDownJump and self.hoverCount < maxHoverCount:
                self.speed -= self.gameSpeed/8 * ((math.cos(2*math.pi*self.hoverCount/(2*maxHoverCount))+1)/2.5+0.2)
                self.hoverCount += 1
            else:
                self.isJumping = False
        else:
            self.speed += 0.07 * self.gameSpeed

        self.rect.y += self.speed

        for i in self.collisionBoxes:
            i.setY(self.rect.y)

        if self.updateCount >= 22 - math.log2(self.gameSpeed) * 2:
            if self.isOnFloor:
                if self.isCrouching:
                    self.currentCrouchImage += 1
                    if self.currentCrouchImage >= len(Player.crouchImages):
                        self.currentCrouchImage = 0
                    self.image = Player.crouchImages[self.currentCrouchImage]
                else:
                    self.currentWalkImage += 1
                    if self.currentWalkImage >= len(Player.walkImages):
                        self.currentWalkImage = 0
                    self.image = Player.walkImages[self.currentWalkImage]
            elif self.isJumping:
                self.currentUpImage += 1
                if self.currentUpImage >= len(Player.upImages):
                    self.currentUpImage = 0
                self.image = Player.upImages[self.currentUpImage]
            else:
                self.currentDownImage += 1
                if self.currentDownImage >= len(Player.downImages):
                    self.currentDownImage = 0
                self.image = Player.downImages[self.currentDownImage]
            self.updateCount = 0