# Creating Stages (more like the part 2 of creating sprites - Shooting game)


import pygame,sys, random

class Crosshair(pygame.sprite.Sprite):
    def __init__(self,picture_path):
        super().__init__()
        self.image = pygame.image.load(picture_path)
        self.rect = self.image.get_rect()
        self.gunshot = pygame.mixer.Sound("Pygame Tasks/gunshot.mp3")
    
    def shoot(self):
        self.gunshot.play()
        pygame.sprite.spritecollide(crosshair,target_group, True)
        
    def update(self):
        self.rect.center = pygame.mouse.get_pos()

class Target(pygame.sprite.Sprite):
    def __init__(self, picture_path, pos_x, pos_y):
        super().__init__()
        self.image = pygame.image.load(picture_path)
        self.rect = self.image.get_rect()
        self.rect.center = [pos_x, pos_y]

class GameState():
    def __init__(self):
        self.state = "intro"

    def intro(self):
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()

            if event.type == pygame.MOUSEBUTTONDOWN:
                self.state = 'main_game'

        #Draw background tiles
        for y in range(num_tiles_y):
            for x in range(num_tiles_x):
                screen.blit(background, (x * background.get_width(), y *background.get_height()))
                screen.blit(ready_text, (880, screen_height / 2))
        
        crosshair_group.draw(screen)
        crosshair_group.update()

        pygame.display.flip()

    def main_game(self):
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                pygame.quit()
                sys.exit()

            if event.type == pygame.MOUSEBUTTONDOWN:
                crosshair.shoot()

            #Draw nackground tiles
        for y in range(num_tiles_y):
            for x in range(num_tiles_x):
                screen.blit(background, (x * background.get_width(), y *background.get_height()))
    
        target_group.draw(screen)
        crosshair_group.draw(screen)
        crosshair_group.update()

        pygame.display.flip()

    def state_manager(self):
        if self.state == 'intro':
            self.intro()
        if self.state == 'main_game':
            self.main_game()

#general setup
pygame.init()
clock = pygame.time.Clock()
game_state = GameState()

#game screen
screen_width = 1910
screen_height = 1030
screen = pygame.display.set_mode((screen_width,screen_height))
background = pygame.image.load("Pygame Tasks/bg_blue.png")
ready_text = pygame.image.load("Pygame Tasks/text_ready.png")
pygame.mouse.set_visible(False)

#crosshair
crosshair = Crosshair("Pygame Tasks/crosshair_red_large.png")
crosshair_group = pygame.sprite.Group()
crosshair_group.add(crosshair)

#calculate number of tiles needed to cover the screen
num_tiles_x = (screen_width // background.get_width()) + 1
num_tiles_y = (screen_height // background.get_height()) + 1


#Target
target_group = pygame.sprite.Group()
for target in range(20):
    new_target = Target("Pygame Tasks/target_red3_outline.png", random.randrange(0, screen_width), random.randrange(0,screen_height))
    target_group.add(new_target)

while True:
    game_state.state_manager()
    clock.tick(60)