changed screen and grid size

This commit is contained in:
Kristofers-Solo 2022-03-29 23:45:44 +03:00
parent d9ea2405fe
commit da9b869434
3 changed files with 216 additions and 217 deletions

View File

@ -1,217 +0,0 @@
# Author - Kristiāns Francis Cagulis
# Date - 11.03.2022
# Title - Snake
from random import randrange, randint
import pygame
import tkinter as tk
from tkinter import messagebox
RED = (255, 0, 0)
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
GREEN = (0, 128, 30)
PURPLE = (170, 0, 255)
RANDOM_COLOR = (randint(0, 255), randint(0, 255), randint(0, 255))
WIDTH = 500
ROWS = 20
class Cube(object):
width = WIDTH
rows = ROWS
def __init__(self, start, color=RANDOM_COLOR):
self.pos = start
self.dirnx = 1
self.dirny = 0
self.color = color
def move(self, dirnx, dirny):
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):
distance = self.width // self.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):
body = []
turns = {}
def __init__(self, color, pos):
self.color = color
self.head = Cube(pos, self.color)
self.body.append(self.head)
self.dirnx = 0
self.dirny = 1
def move(self):
for event in pygame.event.get():
if event.type == pygame.QUIT:
pygame.quit()
keys = pygame.key.get_pressed()
for _ in keys:
if keys[pygame.K_LEFT]:
self.dirnx = -1
self.dirny = 0
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
elif keys[pygame.K_RIGHT]:
self.dirnx = 1
self.dirny = 0
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
elif keys[pygame.K_UP]:
self.dirnx = 0
self.dirny = -1
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
elif keys[pygame.K_DOWN]:
self.dirnx = 0
self.dirny = 1
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
for i, c in enumerate(self.body):
p = c.pos[:]
if p in self.turns:
turn = self.turns[p]
c.move(turn[0], turn[1])
if i == len(self.body) - 1:
self.turns.pop(p)
else:
if c.dirnx == -1 and c.pos[0] <= 0:
c.pos = (c.rows - 1, c.pos[1])
elif c.dirnx == 1 and c.pos[0] >= c.rows - 1:
c.pos = (0, c.pos[1])
elif c.dirny == 1 and c.pos[1] >= c.rows - 1:
c.pos = (c.pos[0], 0)
elif c.dirny == -1 and c.pos[1] <= 0:
c.pos = (c.pos[0], c.rows - 1)
else:
c.move(c.dirnx, c.dirny)
def reset(self, pos):
self.head = Cube(pos)
self.body = []
self.turns = {}
self.dirnx = 0
self.dirny = 1
def add_cube(self):
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])))
elif dx == -1 and dy == 0:
self.body.append(Cube((tail.pos[0] + 1, tail.pos[1])))
elif dx == 0 and dy == 1:
self.body.append(Cube((tail.pos[0], tail.pos[1] - 1)))
elif dx == 0 and dy == -1:
self.body.append(Cube((tail.pos[0], tail.pos[1] + 1)))
self.body[-1].dirnx = dx
self.body[-1].dirny = dy
def draw(self, surface):
for i, c in enumerate(self.body):
if i == 0:
c.draw(surface, True)
else:
c.draw(surface)
def draw_grid(width, rows, surface):
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, width))
pygame.draw.line(surface, WHITE, (0, y), (width, y))
def redraw_window(surface):
surface.fill(BLACK)
s.draw(surface)
snack.draw(surface)
draw_grid(WIDTH, ROWS, surface)
pygame.display.update()
def random_snack(rows, item):
positions = item.body
while True:
x = randrange(rows)
y = randrange(rows)
if len(list(filter(lambda z: z.pos == (x, y), positions))) > 0:
continue
else:
break
return x, y
def message_box(subject, content):
root = tk.Tk()
root.attributes("-topmost", True)
root.withdraw()
messagebox.showinfo(subject, content)
try:
root.destroy()
except:
pass
def main():
win = pygame.display.set_mode((WIDTH, WIDTH))
flag = True
clock = pygame.time.Clock()
global snack
snack = Cube(random_snack(ROWS, s), color=RED)
while flag:
pygame.time.delay(50)
clock.tick(10)
s.move()
if s.body[0].pos == snack.pos:
s.add_cube()
snack = Cube(random_snack(ROWS, s), color=RED)
for i in range(len(s.body)):
if s.body[i].pos in list(map(lambda z: z.pos, s.body[i + 1:])):
print(f"Score: {len(s.body)}")
message_box("You Lost!", f"Your score was {len(s.body)}.\nPlay again")
s.reset((10, 10))
break
redraw_window(win)
pass
s = Snake(PURPLE, (10, 10))
if __name__ == '__main__':
main()

Binary file not shown.

216
pygame/snake/source/snake.py Executable file
View File

@ -0,0 +1,216 @@
# 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)
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
RANDOM_POS = (randint(0, ROWS - 1), randint(0, COLUMNS - 1))
WINDOW = pygame.display.set_mode((WIDTH, HEIGHT))
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
# print(distance)
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 keys[pygame.K_LEFT] or keys[pygame.K_a]: # turn left
self.dirnx = -1
self.dirny = 0
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny] # turn right
elif keys[pygame.K_RIGHT] or keys[pygame.K_d]:
self.dirnx = 1
self.dirny = 0
self.turns[self.head.pos[:]] = [self.dirnx, self.dirny]
elif keys[pygame.K_UP] or 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_DOWN] or 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 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(snake, snack) -> None:
WINDOW.fill(BLACK)
draw_grid(WINDOW)
snake.draw(WINDOW)
snack.draw(WINDOW)
pygame.display.update()
def random_snack(rows, columns, item) -> tuple:
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 = Snake(RANDOM_POS, PURPLE)
snack = Cube(random_snack(ROWS, COLUMNS, snake), color=GREEN)
while run:
clock.tick(FPS)
pygame.time.delay(50)
snake.move()
if snake.body[0].pos == snack.pos:
snake.add_cube()
snack = Cube(random_snack(ROWS, COLUMNS, snake), color=GREEN)
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
# snake.reset(RANDOM_POS)
redraw_window(snake, snack)
set_font = lambda size: pygame.font.SysFont("roboto", 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)
title_label = pygame.font.SysFont("roboto", 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()