-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathshow_map.py
148 lines (139 loc) · 5.93 KB
/
show_map.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
#!/usr/bin/env python3
"""
地図表示
"""
import sys
from PIL import Image, ImageDraw
import const
class ShowMap():
""" 地図の表示 """
# 1マスのサイズ（可変）
size = 20
def __init__(self):
""" コンストラクタ """
pass
def draw_map(self, obj_maze, player=None):
""" 地図を描画 """
img = Image.new('RGB', (const.WIDTH, const.HEIGHT), color=0)
self.draw = ImageDraw.Draw(img)
# 壁の色
COLOR_WALL = (255, 255, 255)
# 扉の色
COLOR_DOOR = (255, 127, 0)
# 部屋の色
COLOR_ROOM = (255, 255, 0)
# イベント天井の色
COLOR_CEILING = (0, 255, 0)
# イベント床の色
COLOR_FLOOR = (0, 0, 255)
# 未到達の色
COLOR_UNKNOWN = (127, 127, 127)
# 枠の描画
self.draw.rectangle(
(0, 0) + (const.WIDTH - 1, const.HEIGHT - 1),
outline=COLOR_WALL)
# 迷路オブジェクトから各内容を取得
maze_wall = obj_maze.maze_wall
maze_door = obj_maze.maze_door
maze_contents = obj_maze.maze_contents
# 1マスのサイズの調整
self.size = const.WIDTH // len(maze_wall[0])
# 壁の大きさ
wl = self.size * 0.95 / 2
# イベントマークの大きさ
ev = self.size * 0.8 / 2
for iy, row_wall in enumerate(maze_wall):
for ix, cell_wall in enumerate(row_wall):
# 通った道のみ表示（Noneの場合、全体表示）
if player is not None:
step = player.mapping[player.floor - 1][iy][ix]
if step == False:
xy = (self.size * ix, self.size * iy) \
+ (self.size * (ix + 1), self.size * (iy + 1))
self.draw.rectangle(
xy,
fill=COLOR_UNKNOWN,
outline=COLOR_UNKNOWN)
continue
# 各セルの中心座標
x = self.size * (ix + 0.5)
y = self.size * (iy + 0.5)
# 壁の描画
if cell_wall & const.MAZE_NORTH != const.MAZE_NONE:
xy = (x - wl, y - wl) + (x + wl, y - wl)
self.draw.line(xy, fill=COLOR_WALL)
if cell_wall & const.MAZE_SOUTH != const.MAZE_NONE:
xy = (x - wl, y + wl) + (x + wl, y + wl)
self.draw.line(xy, fill=COLOR_WALL)
if cell_wall & const.MAZE_WEST != const.MAZE_NONE:
xy = (x - wl, y - wl) + (x - wl, y + wl)
self.draw.line(xy, fill=COLOR_WALL)
if cell_wall & const.MAZE_EAST != const.MAZE_NONE:
xy = (x + wl, y - wl) + (x + wl, y + wl)
self.draw.line(xy, fill=COLOR_WALL)
# 扉の描画
cell_door = maze_door[iy][ix]
if cell_door & const.MAZE_NORTH != const.MAZE_NONE:
xy = (x - wl, y - wl) + (x + wl, y - wl)
self.draw.line(xy, fill=COLOR_DOOR)
if cell_door & const.MAZE_SOUTH != const.MAZE_NONE:
xy = (x - wl, y + wl) + (x + wl, y + wl)
self.draw.line(xy, fill=COLOR_DOOR)
if cell_door & const.MAZE_WEST != const.MAZE_NONE:
xy = (x - wl, y - wl) + (x - wl, y + wl)
self.draw.line(xy, fill=COLOR_DOOR)
if cell_door & const.MAZE_EAST != const.MAZE_NONE:
xy = (x + wl, y - wl) + (x + wl, y + wl)
self.draw.line(xy, fill=COLOR_DOOR)
# イベント
cell_contents = maze_contents[iy][ix]
# イベント天井の描画
if cell_contents & const.EVENT_CEILING != const.MAZE_NONE:
xy = (x, y - ev)
xy += (x - ev, y)
xy += (x + ev, y)
self.draw.polygon(
xy, fill=COLOR_CEILING, outline=COLOR_CEILING)
# イベント床の描画
if cell_contents & const.EVENT_FLOOR != const.MAZE_NONE:
xy = (x, y + ev)
xy += (x - ev, y)
xy += (x + ev, y)
self.draw.polygon(
xy, fill=COLOR_FLOOR, outline=COLOR_FLOOR)
# 部屋の描画
#if cell_contents & const.CONTENTS_ROOM != const.MAZE_NONE:
# xy = (x, y)
# self.draw.point(xy, COLOR_ROOM)
return img
def draw_player(self, px, py, direction):
""" プレイヤーの位置表示 """
# プレイヤーの色
COLOR_PLAYER = (255, 0, 0)
# 大きさ
p = self.size * 0.7 / 2
# 中心座標
x = self.size * (px + 0.5)
y = self.size * (py + 0.5)
# 向き毎に描画
if direction == const.DIR_NORTH:
xy = (x, y - p) + (x - p, y + p) + (x + p, y + p)
elif direction == const.DIR_SOUTH:
xy = (x, y + p) + (x - p, y - p) + (x + p, y - p)
elif direction == const.DIR_EAST:
xy = (x + p, y) + (x - p, y - p) + (x - p, y + p)
elif direction == const.DIR_WEST:
xy = (x - p, y) + (x + p, y - p) + (x + p, y + p)
self.draw.polygon(xy, fill=COLOR_PLAYER, outline=COLOR_PLAYER)
def draw_route(self, obj_search):
""" 経路の表示 """
# 経路の色
COLOR_ROUTE = (255, 0, 255)
# 描画座標
xy = ()
# 探索オブジェクトから経路を取り出す
for route in obj_search.route:
step, x, y = route
xy += (self.size * (x + 0.5), self.size * (y + 0.5))
# 描画
self.draw.line(xy, fill=COLOR_ROUTE)