home · contact · privacy
Fix so-far invisible task name derivation error.
[plomrogue2-experiments] / new / plomrogue / things.py
1 from plomrogue.errors import GameError
2
3
4
5 class ThingBase:
6
7     def __init__(self, world, id_, type_='?', position=[0,0]):
8         self.world = world
9         self.id_ = id_
10         self.type_ = type_
11         self.position = position
12
13
14
15 class Thing(ThingBase):
16
17     def __init__(self, *args, **kwargs):
18         super().__init__(*args, **kwargs)
19         self.set_task('WAIT')
20         self._last_task_result = None
21         self._stencil = None
22
23     def move_towards_target(self, target):
24         dijkstra_map = type(self.world.map_)(self.world.map_.size)
25         n_max = 256
26         dijkstra_map.terrain = [n_max for i in range(dijkstra_map.size_i)]
27         dijkstra_map[target] = 0
28         shrunk = True
29         visible_map = self.get_visible_map()
30         while shrunk:
31             shrunk = False
32             for pos in dijkstra_map:
33                 if visible_map[pos] != '.':
34                     continue
35                 neighbors = dijkstra_map.get_neighbors(tuple(pos))
36                 for direction in neighbors:
37                     yx = neighbors[direction]
38                     if yx is not None and dijkstra_map[yx] < dijkstra_map[pos] - 1:
39                         dijkstra_map[pos] = dijkstra_map[yx] + 1
40                         shrunk = True
41         neighbors = dijkstra_map.get_neighbors(tuple(self.position))
42         n = n_max
43         target_direction = None
44         for direction in neighbors:
45             yx = neighbors[direction]
46             if yx is not None:
47                 n_new = dijkstra_map[yx]
48                 if n_new < n:
49                     n = n_new
50                     target_direction = direction
51         if target_direction:
52             self.set_task('MOVE', (target_direction,))
53
54     def decide_task(self):
55         visible_things = self.get_visible_things()
56         target = None
57         for t in visible_things:
58             if t.type_ == 'human':
59                 target = t.position
60                 break
61         if target is not None:
62             try:
63                 self.move_towards_target(target)
64                 return
65             except GameError:
66                 pass
67         self.set_task('WAIT')
68
69     def set_task(self, task_name, args=()):
70         task_class = self.world.game.tasks[task_name]
71         self.task = task_class(self, args)
72         self.task.check()  # will throw GameError if necessary
73
74     def proceed(self, is_AI=True):
75         """Further the thing in its tasks.
76
77         Decrements .task.todo; if it thus falls to <= 0, enacts method
78         whose name is 'task_' + self.task.name and sets .task =
79         None. If is_AI, calls .decide_task to decide a self.task.
80
81         Before doing anything, ensures an empty map visibility stencil
82         and checks that task is still possible, and aborts it
83         otherwise (for AI things, decides a new task).
84
85         """
86         self._stencil = None
87         try:
88             self.task.check()
89         except GameError as e:
90             self.task = None
91             self._last_task_result = e
92             if is_AI:
93                 try:
94                     self.decide_task()
95                 except GameError:
96                     self.set_task('WAIT')
97             return
98         self.task.todo -= 1
99         if self.task.todo <= 0:
100             self._last_task_result = self.task.do()
101             self.task = None
102         if is_AI and self.task is None:
103             try:
104                 self.decide_task()
105             except GameError:
106                 self.set_task('WAIT')
107
108     def get_stencil(self):
109         if self._stencil is not None:
110             return self._stencil
111         self._stencil = self.world.map_.get_fov_map(self.position)
112         return self._stencil
113
114     def get_visible_map(self):
115         stencil = self.get_stencil()
116         m = self.world.map_.new_from_shape(' ')
117         for pos in m:
118             if stencil[pos] == '.':
119                 m[pos] = self.world.map_[pos]
120         return m
121
122     def get_visible_things(self):
123         stencil = self.get_stencil()
124         visible_things = []
125         for thing in self.world.things:
126             if stencil[thing.position] == '.':
127                 visible_things += [thing]
128         return visible_things