import pygame

from scripts.animation import Animation


class Particle:
    def __init__(self, game, p_type, pos: tuple, velocity=[0, 0], frame=0):
        self.game = game
        self.p_type = p_type
        self.pos = list(pos)
        self.velocity = list(velocity)
        self.frame = frame

        self.animation: Animation = self.game.assets["particles/" + self.p_type].copy()
        self.animation.frame = frame

    def update(self) -> bool:

        kill = False

        if self.animation.done:
            kill = True

        self.pos[0] += self.velocity[0]
        self.pos[1] += self.velocity[1]
        self.animation.update()

        return kill

    def render(self, surface: pygame.Surface, offset: tuple[float, float]=(0, 0)):
        img = self.animation.img()
        surface.blit(img, (
            self.pos[0] - offset[0] - img.get_width() // 2,
            self.pos[1] - offset[1] - img.get_height() // 2,
        ))