home · contact · privacy
7789223b9c9e63abbb24874692ad63cbfc0c0a2c
[plomrogue2-experiments] / new / plomrogue / things.py
1 from plomrogue.errors import GameError
2 from plomrogue.mapping import YX, Map, FovMapHex
3
4
5
6 class ThingBase:
7     type_ = '?'
8
9     def __init__(self, game, id_=None, position=(YX(0,0), YX(0,0))):
10         self.game = game
11         if id_ is None:
12             self.id_ = self.game.new_thing_id()
13         else:
14             self.id_ = id_
15         self.position = position
16
17     @property
18     def position(self):
19         return self._position
20
21     def _position_set(self, pos):
22         """Set self._position to pos.
23
24         We use this setter as core to the @position.setter property
25         method due to property setter subclassing not yet working
26         properly, see <https://bugs.python.org/issue14965>. We will
27         therefore super() _position_set instead of @position.setter in
28         subclasses.
29
30         """
31         self._position = pos
32
33     @position.setter
34     def position(self, pos):
35         self._position_set(pos)
36
37
38
39 class Thing(ThingBase):
40     blocking = False
41     in_inventory = False
42
43     def __init__(self, *args, **kwargs):
44         self.inventory = []
45         self._radius = 8
46         super().__init__(*args, **kwargs)
47
48     def proceed(self):
49         pass
50
51     def _position_set(self, pos):
52         super()._position_set(pos)
53         for t_id in self.inventory:
54             t = self.game.get_thing(t_id)
55             t.position = self.position
56         if not self.id_ == self.game.player_id:
57             return
58         edge_left = self.position[1].x - self._radius
59         edge_right = self.position[1].x + self._radius
60         edge_up = self.position[1].y - self._radius
61         edge_down = self.position[1].y + self._radius
62         if edge_left < 0:
63             self.game.get_map(self.position[0] + YX(1,-1))
64             self.game.get_map(self.position[0] + YX(0,-1))
65             self.game.get_map(self.position[0] + YX(-1,-1))
66         if edge_right >= self.game.map_size.x:
67             self.game.get_map(self.position[0] + YX(1,1))
68             self.game.get_map(self.position[0] + YX(0,1))
69             self.game.get_map(self.position[0] + YX(-1,1))
70         if edge_up < 0:
71             self.game.get_map(self.position[0] + YX(-1,1))
72             self.game.get_map(self.position[0] + YX(-1,0))
73             self.game.get_map(self.position[0] + YX(-1,-1))
74         if edge_down >= self.game.map_size.y:
75             self.game.get_map(self.position[0] + YX(1,1))
76             self.game.get_map(self.position[0] + YX(1,0))
77             self.game.get_map(self.position[0] + YX(1,-1))
78         #alternative
79         #if self.position[1].x < self._radius:
80         #    self.game.get_map(self.position[0] - YX(0,1))
81         #if self.position[1].y < self._radius:
82         #    self.game.get_map(self.position[0] - YX(1,0))
83         #if self.position[1].x > self.game.map_size.x - self._radius:
84         #    self.game.get_map(self.position[0] + YX(0,1))
85         #if self.position[1].y > self.game.map_size.y - self._radius:
86         #    self.game.get_map(self.position[0] + YX(1,0))
87         #if self.position[1].y < self._radius and \
88         #   self.position[1].x <= [pos for pos in
89         #                          diagonal_distance_edge
90         #                          if pos.y == self.position[1].y][0].x:
91         #    self.game.get_map(self.position[0] - YX(1,1))
92
93
94
95 class ThingItem(Thing):
96     pass
97
98
99
100 class ThingFood(ThingItem):
101     type_ = 'food'
102
103
104
105 class ThingAnimate(Thing):
106     blocking = True
107
108     def __init__(self, *args, **kwargs):
109         super().__init__(*args, **kwargs)
110         self.set_task('WAIT')
111         self._last_task_result = None
112         self.unset_surroundings()
113         self.close_maps = ()
114
115     def _position_set(self, pos):
116         """For player we need to update .close_maps on every move via the
117            self.surroundings property method, to keep their reality
118            bubble in sync with their movement.
119
120         """
121         super()._position_set(pos)
122         if self.id_ == self.game.player_id:
123             if not hasattr(self, '_surroundings'):
124                 self._surroundings = None
125             self.surroundings
126
127     def move_on_dijkstra_map(self, own_pos, targets):
128         visible_map = self.get_visible_map()
129         dijkstra_map = Map(visible_map.size,
130                            start_indented=visible_map.start_indented)
131         n_max = 256
132         dijkstra_map.terrain = [n_max for i in range(dijkstra_map.size_i)]
133         for target in targets:
134             dijkstra_map[target] = 0
135         shrunk = True
136         get_neighbors = self.game.map_geometry.get_neighbors
137         while shrunk:
138             shrunk = False
139             for pos in dijkstra_map:
140                 if visible_map[pos] != '.':
141                     continue
142                 neighbors = get_neighbors((YX(0,0), pos), dijkstra_map.size,
143                                           dijkstra_map.start_indented)
144                 for direction in neighbors:
145                     big_yx, small_yx = neighbors[direction]
146                     if big_yx == YX(0,0) and \
147                        dijkstra_map[small_yx] < dijkstra_map[pos] - 1:
148                         dijkstra_map[pos] = dijkstra_map[small_yx] + 1
149                         shrunk = True
150         #print('DEBUG DIJKSTRA ---------------------', self.id_, self.position)
151         #for y, line in dijkstra_map.lines():
152         #    line_to_print = []
153         #    for x in line:
154         #        line_to_print += ['%3s' % x]
155         #    print(' '.join(line_to_print))
156         neighbors = get_neighbors((YX(0,0), own_pos), dijkstra_map.size,
157                                   dijkstra_map.start_indented)
158         n = n_max
159         target_direction = None
160         for direction in sorted(neighbors.keys()):
161             big_yx, small_yx = neighbors[direction]
162             if big_yx == (0,0):
163                 n_new = dijkstra_map[small_yx]
164                 if n_new < n:
165                     n = n_new
166                     target_direction = direction
167         return target_direction
168
169     #def hunt_player(self):
170     #    visible_things = self.get_visible_things()
171     #    target = None
172     #    for t in visible_things:
173     #        if t.type_ == 'human':
174     #            target = t.position[1] - self.view_offset
175     #            break
176     #    if target is not None:
177     #        try:
178     #            offset_self_pos = self.position[1] - self.view_offset
179     #            target_dir = self.move_on_dijkstra_map(offset_self_pos,
180     #                                                   [target])
181     #            if target_dir is not None:
182     #                self.set_task('MOVE', (target_dir,))
183     #                return True
184     #        except GameError:
185     #            pass
186     #    return False
187
188     def hunt_food_satisfaction(self):
189         for id_ in self.inventory:
190             t = self.game.get_thing(id_)
191             if t.type_ == 'food':
192                 self.set_task('EAT', (id_,))
193                 return True
194         for id_ in self.get_pickable_items():
195             t = self.game.get_thing(id_)
196             if t.type_ == 'food':
197                 self.set_task('PICKUP', (id_,))
198                 return True
199         visible_things = self.get_visible_things()
200         food_targets = []
201         for t in visible_things:
202             if t.type_ == 'food':
203                 food_targets += [self.game.map_geometry.pos_in_view(t.position,
204                                                                     self.view_offset,
205                                                                     self.game.map_size)]
206         offset_self_pos = self.game.map_geometry.pos_in_view(self.position,
207                                                              self.view_offset,
208                                                              self.game.map_size)
209         target_dir = self.move_on_dijkstra_map(offset_self_pos,
210                                                food_targets)
211         if target_dir:
212             try:
213                 self.set_task('MOVE', (target_dir,))
214                 return True
215             except GameError:
216                 pass
217         return False
218
219     def decide_task(self):
220         #if not self.hunt_player():
221         if not self.hunt_food_satisfaction():
222             self.set_task('WAIT')
223
224     def set_task(self, task_name, args=()):
225         task_class = self.game.tasks[task_name]
226         self.task = task_class(self, args)
227         self.task.check()  # will throw GameError if necessary
228
229     def proceed(self, is_AI=True):
230         """Further the thing in its tasks, decrease its health.
231
232         First, ensures an empty map, decrements .health and kills
233         thing if crossing zero (removes from self.game.things for AI
234         thing, or unsets self.game.player_is_alive for player thing);
235         then checks that self.task is still possible and aborts if
236         otherwise (for AI things, decides a new task).
237
238         Then decrements .task.todo; if it thus falls to <= 0, enacts
239         method whose name is 'task_' + self.task.name and sets .task =
240         None. If is_AI, calls .decide_task to decide a self.task.
241
242         """
243         self.unset_surroundings()
244         self.health -= 1
245         if self.health <= 0:
246             if self is self.game.player:
247                 self.game.player_is_alive = False
248             else:
249                 # TODO: Handle inventory.
250                 del self.game.things[self.game.things.index(self)]
251             return
252         try:
253             self.task.check()
254         except GameError as e:
255             self.task = None
256             self._last_task_result = e
257             if is_AI:
258                 try:
259                     self.decide_task()
260                 except GameError:
261                     self.set_task('WAIT')
262             return
263         self.task.todo -= 1
264         if self.task.todo <= 0:
265             self._last_task_result = self.task.do()
266             self.task = None
267         if is_AI and self.task is None:
268             try:
269                 self.decide_task()
270             except GameError:
271                 self.set_task('WAIT')
272
273     def unset_surroundings(self):
274         self._stencil = None
275         self._surroundings = None
276
277     @property
278     def view_offset(self):
279         return self.game.map_geometry.get_view_offset(self.game.map_size,
280                                                       self.position,
281                                                       self._radius)
282
283     @property
284     def surroundings(self):
285         if self._surroundings is not None:
286             return self._surroundings
287         s, close_maps = self.\
288             game.map_geometry.get_view_and_seen_maps(self.game.map_size,
289                                                      self.game.get_map,
290                                                      self._radius,
291                                                      self.view_offset)
292         self.close_maps = close_maps
293         self._surroundings = s
294         return self._surroundings
295
296     def get_stencil(self):
297         if self._stencil is not None:
298             return self._stencil
299         m = Map(self.surroundings.size, ' ', self.surroundings.start_indented)
300         for pos in self.surroundings:
301             if self.surroundings[pos] in {'.', '~'}:
302                 m[pos] = '.'
303         fov_center = YX((m.size.y) // 2, m.size.x // 2)
304         self._stencil = FovMapHex(m, fov_center)
305         return self._stencil
306
307     def get_visible_map(self):
308         stencil = self.get_stencil()
309         m = Map(self.surroundings.size, ' ', self.surroundings.start_indented)
310         for pos in m:
311             if stencil[pos] == '.':
312                 m[pos] = self.surroundings[pos]
313         return m
314
315     def get_visible_things(self):
316         stencil = self.get_stencil()
317         visible_things = []
318         for thing in self.game.things:
319             pos = self.game.map_geometry.pos_in_view(thing.position,
320                                                      self.view_offset,
321                                                      self.game.map_size)
322             if pos.y < 0 or pos.x < 0 or\
323                pos.y >= stencil.size.y or pos.x >= stencil.size.x:
324                 continue
325             if (not thing.in_inventory) and stencil[pos] == '.':
326                 visible_things += [thing]
327         return visible_things
328
329     def get_pickable_items(self):
330         pickable_ids = []
331         visible_things = self.get_visible_things()
332         neighbor_fields = self.game.map_geometry.get_neighbors(self.position,
333                                                                self.game.map_size)
334         for t in [t for t in visible_things
335                   if isinstance(t, ThingItem) and
336                   (t.position == self.position or
337                    t.position in neighbor_fields.values())]:
338             pickable_ids += [t.id_]
339         return pickable_ids
340
341
342
343 class ThingHuman(ThingAnimate):
344     type_ = 'human'
345     health = 100
346
347
348
349 class ThingMonster(ThingAnimate):
350     type_ = 'monster'
351     health = 50