import pygame
import Flower
import Beehive
import Frames


white = (255, 255, 255)

#setting caption and icon
pygame.display.set_caption('Smart Bee')
setIcon = pygame.image.load('spritesNtiles/bee64.png')
pygame.display.set_icon(setIcon)

#
###
########################################


#SHORT = pygame.image.load('spritesNtiles/shortGrass64.png') #0
#TALL = pygame.image.load('spritesNtiles/tallGrass64.png') #1
#TREE = pygame.image.load('spritesNtiles/dirtTree64.png') #10


### tilemap
SHORT = 0
TALL = 1
TREE = 10
BEE = pygame.image.load('spritesNtiles/bee64.png')

tiles = {
    SHORT: pygame.image.load('spritesNtiles/shortGrass64.png'),
    TALL: pygame.image.load('spritesNtiles/tallGrass64.png'),
    TREE: pygame.image.load('spritesNtiles/dirtTree64.png')
}

tilemap = [
    [SHORT, SHORT, SHORT, TREE, SHORT, TALL, TALL, SHORT, SHORT, TREE, SHORT, SHORT, TALL, SHORT, SHORT],
    [SHORT, SHORT, SHORT, TALL, TALL, SHORT, SHORT, TREE, SHORT, SHORT, SHORT, SHORT, SHORT, SHORT, SHORT],
    [SHORT, SHORT, SHORT, SHORT, TALL, SHORT, SHORT, SHORT, SHORT, SHORT, SHORT, SHORT, TALL, SHORT, SHORT],
    [TALL, TREE, SHORT, SHORT, SHORT, TREE, SHORT, SHORT, SHORT, SHORT, SHORT, TALL, SHORT, SHORT, TREE],
    [SHORT, SHORT, SHORT, SHORT, SHORT, SHORT, SHORT, TREE, SHORT, TREE, SHORT, SHORT, SHORT, SHORT, SHORT],
    [SHORT, SHORT, SHORT, SHORT, TALL, SHORT, SHORT, SHORT, SHORT, TALL, SHORT, SHORT, SHORT, SHORT, TALL],
    [SHORT, SHORT, TALL, SHORT, SHORT, SHORT, SHORT, SHORT, TREE, SHORT, SHORT, TREE, SHORT, SHORT, SHORT],
    [SHORT, TREE, SHORT, TREE, SHORT, SHORT, SHORT, SHORT, TALL, TALL, SHORT, SHORT, SHORT, TALL, SHORT],
    [SHORT, SHORT, SHORT, SHORT, TALL, SHORT, SHORT, SHORT, SHORT, TREE, SHORT, SHORT, SHORT, SHORT, SHORT],
    [TALL, SHORT, SHORT, SHORT, SHORT, SHORT, TREE, SHORT, SHORT, SHORT, SHORT, SHORT, SHORT, SHORT, TALL],
    [SHORT, SHORT, SHORT, SHORT, SHORT, TREE, SHORT, SHORT, SHORT, TALL, SHORT, SHORT, SHORT, SHORT, SHORT],
    [SHORT, SHORT, SHORT, SHORT, TALL, SHORT, SHORT, SHORT, SHORT, SHORT, SHORT, SHORT, TALL, SHORT, SHORT],

]
tilemapSizeY = 12
tilemapSizeX = 15
tileSize = 64
###


#calculate the window size
disX = tilemapSizeX * tileSize
disY = tilemapSizeY * tileSize

# tilemap = [
#     [0, 0, 0, 10, 0, 1, 1, 0, 0, 10],
#     [0, 0, 0, 1, 1, 0, 0, 10, 0, 0],
#     [0, 0, 0, 0, 1, 0, 0, 0, 0, 0],
#     [1, 10, 0, 0, 0, 10, 0, 0, 0, 0],
#     [0, 0, 0, 0, 0, 0, 0, 10, 0, 10],
#     [0, 0, 0, 0, 1, 0, 0, 0, 0, 1],
#     [0, 0, 1, 0, 0, 0, 0, 0, 10, 0],
#     [0, 10, 0, 10, 0, 0, 0, 0, 1, 1],
#     [0, 0, 0, 0, 1, 0, 0, 0, 0, 10],
#     [1, 0, 0, 0, 0, 0, 10, 0, 0, 0]
# ]


pygame.init()
dis = pygame.display.set_mode((disX, disY))
clock = pygame.time.Clock()

#position of the bee and pos changings
XChanges = 0
XPosBee = 64

YChanges = 0
YPosBee = 64

while True:

    for event in pygame.event.get():
        #check to quit
        if (event.type == pygame.QUIT) or (event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE):
            # game_over = True
            pygame.quit()
            quit()

    #printing map
    for i in range(len(tilemap)):
        for j in range(len(tilemap[i])):
            dis.blit(tiles[tilemap[i][j]], (j * tileSize, i * tileSize))

        #movement speed
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_UP:
                XChanges = 0
                YChanges = -16
            if event.key == pygame.K_DOWN:
                XChanges = 0
                YChanges = 16
            if event.key == pygame.K_LEFT:
                XChanges = -16
                YChanges = 0
            if event.key == pygame.K_RIGHT:
                XChanges = 16
                YChanges = 0
        #stopmovement when a key is up
        if event.type == pygame.KEYUP:
            XChanges = 0
            YChanges = 0
    #calculating position of the bee
    if disX >= XPosBee >= 0:
        XPosBee = XPosBee + XChanges
    if disY >= YPosBee >= 0:
        YPosBee = YPosBee + YChanges

    #left and up restriction
    if XPosBee <= 0:
        XPosBee = 0
    elif XPosBee >= disX:
        XPosBee = disX
    elif YPosBee <= 0:
        YPosBee = 0
    elif YPosBee >= disY:
        YPosBee = disY

    #beatiful bee
    dis.blit(BEE, (XPosBee, YPosBee))

    #updating
    # pygame.display.flip()
    pygame.display.update()
    clock.tick(10000)