import pygame
import numpy
import random
import math
pygame.init()
clock = pygame.time.Clock()
screen = pygame.display.set_mode((377, 377))
grid = numpy.zeros((15, 15), dtype="int8")
for i in range(4):
grid[7][2+i] = 2
apple_pos = [7, 11]
grid[apple_pos[0]][apple_pos[1]] = 5
snake = [[7, 2], [7, 5]]
d = [2, 2]
direction_list = []
keys = [pygame.K_UP, pygame.K_RIGHT, pygame.K_DOWN, pygame.K_LEFT]
colors = [(0, 150, 150), (0, 255, 0), (255, 0, 0)]
frames = 0
moving = False
while True:
frames += 1
for event in pygame.event.get():
if event.type == pygame.QUIT:
quit()
if event.type == pygame.KEYDOWN:
for i in range(len(keys)):
if event.key == keys[i]:
moving = True
direction_list.append(i+1)
if moving and frames >= 10:
frames = 0
if len(direction_list) > 0:
d[1] = direction_list[0]
direction_list.pop(0)
d[0] = grid[snake[0][0]][snake[0][1]]
last_pos = [snake[0][:], snake[1][:]]
for i in range(2):
snake[i][round((d[i] % 2-1)*-1)] += round(5/24*d[i]**4 - 25/12*d[i]**3 + 151/24*d[i]**2 - 65/12*d[i])
grid[last_pos[i][0]][last_pos[i][1]] = d[i] * i
if i:
for n in range(2):
if snake[i][n] < 0 or snake[i][n] > 14:
quit()
if 5 > grid[snake[i][0]][snake[i][1]] > 0:
quit()
if snake[i] == apple_pos:
grid[last_pos[0][0]][last_pos[0][1]] = d[0]
snake[0] = last_pos[0]
while grid[apple_pos[0]][apple_pos[1]] != 0:
apple_pos = [random.randint(0, 14), random.randint(0, 14)]
grid[apple_pos[0]][apple_pos[1]] = 5
grid[snake[i][0]][snake[i][1]] = d[1]
for row in range(len(grid)):
for pos in range(len(grid[row])):
pygame.draw.rect(screen, colors[round((math.log(grid[row][pos]+1))-0.15)], (25*pos + 2, 25*row + 2, 23, 23))
clock.tick(60)
pygame.display.update()