7 class Map(game_common.Map):
9 def __getitem__(self, yx):
10 return self.terrain[self.get_position_index(yx)]
12 def __setitem__(self, yx, c):
13 pos_i = self.get_position_index(yx)
14 self.terrain = self.terrain[:pos_i] + c + self.terrain[pos_i + 1:]
17 """Iterate over YX position coordinates."""
18 for y in range(self.size[0]):
19 for x in range(self.size[1]):
24 return self.__class__.__name__[3:]
28 for y in range(self.size[0]):
29 yield (y, self.terrain[y * width:(y + 1) * width])
31 # The following is used nowhere, so not implemented.
33 # for y in range(self.size[0]):
34 # for x in range(self.size[1]):
35 # yield ([y, x], self.terrain[self.get_position_index([y, x])])
37 def get_directions(self):
39 for name in dir(self):
40 if name[:5] == 'move_':
41 directions += [name[5:]]
44 def new_from_shape(self, init_char):
46 new_map = copy.deepcopy(self)
48 new_map[pos] = init_char
51 def move(self, start_pos, direction):
52 mover = getattr(self, 'move_' + direction)
53 new_pos = mover(start_pos)
54 if new_pos[0] < 0 or new_pos[1] < 0 or \
55 new_pos[0] >= self.size[0] or new_pos[1] >= self.size[1]:
56 raise server_.game.GameError('would move outside map bounds')
59 def move_LEFT(self, start_pos):
60 return [start_pos[0], start_pos[1] - 1]
62 def move_RIGHT(self, start_pos):
63 return [start_pos[0], start_pos[1] + 1]
68 def are_neighbors(self, pos_1, pos_2):
69 if pos_1[0] == pos_2[0] and abs(pos_1[1] - pos_2[1]) <= 1:
71 elif abs(pos_1[0] - pos_2[0]) == 1:
73 if pos_2[1] in (pos_1[1], pos_1[1] - 1):
75 elif pos_2[1] in (pos_1[1], pos_1[1] + 1):
79 def move_UPLEFT(self, start_pos):
80 if start_pos[0] % 2 == 0:
81 return [start_pos[0] - 1, start_pos[1] - 1]
83 return [start_pos[0] - 1, start_pos[1]]
85 def move_UPRIGHT(self, start_pos):
86 if start_pos[0] % 2 == 0:
87 return [start_pos[0] - 1, start_pos[1]]
89 return [start_pos[0] - 1, start_pos[1] + 1]
91 def move_DOWNLEFT(self, start_pos):
92 if start_pos[0] % 2 == 0:
93 return [start_pos[0] + 1, start_pos[1] - 1]
95 return [start_pos[0] + 1, start_pos[1]]
97 def move_DOWNRIGHT(self, start_pos):
98 if start_pos[0] % 2 == 0:
99 return [start_pos[0] + 1, start_pos[1]]
101 return [start_pos[0] + 1, start_pos[1] + 1]
104 class MapSquare(Map):
106 def are_neighbors(self, pos_1, pos_2):
107 return abs(pos_1[0] - pos_2[0]) <= 1 and abs(pos_1[1] - pos_2[1] <= 1)
109 def move_UP(self, start_pos):
110 return [start_pos[0] - 1, start_pos[1]]
112 def move_DOWN(self, start_pos):
113 return [start_pos[0] + 1, start_pos[1]]
116 def get_map_class(geometry):
117 return globals()['Map' + geometry]
120 map_manager = game_common.MapManager(globals())