plomlompom.com Git - plomrogue2-experiments/blob - game_common.py

   1 from parser import ArgError
   2
   3
   4 class Map:
   5
   6     def __init__(self, size=(0, 0), terrain=''):
   7         self.size = size
   8         self.terrain = terrain
   9
  10     def set_line(self, y, line):
  11         height_map = self.size[0]
  12         width_map = self.size[1]
  13         if y >= height_map:
  14             raise ArgError('too large row number %s' % y)
  15         width_line = len(line)
  16         if width_line > width_map:
  17             raise ArgError('too large map line width %s' % width_line)
  18         self.terrain = self.terrain[:y * width_map] + line +\
  19                        self.terrain[(y + 1) * width_map:]
  20
  21     def set_size(self, yx):
  22         y, x = yx
  23         self.size = (y, x)
  24         self.terrain = ''
  25         for y in range(self.size[0]):
  26             self.terrain += '?' * self.size[1]
  27
  28     def get_position_index(self, yx):
  29         return yx[0] * self.size[1] + yx[1]
  30
  31
  32 class World:
  33
  34     def __init__(self):
  35         self.turn = 0
  36         self.map_ = Map()
  37         self.things = []
  38         self.Thing = Thing  # child classes may use an extended Thing class here
  39
  40     def get_thing(self, id_):
  41         for thing in self.things:
  42             if id_ == thing.id_:
  43                 return thing
  44         t = self.Thing(self, id_)
  45         self.things += [t]
  46         return t
  47
  48
  49 class Thing:
  50
  51     def __init__(self, world, id_):
  52         self.world = world
  53         self.id_ = id_
  54         self.type_ = '?'
  55         self.position = [0,0]
  56
  57
  58 class CommonCommandsMixin:
  59
  60     def cmd_MAP_SIZE(self, yx):
  61         """Set self.map_size to yx, redraw self.terrain_map as '?' cells."""
  62         self.world.map_.set_size(yx)
  63     cmd_MAP_SIZE.argtypes = 'yx_tuple:nonneg'
  64
  65     def cmd_THING_TYPE(self, i, type_):
  66         t = self.world.get_thing(i)
  67         t.type_ = type_
  68     cmd_THING_TYPE.argtypes = 'int:nonneg string'
  69
  70     def cmd_THING_POS(self, i, yx):
  71         t = self.world.get_thing(i)
  72         t.position = list(yx)
  73     cmd_THING_POS.argtypes = 'int:nonneg yx_tuple:nonneg'