import pygame
import sprites
import setup
import functions

pygame.init()

SCREEN_WIDTH = 1800
SCREEN_HEIGHT = 1000

screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
clock = pygame.time.Clock()

player = pygame.sprite.GroupSingle()
player.add(sprites.Player())

streets_scene_1 = pygame.sprite.Group()
streets_list_scene_1 = setup.setup_streets_scene_1()
streets_scene_2 = pygame.sprite.Group()
streets_list_scene_2 = setup.setup_streets_scene_2()


lanterns_scene_1 = pygame.sprite.Group()
detectors_scene_1 = pygame.sprite.Group()
lantern_list_scene_1, detectors_list_scene_1 = setup.setup_lanterns_detectors_scene_1()

lanterns_scene_2 = pygame.sprite.Group()
detectors_scene_2 = pygame.sprite.Group()
lantern_list_scene_2, detectors_list_scene_2 = setup.setup_lanterns_detectors_scene_2()

current_scene = "scene_1"
    
for item in lantern_list_scene_1:
    lanterns_scene_1.add(item)
for item in detectors_list_scene_1:
    detectors_scene_1.add(item)
for item in streets_list_scene_1:
    streets_scene_1.add(item)
for item in streets_list_scene_2:
    streets_scene_2.add(item)
for item in lantern_list_scene_2:
    lanterns_scene_2.add(item)
for item in detectors_list_scene_2:
    detectors_scene_2.add(item)



if __name__ == "__main__":
    run = True
    while run:
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                run = False
        
        if current_scene == "scene_1":
            functions.scene_1(screen, detectors_scene_1, streets_scene_1, lanterns_scene_1, player)
        if current_scene == "scene_2":
            functions.scene_2(screen, detectors_scene_2, streets_scene_2, lanterns_scene_2, player)

        current_scene = player.sprite.left_scene(current_scene)

        pygame.display.update()
        clock.tick(60)

    pygame.quit()