--- /dev/null
+from server.config.world_data import world_db
+
+
+def eat_vs_hunger_threshold(thingtype):
+ """Return satiation cost of eating for type. Good food for it must be >."""
+ from server.world import hunger_per_turn
+ hunger_unit = hunger_per_turn(thingtype)
+ actiontype = [id for id in world_db["ThingActions"]
+ if world_db["ThingActions"][id]["TA_NAME"] == "use"][0]
+ return world_db["ThingActions"][actiontype]["TA_EFFORT"] * hunger_unit
+
+
+def get_dir_to_target(t, filter):
+ """Try to set T_COMMAND/T_ARGUMENT for move to "filter"-determined target.
+
+ The path-wise nearest target is chosen, via the shortest available path.
+ Target must not be t. On succcess, return positive value, else False.
+ Filters:
+ "a": Thing in FOV is animate, but of ThingType, starts out weaker than t
+ is, and its corpse would be healthy food for t
+ "f": move away from an enemy – any visible actor whose thing type has more
+ TT_LIFEPOINTS than t LIFEPOINTS, and might find t's corpse healthy
+ food – if it is closer than n steps, where n will shrink as t's hunger
+ grows; if enemy is too close, move towards (attack) the enemy instead;
+ if no fleeing is possible, nor attacking useful, wait; don't tread on
+ non-enemies for fleeing
+ "c": Thing in memorized map is consumable of sufficient nutrition for t
+ "s": memory map cell with greatest-reachable degree of unexploredness
+ """
+ from server.utils import rand, libpr, c_pointer_to_bytearray
+
+ def zero_score_map_where_char_on_memdepthmap(c):
+ # OUTSOURCED FOR PERFORMANCE REASONS TO libplomrogue.so:
+ # for i in [i for i in range(world_db["MAP_LENGTH"] ** 2)
+ # if t["T_MEMDEPTHMAP"][i] == mem_depth_c[0]]:
+ # set_map_score(i, 0)
+ 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_at_thingpos(id, score):
+ pos = world_db["Things"][id]["T_POSY"] * world_db["MAP_LENGTH"] \
+ + world_db["Things"][id]["T_POSX"]
+ set_map_score(pos, score)
+
+ 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 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 animate_in_fov(Thing):
+ if Thing["carried"] or Thing == t or not Thing["T_LIFEPOINTS"]:
+ return False
+ pos = Thing["T_POSY"] * world_db["MAP_LENGTH"] + Thing["T_POSX"]
+ if ord("v") == t["fovmap"][pos]:
+ return True
+
+ def good_attack_target(v):
+ eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
+ type = world_db["ThingTypes"][v["T_TYPE"]]
+ type_corpse = world_db["ThingTypes"][type["TT_CORPSE_ID"]]
+ if t["T_LIFEPOINTS"] > type["TT_LIFEPOINTS"] \
+ and type_corpse["TT_TOOL"] == "food" \
+ and type_corpse["TT_TOOLPOWER"] > eat_cost:
+ return True
+ return False
+
+ def good_flee_target(m):
+ own_corpse_id = world_db["ThingTypes"][t["T_TYPE"]]["TT_CORPSE_ID"]
+ corpse_type = world_db["ThingTypes"][own_corpse_id]
+ targetness = 0 if corpse_type["TT_TOOL"] != "food" \
+ else corpse_type["TT_TOOLPOWER"]
+ type = world_db["ThingTypes"][m["T_TYPE"]]
+ if t["T_LIFEPOINTS"] < type["TT_LIFEPOINTS"] \
+ and targetness > eat_vs_hunger_threshold(m["T_TYPE"]):
+ return True
+ return False
+
+ def seeing_thing():
+ if t["fovmap"] and "a" == filter:
+ for id in world_db["Things"]:
+ if animate_in_fov(world_db["Things"][id]):
+ if good_attack_target(world_db["Things"][id]):
+ return True
+ elif t["fovmap"] and "f" == filter:
+ for id in world_db["Things"]:
+ if animate_in_fov(world_db["Things"][id]):
+ if good_flee_target(world_db["Things"][id]):
+ return True
+ elif t["T_MEMMAP"] and "c" == filter:
+ eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
+ for mt in t["T_MEMTHING"]:
+ if ' ' != chr(t["T_MEMMAP"][(mt[1] * world_db["MAP_LENGTH"])
+ + mt[2]]) \
+ and world_db["ThingTypes"][mt[0]]["TT_TOOL"] == "food" \
+ and world_db["ThingTypes"][mt[0]]["TT_TOOLPOWER"] \
+ > eat_cost:
+ return True
+ return False
+
+ def set_cells_passable_on_memmap_to_65534_on_scoremap():
+ # OUTSOURCED FOR PERFORMANCE REASONS TO libplomrogue.so:
+ # ord_dot = ord(".")
+ # memmap = t["T_MEMMAP"]
+ # for i in [i for i in range(world_db["MAP_LENGTH"] ** 2)
+ # if ord_dot == memmap[i]]:
+ # set_map_score(i, 65534) # i.e. 65535-1
+ map = c_pointer_to_bytearray(t["T_MEMMAP"])
+ if libpr.set_cells_passable_on_memmap_to_65534_on_scoremap(map):
+ raise RuntimeError("No score map allocated for set_cells_passable"
+ "_on_memmap_to_65534_on_scoremap().")
+
+ def init_score_map():
+ test = libpr.init_score_map()
+ if test:
+ raise RuntimeError("Malloc error in init_score_map().")
+ ord_v = ord("v")
+ ord_blank = ord(" ")
+ set_cells_passable_on_memmap_to_65534_on_scoremap()
+ if "a" == filter:
+ for id in world_db["Things"]:
+ if animate_in_fov(world_db["Things"][id]) \
+ and good_attack_target(world_db["Things"][id]):
+ set_map_score_at_thingpos(id, 0)
+ elif "f" == filter:
+ for id in world_db["Things"]:
+ if animate_in_fov(world_db["Things"][id]) \
+ and good_flee_target(world_db["Things"][id]):
+ set_map_score_at_thingpos(id, 0)
+ elif "c" == filter:
+ eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
+ for mt in [mt for mt in t["T_MEMTHING"]
+ if ord_blank != t["T_MEMMAP"][mt[1]
+ * world_db["MAP_LENGTH"]
+ + mt[2]]
+ if world_db["ThingTypes"][mt[0]]["TT_TOOL"] == "food"
+ if world_db["ThingTypes"][mt[0]]["TT_TOOLPOWER"]
+ > eat_cost]:
+ set_map_score(mt[1] * world_db["MAP_LENGTH"] + mt[2], 0)
+ elif "s" == filter:
+ zero_score_map_where_char_on_memdepthmap(mem_depth_c[0])
+ if "a" != filter:
+ for id in world_db["Things"]:
+ if animate_in_fov(world_db["Things"][id]):
+ if "f" == filter:
+ pos = world_db["Things"][id]["T_POSY"] \
+ * world_db["MAP_LENGTH"] \
+ + world_db["Things"][id]["T_POSX"]
+ if 0 == get_map_score(pos):
+ continue
+ set_map_score_at_thingpos(id, 65535)
+
+ 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["T_POSY"] * world_db["MAP_LENGTH"] + t["T_POSX"]
+ neighbors = get_neighbor_scores(dirs, eye_pos)
+ minmax_start = 0 if "f" == filter else 65535 - 1
+ minmax_neighbor = minmax_start
+ for i in range(len(dirs)):
+ if ("f" == filter and get_map_score(eye_pos) < neighbors[i] and
+ minmax_neighbor < neighbors[i] and 65535 != neighbors[i]) \
+ or ("f" != filter 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 "f" == filter:
+ distance = get_map_score(eye_pos)
+ fear_distance = world_db["MAP_LENGTH"]
+ if t["T_SATIATION"] < 0 and math.sqrt(-t["T_SATIATION"]) > 0:
+ fear_distance = fear_distance / math.sqrt(-t["T_SATIATION"])
+ attack_distance = 1
+ if not dir_to_target:
+ if attack_distance >= distance:
+ dir_to_target = rand_target_dir(neighbors,
+ distance - 1, dirs)
+ elif fear_distance >= distance:
+ t["T_COMMAND"] = [id for id in world_db["ThingActions"]
+ if
+ world_db["ThingActions"][id]["TA_NAME"]
+ == "wait"][0]
+ return 1
+ elif dir_to_target and fear_distance < distance:
+ dir_to_target = 0
+ return dir_to_target
+
+ dir_to_target = False
+ mem_depth_c = b' '
+ run_i = 9 + 1 if "s" == filter else 1
+ while run_i and not dir_to_target and ("s" == filter 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.dijkstra_map():
+ raise RuntimeError("No score map allocated for dijkstra_map().")
+ dir_to_target = get_dir_from_neighbors()
+ libpr.free_score_map()
+ if dir_to_target and str == type(dir_to_target):
+ t["T_COMMAND"] = [id for id in world_db["ThingActions"]
+ if world_db["ThingActions"][id]["TA_NAME"]
+ == "move"][0]
+ t["T_ARGUMENT"] = ord(dir_to_target)
+ return dir_to_target
+
+
+def standing_on_food(t):
+ """Return True/False whether t is standing on healthy consumable."""
+ eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
+ for id in [id for id in world_db["Things"] if world_db["Things"][id] != t
+ if not world_db["Things"][id]["carried"]
+ if world_db["Things"][id]["T_POSY"] == t["T_POSY"]
+ if world_db["Things"][id]["T_POSX"] == t["T_POSX"]
+ if world_db["ThingTypes"][world_db["Things"][id]["T_TYPE"]]
+ ["TT_TOOL"] == "food"
+ if world_db["ThingTypes"][world_db["Things"][id]["T_TYPE"]]
+ ["TT_TOOLPOWER"] > eat_cost]:
+ return True
+ return False
+
+
+def get_inventory_slot_to_consume(t):
+ """Return invent. slot of healthiest consumable(if any healthy),else -1."""
+ cmp_food = -1
+ selection = -1
+ i = 0
+ eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
+ for id in t["T_CARRIES"]:
+ type = world_db["Things"][id]["T_TYPE"]
+ if world_db["ThingTypes"][type]["TT_TOOL"] == "food" \
+ and world_db["ThingTypes"][type]["TT_TOOLPOWER"]:
+ nutvalue = world_db["ThingTypes"][type]["TT_TOOLPOWER"]
+ tmp_cmp = abs(t["T_SATIATION"] + nutvalue - eat_cost)
+ if (cmp_food < 0 and tmp_cmp < abs(t["T_SATIATION"])) \
+ or tmp_cmp < cmp_food:
+ cmp_food = tmp_cmp
+ selection = i
+ i += 1
+ return selection
+
+
+def ai(t):
+ """Determine next command/argment for actor t via AI algorithms."""
+ t["T_COMMAND"] = [id for id in world_db["ThingActions"]
+ if world_db["ThingActions"][id]["TA_NAME"] == "wait"][0]
+ if get_dir_to_target(t, "f"):
+ return
+ sel = get_inventory_slot_to_consume(t)
+ if -1 != sel:
+ t["T_COMMAND"] = [id for id in world_db["ThingActions"]
+ if world_db["ThingActions"][id]["TA_NAME"]
+ == "use"][0]
+ t["T_ARGUMENT"] = sel
+ elif standing_on_food(t):
+ t["T_COMMAND"] = [id for id in world_db["ThingActions"]
+ if world_db["ThingActions"][id]["TA_NAME"]
+ == "pick_up"][0]
+ else:
+ going_to_known_food_spot = get_dir_to_target(t, "c")
+ if not going_to_known_food_spot:
+ aiming_for_walking_food = get_dir_to_target(t, "a")
+ if not aiming_for_walking_food:
+ get_dir_to_target(t, "s")