+def get_dir_to_target(t, target):
+
+ from server.utils import rand, libpr, c_pointer_to_bytearray
+ from server.config.world_data import symbols_passable
+
+ def get_map_score(pos):
+ result = libpr.get_map_score(pos)
+ if result < 0:
+ raise RuntimeError("No score map allocated for get_map_score().")
+ return result
+
+ def zero_score_map_where_char_on_memdepthmap(c):
+ map = c_pointer_to_bytearray(t["T_MEMDEPTHMAP"])
+ if libpr.zero_score_map_where_char_on_memdepthmap(c, map):
+ raise RuntimeError("No score map allocated for "
+ "zero_score_map_where_char_on_memdepthmap().")
+
+ def set_map_score(pos, score):
+ test = libpr.set_map_score(pos, score)
+ if test:
+ raise RuntimeError("No score map allocated for set_map_score().")
+
+ def set_movement_cost_map():
+ copy_memmap = t["T_MEMMAP"][:]
+ copy_memmap.replace(b' ', b'4')
+ memmap = c_pointer_to_bytearray(copy_memmap)
+ if libpr.TCE_set_movement_cost_map(memmap):
+ raise RuntimeError("No movement cost map allocated for "
+ "set_movement_cost_map().")
+
+ def animates_in_fov(maplength):
+ return [Thing for Thing in world_db["Things"].values()
+ if Thing["T_LIFEPOINTS"] and 118 == t["fovmap"][Thing["pos"]]
+ and (not Thing == t)]
+
+ def seeing_thing():
+ def exists(gen):
+ try:
+ next(gen)
+ except StopIteration:
+ return False
+ return True
+ mapsize = world_db["MAP_LENGTH"] ** 2
+ if target == "food" and t["T_MEMMAP"]:
+ return exists(pos for pos in range(mapsize)
+ if ord("2") < t["T_MEMMAP"][pos] < ord("5"))
+ elif target == "fluid_certain" and t["fovmap"]:
+ return exists(pos for pos in range(mapsize)
+ if t["fovmap"] == ord("v")
+ if world_db["MAP"][pos] == ord("0")
+ if world_db["wetmap"][pos] > ord("0"))
+ elif target == "crack" and t["T_MEMMAP"]:
+ return exists(pos for pos in range(mapsize)
+ if t["T_MEMMAP"][pos] == ord("-"))
+ elif target == "fluid_potential" and t["T_MEMMAP"] and t["fovmap"]:
+ return exists(pos for pos in range(mapsize)
+ if t["T_MEMMAP"][pos] == ord("0")
+ if t["fovmap"] != ord("v"))
+ elif target == "space" and t["T_MEMMAP"] and t["fovmap"]:
+ return exists(pos for pos in range(mapsize)
+ if ord("-") <= t["T_MEMMAP"][pos] <= ord("2")
+ if (t["fovmap"] != ord("v")
+ or world_db["terrain_fullness"](pos) < 5))
+ elif target in {"hunt", "flee"} and t["fovmap"]:
+ return exists(Thing for
+ Thing in animates_in_fov(world_db["MAP_LENGTH"])) \
+ or exists(pos for pos in range(mapsize)
+ if world_db["soundmap"][pos] > ord("0")
+ if t["fovmap"][pos] != ord("v"))
+ return False
+
+ def init_score_map():
+ mapsize = world_db["MAP_LENGTH"] ** 2
+ test = libpr.TCE_init_score_map()
+ [set_map_score(pos, 65535) for pos in range(mapsize)
+ if chr(t["T_MEMMAP"][pos]) in "5*&"]
+ set_movement_cost_map()
+ if test:
+ raise RuntimeError("Malloc error in init_score_map().")
+ if target == "food" and t["T_MEMMAP"]:
+ [set_map_score(pos, 0) for pos in range(mapsize)
+ if ord("2") < t["T_MEMMAP"][pos] < ord("5")]
+ elif target == "fluid_certain" and t["fovmap"]:
+ [set_map_score(pos, 0) for pos in range(mapsize)
+ if t["fovmap"] == ord("v")
+ if world_db["MAP"][pos] == ord("0")
+ if world_db["wetmap"][pos] > ord("0")]
+ elif target == "crack" and t["T_MEMMAP"]:
+ [set_map_score(pos, 0) for pos in range(mapsize)
+ if t["T_MEMMAP"][pos] == ord("-")]
+ elif target == "fluid_potential" and t["T_MEMMAP"] and t["fovmap"]:
+ [set_map_score(pos, 0) for pos in range(mapsize)
+ if t["T_MEMMAP"][pos] == ord("0")
+ if t["fovmap"] != ord("v")]
+ elif target == "space" and t["T_MEMMAP"] and t["fovmap"]:
+ [set_map_score(pos, 0) for pos in range(mapsize)
+ if ord("-") <= t["T_MEMMAP"][pos] <= ord("2")
+ if (t["fovmap"] != ord("v")
+ or world_db["terrain_fullness"](pos) < 5)]
+ elif target == "search":
+ zero_score_map_where_char_on_memdepthmap(mem_depth_c[0])
+ elif target in {"hunt", "flee"}:
+ [set_map_score(Thing["pos"], 0) for
+ Thing in animates_in_fov(world_db["MAP_LENGTH"])]
+ [set_map_score(pos, 0) for pos in range(mapsize)
+ if world_db["soundmap"][pos] > ord("0")
+ if t["fovmap"][pos] != ord("v")]
+
+ def rand_target_dir(neighbors, cmp, dirs):
+ candidates = []
+ n_candidates = 0
+ for i in range(len(dirs)):
+ if cmp == neighbors[i]:
+ candidates.append(dirs[i])
+ n_candidates += 1
+ return candidates[rand.next() % n_candidates] if n_candidates else 0
+
+ def get_neighbor_scores(dirs, eye_pos):
+ scores = []
+ if libpr.ready_neighbor_scores(eye_pos):
+ raise RuntimeError("No score map allocated for " +
+ "ready_neighbor_scores.()")
+ for i in range(len(dirs)):
+ scores.append(libpr.get_neighbor_score(i))
+ return scores
+
+ def get_dir_from_neighbors():
+ import math
+ dir_to_target = False
+ dirs = "edcxsw"
+ eye_pos = t["pos"]
+ neighbors = get_neighbor_scores(dirs, eye_pos)
+ minmax_start = 0 if "flee" == target else 65535 - 1
+ minmax_neighbor = minmax_start
+ for i in range(len(dirs)):
+ if ("flee" == target and get_map_score(t["pos"]) < neighbors[i] and
+ minmax_neighbor < neighbors[i] and 65535 != neighbors[i]) \
+ or ("flee" != target and minmax_neighbor > neighbors[i]):
+ minmax_neighbor = neighbors[i]
+ if minmax_neighbor != minmax_start:
+ dir_to_target = rand_target_dir(neighbors, minmax_neighbor, dirs)
+ if "flee" == target:
+ distance = get_map_score(t["pos"])
+ fear_distance = 5
+ attack_distance = 1
+ if not dir_to_target:
+ if attack_distance >= distance:
+ dir_to_target = rand_target_dir(neighbors,
+ distance - 1, dirs)
+ elif dir_to_target and fear_distance < distance:
+ dir_to_target = 0
+ return dir_to_target, minmax_neighbor
+
+ dir_to_target = False
+ mem_depth_c = b' '
+ run_i = 9 + 1 if "search" == target else 1
+ minmax_neighbor = 0
+ while run_i and not dir_to_target and \
+ ("search" == target or seeing_thing()):
+ run_i -= 1
+ init_score_map()
+ mem_depth_c = b'9' if b' ' == mem_depth_c \
+ else bytes([mem_depth_c[0] - 1])
+ if libpr.TCE_dijkstra_map_with_movement_cost():
+ raise RuntimeError("No score map allocated for dijkstra_map().")
+ dir_to_target, minmax_neighbor = get_dir_from_neighbors()
+ libpr.free_score_map()
+ if dir_to_target and str == type(dir_to_target):
+ action = "move"
+ from server.utils import mv_yx_in_dir_legal
+ move_result = mv_yx_in_dir_legal(dir_to_target, t["T_POSY"],
+ t["T_POSX"])
+ if 1 != move_result[0]:
+ return False, 0
+ pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
+ hitted = [tid for tid in world_db["Things"]
+ if world_db["Things"][tid]["pos"] == pos]
+ if world_db["MAP"][pos] > ord("2") or len(hitted) > 0:
+ action = "eat"
+ t["T_COMMAND"] = [taid for taid in world_db["ThingActions"]
+ if world_db["ThingActions"][taid]["TA_NAME"]
+ == action][0]
+ t["T_ARGUMENT"] = ord(dir_to_target)
+ return dir_to_target, minmax_neighbor
+world_db["get_dir_to_target"] = get_dir_to_target
+
+
+def terrain_fullness(pos):
+ wetness = world_db["wetmap"][pos] - ord("0")
+ if chr(world_db["MAP"][pos]) in "-+":
+ height = 0
+ else:
+ height = world_db["MAP"][pos] - ord("0")
+ return wetness + height
+world_db["terrain_fullness"] = terrain_fullness
+
+
+def ai(t):
+
+ if t["T_LIFEPOINTS"] == 0:
+ return
+
+ def standing_on_fluid(t):
+ if world_db["MAP"][t["pos"]] == ord("0") and \
+ world_db["wetmap"][t["pos"]] > ord("0"):
+ return True
+ else:
+ return False
+
+ def thing_action_id(name):
+ return [taid for taid in world_db["ThingActions"]
+ if world_db["ThingActions"][taid]
+ ["TA_NAME"] == name][0]
+
+ t["T_COMMAND"] = thing_action_id("wait")
+ needs = {
+ "fix_cracks": 24,
+ "flee": 24,
+ "safe_pee": (world_db["terrain_fullness"](t["pos"]) * t["T_BLADDER"]) / 4,
+ "safe_drop": (world_db["terrain_fullness"](t["pos"]) * t["T_BOWEL"]) / 4,
+ "food": 33 - t["T_STOMACH"],
+ "fluid_certain": 33 - t["T_KIDNEY"],
+ "fluid_potential": 32 - t["T_KIDNEY"],
+ "search": 1,
+ }
+ from operator import itemgetter
+ needs = sorted(needs.items(), key=itemgetter(1,0))
+ needs.reverse()
+ for need in needs:
+ if need[1] > 0:
+ if need[0] == "fix_cracks":
+ if world_db["MAP"][t["pos"]] == ord("-") and \
+ t["T_BOWEL"] > 0 and \
+ world_db["terrain_fullness"](t["pos"]) <= 3:
+ t["T_COMMAND"] = thing_action_id("drop")
+ return
+ elif world_db["get_dir_to_target"](t, "crack"):
+ return
+ if need[0] in {"fluid_certain", "fluid_potential"}:
+ if standing_on_fluid(t):
+ t["T_COMMAND"] = thing_action_id("drink")
+ return
+ elif t["T_BLADDER"] > 0 and \
+ world_db["MAP"][t["pos"]] == ord("0"):
+ t["T_COMMAND"] = thing_action_id("pee")
+ return
+ elif need[0] in {"safe_pee", "safe_drop"}:
+ action_name = need[0][len("safe_"):]
+ if world_db["terrain_fullness"](t["pos"]) <= 3:
+ t["T_COMMAND"] = thing_action_id(action_name)
+ return
+ test = world_db["get_dir_to_target"](t, "space")
+ if test[0]:
+ if test[1] < 5:
+ return
+ elif world_db["terrain_fullness"](t["pos"]) < 5:
+ t["T_COMMAND"] = thing_action_id(action_name)
+ return
+ if t["T_STOMACH"] < 32 and \
+ world_db["get_dir_to_target"](t, "food")[0]:
+ return
+ continue
+ if need[0] in {"fluid_certain", "fluid_potential", "food"}:
+ if world_db["get_dir_to_target"](t, need[0])[0]:
+ return
+ elif world_db["get_dir_to_target"](t, "hunt")[0]:
+ return
+ elif need[0] != "food" and t["T_STOMACH"] < 32 and \
+ world_db["get_dir_to_target"](t, "food")[0]:
+ return
+ elif world_db["get_dir_to_target"](t, need[0])[0]:
+ return
+world_db["ai"] = ai
+
+