1 from server.config.world_data import world_db
4 def eat_vs_hunger_threshold(thingtype):
5 """Return satiation cost of eating for type. Good food for it must be >."""
6 from server.world import hunger_per_turn
7 hunger_unit = hunger_per_turn(thingtype)
8 actiontype = [id for id in world_db["ThingActions"]
9 if world_db["ThingActions"][id]["TA_NAME"] == "use"][0]
10 return world_db["ThingActions"][actiontype]["TA_EFFORT"] * hunger_unit
13 def get_dir_to_target(t, filter):
14 """Try to set T_COMMAND/T_ARGUMENT for move to "filter"-determined target.
16 The path-wise nearest target is chosen, via the shortest available path.
17 Target must not be t. On succcess, return positive value, else False.
19 "a": Thing in FOV is animate, but of ThingType, starts out weaker than t
20 is, and its corpse would be healthy food for t
21 "f": move away from an enemy – any visible actor whose thing type has more
22 TT_LIFEPOINTS than t LIFEPOINTS, and might find t's corpse healthy
23 food – if it is closer than n steps, where n will shrink as t's hunger
24 grows; if enemy is too close, move towards (attack) the enemy instead;
25 if no fleeing is possible, nor attacking useful, wait; don't tread on
26 non-enemies for fleeing
27 "c": Thing in memorized map is consumable of sufficient nutrition for t
28 "s": memory map cell with greatest-reachable degree of unexploredness
30 from server.utils import rand, libpr, c_pointer_to_bytearray
32 def zero_score_map_where_char_on_memdepthmap(c):
33 # OUTSOURCED FOR PERFORMANCE REASONS TO libplomrogue.so:
34 # for i in [i for i in range(world_db["MAP_LENGTH"] ** 2)
35 # if t["T_MEMDEPTHMAP"][i] == mem_depth_c[0]]:
37 map = c_pointer_to_bytearray(t["T_MEMDEPTHMAP"])
38 if libpr.zero_score_map_where_char_on_memdepthmap(c, map):
39 raise RuntimeError("No score map allocated for "
40 "zero_score_map_where_char_on_memdepthmap().")
42 def set_map_score_at_thingpos(id, score):
43 pos = world_db["Things"][id]["T_POSY"] * world_db["MAP_LENGTH"] \
44 + world_db["Things"][id]["T_POSX"]
45 set_map_score(pos, score)
47 def set_map_score(pos, score):
48 test = libpr.set_map_score(pos, score)
50 raise RuntimeError("No score map allocated for set_map_score().")
52 def get_map_score(pos):
53 result = libpr.get_map_score(pos)
55 raise RuntimeError("No score map allocated for get_map_score().")
58 def animate_in_fov(Thing):
59 if Thing["carried"] or Thing == t or not Thing["T_LIFEPOINTS"]:
61 pos = Thing["T_POSY"] * world_db["MAP_LENGTH"] + Thing["T_POSX"]
62 if ord("v") == t["fovmap"][pos]:
65 def good_attack_target(v):
66 eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
67 type = world_db["ThingTypes"][v["T_TYPE"]]
68 type_corpse = world_db["ThingTypes"][type["TT_CORPSE_ID"]]
69 if t["T_LIFEPOINTS"] > type["TT_LIFEPOINTS"] \
70 and type_corpse["TT_TOOL"] == "food" \
71 and type_corpse["TT_TOOLPOWER"] > eat_cost:
75 def good_flee_target(m):
76 own_corpse_id = world_db["ThingTypes"][t["T_TYPE"]]["TT_CORPSE_ID"]
77 corpse_type = world_db["ThingTypes"][own_corpse_id]
78 targetness = 0 if corpse_type["TT_TOOL"] != "food" \
79 else corpse_type["TT_TOOLPOWER"]
80 type = world_db["ThingTypes"][m["T_TYPE"]]
81 if t["T_LIFEPOINTS"] < type["TT_LIFEPOINTS"] \
82 and targetness > eat_vs_hunger_threshold(m["T_TYPE"]):
87 if t["fovmap"] and "a" == filter:
88 for id in world_db["Things"]:
89 if animate_in_fov(world_db["Things"][id]):
90 if good_attack_target(world_db["Things"][id]):
92 elif t["fovmap"] and "f" == filter:
93 for id in world_db["Things"]:
94 if animate_in_fov(world_db["Things"][id]):
95 if good_flee_target(world_db["Things"][id]):
97 elif t["T_MEMMAP"] and "c" == filter:
98 eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
99 for mt in t["T_MEMTHING"]:
100 if ' ' != chr(t["T_MEMMAP"][(mt[1] * world_db["MAP_LENGTH"])
102 and world_db["ThingTypes"][mt[0]]["TT_TOOL"] == "food" \
103 and world_db["ThingTypes"][mt[0]]["TT_TOOLPOWER"] \
108 def set_cells_passable_on_memmap_to_65534_on_scoremap():
109 # OUTSOURCED FOR PERFORMANCE REASONS TO libplomrogue.so:
111 # memmap = t["T_MEMMAP"]
112 # for i in [i for i in range(world_db["MAP_LENGTH"] ** 2)
113 # if ord_dot == memmap[i]]:
114 # set_map_score(i, 65534) # i.e. 65535-1
115 map = c_pointer_to_bytearray(t["T_MEMMAP"])
116 if libpr.set_cells_passable_on_memmap_to_65534_on_scoremap(map):
117 raise RuntimeError("No score map allocated for set_cells_passable"
118 "_on_memmap_to_65534_on_scoremap().")
120 def init_score_map():
121 test = libpr.init_score_map()
123 raise RuntimeError("Malloc error in init_score_map().")
126 set_cells_passable_on_memmap_to_65534_on_scoremap()
128 for id in world_db["Things"]:
129 if animate_in_fov(world_db["Things"][id]) \
130 and good_attack_target(world_db["Things"][id]):
131 set_map_score_at_thingpos(id, 0)
133 for id in world_db["Things"]:
134 if animate_in_fov(world_db["Things"][id]) \
135 and good_flee_target(world_db["Things"][id]):
136 set_map_score_at_thingpos(id, 0)
138 eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
139 for mt in [mt for mt in t["T_MEMTHING"]
140 if ord_blank != t["T_MEMMAP"][mt[1]
141 * world_db["MAP_LENGTH"]
143 if world_db["ThingTypes"][mt[0]]["TT_TOOL"] == "food"
144 if world_db["ThingTypes"][mt[0]]["TT_TOOLPOWER"]
146 set_map_score(mt[1] * world_db["MAP_LENGTH"] + mt[2], 0)
148 zero_score_map_where_char_on_memdepthmap(mem_depth_c[0])
150 for id in world_db["Things"]:
151 if animate_in_fov(world_db["Things"][id]):
153 pos = world_db["Things"][id]["T_POSY"] \
154 * world_db["MAP_LENGTH"] \
155 + world_db["Things"][id]["T_POSX"]
156 if 0 == get_map_score(pos):
158 set_map_score_at_thingpos(id, 65535)
160 def rand_target_dir(neighbors, cmp, dirs):
163 for i in range(len(dirs)):
164 if cmp == neighbors[i]:
165 candidates.append(dirs[i])
167 return candidates[rand.next() % n_candidates] if n_candidates else 0
169 def get_neighbor_scores(dirs, eye_pos):
171 if libpr.ready_neighbor_scores(eye_pos):
172 raise RuntimeError("No score map allocated for " +
173 "ready_neighbor_scores.()")
174 for i in range(len(dirs)):
175 scores.append(libpr.get_neighbor_score(i))
178 def get_dir_from_neighbors():
180 dir_to_target = False
182 eye_pos = t["T_POSY"] * world_db["MAP_LENGTH"] + t["T_POSX"]
183 neighbors = get_neighbor_scores(dirs, eye_pos)
184 minmax_start = 0 if "f" == filter else 65535 - 1
185 minmax_neighbor = minmax_start
186 for i in range(len(dirs)):
187 if ("f" == filter and get_map_score(eye_pos) < neighbors[i] and
188 minmax_neighbor < neighbors[i] and 65535 != neighbors[i]) \
189 or ("f" != filter and minmax_neighbor > neighbors[i]):
190 minmax_neighbor = neighbors[i]
191 if minmax_neighbor != minmax_start:
192 dir_to_target = rand_target_dir(neighbors, minmax_neighbor, dirs)
194 distance = get_map_score(eye_pos)
195 fear_distance = world_db["MAP_LENGTH"]
196 if t["T_SATIATION"] < 0 and math.sqrt(-t["T_SATIATION"]) > 0:
197 fear_distance = fear_distance / math.sqrt(-t["T_SATIATION"])
199 if not dir_to_target:
200 if attack_distance >= distance:
201 dir_to_target = rand_target_dir(neighbors,
203 elif fear_distance >= distance:
204 t["T_COMMAND"] = [id for id in world_db["ThingActions"]
206 world_db["ThingActions"][id]["TA_NAME"]
209 elif dir_to_target and fear_distance < distance:
213 dir_to_target = False
215 run_i = 9 + 1 if "s" == filter else 1
216 while run_i and not dir_to_target and ("s" == filter or seeing_thing()):
219 mem_depth_c = b'9' if b' ' == mem_depth_c \
220 else bytes([mem_depth_c[0] - 1])
221 if libpr.dijkstra_map():
222 raise RuntimeError("No score map allocated for dijkstra_map().")
223 dir_to_target = get_dir_from_neighbors()
224 libpr.free_score_map()
225 if dir_to_target and str == type(dir_to_target):
226 t["T_COMMAND"] = [id for id in world_db["ThingActions"]
227 if world_db["ThingActions"][id]["TA_NAME"]
229 t["T_ARGUMENT"] = ord(dir_to_target)
233 def standing_on_food(t):
234 """Return True/False whether t is standing on healthy consumable."""
235 eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
236 for id in [id for id in world_db["Things"] if world_db["Things"][id] != t
237 if not world_db["Things"][id]["carried"]
238 if world_db["Things"][id]["T_POSY"] == t["T_POSY"]
239 if world_db["Things"][id]["T_POSX"] == t["T_POSX"]
240 if world_db["ThingTypes"][world_db["Things"][id]["T_TYPE"]]
241 ["TT_TOOL"] == "food"
242 if world_db["ThingTypes"][world_db["Things"][id]["T_TYPE"]]
243 ["TT_TOOLPOWER"] > eat_cost]:
248 def get_inventory_slot_to_consume(t):
249 """Return invent. slot of healthiest consumable(if any healthy),else -1."""
253 eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
254 for id in t["T_CARRIES"]:
255 type = world_db["Things"][id]["T_TYPE"]
256 if world_db["ThingTypes"][type]["TT_TOOL"] == "food" \
257 and world_db["ThingTypes"][type]["TT_TOOLPOWER"]:
258 nutvalue = world_db["ThingTypes"][type]["TT_TOOLPOWER"]
259 tmp_cmp = abs(t["T_SATIATION"] + nutvalue - eat_cost)
260 if (cmp_food < 0 and tmp_cmp < abs(t["T_SATIATION"])) \
261 or tmp_cmp < cmp_food:
269 """Determine next command/argment for actor t via AI algorithms."""
270 t["T_COMMAND"] = [id for id in world_db["ThingActions"]
271 if world_db["ThingActions"][id]["TA_NAME"] == "wait"][0]
272 if get_dir_to_target(t, "f"):
274 sel = get_inventory_slot_to_consume(t)
276 t["T_COMMAND"] = [id for id in world_db["ThingActions"]
277 if world_db["ThingActions"][id]["TA_NAME"]
279 t["T_ARGUMENT"] = sel
280 elif standing_on_food(t):
281 t["T_COMMAND"] = [id for id in world_db["ThingActions"]
282 if world_db["ThingActions"][id]["TA_NAME"]
285 going_to_known_food_spot = get_dir_to_target(t, "c")
286 if not going_to_known_food_spot:
287 aiming_for_walking_food = get_dir_to_target(t, "a")
288 if not aiming_for_walking_food:
289 get_dir_to_target(t, "s")