changed screen and grid size

pygame/snake/snake.py

@@ -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()

pygame/snake/source/snake.py

@@ -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()