plomlompom.com Git - plomrogue2/blob - plomrogue/things.py

   1 from plomrogue.errors import GameError
   2 from plomrogue.mapping import YX
   3
   4
   5
   6 class ThingBase:
   7     type_ = '?'
   8
   9     def __init__(self, game, id_=0, position=(YX(0, 0), YX(0, 0))):
  10         self.game = game
  11         if id_ == 0:
  12             self.id_ = self.game.new_thing_id()
  13         else:
  14             self.id_ = id_
  15         self.position = position
  16
  17
  18
  19 class Thing(ThingBase):
  20     blocking = False
  21     portable = False
  22     protection = '.'
  23
  24     def __init__(self, *args, **kwargs):
  25         super().__init__(*args, **kwargs)
  26
  27     def proceed(self):
  28         pass
  29
  30     @property
  31     def type_(self):
  32         return self.__class__.get_type()
  33
  34     @classmethod
  35     def get_type(cls):
  36         return cls.__name__[len('Thing_'):]
  37
  38
  39
  40 class Thing_Item(Thing):
  41     symbol_hint = 'i'
  42     portable = True
  43
  44
  45
  46 class ThingSpawner(Thing):
  47     symbol_hint = 'S'
  48
  49     def proceed(self):
  50         for t in [t for t in self.game.things
  51                   if t != self and t.position == self.position]:
  52             return
  53         t = self.game.thing_types[self.child_type](self.game,
  54                                                    position=self.position)
  55         self.game.things += [t]
  56         self.game.changed = True
  57
  58
  59
  60 class Thing_ItemSpawner(ThingSpawner):
  61     child_type = 'Item'
  62
  63
  64
  65 class Thing_SpawnPointSpawner(ThingSpawner):
  66     child_type = 'SpawnPoint'
  67
  68
  69
  70 class Thing_SpawnPoint(Thing):
  71     symbol_hint = 's'
  72     portable = True
  73     name = 'username'
  74
  75
  76
  77 class Thing_DoorSpawner(ThingSpawner):
  78     child_type = 'Door'
  79
  80
  81
  82 class Thing_Door(Thing):
  83     symbol_hint = 'D'
  84     blocking = False
  85     portable = True
  86
  87     def open(self):
  88         self.blocking = False
  89         self.portable = True
  90         del self.thing_char
  91
  92     def close(self):
  93         self.blocking = True
  94         self.portable = False
  95         self.thing_char = '#'
  96
  97
  98
  99 class ThingAnimate(Thing):
 100     blocking = True
 101
 102     def __init__(self, *args, **kwargs):
 103         super().__init__(*args, **kwargs)
 104         self.next_tasks = []
 105         self.set_task('WAIT')
 106         self._fov = None
 107
 108     def set_task(self, task_name, args=()):
 109         task_class = self.game.tasks[task_name]
 110         self.task = task_class(self, args)
 111         self.task.check()  # will throw GameError if necessary
 112
 113     def set_next_task(self, task_name, args=()):
 114         task_class = self.game.tasks[task_name]
 115         self.next_tasks += [task_class(self, args)]
 116
 117     def get_next_task(self):
 118         if len(self.next_tasks) > 0:
 119             task = self.next_tasks.pop(0)
 120             task.check()
 121             return task
 122         else:
 123             return None
 124
 125     def proceed(self):
 126         self._fov = None
 127         if self.task is None:
 128             self.task = self.get_next_task()
 129             return
 130
 131         try:
 132             self.task.check()
 133         except GameError as e:
 134             self.task = None
 135             raise e
 136         self.task.todo -= 1
 137         if self.task.todo <= 0:
 138             self._last_task_result = self.task.do()
 139             self.game.changed = True
 140             self.task = self.get_next_task()
 141
 142     @property
 143     def fov_stencil(self):
 144         if self._fov:
 145             return self._fov
 146         fov_map_class = self.game.map_geometry.fov_map_class
 147         self._fov = fov_map_class(self.game.things, self.game.maps, self.position,
 148                                   12, self.game.get_map)
 149         return self._fov
 150
 151     def fov_test(self, big_yx, little_yx):
 152         test_position = self.fov_stencil.target_yx(big_yx, little_yx)
 153         if self.fov_stencil.inside(test_position):
 154             if self.fov_stencil[test_position] == '.':
 155                 return True
 156         return False
 157
 158     def fov_stencil_map(self, map_type='normal'):
 159         visible_terrain = ''
 160         for yx in self.fov_stencil:
 161             if self.fov_stencil[yx] == '.':
 162                 big_yx, little_yx = self.fov_stencil.source_yxyx(yx)
 163                 map_ = self.game.get_map(big_yx, map_type)
 164                 visible_terrain += map_[little_yx]
 165             else:
 166                 visible_terrain += ' '
 167         return visible_terrain
 168
 169
 170
 171 class Thing_Player(ThingAnimate):
 172     symbol_hint = '@'
 173
 174     def __init__(self, *args, **kwargs):
 175         super().__init__(*args, **kwargs)
 176         self.carrying = None