import pygame

from PositionScale import PositionScale


class Sprite:
    def __init__(self, spritesheet):
        self.spritesheet = spritesheet

        self.animation_state = list(self.spritesheet.animations.keys())[0]
        self.animation_frame = 0
        self.animation_delay = 0

        self.position_scale = PositionScale()

        self.visible = True
        self.animated = True

        self.image = None

    def tick(self, dt):
        animation = self.spritesheet.animations[self.animation_state]

        if self.animated:
            self.animation_delay += dt

            while self.animation_delay >= animation["delays"][self.animation_frame % len(animation["delays"])]:
                self.animation_delay -= animation["delays"][self.animation_frame % len(animation["delays"])]
                self.animation_frame = (self.animation_frame + 1) % len(animation["images"])

        self.image = animation["images"][self.animation_frame % len(animation["images"])]

    def set_animation_state(self, state):
        if state in self.spritesheet.animations:
            self.animation_state = state
            self.animation_delay = 0
            self.tick(0)

    def set_animation_frame(self, frame):
        self.animation_frame = frame
        self.tick(0)

    def draw(self, screen, screen_transform):
        if not self.visible:
            return

        if self.image is not None:
            target_position = screen_transform.apply_scale_to_position()

            target_scale = (
                screen_transform.scale[0] * self.position_scale.scale[0],
                screen_transform.scale[1] * self.position_scale.scale[1]
            )
            target_size = (
                int(target_scale[0] * self.image.get_width()),
                int(target_scale[1] * self.image.get_height())
            )

            target_image = self.get_scaled_image(self.image, target_size)

            screen.blit(target_image, target_position)

    def get_scaled_image(self, image, resize):
        return pygame.transform.scale(image, resize)

    def dump(self, file):
        # re-attach all the images to a single surface and save it to a file
        width = 0
        height = 0

        for animation in self.spritesheet.animations.values():
            max_height = 0
            total_width = 0
            for image in animation["images"]:
                total_width += image.get_width()
                max_height = max(max_height, image.get_height())
            width = max(width, total_width)
            height += max_height

        sheet = pygame.Surface((width, height))

        x = 0
        y = 0
        for animation in self.spritesheet.animations.values():
            max_height = 0
            for image in animation["images"]:
                sheet.blit(image, (x, y))
                x += image.get_width()
                max_height = max(max_height, image.get_height())
            y += max_height
            x = 0

        pygame.image.save(sheet, file)