temp/py_stepik
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add maze

Browse Source
This commit is contained in:
Dmitry Belyaev 2024-03-30 16:17:42 +03:00
parent efa19e3398
commit 5a4bc5f225
1 changed files with 200 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

220
mod_pygame/moveimg.py
View File

@ -14,6 +14,7 @@ import tempfile
import urllib.request
from abc import ABC, abstractmethod
from contextlib import contextmanager
from random import choice, randrange
from typing import NamedTuple, Optional
import PIL
@ -148,6 +149,75 @@ def choose_plural(amount, declensions):
return f"{amount} {declensions[i]}"
def maze_gen(row=4, col=4):
"""генератор карты
взял с коментария
https://stepik.org/lesson/502494/step/3?discussion=6527620&unit=494196
"""
row = max(2 * row + 1, 3)
col = max(2 * col + 1, 3)
maze = [[2] * col]
maze.extend([[2] + [1] * (col - 2) + [2] for _ in range(row - 2)])
maze.append(maze[0])
curr = (randrange(1, len(maze) - 1, 2), randrange(1, len(maze[0]) - 1, 2))
path = [curr]
maze[curr[0]][curr[1]] = 0
while path:
nexts = [
(r1, c1, r2, c2)
for r, c in zip((1, 0, -1, 0), (0, 1, 0, -1))
if (
(r1 := curr[0] + r) is None
or (c1 := curr[1] + c) is None
or (r2 := curr[0] + 2 * r) is None
or (c2 := curr[1] + 2 * c) is None
or 1 == maze[r1][c1] == maze[r2][c2]
)
]
if len(nexts):
r1, c1, r2, c2 = choice(nexts)
maze[r1][c1] = maze[r2][c2] = 0
path.append((r2, c2))
else:
curr = path.pop()
upd = {
("22", "20"): (None, "00"),
("02", "22"): ("00", None),
("11101", "00101", "11111", "10100", "10111"): (
None,
None,
None,
"10001",
None,
),
("10111", "10100", "11111", "00101", "11101"): (
None,
"10001",
None,
None,
None,
),
}
for key in upd:
for i in range(len(maze) - len(key) + 1):
for j in range(len(maze[0]) - len(key[0]) + 1):
if all(
maze[i + k][j : j + len(v)] == list(map(int, v))
for k, v in enumerate(key)
):
for k, v in filter(lambda x: x[1], enumerate(upd[key])):
maze[i + k][j : j + len(v)] = list(map(int, v))
return maze
def get_maze_sz(maze: list[list[int]]) -> Coords:
return Coords(len(maze[0]), len(maze))
class GameObject(ABC):
"""обобщение игрового элемента"""
@ -162,7 +232,7 @@ class GameObject(ABC):
assets: dict | None = None,
):
self.parent = parent
self._coords = coords
self.rect = pygame.Rect(coords, coords)
self.assets = assets or (parent.assets if parent else None)
self._surface = None
@ -170,20 +240,24 @@ class GameObject(ABC):
def surface(self) -> pygame.Surface | None:
return self._surface or (self.parent.surface if self.parent else None)
@property
def scene(self):
return self.parent.scene if self.parent else self
def get_coords(self) -> Coords:
return self._coords
return Coords(*self.rect.topleft)
def set_coords(self, coords: Coords):
if self.parent:
if self.parent.surface:
if not (
coords.x >= 0
and coords.x < self.parent.surface.get_width()
and coords.x + self.rect.width <= self.parent.surface.get_width()
and coords.y >= 0
and coords.y < self.parent.surface.get_height()
and coords.y + self.rect.height < self.parent.surface.get_height()
):
return
self._coords = coords
self.rect.topleft = coords
@abstractmethod
def draw(self):
@ -204,23 +278,46 @@ class Hero(GameObject):
assets: dict | None = None,
):
super().__init__(coords, parent, assets)
screen_sz = Coords(*parent.surface.get_size())
ghost_sz = screen_sz // 10
resize_img(self.assets, "ghost.png", ghost_sz)
self._surface = pygame.image.load(self.assets["ghost.png"])
self.rect = self.surface.get_rect()
self.coords = coords
def set_coords(self, coords: Coords):
super().set_coords(coords)
self.rect.topleft = self.coords
self.rect.topleft = coords
def draw(self):
self.parent.surface.blit(self.surface, self.rect)
def _check_collision(self, coords):
"""Проверка пересечения со стенами"""
new_rect = self.rect.copy()
new_rect.topleft = coords
return self.scene.walls.check_collision(new_rect)
def _reduce_step(self, coords):
"""Уменьшение шага движения, с целью подойти вплотную к стене"""
delta = coords - self.coords
dx, dy = 0, 0
if abs(delta.x) > 1:
dx = 1 * (delta.x < 0 or -1)
if abs(delta.y) > 1:
dy = 1 * (delta.y < 0 or -1)
return coords + Coords(dx, dy)
def set_coords(self, coords: Coords):
# проверка колизии
has_collision = self._check_collision(coords)
if not has_collision:
return super().set_coords(coords)
# уменьшение шага
while has_collision and coords != self.coords:
coords_new = self._reduce_step(coords)
if coords_new == coords:
return # не могу уменьшить шаг
coords = coords_new
has_collision = self._check_collision(coords)
super().set_coords(coords)
def handle_event(self, event: pygame.event.Event):
delta = 10
delta = 30
if event.type == pygame.KEYDOWN:
match event.key:
case pygame.K_UP | pygame.K_w:
@ -233,25 +330,106 @@ class Hero(GameObject):
self.coords += Coords(1, 0) * delta
class WallBlock(GameObject):
"""объект елемента стены"""
def __init__(
self,
coords: Coords,
parent: GameObject,
assets: dict | None = None,
):
super().__init__(coords, parent, assets)
self._surface = pygame.image.load(self.assets["brick.png"])
self.rect = self.surface.get_rect()
self.rect.topleft = coords
def draw(self):
self.parent.surface.blit(self.surface, self.rect)
def handle_event(self, event: pygame.event.Event):
...
class Walls(GameObject):
"""объект стен"""
def __init__(
self,
parent: GameObject,
maze: list[list[int]],
box_sz: Coords,
assets: dict | None = None,
):
super().__init__(Coords.zero(), parent, assets)
self.box_sz = box_sz
self.blocks = [
WallBlock(Coords(j, i).transform(box_sz), self, self.assets)
for i, row in enumerate(maze)
for j, item in enumerate(row)
if item > 0
]
def draw(self):
for block in self.blocks:
block.draw()
def check_collision(self, rect: pygame.Rect) -> bool:
for block in self.blocks:
if block.rect.colliderect(rect):
return True
return False
def handle_event(self, event: pygame.event.Event):
for block in self.blocks:
block.handle_event(event)
class Scene(GameObject):
"""основной игровой объект"""
# кнопки для выхода из игры
exit_keys = (pygame.K_ESCAPE, pygame.K_q)
def __init__(self, assets: dict, sz: Coords):
def __init__(self, assets: dict, screen_sz: Coords, maze_sz: Coords):
super().__init__(Coords.zero(), None, assets)
self._surface = pygame.display.set_mode(sz)
self.hero = Hero(Coords(100, 100), self)
resize_img(assets, "bg1k.png", sz)
self.maze = maze_gen(*maze_sz)
maze_sz = get_maze_sz(self.maze)
box_sz = screen_sz // get_maze_sz(self.maze)
self.box_sz = box_sz
resize_img(self.assets, "brick.png", box_sz)
hero_sz = Coords(*map(int, box_sz * 0.8))
resize_img(self.assets, "ghost.png", hero_sz)
hero_y_offset = (box_sz.y - hero_sz.y) // 2 + box_sz.y
self._surface = pygame.display.set_mode(screen_sz)
self.rect = self._surface.get_rect()
self.hero = Hero(Coords(0, hero_y_offset), self)
resize_img(assets, "bg1k.png", screen_sz)
self.background = pygame.image.load(self.assets["bg1k.png"])
self.done = False
self.maze = maze_gen(6, 6)
self.walls = Walls(self, self.maze, box_sz)
print(self.get_exit_rect())
print(self.rect)
def get_exit_rect(self) -> pygame.Rect:
maze_sz = get_maze_sz(self.maze)
coords = (maze_sz - Coords(1, 2)) * self.box_sz
rect = pygame.Rect(coords, coords)
rect.width, rect.height = self.box_sz
return rect
def draw(self):
if self.done:
return
self.surface.blit(self.background, self.coords)
pygame.draw.rect(self._surface, pygame.Color("#42c53d25"), self.get_exit_rect())
self.hero.draw()
self.walls.draw()
def handle_event(self, event: pygame.event.Event):
if self.done:
@ -276,9 +454,10 @@ class Scene(GameObject):
def game(assets):
pygame.init()
screen_sz = Coords(1000, 1000)
maze_sz = Coords(6, 6)
pygame.display.set_caption("Движение рисунка на Pygame")
scene = Scene(assets, screen_sz)
scene = Scene(assets, screen_sz, maze_sz)
scene.event_loop()
pygame.quit()
@ -288,6 +467,7 @@ def main():
assets = [
"bg1k.png",
"ghost.png",
"brick.png",
]
with get_assets(assets) as assets:
game(assets)