plomlompom.com Git - plomrogue2-experiments/blob - server_/map_.py

   1 import sys
   2 sys.path.append('../')
   3 import game_common
   4 import server_.game
   5
   6
   7 class Map(game_common.Map):
   8
   9     def __getitem__(self, yx):
  10         return self.terrain[self.get_position_index(yx)]
  11
  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:]
  15
  16     def __iter__(self):
  17         """Iterate over YX position coordinates."""
  18         for y in range(self.size[0]):
  19             for x in range(self.size[1]):
  20                 yield [y, x]
  21
  22     def lines(self):
  23         width = self.size[1]
  24         for y in range(self.size[0]):
  25             yield (y, self.terrain[y * width:(y + 1) * width])
  26
  27     # The following is used nowhere, so not implemented.
  28     #def items(self):
  29     #    for y in range(self.size[0]):
  30     #        for x in range(self.size[1]):
  31     #            yield ([y, x], self.terrain[self.get_position_index([y, x])])
  32
  33     def get_directions(self):
  34         directions = []
  35         for name in dir(self):
  36             if name[:5] == 'move_':
  37                 directions += [name[5:]]
  38         return directions
  39
  40     def new_from_shape(self, init_char):
  41         import copy
  42         new_map = copy.deepcopy(self)
  43         for pos in new_map:
  44             new_map[pos] = init_char
  45         return new_map
  46
  47     def move(self, start_pos, direction):
  48         mover = getattr(self, 'move_' + direction)
  49         new_pos = mover(start_pos)
  50         if new_pos[0] < 0 or new_pos[1] < 0 or \
  51                 new_pos[0] >= self.size[0] or new_pos[1] >= self.size[1]:
  52             raise server_.game.GameError('would move outside map bounds')
  53         return new_pos
  54
  55     def move_LEFT(self, start_pos):
  56         return [start_pos[0], start_pos[1] - 1]
  57
  58     def move_RIGHT(self, start_pos):
  59         return [start_pos[0], start_pos[1] + 1]
  60
  61
  62 class MapHex(Map):
  63
  64     def are_neighbors(self, pos_1, pos_2):
  65         if pos_1[0] == pos_2[0] and abs(pos_1[1] - pos_2[1]) <= 1:
  66             return True
  67         elif abs(pos_1[0] - pos_2[0]) == 1:
  68             if pos_1[0] % 2 == 0:
  69                 if pos_2[1] in (pos_1[1], pos_1[1] - 1):
  70                     return True
  71             elif pos_2[1] in (pos_1[1], pos_1[1] + 1):
  72                 return True
  73         return False
  74
  75     def move_UPLEFT(self, start_pos):
  76         if start_pos[0] % 2 == 0:
  77             return [start_pos[0] - 1, start_pos[1] - 1]
  78         else:
  79             return [start_pos[0] - 1, start_pos[1]]
  80
  81     def move_UPRIGHT(self, start_pos):
  82         if start_pos[0] % 2 == 0:
  83             return [start_pos[0] - 1, start_pos[1]]
  84         else:
  85             return [start_pos[0] - 1, start_pos[1] + 1]
  86
  87     def move_DOWNLEFT(self, start_pos):
  88         if start_pos[0] % 2 == 0:
  89             return [start_pos[0] + 1, start_pos[1] - 1]
  90         else:
  91             return [start_pos[0] + 1, start_pos[1]]
  92
  93     def move_DOWNRIGHT(self, start_pos):
  94         if start_pos[0] % 2 == 0:
  95             return [start_pos[0] + 1, start_pos[1]]
  96         else:
  97             return [start_pos[0] + 1, start_pos[1] + 1]
  98
  99
 100 class MapSquare(Map):
 101
 102     def are_neighbors(self, pos_1, pos_2):
 103         return abs(pos_1[0] - pos_2[0]) <= 1 and abs(pos_1[1] - pos_2[1] <= 1)
 104
 105     def move_UP(self, start_pos):
 106         return [start_pos[0] - 1, start_pos[1]]
 107
 108     def move_DOWN(self, start_pos):
 109         return [start_pos[0] + 1, start_pos[1]]
 110
 111
 112 def get_map_class(geometry):
 113     return globals()['Map' + geometry]