snake_coletor.py

import pygame, random
import json
import time
from pygame.locals import *

NAME = 'weuler'

BOARD_SIZE = 600

TILE_SIZE = 20

def on_grid_random():
    x = random.randint(TILE_SIZE,570)
    y = random.randint(TILE_SIZE,570)
    return (x//TILE_SIZE * TILE_SIZE, y//TILE_SIZE * TILE_SIZE)

def collision(c1, c2):
    return (c1[0] == c2[0]) and (c1[1] == c2[1])

def walk(head, vector):
    '''
    Given a head position and a vector, walks the snake by 10px on that direction
    '''
    return (head[0] + TILE_SIZE * vector[0], head[1] + TILE_SIZE * vector[1])

def get_ambient_data(head, food, body):
    '''
    Given head position (tuple x,y) , check for:
        - Distance to wall
        - Distance to body (if found)
        - If there is food in that direction (boolean)

    Returns :
        - 1 / Distance to wall
        - 1 / Distance to body
        - 1 or 0 depending on food on that direction
    
    Convention: first 3 is UP, then RIGHT, DOWN, LEFT
    
    This will generate 12 values that will be input into the RN
    '''
    # answer = [0,0,0,0,0,0,0,0,0,0,0,0]

    # #distance to top wall
    # answer[0] = 1/head[1]
    # #distance to right wall
    # answer[3] = 1/(600 - head[0])
    # #distance to down wall
    # answer[6] = 1/(600 - head[1])
    # #distance to left wall
    # answer[9] = 1/(head[0])

    # for b in body[1:]:
    #     if b[0] == head[0] and b[1] > head[1]:
    #         #body found above head
    #         answer[1] = 1/(b[1] - head[1])
        
    #     if b[0] == head[0] and b[1] < head[1]:
    #         #body found below head
    #         answer[7] = 1/(head[1] - b[1])
        
    #     if b[1] == head[1] and b[0] > head[0]:
    #         #body found right to head
    #         answer[4] = 1/(b[0] - head[0])
        
    #     if b[1] == head[1] and b[0] < head[0]:
    #         #body found left to head
    #         answer[10] = 1/(head[0] - b[0])

    # if food[0] == head[0] and food[1] > head[1]:
    #     #food found above head
    #     answer[2] = 1
    
    # if food[0] == head[0] and food[1] < head[1]:
    #     #food found below head
    #     answer[8] = 1
    
    # if food[1] == head[1] and food[0] > head[0]:
    #     #food found right to head
    #     answer[5] = 1
    
    # if food[1] == head[1] and food[0] < head[0]:
    #     #food found left to head
    #     answer[11] = 1

    # return answer

    answer = [0,0,0,0]

    answer[0] = head[0]/600
    answer[1] = head[1]/600
    answer[2] = food[0]/600
    answer[3] = food[1]/600
    
    return answer

UP = 0
RIGHT = 1
DOWN = 2
LEFT = 3

pygame.init()
screen = pygame.display.set_mode((600,600))
pygame.display.set_caption('Snake')

snake = [(200, 200), (200 + TILE_SIZE, 200), (200 + TILE_SIZE * 2,200)]
snake_skin = pygame.Surface((TILE_SIZE,TILE_SIZE))
snake_skin.fill((25,255,25))

wall = pygame.Surface((TILE_SIZE,TILE_SIZE))
wall.fill((255,255,255))

apple_pos = on_grid_random()
apple = pygame.Surface((TILE_SIZE,TILE_SIZE))
apple.fill((255,0,0))

font = pygame.font.SysFont('Arial', 30)
score_text = font.render('Score: ', False, (255,255,255))

my_direction = LEFT

clock = pygame.time.Clock()

score = 0

report = []

game_over = False

while True:
    clock.tick(10)
    for event in pygame.event.get():
        if event.type == QUIT:
            pygame.quit()

        if event.type == KEYDOWN:
            if event.key == K_UP and my_direction != DOWN:
                my_direction = UP
            if event.key == K_DOWN and my_direction != UP:
                my_direction = DOWN
            if event.key == K_LEFT and my_direction != RIGHT:
                my_direction = LEFT
            if event.key == K_RIGHT and my_direction != LEFT:
                my_direction = RIGHT

    if collision(snake[0], apple_pos):
        score+=1
        apple_pos = on_grid_random()
        snake.append((0,0))

    for i in range(len(snake) - 1, 0, -1):
        snake[i] = (snake[i-1][0], snake[i-1][1])

    info = []

    if my_direction == UP:
        snake[0] = (snake[0][0], snake[0][1] - TILE_SIZE)
        info.append([1,0,0,0])
    if my_direction == DOWN:
        snake[0] = (snake[0][0], snake[0][1] + TILE_SIZE)
        info.append([0,0,1,0])
    if my_direction == RIGHT:
        snake[0] = (snake[0][0] + TILE_SIZE, snake[0][1])
        info.append([0,1,0,0])
    if my_direction == LEFT:
        snake[0] = (snake[0][0] - TILE_SIZE, snake[0][1])
        info.append([0,0,0,1])

    for b in snake[1:]:
        if collision(snake[0], b):
            #collision between head and body
            game_over = True
            break
    
    if (snake[0][0] <= 0 or snake[0][0] >= 600 - TILE_SIZE or snake[0][1] <= 0 or snake[0][1] >= 600 - TILE_SIZE):
        game_over = True

    if game_over:
        break

    info.append(get_ambient_data(snake[0], apple_pos, snake))

    screen.fill((0,0,0))
    screen.blit(apple, apple_pos)
    for pos in snake:
        screen.blit(snake_skin,pos)
    
    for x in range(0,600,TILE_SIZE):
        screen.blit(wall,(0, x))
        screen.blit(wall,(600 - TILE_SIZE, x))
        screen.blit(wall,(x, 0))
        screen.blit(wall,(x, 600 - TILE_SIZE))
    
    score_value = font.render(str(score), False, (255,255,255))
    screen.blit(score_text,(20,550))
    screen.blit(score_value,(100,550))
    
    info.reverse()
    report.append(info)
    pygame.display.update()

out_name = 'base-' + NAME + '-' + str(int(time.time()))
print('Game info to %s' % out_name)
with open(out_name, 'w') as out:
    json.dump(report, out)