import pygame
from setting import pos_bullet_list


class Bullet(pygame.sprite.Sprite):
    def __init__(self, screen, pos_index):
        super(Bullet, self).__init__()
        self.screen = screen
        self.pos_index = pos_index
        self.pos_x = pos_bullet_list[self.pos_index][0]
        self.pos_y = pos_bullet_list[self.pos_index][1]
        self.image_index = 0
        self.image_list = []
        # self.image = ["red", "green", "blue", "yellow"]
        for i in range(6):
            image_filename = "./images/rest/bullet" + str(i) + ".png"
            image = pygame.image.load(image_filename)
            image = pygame.transform.scale(image, (image.get_width()//4, image.get_height()//4))
            self.image_list.append(image)
        # self.rect = pygame.Rect(self.pos_x, self.pos_y, 50, 45)
        self.rect = self.image_list[self.image_index].get_rect(left=self.pos_x, top=self.pos_y)
        self.frequency = 0
        self.MAX_FREQUENCY = 2

    def move(self):
        self.frequency += 1
        if self.frequency == self.MAX_FREQUENCY:
            self.frequency = 0
            self.pos_x += 100  # 30
            self.rect.x = self.pos_x
            if self.rect.x > 1200:
                self.kill()

    def get_pos(self):
        return self.pos_x, self.pos_y

    def update(self):
        # screen.fill(self.image[self.image_index], self.rect)
        # self.screen.blit(self.image_list[self.pos_index], (self.pos_x, self.pos_y))
        self.move()
        self.image = self.image_list[self.image_index]
        self.rect.topleft = (self.pos_x, self.pos_y)


class ReturnBullet(pygame.sprite.Sprite):
    def __init__(self, pos_x, pos_y):
        super(ReturnBullet, self).__init__()
        self.pos_x = pos_x - 20
        self.pos_y = pos_y
        self.image_index = 0
        self.image_list = []
        # self.image = ["red", "green", "blue", "yellow"]
        for i in range(3):
            image_filename = "./images/rest/return_bullet" + str(i) + ".png"
            image = pygame.image.load(image_filename)
            # image = pygame.transform.scale(image, (image.get_width()//4, image.get_height()//4))
            self.image_list.append(image)
        # self.rect = pygame.Rect(self.pos_x, self.pos_y, 50, 45)
        self.rect = self.image_list[self.image_index].get_rect(left=self.pos_x, top=self.pos_y)
        self.frequency = 0
        self.MAX_FREQUENCY = 2

    def move(self):
        self.image_index += 1
        if self.image_index >= 3:
            self.image_index = 0
        self.frequency += 1
        if self.frequency == self.MAX_FREQUENCY:
            self.frequency = 0
            self.pos_x -= 50  # 30
            self.rect.x = self.pos_x
            if self.rect.x < 90:
                self.kill()

    def get_pos(self):
        return self.pos_x

    def update(self):
        # screen.fill(self.image[self.image_index], self.rect)
        self.move()
        self.image = self.image_list[self.image_index]
        self.rect.topleft = (self.pos_x, self.pos_y)
        # self.screen.blit(self.image_list[self.image_index], (self.pos_x, self.pos_y))