from physics.CollisionDirection import CollisionDirection
from physics.TickData import TickData
from sprite.DynamicSprite import DynamicSprite
from sprite.Spritesheet import Spritesheet
from ui_elements.KeyManager import KeyManager


class PlayerSprite(DynamicSprite):
    def __init__(self, spritesheet: Spritesheet):
        super().__init__(spritesheet)

        self.jump_time = -1
        self.allowed_jump_time = 12

        self.acceleration_horizontal = 2

        self.deceleration_horizontal_air = 0.2
        self.deceleration_horizontal_ground = 0.4
        self.gravity = 9.81 / 10

        self.max_motion_horizontal_via_input = 5

    def tick(self, tick_data: TickData):
        super().tick(tick_data)

        if tick_data.key_manager.is_keymap_down(KeyManager.KEY_RIGHT):
            if self.motion[0] < self.max_motion_horizontal_via_input:
                self.motion = (self.motion[0] + self.acceleration_horizontal, self.motion[1])

        if tick_data.key_manager.is_keymap_down(KeyManager.KEY_LEFT):
            if self.motion[0] > -self.max_motion_horizontal_via_input:
                self.motion = (self.motion[0] - self.acceleration_horizontal, self.motion[1])

        if tick_data.key_manager.is_keymap_down(KeyManager.KEY_UP):
            if self.jump_time < 0 and self.get_collides_with_direction(CollisionDirection.BOTTOM):
                self.jump_time = self.allowed_jump_time
                self.motion = (self.motion[0], self.motion[1] - 6)
            if self.jump_time >= 0:
                self.motion = (self.motion[0], self.motion[1] - 0.55)
        if self.jump_time >= 0:
            self.jump_time -= 1