import sys,pygame
from pygame.locals import *
pygame.init()
screen = pygame.display.set_mode((1024,768),0,0)
myfont = pygame.font.SysFont("monospace", 36)

def gen_map (w,h):
    global Matrix
    Matrix = [[(0) for x in range(w)] for x in range(h)]
def color_map(x,y):
    global Matrix
    Matrix[y][x] = 1
global  playtime
playtime =  0
formiga=[]
t = 0
def pos_direction_ant (x,y,d):
    global t
    formiga.append([])
    formiga[t].append(x)
    formiga[t].append(y)
    formiga[t].append(d)
    t+=1

def check (n):
    global formiga
    global inte
    global playtime
    print(n)
    pygame.draw.rect(background, (92, 0, 92), Rect((formiga[n][1] * 8, formiga[n][0] * 8), (8, 8)))
    if (Matrix[formiga[n][1]][formiga[n][0]] == 0 and playtime>= 0.01 ):
        Matrix[formiga[n][1]][formiga[n][0]] = 1
        formiga[n][2] = (formiga[n][2] + 1) % 4
        if formiga[n][2] == 0:
            formiga[n][0] -= 1
        if formiga[n][2] == 1:
            formiga[n][1] += 1
        if formiga[n][2] == 2:
            formiga[n][0] += 1
        if formiga[n][2] == 3:
            formiga[n][1] -= 1
        pygame.draw.rect(background, (92, 0, 92), Rect((formiga[n][1] * 8, formiga[n][0] * 8), (8, 8)))
        playtime = 0.0
        inte += 1
    if (Matrix[formiga[n][1]][formiga[n][0]] == 1 and playtime>= 0.01):
        Matrix[formiga[n][1]][formiga[n][0]] = 0
        formiga[n][2] = (formiga[n][2] - 1) % 4
        if formiga[n][2] == 0:
            formiga[n][0] -= 1
        if formiga[n][2] == 1:
            formiga[n][1] += 1
        if formiga[n][2] == 2:
            formiga[n][0] += 1
        if formiga[n][2] == 3:
            formiga[n][1] -= 1
        pygame.draw.rect(background, (92, 0, 92), Rect((formiga[n][1] * 8, formiga[n][0] * 8), (8, 8)))
        playtime = 0.0
        inte += 1

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

gen_map(96,128) #width,hieght
pos_direction_ant (50,50,1)
pos_direction_ant(20,20,0)# x position,y position, direction (1 up,2 right,3 down,4 left)
color_map(10,10)         #  color x,y rect if it needed
print(t)
iteration = 0
inte = 0
while iteration !=1000 :
    for event in pygame.event.get():
        if event.type == QUIT:
            pygame.quit()

    background = pygame.Surface((screen.get_size()))
    background.fill((255, 255, 255))
    milliseconds = clock.tick(FPS)
    seconds = milliseconds / 1000.0
    playtime += milliseconds / 1000.0

    screen.lock()

    for i in range(96):
        for j in range(128):
            if Matrix[j][i] == 0:
                pygame.draw.rect(background, (255, 255, 255), Rect((j * 8, i * 8), (8, 8)))
            else:
                pygame.draw.rect(background, (0, 0, 0), Rect((j * 8, i * 8), (8, 8)))

    i = 128
    while i != 0:
        pygame.draw.line(background, (92, 51, 23), (i * 8, 0), (i * 8, 768))
        i = i - 1

    i = 96
    while i != 0:
        pygame.draw.line(background, (92, 51, 23), (0, i * 8), (1024, i * 8))
        i = i - 1




    check(0)
    check(1)
    print(*formiga)
    strings = str(inte)
    label = myfont.render(strings, 1, (0, 0, 0))

    screen.unlock()
    screen.blit(background, (0, 0))
    screen.blit(label, (0, 0))
    pygame.display.update()
    iteration += 1