277 lines
8.0 KiB
Python
277 lines
8.0 KiB
Python
|
# -----------------------------------------------------------------------------
|
||
|
|
#
|
||
|
|
# Snake - 2D
|
||
|
|
# Language - Python
|
||
|
|
# Modules - pygame, sys, random, copy, time
|
||
|
|
#
|
||
|
|
# Controls - Arrow Keys
|
||
|
|
#
|
||
|
|
# By - Jatin Kumar Mandav
|
||
|
|
#
|
||
|
|
# Website - https://jatinmandav.wordpress.com
|
||
|
|
#
|
||
|
|
# YouTube Channel - https://www.youtube.com/channel/UCdpf6Lz3V357cIZomPwjuFQ
|
||
|
|
# Twitter - @jatinmandav
|
||
|
|
#
|
||
|
|
# -----------------------------------------------------------------------------
|
||
|
|
import pygame
|
||
|
|
import sys
|
||
|
|
import copy
|
||
|
|
import random
|
||
|
|
import time
|
||
|
|
|
||
|
|
pygame.init()
|
||
|
|
|
||
|
|
width = 500
|
||
|
|
height = 500
|
||
|
|
scale = 10
|
||
|
|
score = 0
|
||
|
|
|
||
|
|
food_x = 10
|
||
|
|
food_y = 10
|
||
|
|
|
||
|
|
display = pygame.display.set_mode((width, height))
|
||
|
|
pygame.display.set_caption("Snake Game")
|
||
|
|
clock = pygame.time.Clock()
|
||
|
|
|
||
|
|
background = (23, 32, 42)
|
||
|
|
snake_colour = (236, 240, 241)
|
||
|
|
food_colour = (148, 49, 38)
|
||
|
|
snake_head = (247, 220, 111)
|
||
|
|
|
||
|
|
|
||
|
|
# ----------- Snake Class ----------------
|
||
|
|
# self.history[0][0] is the location of the head of the snake
|
||
|
|
|
||
|
|
class Snake:
|
||
|
|
def __init__(self, x_start, y_start):
|
||
|
|
self.x = x_start
|
||
|
|
self.y = y_start
|
||
|
|
self.w = 10
|
||
|
|
self.h = 10
|
||
|
|
self.x_dir = 1
|
||
|
|
self.y_dir = 0
|
||
|
|
self.history = [[self.x, self.y]]
|
||
|
|
self.length = 1
|
||
|
|
|
||
|
|
def reset(self):
|
||
|
|
self.x = width/2-scale
|
||
|
|
self.y = height/2-scale
|
||
|
|
self.w = 10
|
||
|
|
self.h = 10
|
||
|
|
self.x_dir = 1
|
||
|
|
self.y_dir = 0
|
||
|
|
self.history = [[self.x, self.y]]
|
||
|
|
self.length = 1
|
||
|
|
|
||
|
|
#function to show the body of snake
|
||
|
|
def show(self):
|
||
|
|
for i in range(self.length):
|
||
|
|
if not i == 0:
|
||
|
|
pygame.draw.rect(display, snake_colour, (self.history[i][0], self.history[i][1], self.w, self.h))
|
||
|
|
else:
|
||
|
|
pygame.draw.rect(display, snake_head, (self.history[i][0], self.history[i][1], self.w, self.h))
|
||
|
|
|
||
|
|
|
||
|
|
def check_eaten(self):
|
||
|
|
if abs(self.history[0][0] - food_x) < scale and abs(self.history[0][1] - food_y) < scale:
|
||
|
|
return True
|
||
|
|
|
||
|
|
def grow(self):
|
||
|
|
self.length += 1
|
||
|
|
self.history.append(self.history[self.length-2])
|
||
|
|
|
||
|
|
def death(self):
|
||
|
|
i = self.length - 1
|
||
|
|
while i > 0:
|
||
|
|
if abs(self.history[0][0] - self.history[i][0]) < self.w and abs(self.history[0][1] - self.history[i][1]) < self.h and self.length > 2:
|
||
|
|
return True
|
||
|
|
i -= 1
|
||
|
|
|
||
|
|
def update(self):
|
||
|
|
i = self.length - 1
|
||
|
|
while i > 0:
|
||
|
|
self.history[i] = copy.deepcopy(self.history[i-1])
|
||
|
|
i -= 1
|
||
|
|
self.history[0][0] += self.x_dir*scale
|
||
|
|
self.history[0][1] += self.y_dir*scale
|
||
|
|
|
||
|
|
def autoplay(self):
|
||
|
|
|
||
|
|
if abs(food_x-self.history[0][0]) < 10 and abs(food_y-self.history[0][1]) < 10:
|
||
|
|
# if self.check_eaten():
|
||
|
|
# food.new_location()
|
||
|
|
# score += 1
|
||
|
|
# self.grow()
|
||
|
|
print("")
|
||
|
|
elif abs(food_x-self.history[0][0]) < 10:
|
||
|
|
|
||
|
|
# if self.y_dir==1 or self.y_dir==-1:
|
||
|
|
# self.y_dir=0
|
||
|
|
# self.x_dir=1
|
||
|
|
if self.x_dir==1 or self.x_dir==-1:
|
||
|
|
if food_y>self.history[0][1]:
|
||
|
|
self.y_dir=1
|
||
|
|
else:
|
||
|
|
self.y_dir=-1
|
||
|
|
self.x_dir=0
|
||
|
|
elif abs(food_y-self.history[0][1]) < 10 :
|
||
|
|
|
||
|
|
# if self.x_dir==1 or self.x_dir==-1:
|
||
|
|
# self.x_dir=0
|
||
|
|
# self.y_dir=1
|
||
|
|
if self.y_dir==1 or self.y_dir==-1:
|
||
|
|
self.y_dir=0
|
||
|
|
if food_x>self.history[0][0]:
|
||
|
|
self.x_dir=1
|
||
|
|
else:
|
||
|
|
self.x_dir=-1
|
||
|
|
|
||
|
|
|
||
|
|
elif food_x-self.history[0][0] >= 10 and food_y-self.history[0][1] >= 10:
|
||
|
|
|
||
|
|
if self.x_dir==-1:
|
||
|
|
self.y_dir=1
|
||
|
|
self.x_dir=0
|
||
|
|
elif self.y_dir==-1:
|
||
|
|
self.y_dir=0
|
||
|
|
self.x_dir=1
|
||
|
|
elif self.history[0][0]-food_x >= 10 and food_y-self.history[0][1] >= 10:
|
||
|
|
|
||
|
|
if self.x_dir==1:
|
||
|
|
self.y_dir=1
|
||
|
|
self.x_dir=0
|
||
|
|
elif self.y_dir==1:
|
||
|
|
self.y_dir=0
|
||
|
|
self.x_dir=-1
|
||
|
|
elif self.history[0][0]-food_x >= 10 and self.history[0][1]-food_y >= 10:
|
||
|
|
|
||
|
|
if self.x_dir==1:
|
||
|
|
self.y_dir=-1
|
||
|
|
self.x_dir=0
|
||
|
|
elif self.y_dir==1:
|
||
|
|
self.y_dir=0
|
||
|
|
self.x_dir=-1
|
||
|
|
|
||
|
|
elif food_x-self.history[0][0] >= 10 and self.history[0][1]-food_y >= 10:
|
||
|
|
|
||
|
|
if self.x_dir==-1:
|
||
|
|
self.y_dir=-1
|
||
|
|
self.x_dir=0
|
||
|
|
elif self.y_dir==1:
|
||
|
|
self.y_dir=0
|
||
|
|
self.x_dir=1
|
||
|
|
|
||
|
|
self.update()
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
# ----------- Food Class --------------
|
||
|
|
class Food:
|
||
|
|
def new_location(self):
|
||
|
|
global food_x, food_y
|
||
|
|
food_x = random.randrange(1, width/scale-1)*scale
|
||
|
|
food_y = random.randrange(1, height/scale-1)*scale
|
||
|
|
|
||
|
|
def show(self):
|
||
|
|
pygame.draw.rect(display, food_colour, (food_x, food_y, scale, scale))
|
||
|
|
|
||
|
|
|
||
|
|
def show_score():
|
||
|
|
font = pygame.font.SysFont("Copperplate Gothic Bold", 20)
|
||
|
|
text = font.render("Score: " + str(score), True, snake_colour)
|
||
|
|
display.blit(text, (scale, scale))
|
||
|
|
|
||
|
|
|
||
|
|
# ----------- Main Game Loop -------------
|
||
|
|
def gameLoop():
|
||
|
|
loop = True
|
||
|
|
|
||
|
|
global score
|
||
|
|
|
||
|
|
snake = Snake(width/2, height/2) #starting from mid of grid
|
||
|
|
food = Food()
|
||
|
|
food.new_location()
|
||
|
|
ap=False
|
||
|
|
|
||
|
|
while loop:
|
||
|
|
|
||
|
|
display.fill(background)
|
||
|
|
snake.show()
|
||
|
|
food.show()
|
||
|
|
show_score()
|
||
|
|
|
||
|
|
for event in pygame.event.get():
|
||
|
|
if event.type == pygame.QUIT:
|
||
|
|
pygame.quit()
|
||
|
|
sys.exit()
|
||
|
|
|
||
|
|
if event.type == pygame.KEYDOWN:
|
||
|
|
if event.key == pygame.K_q:
|
||
|
|
pygame.quit()
|
||
|
|
sys.exit()
|
||
|
|
if event.key==pygame.K_SPACE: #autoplay start
|
||
|
|
ap=True
|
||
|
|
if event.key==pygame.K_TAB: #autoplay end
|
||
|
|
ap=False
|
||
|
|
else:
|
||
|
|
if snake.y_dir == 0:
|
||
|
|
if event.key == pygame.K_UP:
|
||
|
|
snake.x_dir = 0
|
||
|
|
snake.y_dir = -1
|
||
|
|
if event.key == pygame.K_DOWN:
|
||
|
|
snake.x_dir = 0
|
||
|
|
snake.y_dir = 1
|
||
|
|
|
||
|
|
if snake.x_dir == 0:
|
||
|
|
if event.key == pygame.K_LEFT:
|
||
|
|
snake.x_dir = -1
|
||
|
|
snake.y_dir = 0
|
||
|
|
if event.key == pygame.K_RIGHT:
|
||
|
|
snake.x_dir = 1
|
||
|
|
snake.y_dir = 0
|
||
|
|
|
||
|
|
|
||
|
|
if ap:
|
||
|
|
snake.autoplay()
|
||
|
|
else:
|
||
|
|
snake.update()
|
||
|
|
|
||
|
|
|
||
|
|
if snake.check_eaten():
|
||
|
|
food.new_location()
|
||
|
|
score += 1
|
||
|
|
snake.grow()
|
||
|
|
|
||
|
|
|
||
|
|
if snake.death():
|
||
|
|
score = 0
|
||
|
|
font = pygame.font.SysFont("Copperplate Gothic Bold", 50)
|
||
|
|
text = font.render("Game Over!", True, snake_colour)
|
||
|
|
display.blit(text, (width/2-50, height/2))
|
||
|
|
pygame.display.update()
|
||
|
|
time.sleep(3)
|
||
|
|
snake.reset()
|
||
|
|
|
||
|
|
|
||
|
|
#updating the values if snake goes out of board
|
||
|
|
if snake.history[0][0] > width:
|
||
|
|
snake.history[0][0] = 0
|
||
|
|
if snake.history[0][0] < 0:
|
||
|
|
snake.history[0][0] = width
|
||
|
|
|
||
|
|
if snake.history[0][1] > height:
|
||
|
|
snake.history[0][1] = 0
|
||
|
|
if snake.history[0][1] < 0:
|
||
|
|
snake.history[0][1] = height
|
||
|
|
|
||
|
|
pygame.display.update()
|
||
|
|
clock.tick(10) #at most 10 frames should pass in 1 sec, it is used to control the speed of snake
|
||
|
|
|
||
|
|
gameLoop()
|