mirror of
https://github.com/kristoferssolo/School.git
synced 2025-10-21 20:10:38 +00:00
248 lines
6.5 KiB
Python
Executable File
248 lines
6.5 KiB
Python
Executable File
# Author - Kristiāns Francis Cagulis
|
|
# Date - 28.03.2022
|
|
# Title - Snake
|
|
|
|
import pygame
|
|
from random import randrange, randint
|
|
|
|
RED = (255, 0, 0)
|
|
BLACK = (0, 0, 0)
|
|
WHITE = (255, 255, 255)
|
|
GREEN = (0, 128, 30)
|
|
PURPLE = (170, 0, 255)
|
|
BLUE = (85, 85, 255)
|
|
RANDOM_COLOR = (randint(0, 255), randint(0, 255), randint(0, 255))
|
|
|
|
SQUARE_SIZE = 30
|
|
ROWS, COLUMNS = 30, 20
|
|
WIDTH, HEIGHT = ROWS * SQUARE_SIZE, COLUMNS * SQUARE_SIZE
|
|
|
|
WINDOW = pygame.display.set_mode((WIDTH, HEIGHT))
|
|
pygame.font.init()
|
|
|
|
|
|
class Cube(object):
|
|
|
|
def __init__(self, start, color=RANDOM_COLOR) -> None:
|
|
self.pos = start
|
|
self.dirnx = 1
|
|
self.dirny = 0
|
|
self.color = color
|
|
|
|
def move(self, dirnx, dirny) -> None:
|
|
self.dirnx = dirnx
|
|
self.dirny = dirny
|
|
self.pos = (self.pos[0] + self.dirnx, self.pos[1] + self.dirny)
|
|
|
|
# eyes
|
|
def draw(self, surface, eyes=False) -> None:
|
|
distance = WIDTH // ROWS
|
|
i = self.pos[0]
|
|
j = self.pos[1]
|
|
|
|
pygame.draw.rect(surface, self.color, (i * distance + 1, j * distance + 1, distance - 2, distance - 2))
|
|
if eyes:
|
|
center = distance // 2
|
|
radius = 3
|
|
circle_middle = (i * distance + center - radius, j * distance + 8)
|
|
circle_middle_2 = (i * distance + distance - radius * 2, j * distance + 8)
|
|
pygame.draw.circle(surface, BLACK, circle_middle, radius)
|
|
pygame.draw.circle(surface, BLACK, circle_middle_2, radius)
|
|
|
|
|
|
class Snake(object):
|
|
|
|
def __init__(self, pos, color) -> None:
|
|
self.color = color
|
|
self.head = Cube(pos, self.color)
|
|
self.body = []
|
|
self.body.append(self.head)
|
|
self.turns = {}
|
|
self.dirnx = 0
|
|
self.dirny = 1
|
|
|
|
def move(self) -> None:
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
pygame.quit()
|
|
keys = pygame.key.get_pressed()
|
|
if self.color == PURPLE:
|
|
if keys[pygame.K_LEFT]: # turn left
|
|
self.dirnx = -1
|
|
self.dirny = 0
|
|
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
|
|
|
elif keys[pygame.K_RIGHT]: # turn right
|
|
self.dirnx = 1
|
|
self.dirny = 0
|
|
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
|
|
|
elif keys[pygame.K_UP]: # turn up
|
|
self.dirnx = 0
|
|
self.dirny = -1
|
|
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
|
|
|
elif keys[pygame.K_DOWN]: # turn down
|
|
self.dirnx = 0
|
|
self.dirny = 1
|
|
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
|
|
|
elif self.color == BLUE:
|
|
if keys[pygame.K_a]: # turn left
|
|
self.dirnx = -1
|
|
self.dirny = 0
|
|
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
|
|
|
elif keys[pygame.K_d]: # turn right
|
|
self.dirnx = 1
|
|
self.dirny = 0
|
|
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
|
|
|
elif keys[pygame.K_w]: # turn up
|
|
self.dirnx = 0
|
|
self.dirny = -1
|
|
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
|
|
|
elif keys[pygame.K_s]: # turn down
|
|
self.dirnx = 0
|
|
self.dirny = 1
|
|
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
|
|
|
|
for i, head in enumerate(self.body):
|
|
head_pos = head.pos[:]
|
|
if head_pos in self.turns:
|
|
turn = self.turns[head_pos]
|
|
head.move(turn[0], turn[1])
|
|
if i == len(self.body) - 1:
|
|
self.turns.pop(head_pos)
|
|
|
|
else: # move player to other screen size
|
|
if head.dirnx == -1 and head.pos[0] <= 0: # left to right
|
|
head.pos = (ROWS - 1, head.pos[1])
|
|
|
|
elif head.dirnx == 1 and head.pos[0] >= ROWS - 1: # right to left
|
|
head.pos = (0, head.pos[1])
|
|
|
|
elif head.dirny == 1 and head.pos[1] >= COLUMNS - 1: # bottom to top
|
|
head.pos = (head.pos[0], 0)
|
|
|
|
elif head.dirny == -1 and head.pos[1] <= 0: # top to bottom
|
|
head.pos = (head.pos[0], COLUMNS - 1)
|
|
|
|
else:
|
|
head.move(head.dirnx, head.dirny)
|
|
|
|
def reset(self, pos) -> None:
|
|
self.head = Cube(pos, self.color)
|
|
self.body = []
|
|
self.turns = {}
|
|
self.dirnx = 0
|
|
self.dirny = 1
|
|
|
|
def add_cube(self) -> None:
|
|
tail = self.body[-1]
|
|
dx, dy = tail.dirnx, tail.dirny
|
|
|
|
if dx == 1 and dy == 0:
|
|
self.body.append(Cube((tail.pos[0] - 1, tail.pos[1]), self.color))
|
|
elif dx == -1 and dy == 0:
|
|
self.body.append(Cube((tail.pos[0] + 1, tail.pos[1]), self.color))
|
|
elif dx == 0 and dy == 1:
|
|
self.body.append(Cube((tail.pos[0], tail.pos[1] - 1), self.color))
|
|
elif dx == 0 and dy == -1:
|
|
self.body.append(Cube((tail.pos[0], tail.pos[1] + 1), self.color))
|
|
|
|
self.body[-1].dirnx = dx
|
|
self.body[-1].dirny = dy
|
|
|
|
def remove_cube(self) -> None:
|
|
self.body.pop(-1)
|
|
|
|
def draw(self, surface) -> None:
|
|
for i, head in enumerate(self.body):
|
|
if i == 0:
|
|
head.draw(surface, True)
|
|
|
|
else:
|
|
head.draw(surface)
|
|
|
|
|
|
def draw_grid(surface) -> None:
|
|
size_between = WIDTH // ROWS
|
|
x = 0
|
|
y = 0
|
|
for _ in range(ROWS):
|
|
x += size_between
|
|
y += size_between
|
|
|
|
pygame.draw.line(surface, WHITE, (x, 0), (x, HEIGHT))
|
|
pygame.draw.line(surface, WHITE, (0, y), (WIDTH, y))
|
|
|
|
|
|
def redraw_window(sprites: list) -> None:
|
|
WINDOW.fill(BLACK)
|
|
draw_grid(WINDOW)
|
|
for sprite in sprites:
|
|
sprite.draw(WINDOW)
|
|
pygame.display.update()
|
|
|
|
|
|
def random_snack(items, rows=ROWS, columns=COLUMNS) -> tuple:
|
|
for item in items:
|
|
positions = item.body
|
|
while True:
|
|
x = randrange(rows)
|
|
y = randrange(columns)
|
|
if len(list(filter(lambda z: z.pos == (x, y), positions))) > 0:
|
|
continue
|
|
else:
|
|
break
|
|
return x, y
|
|
|
|
|
|
def main() -> None:
|
|
FPS = 10
|
|
run = True
|
|
clock = pygame.time.Clock()
|
|
snake_one = Snake((randint(0, ROWS - 1), randint(0, COLUMNS - 1)), PURPLE)
|
|
snake_two = Snake((randint(0, ROWS - 1), randint(0, COLUMNS - 1)), BLUE)
|
|
snakes = [snake_one, snake_two]
|
|
snack = Cube(random_snack(snakes), color=GREEN)
|
|
poison = Cube(random_snack(snakes), color=RED)
|
|
while run:
|
|
clock.tick(FPS)
|
|
pygame.time.delay(50)
|
|
for snake in snakes:
|
|
snake.move()
|
|
if snake.body[0].pos == snack.pos:
|
|
snake.add_cube()
|
|
snack = Cube(random_snack(snakes), color=GREEN)
|
|
if snake.body[0].pos == poison.pos:
|
|
snake.remove_cube()
|
|
poison = Cube(random_snack(snakes), color=RED)
|
|
for i in range(len(snake.body)):
|
|
if snake.body[i].pos in list(map(lambda z: z.pos, snake.body[i + 1:])):
|
|
print(f"Score: {len(snake.body)}")
|
|
run = False
|
|
redraw_window(list(set(snakes + [snack, poison])))
|
|
|
|
|
|
set_font = lambda size: pygame.font.SysFont("arial", size) # sets font size
|
|
|
|
|
|
def main_menu() -> None:
|
|
while True:
|
|
WINDOW.fill(BLACK)
|
|
title_label = set_font(50).render("Press any key to start...", 1, WHITE)
|
|
WINDOW.blit(title_label, (WIDTH / 2 - title_label.get_width() / 2, HEIGHT / 2 - title_label.get_height() / 2))
|
|
pygame.display.update()
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
quit()
|
|
if event.type == pygame.KEYDOWN:
|
|
main()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main_menu()
|