home · contact · privacy
Server, plugin: Refactor actor_drop plugin hooking.
[plomrogue] / plugins / server / PleaseTheIslandGod.py
index 93e69810c160736242531db987eeffb2cacb4e2e..b3afc6a3888808947bc823a1625a989dbbf4ee64 100644 (file)
@@ -4,10 +4,9 @@
 
 
 from server.io import log, strong_write
-from server.config.world_data import world_db, symbols_passable, directions_db
+from server.config.world_data import world_db, directions_db
 from server.utils import mv_yx_in_dir_legal, rand, id_setter
 from server.config.io import io_db
-from server.new_thing import new_Thing
 
 def pos_test(type, y, x):
     pos = y * world_db["MAP_LENGTH"] + x;
@@ -47,116 +46,55 @@ def make_map():
             world_db["altar"] = (y, x)
             altar_placed = True
 
-def thingproliferation(t, prol_map):
-    from server.new_thing import new_Thing
-    global directions_db, mv_yx_in_dir_legal, rand
-    prolscore = world_db["ThingTypes"][t["T_TYPE"]]["TT_PROLIFERATE"]
-    if prolscore and \
-      (world_db["ThingTypes"][t["T_TYPE"]]["TT_LIFEPOINTS"] == 0 or
-       t["T_LIFEPOINTS"] >= 0.9 *
-                        world_db["ThingTypes"][t["T_TYPE"]]["TT_LIFEPOINTS"]) \
-       and \
-      (1 == prolscore or 1 == (rand.next() % prolscore)):
-        candidates = []
-        for dir in [directions_db[key] for key in directions_db]:
-            mv_result = mv_yx_in_dir_legal(dir, t["T_POSY"], t["T_POSX"])
-            pos = mv_result[1] * world_db["MAP_LENGTH"] + mv_result[2]
-            if mv_result[0] and \
-               (ord(":") == prol_map[pos]
-                or (world_db["ThingTypes"][t["T_TYPE"]]["TT_LIFEPOINTS"]
-                    and ord(".") == prol_map[pos])):
-                candidates.append((mv_result[1], mv_result[2]))
-        if len(candidates):
-            i = rand.next() % len(candidates)
-            id = id_setter(-1, "Things")
-            newT = new_Thing(t["T_TYPE"], (candidates[i][0], candidates[i][1]))
-            world_db["Things"][id] = newT
-            if (world_db["FAVOR_STAGE"] > 0
-                and t["T_TYPE"] == world_db["PLANT_0"]):
-                world_db["GOD_FAVOR"] += 5
-            elif t["T_TYPE"] == world_db["PLANT_1"]:
-                world_db["GOD_FAVOR"] += 25
-            elif world_db["FAVOR_STAGE"] >= 4 and \
-                t["T_TYPE"] == world_db["ANIMAL_1"]:
-                log("The Island God SMILES upon a new-born bear baby.")
-                world_db["GOD_FAVOR"] += 750
-
-def ai(t):
-    from server.ai import get_dir_to_target, get_inventory_slot_to_consume, \
-        standing_on_food
-    t["T_COMMAND"] = [id for id in world_db["ThingActions"]
-                      if world_db["ThingActions"][id]["TA_NAME"] == "wait"][0]
-    eating = len([id for id in world_db["ThingActions"]
-                  if world_db["ThingActions"][id]["TA_NAME"] == "use"]) > 0
-    picking = len([id for id in world_db["ThingActions"]
-                   if world_db["ThingActions"][id]["TA_NAME"] == "pickup"]) > 0
-    if eating and picking:
-        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) and (len(t["T_CARRIES"]) <
-                world_db["ThingTypes"][t["T_TYPE"]]["TT_STORAGE"]):
-                t["T_COMMAND"] = [id for id in world_db["ThingActions"]
-                                  if world_db["ThingActions"][id]["TA_NAME"]
-                                  == "pickup"][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")
+def field_spreadable(c, t):
+    return ":" == c or (world_db["ThingTypes"][t["T_TYPE"]]["TT_LIFEPOINTS"]
+                        and "." == c)
+
+def thingprol_plugin_conditions(t):
+    tt = world_db["ThingTypes"][t["T_TYPE"]]
+    return (tt["TT_LIFEPOINTS"] == 0 or \
+            t["T_LIFEPOINTS"] >= 0.9 * tt["TT_LIFEPOINTS"])
+
+def thingprol_plugin_post_create_hook(t):
+    tt = world_db["ThingTypes"][t["T_TYPE"]]
+    if (world_db["FAVOR_STAGE"] > 0 and t["T_TYPE"] == world_db["PLANT_0"]):
+        world_db["GOD_FAVOR"] += 5
+    elif t["T_TYPE"] == world_db["PLANT_1"]:
+        world_db["GOD_FAVOR"] += 25
+    elif world_db["FAVOR_STAGE"] >= 4 and t["T_TYPE"] == world_db["ANIMAL_1"]:
+        log("The Island God SMILES upon a new-born bear baby.")
+        world_db["GOD_FAVOR"] += 750
+
+def ai_hook_pickup_test(t):
+    return len(t["T_CARRIES"]) < \
+        world_db["ThingTypes"][t["T_TYPE"]]["TT_STORAGE"]
 
 def actor_pickup(t):
-    from server.ai import eat_vs_hunger_threshold
     used_slots = len(t["T_CARRIES"])
     if used_slots < world_db["ThingTypes"][t["T_TYPE"]]["TT_STORAGE"]:
-        ids = [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 len(ids):
-            lowest_tid = -1
-            eat_cost = eat_vs_hunger_threshold(t["T_TYPE"])
-            for iid in ids:
-                tid = world_db["Things"][iid]["T_TYPE"] 
-                if lowest_tid == -1 or tid < lowest_tid:
-                    if (t != world_db["Things"][0] and
-                        (world_db["ThingTypes"][tid]["TT_TOOL"] != "food"
-                         or (world_db["ThingTypes"][tid]["TT_TOOLPOWER"]
-                             <= eat_cost))):
-                        continue
-                    id = iid
-                    lowest_tid = tid
-            world_db["Things"][id]["carried"] = True
-            ty = world_db["Things"][id]["T_TYPE"]
-            if (t != world_db["Things"][0]
-                and world_db["Things"][id]["T_PLAYERDROP"]
-                and world_db["ThingTypes"][ty]["TT_TOOL"] == "food"):
-                score = int(world_db["ThingTypes"][ty]["TT_TOOLPOWER"] / 32)
+        from server.actions import actor_pickup
+        t_picked = actor_pickup(t)
+        if t_picked != None:
+            ty = world_db["ThingTypes"][t_picked["T_TYPE"]]
+            if t != world_db["Things"][0] and t_picked["T_PLAYERDROP"] \
+                    and ty["TT_TOOL"] == "food":
+                score = int(ty["TT_TOOLPOWER"] / 32)
                 world_db["GOD_FAVOR"] += score
-                world_db["Things"][id]["T_PLAYERDROP"] = 0
-            t["T_CARRIES"].append(id)
-            if t == world_db["Things"][0]:
-                log("You PICK UP an object.")
+                t_picked["T_PLAYERDROP"] = 0
     elif t == world_db["Things"][0]:
         log("CAN'T pick up object: No storage room to carry more.")
 
+def actor_pickup_test_hook(t, tid):
+    from server.ai import eat_vs_hunger_threshold
+    tt = world_db["ThingTypes"][tid]
+    return not (t != world_db["Things"][0] and (tt["TT_TOOL"] != "food" or
+            (tt["TT_TOOLPOWER"] <= eat_vs_hunger_threshold(t["T_TYPE"]))))
 
 def actor_drop(t):
-    """Make t rop Thing from inventory to ground indexed by T_ARGUMENT."""
-    if len(t["T_CARRIES"]):
-        id = t["T_CARRIES"][t["T_ARGUMENT"]]
-        t["T_CARRIES"].remove(id)
-        world_db["Things"][id]["carried"] = False
-        if t == world_db["Things"][0]:
-            log("You DROP an object.")
-            world_db["Things"][id]["T_PLAYERDROP"] = 1
-
+    from server.actions import actor_drop
+    dropped = actor_drop(t)
+    if dropped != None:
+        dropped["T_PLAYERDROP"] = 1
 
 def actor_use(t):
     if len(t["T_CARRIES"]):
@@ -370,6 +308,7 @@ def actor_move(t):
     from server.build_fov_map import build_fov_map
     from server.config.misc import decrement_lifepoints_func
     from server.new_thing import new_Thing
+    global mv_yx_in_dir_legal
     passable = False
     move_result = mv_yx_in_dir_legal(chr(t["T_ARGUMENT"]),
                                      t["T_POSY"], t["T_POSX"])
@@ -429,6 +368,7 @@ def actor_move(t):
                         build_fov_map(t)
                     return
         passable = chr(world_db["MAP"][pos]) in symbols_passable
+    global directions_db
     dir = [dir for dir in directions_db
            if directions_db[dir] == chr(t["T_ARGUMENT"])][0]
     if passable:
@@ -515,7 +455,7 @@ def play_move(str_arg):
             print("Illegal move direction string.")
             return
         dir = ord(directions_db[str_arg])
-        from server.utils import mv_yx_in_dir_legal
+        global mv_yx_in_dir_legal
         move_result = mv_yx_in_dir_legal(chr(dir), t["T_POSY"], t["T_POSX"])
         if 1 == move_result[0]:
             pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
@@ -717,7 +657,7 @@ server.config.actions.action_db["actor_move"] = actor_move
 server.config.actions.action_db["actor_pickup"] = actor_pickup
 server.config.actions.action_db["actor_drop"] = actor_drop
 server.config.actions.action_db["actor_use"] = actor_use
-server.config.actions.ai_func = ai
+server.config.actions.actor_pickup_test_hook = actor_pickup_test_hook
 
 from server.config.commands import commands_db
 commands_db["TT_ID"] = (1, False, command_ttid)
@@ -741,7 +681,6 @@ commands_db["pickup"] = (0, False, play_pickup)
 
 import server.config.misc
 server.config.misc.make_map_func = make_map
-server.config.misc.thingproliferation_func = thingproliferation
 server.config.misc.decrement_lifepoints_func = decrement_lifepoints
 server.config.misc.calc_effort_func = calc_effort
 
@@ -749,3 +688,13 @@ import server.config.make_world_helpers
 server.config.make_world_helpers.pos_test_func = pos_test
 server.config.make_world_helpers.world_makable_func = world_makable
 server.config.make_world_helpers.make_map_func = make_map
+
+import server.config.thingproliferation
+server.config.thingproliferation.field_spreadable = field_spreadable
+server.config.thingproliferation.thingprol_plugin_conditions = \
+    thingprol_plugin_conditions
+server.config.thingproliferation.thingprol_plugin_post_create_hook = \
+    thingprol_plugin_post_create_hook
+
+import server.config.ai
+server.config.ai.ai_hook_pickup = ai_hook_pickup_test