import pygame
import button

pygame.init()

clock = pygame.time.Clock()
FPS = 60

#game window
SCREEN_WIDTH = 800
SCREEN_HEIGHT = 640
LOWER_MARGIN = 100
SIDE_MARGIN = 300

screen = pygame.display.set_mode((SCREEN_WIDTH + SIDE_MARGIN, SCREEN_HEIGHT + LOWER_MARGIN))
pygame.display.set_caption('Level Editor')

#deaf game variables ("sorry what was that? I can't hear you.")
ROWS = 16
MAX_COLS = 150 
TILE_SIZE = SCREEN_HEIGHT // ROWS
TILE_TYPES = 21
scroll_left = False
scroll_right = False
scroll = 0 
scroll_speed = 1

#def some color needed
GREEN = (144, 201, 120)
WHITE = (255, 255, 255)
RED = (200, 25, 25)

#load images
pine1_img = pygame.image.load('level editor/background/pine1.png').convert_alpha()
pine2_img = pygame.image.load('level editor/background/pine2.png').convert_alpha()
mountain_img = pygame.image.load('level editor/background/mountain.png').convert_alpha()
sky_img = pygame.image.load('level editor/background/sky_cloud.png').convert_alpha()
#storing them tiles in that double chest yk
img_list = []
for x in range(TILE_TYPES):
    img = pygame.image.load(f'level editor/tile/{x}.png')
    img = pygame.transform.scale(img, (TILE_SIZE, TILE_SIZE))
    img_list.append(img)

#creating functions for drawing the background
def draw_bg():
    screen.fill(GREEN)
    width = sky_img.get_width()
    for x in range(4):
        screen.blit(sky_img, ((x * width) - scroll * 0.5, 0))
        screen.blit(mountain_img, ((x * width) - scroll * 0.6, SCREEN_HEIGHT - mountain_img.get_height() - 300))
        screen.blit(pine1_img, ((x * width) - scroll * 0.7, SCREEN_HEIGHT - pine1_img.get_height() - 150))
        screen.blit(pine2_img, ((x * width) - scroll * 0.8, SCREEN_HEIGHT - pine2_img.get_height()))

#draw grid. But with what stationary?
def draw_grid():
    #verticval lines. wheres the horizontal ones?
    for c in range(MAX_COLS + 1):
        pygame.draw.line(screen, WHITE, (c * TILE_SIZE - scroll, 0), (c * TILE_SIZE - scroll, SCREEN_HEIGHT))
    #Here they are!
    for c in range(ROWS + 1):
        pygame.draw.line(screen, WHITE, (0, c * TILE_SIZE), (SCREEN_WIDTH, c * TILE_SIZE))

#creating me some buttons
#Listing them buttons 
button_list = []
button_col = 0
button_row = 0
for i in range(len(img_list)):
    tile_button = button.Button(SCREEN_WIDTH + (75 * button_col) + 50, 75 * button_row + 50, img_list[i], 1)
    button_list.append(tile_button)
    button_col += 1
    if button_col == 3:
        button_row += 1
        button_col = 0



run = True
while run:

    clock.tick(FPS)

    draw_bg()
    draw_grid()
    
    #draw them panels (its for the tiles)
    pygame.draw.rect(screen, GREEN, (SCREEN_WIDTH, 0, SCREEN_HEIGHT, 0))
    
    

    #map the scroll
    if scroll_left == True and scroll > 0:
        scroll -= 5 * scroll_speed
    if scroll_right == True:
        scroll += 5 * scroll_speed


    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            run = False

        #Keyboard presses. oh can you play fur elise?
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_LEFT:
                scroll_left = True
            if event.key == pygame.K_RIGHT:
                scroll_right = True
            if event.key == pygame.K_RSHIFT:
                scroll_speed = 5

        if event.type == pygame.KEYUP:
            if event.key == pygame.K_LEFT:
                scroll_left = False
            if event.key == pygame.K_RIGHT:
                scroll_right = False
            if event.key == pygame.K_RSHIFT:
                scroll_speed = 1


    pygame.display.update()
    
pygame.quit()