1 # This file is part of PlomRogue. PlomRogue is licensed under the GPL version 3
2 # or any later version. For details on its copyright, license, and warranties,
3 # see the file NOTICE in the root directory of the PlomRogue source package.
6 from server.config.world_data import world_db
9 def get_dir_to_target(t, filter):
10 """Try to set T_COMMAND/T_ARGUMENT for move to "filter"-determined target.
12 The path-wise nearest target is chosen, via the shortest available path.
13 Target must not be t. On succcess, return positive value, else False.
15 "a": Thing in FOV is animate, but of ThingType, starts out weaker than t
16 is, and its corpse would be healthy food for t
17 "f": move away from an enemy – any visible actor whose thing type has more
18 TT_LIFEPOINTS than t LIFEPOINTS, and might find t's corpse healthy
19 food – if it is closer than n steps, where n will shrink as t's hunger
20 grows; if enemy is too close, move towards (attack) the enemy instead;
21 if no fleeing is possible, nor attacking useful, wait; don't tread on
22 non-enemies for fleeing
23 "c": Thing in memorized map is consumable of sufficient nutrition for t
24 "s": memory map cell with greatest-reachable degree of unexploredness
26 from server.utils import rand, libpr, c_pointer_to_bytearray, \
28 from server.config.world_data import symbols_passable
29 tt = world_db["ThingTypes"][t["T_TYPE"]]
31 def zero_score_map_where_char_on_memdepthmap(c):
32 # OUTSOURCED FOR PERFORMANCE REASONS TO libplomrogue.so:
33 # for i in [i for i in range(world_db["MAP_LENGTH"] ** 2)
34 # if t["T_MEMDEPTHMAP"][i] == mem_depth_c[0]]:
36 map = c_pointer_to_bytearray(t["T_MEMDEPTHMAP"])
37 if libpr.zero_score_map_where_char_on_memdepthmap(c, map):
38 raise RuntimeError("No score map allocated for "
39 "zero_score_map_where_char_on_memdepthmap().")
41 def set_map_score(pos, score):
42 test = libpr.set_map_score(pos, score)
44 raise RuntimeError("No score map allocated for set_map_score().")
46 def get_map_score(pos):
47 result = libpr.get_map_score(pos)
49 raise RuntimeError("No score map allocated for get_map_score().")
52 def animates_in_fov(maplength):
53 return [Thing for Thing in world_db["Things"].values()
54 if Thing["T_LIFEPOINTS"] and not Thing["carried"]
55 and 118 == t["fovmap"][Thing["pos"]] and not Thing == t]
57 def good_attack_target(v):
58 eat_cost = tt["eat_vs_hunger_threshold"]
59 ty = world_db["ThingTypes"][v["T_TYPE"]]
60 type_corpse = world_db["ThingTypes"][ty["TT_CORPSE_ID"]]
61 if t["T_LIFEPOINTS"] > ty["TT_LIFEPOINTS"] \
62 and type_corpse["TT_TOOL"] == "food" \
63 and type_corpse["TT_TOOLPOWER"] > eat_cost:
67 def good_flee_target(m):
68 corpse_type = world_db["ThingTypes"][tt["TT_CORPSE_ID"]]
69 targetness = 0 if corpse_type["TT_TOOL"] != "food" \
70 else corpse_type["TT_TOOLPOWER"]
71 type = world_db["ThingTypes"][m["T_TYPE"]]
72 if t["T_LIFEPOINTS"] < type["TT_LIFEPOINTS"] \
73 and targetness > type["eat_vs_hunger_threshold"]:
84 maplength = world_db["MAP_LENGTH"]
85 if t["fovmap"] and "a" == filter:
86 return exists(Thing for Thing in animates_in_fov(maplength)
87 if good_attack_target(Thing))
88 elif t["fovmap"] and "f" == filter:
89 return exists(Thing for Thing in animates_in_fov(maplength)
90 if good_flee_target(Thing))
91 elif t["T_MEMMAP"] and "c" == filter:
92 eat_cost = tt["eat_vs_hunger_threshold"]
94 map_len = world_db["MAP_LENGTH"]
95 return exists(mt for mt in t["T_MEMTHING"]
96 if ord_blank != t["T_MEMMAP"][mt[1] * map_len + mt[2]]
97 and world_db["ThingTypes"][mt[0]]["TT_TOOL"] == "food"
98 and world_db["ThingTypes"][mt[0]]["TT_TOOLPOWER"]
102 def set_cells_passable_on_memmap_to_65534_on_scoremap():
103 # OUTSOURCED FOR PERFORMANCE REASONS TO libplomrogue.so:
104 # memmap = t["T_MEMMAP"]
105 # for i in [i for i in range(world_db["MAP_LENGTH"] ** 2)
106 # if memmap[i] in symbols_passable]:
107 # set_map_score(i, 65534) # i.e. 65535-1
108 scoremap = c_pointer_to_bytearray(t["T_MEMMAP"])
109 passable_string = c_pointer_to_string(symbols_passable)
110 if libpr.set_cells_passable_on_memmap_to_65534_on_scoremap(scoremap,
112 raise RuntimeError("No score map allocated for set_cells_passable"
113 "_on_memmap_to_65534_on_scoremap().")
115 def init_score_map():
116 test = libpr.init_score_map()
118 raise RuntimeError("Malloc error in init_score_map().")
121 set_cells_passable_on_memmap_to_65534_on_scoremap()
122 maplen = world_db["MAP_LENGTH"]
124 [set_map_score(Thing["pos"], 0) for
125 Thing in animates_in_fov(maplen) if good_attack_target(Thing)]
127 [set_map_score(Thing["pos"], 0) for
128 Thing in animates_in_fov(maplen) if good_flee_target(Thing)]
130 eat_cost = tt["eat_vs_hunger_threshold"]
132 [set_map_score(mt[1] * maplen + mt[2], 0)
133 for mt in t["T_MEMTHING"]
134 if ord_blank != t["T_MEMMAP"][mt[1] * maplen + mt[2]]
135 if world_db["ThingTypes"][mt[0]]["TT_TOOL"] == "food"
136 if world_db["ThingTypes"][mt[0]]["TT_TOOLPOWER"] > eat_cost]
138 zero_score_map_where_char_on_memdepthmap(mem_depth_c[0])
140 [set_map_score(Thing["pos"], 65535)
141 for Thing in animates_in_fov(maplen)
142 if get_map_score(Thing["pos"])]
144 [set_map_score(Thing["pos"], 65535)
145 for Thing in animates_in_fov(maplen)]
147 def rand_target_dir(neighbors, cmp, dirs):
150 for i in range(len(dirs)):
151 if cmp == neighbors[i]:
152 candidates.append(dirs[i])
154 return candidates[rand.next() % n_candidates] if n_candidates else 0
156 def get_neighbor_scores(dirs, eye_pos):
158 if libpr.ready_neighbor_scores(eye_pos):
159 raise RuntimeError("No score map allocated for " +
160 "ready_neighbor_scores.()")
161 for i in range(len(dirs)):
162 scores.append(libpr.get_neighbor_score(i))
165 def get_dir_from_neighbors():
167 dir_to_target = False
170 neighbors = get_neighbor_scores(dirs, eye_pos)
171 minmax_start = 0 if "f" == filter else 65535 - 1
172 minmax_neighbor = minmax_start
173 for i in range(len(dirs)):
174 if ("f" == filter and get_map_score(eye_pos) < neighbors[i] and
175 minmax_neighbor < neighbors[i] and 65535 != neighbors[i]) \
176 or ("f" != filter and minmax_neighbor > neighbors[i]):
177 minmax_neighbor = neighbors[i]
178 if minmax_neighbor != minmax_start:
179 dir_to_target = rand_target_dir(neighbors, minmax_neighbor, dirs)
181 distance = get_map_score(eye_pos)
182 fear_distance = world_db["MAP_LENGTH"]
183 if t["T_SATIATION"] < 0 and math.sqrt(-t["T_SATIATION"]) > 0:
184 fear_distance = fear_distance / math.sqrt(-t["T_SATIATION"])
186 if not dir_to_target:
187 if attack_distance >= distance:
188 dir_to_target = rand_target_dir(neighbors,
190 elif fear_distance >= distance:
191 t["T_COMMAND"] = [id for id in world_db["ThingActions"]
193 world_db["ThingActions"][id]["TA_NAME"]
196 elif dir_to_target and fear_distance < distance:
200 dir_to_target = False
202 run_i = 9 + 1 if "s" == filter else 1
203 while run_i and not dir_to_target and ("s" == filter or seeing_thing()):
206 mem_depth_c = b'9' if b' ' == mem_depth_c \
207 else bytes([mem_depth_c[0] - 1])
208 if libpr.dijkstra_map():
209 raise RuntimeError("No score map allocated for dijkstra_map().")
210 dir_to_target = get_dir_from_neighbors()
211 libpr.free_score_map()
212 if dir_to_target and str == type(dir_to_target):
213 t["T_COMMAND"] = [id for id in world_db["ThingActions"]
214 if world_db["ThingActions"][id]["TA_NAME"]
216 t["T_ARGUMENT"] = ord(dir_to_target)
220 def standing_on_food(t):
221 """Return True/False whether t is standing on healthy consumable."""
222 tt = world_db["ThingTypes"][t["T_TYPE"]]
223 eat_cost = tt["eat_vs_hunger_threshold"]
224 for id in [id for id in world_db["Things"] if world_db["Things"][id] != t
225 if not world_db["Things"][id]["carried"]
226 if world_db["Things"][id]["pos"] == t["pos"]
227 if world_db["ThingTypes"][world_db["Things"][id]["T_TYPE"]]
228 ["TT_TOOL"] == "food"
229 if world_db["ThingTypes"][world_db["Things"][id]["T_TYPE"]]
230 ["TT_TOOLPOWER"] > eat_cost]:
235 def get_inventory_slot_to_consume(t):
236 """Return invent. slot of healthiest consumable(if any healthy),else -1."""
240 tt = world_db["ThingTypes"][t["T_TYPE"]]
241 eat_cost = tt["eat_vs_hunger_threshold"]
242 for id in t["T_CARRIES"]:
243 type = world_db["Things"][id]["T_TYPE"]
244 if world_db["ThingTypes"][type]["TT_TOOL"] == "food" \
245 and world_db["ThingTypes"][type]["TT_TOOLPOWER"]:
246 nutvalue = world_db["ThingTypes"][type]["TT_TOOLPOWER"]
247 tmp_cmp = abs(t["T_SATIATION"] + nutvalue - eat_cost)
248 if (cmp_food < 0 and tmp_cmp < abs(t["T_SATIATION"])) \
249 or tmp_cmp < cmp_food:
257 """Determine next command/argment for actor t via AI algorithms."""
258 t["T_COMMAND"] = [id for id in world_db["ThingActions"]
259 if world_db["ThingActions"][id]["TA_NAME"] == "wait"][0]
260 eating = len([id for id in world_db["ThingActions"]
261 if world_db["ThingActions"][id]["TA_NAME"] == "use"]) > 0
262 picking = len([id for id in world_db["ThingActions"]
263 if world_db["ThingActions"][id]["TA_NAME"] == "pickup"]) > 0
264 if eating and picking:
265 if get_dir_to_target(t, "f"):
267 sel = get_inventory_slot_to_consume(t)
268 from server.config.ai import ai_hook_pickup_test
270 t["T_COMMAND"] = [id for id in world_db["ThingActions"]
271 if world_db["ThingActions"][id]["TA_NAME"]
273 t["T_ARGUMENT"] = sel
274 elif standing_on_food(t) and ai_hook_pickup_test(t):
275 t["T_COMMAND"] = [id for id in world_db["ThingActions"]
276 if world_db["ThingActions"][id]["TA_NAME"]
279 going_to_known_food_spot = get_dir_to_target(t, "c")
280 if not going_to_known_food_spot:
281 aiming_for_walking_food = get_dir_to_target(t, "a")
282 if not aiming_for_walking_food:
283 get_dir_to_target(t, "s")