home · contact · privacy
Server, plugin: Refactor thingproliferation (plugin integration).
[plomrogue] / plugins / server / PleaseTheIslandGod.py
index f4f5e6ee9ef8a6c6c9f46d84cb32bafabcffd49d..e4ca0b3ad1054073cd2ea4f5128a132f6836dfa4 100644 (file)
 
 
 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 make_world(seed):
-    from server.update_map_memory import update_map_memory
-    from server.config.misc import make_map_func
-    from server.utils import libpr
-
-    def free_pos(plant=False):
-        i = 0
-        while 1:
-            err = "Space to put thing on too hard to find. Map too small?"
-            while 1:
-                y = rand.next() % world_db["MAP_LENGTH"]
-                x = rand.next() % world_db["MAP_LENGTH"]
-                pos = y * world_db["MAP_LENGTH"] + x;
-                if (not plant
-                    and "." == chr(world_db["MAP"][pos])) \
-                   or ":" == chr(world_db["MAP"][pos]):
-                    break
-                i += 1
-                if i == 65535:
-                    raise SystemExit(err)
-            pos_clear = (0 == len([id for id in world_db["Things"]
-                                   if world_db["Things"][id]["T_LIFEPOINTS"]
-                                   if world_db["Things"][id]["T_POSY"] == y
-                                   if world_db["Things"][id]["T_POSX"] == x]))
-            if pos_clear:
-                break
-        return (y, x)
-
-    rand.seed = seed
-    if world_db["MAP_LENGTH"] < 1:
-        print("Ignoring: No map length >= 1 defined.")
-        return
-    libpr.set_maplength(world_db["MAP_LENGTH"])
-    player_will_be_generated = False
-    playertype = world_db["PLAYER_TYPE"]
-    for ThingType in world_db["ThingTypes"]:
-        if playertype == ThingType:
-            if 0 < world_db["ThingTypes"][ThingType]["TT_START_NUMBER"]:
-                player_will_be_generated = True
-            break
-    if not player_will_be_generated:
-        print("Ignoring: No player type with start number >0 defined.")
-        return
-    wait_action = False
-    for ThingAction in world_db["ThingActions"]:
-        if "wait" == world_db["ThingActions"][ThingAction]["TA_NAME"]:
-            wait_action = True
-    if not wait_action:
-        print("Ignoring beyond SEED_MAP: " +
-              "No thing action with name 'wait' defined.")
-        return
+
+def pos_test(type, y, x):
+    pos = y * world_db["MAP_LENGTH"] + x;
+    plant = world_db["ThingTypes"][type]["TT_PROLIFERATE"]
+    return (not plant or ":" == chr(world_db["MAP"][pos]))
+
+def world_makable():
+    from server.world_makable import world_makable
+    playertype = world_makable()
     for name in world_db["specials"]:
         if world_db[name] not in world_db["ThingTypes"]:
             print("Ignoring: No valid " + name + " set.")
-            return
-    world_db["Things"] = {}
-    make_map()
-    world_db["WORLD_ACTIVE"] = 1
-    world_db["TURN"] = 1
-    for i in range(world_db["ThingTypes"][playertype]["TT_START_NUMBER"]):
-        id = id_setter(-1, "Things")
-        world_db["Things"][id] = new_Thing(playertype, free_pos())
-    if not world_db["Things"][0]["fovmap"]:
-        empty_fovmap = bytearray(b" " * world_db["MAP_LENGTH"] ** 2)
-        world_db["Things"][0]["fovmap"] = empty_fovmap
-    update_map_memory(world_db["Things"][0])
-    for type in world_db["ThingTypes"]:
-        for i in range(world_db["ThingTypes"][type]["TT_START_NUMBER"]):
-            if type != playertype:
-                id = id_setter(-1, "Things")
-                plantness = world_db["ThingTypes"][type]["TT_PROLIFERATE"]
-                world_db["Things"][id] = new_Thing(type, free_pos(plantness))
-    strong_write(io_db["file_out"], "NEW_WORLD\n")
-
-def thingproliferation(t, prol_map):
-    from server.new_thing import new_Thing
-    global directions_db, mv_yx_in_dir_legal
-    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
+            return -1
+    return playertype
 
 def make_map():
+    from server.make_map import make_map, is_neighbor, new_pos
     global rand
-
-    def is_neighbor(coordinates, type):
-        y = coordinates[0]
-        x = coordinates[1]
-        length = world_db["MAP_LENGTH"]
-        ind = y % 2
-        diag_west = x + (ind > 0)
-        diag_east = x + (ind < (length - 1))
-        pos = (y * length) + x
-        if (y > 0 and diag_east
-            and type == chr(world_db["MAP"][pos - length + ind])) \
-           or (x < (length - 1)
-               and type == chr(world_db["MAP"][pos + 1])) \
-           or (y < (length - 1) and diag_east
-               and type == chr(world_db["MAP"][pos + length + ind])) \
-           or (y > 0 and diag_west
-               and type == chr(world_db["MAP"][pos - length - (not ind)])) \
-           or (x > 0
-               and type == chr(world_db["MAP"][pos - 1])) \
-           or (y < (length - 1) and diag_west
-               and type == chr(world_db["MAP"][pos + length - (not ind)])):
-            return True
-        return False
-
-    world_db["MAP"] = bytearray(b'~' * (world_db["MAP_LENGTH"] ** 2))
+    make_map()
     length = world_db["MAP_LENGTH"]
-    add_half_width = (not (length % 2)) * int(length / 2)
-    world_db["MAP"][int((length ** 2) / 2) + add_half_width] = ord(".")
-    while (1):
-        y = rand.next() % length
-        x = rand.next() % length
-        pos = (y * length) + x
-        if "~" == chr(world_db["MAP"][pos]) and is_neighbor((y, x), "."):
-            if y == 0 or y == (length - 1) or x == 0 or x == (length - 1):
-                break
-            world_db["MAP"][pos] = ord(".")
-    n_trees = int((length ** 2) / 16)
-    i_trees = 0
-    while (i_trees <= n_trees):
-        single_allowed = rand.next() % 32
-        y = rand.next() % length
-        x = rand.next() % length
-        pos = (y * length) + x
-        if "." == chr(world_db["MAP"][pos]) \
-          and ((not single_allowed) or is_neighbor((y, x), "X")):
-            world_db["MAP"][pos] = ord("X")
-            i_trees += 1
     n_colons = int((length ** 2) / 16)
     i_colons = 0
     while (i_colons <= n_colons):
         single_allowed = rand.next() % 256
-        y = rand.next() % length
-        x = rand.next() % length
-        pos = (y * length) + x
+        y, x, pos = new_pos()
         if ("." == chr(world_db["MAP"][pos])
           and ((not single_allowed) or is_neighbor((y, x), ":"))):
             world_db["MAP"][pos] = ord(":")
             i_colons += 1
     altar_placed = False
     while not altar_placed:
-        y = rand.next() % length
-        x = rand.next() % length
-        pos = (y * length) + x
+        y, x, pos = new_pos()
         if (("." == chr(world_db["MAP"][pos]
              or ":" == chr(world_db["MAP"][pos]))
             and not is_neighbor((y, x), "X"))):
@@ -189,6 +46,25 @@ def make_map():
             world_db["altar"] = (y, x)
             altar_placed = True
 
+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(t):
     from server.ai import get_dir_to_target, get_inventory_slot_to_consume, \
         standing_on_food
@@ -478,6 +354,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"])
@@ -507,40 +384,37 @@ def actor_move(t):
             return
         if (ord("X") == world_db["MAP"][pos]
             or ord("|") == world_db["MAP"][pos]):
-            carries_axe = False
             for id in t["T_CARRIES"]:
                 type = world_db["Things"][id]["T_TYPE"]
                 if world_db["ThingTypes"][type]["TT_TOOL"] == "axe":
-                    carries_axe = True
-                    break
-            if carries_axe:
-                axe_name = world_db["ThingTypes"][type]["TT_NAME"]
-                if t == world_db["Things"][0]:
-                    log("With your " + axe_name + ", you chop!")
-                    if ord("X") == world_db["MAP"][pos]:
-                        world_db["GOD_FAVOR"] -= 1
-                chop_power = world_db["ThingTypes"][type]["TT_TOOLPOWER"]
-
-                case_X = world_db["MAP"][pos] == ord("X")
-                if (chop_power > 0
-                    and ((case_X and
-                          0 == int(rand.next() / chop_power))
-                    or (not case_X and
-                             0 == int(rand.next() / (3 * chop_power))))):
+                    axe_name = world_db["ThingTypes"][type]["TT_NAME"]
                     if t == world_db["Things"][0]:
-                        log("You chop it DOWN.")
+                        log("With your " + axe_name + ", you chop!")
                         if ord("X") == world_db["MAP"][pos]:
-                            world_db["GOD_FAVOR"] -= 10
-                    world_db["MAP"][pos] = ord(".")
-                    i = 3 if case_X else 1
-                    for i in range(i):
-                        id = id_setter(-1, "Things")
-                        world_db["Things"][id] = \
-                          new_Thing(world_db["LUMBER"],
-                                    (move_result[1], move_result[2]))
-                    build_fov_map(t)
-                return
+                            world_db["GOD_FAVOR"] -= 1
+                    chop_power = world_db["ThingTypes"][type]["TT_TOOLPOWER"]
+
+                    case_X = world_db["MAP"][pos] == ord("X")
+                    if (chop_power > 0
+                        and ((case_X and
+                              0 == int(rand.next() / chop_power))
+                        or (not case_X and
+                                 0 == int(rand.next() / (3 * chop_power))))):
+                        if t == world_db["Things"][0]:
+                            log("You chop it DOWN.")
+                            if ord("X") == world_db["MAP"][pos]:
+                                world_db["GOD_FAVOR"] -= 10
+                        world_db["MAP"][pos] = ord(".")
+                        i = 3 if case_X else 1
+                        for i in range(i):
+                            id = id_setter(-1, "Things")
+                            world_db["Things"][id] = \
+                              new_Thing(world_db["LUMBER"],
+                                        (move_result[1], move_result[2]))
+                        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:
@@ -580,41 +454,44 @@ def command_worldactive(worldactive_string):
             else:
                 print("World already active.")
         elif 0 == world_db["WORLD_ACTIVE"]:
-            wait_exists = False
             for ThingAction in world_db["ThingActions"]:
                 if "wait" == world_db["ThingActions"][ThingAction]["TA_NAME"]:
-                    wait_exists = True
                     break
-            player_exists = False
+            else:
+                print("Ignored: No wait action defined for world to activate.")
+                return
             for Thing in world_db["Things"]:
                 if 0 == Thing:
-                    player_exists = True
                     break
-            altar_found = False
+            else:
+                print("Ignored: No player defined for world to activate.")
+                return
             if world_db["MAP"]:
                 pos = world_db["MAP"].find(b'_')
                 if pos > 0:
                     y = int(pos / world_db["MAP_LENGTH"])
                     x = pos % world_db["MAP_LENGTH"]
                     world_db["altar"] = (y, x)
-                    altar_found = True
-            specials_set = True
+                else:
+                    print("Ignored: No altar defined for world to activate.")
+                    return
+            else:
+                print("Ignored: No map defined for world to activate.")
+                return
             for name in world_db["specials"]:
                 if world_db[name] not in world_db["ThingTypes"]:
-                    specials_set = False
-            if altar_found and wait_exists and player_exists and \
-                    world_db["MAP"] and specials_set:
-                for id in world_db["Things"]:
-                    if world_db["Things"][id]["T_LIFEPOINTS"]:
-                        build_fov_map(world_db["Things"][id])
-                        if 0 == id:
-                            update_map_memory(world_db["Things"][id], False)
-                if not world_db["Things"][0]["T_LIFEPOINTS"]:
-                    empty_fovmap = bytearray(b" " * world_db["MAP_LENGTH"] ** 2)
-                    world_db["Things"][0]["fovmap"] = empty_fovmap
-                world_db["WORLD_ACTIVE"] = 1
-            else:
-                print("Ignoring: Not all conditions for world activation met.")
+                    print("Ignored: Not all specials set for world to "
+                          "activate.")
+                    return
+            for id in world_db["Things"]:
+                if world_db["Things"][id]["T_LIFEPOINTS"]:
+                    build_fov_map(world_db["Things"][id])
+                    if 0 == id:
+                        update_map_memory(world_db["Things"][id], False)
+            if not world_db["Things"][0]["T_LIFEPOINTS"]:
+                empty_fovmap = bytearray(b" " * world_db["MAP_LENGTH"] ** 2)
+                world_db["Things"][0]["fovmap"] = empty_fovmap
+            world_db["WORLD_ACTIVE"] = 1
 
 def play_move(str_arg):
     if action_exists("move"):
@@ -624,7 +501,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]
@@ -796,33 +673,21 @@ def play_pickup():
 
 strong_write(io_db["file_out"], "PLUGIN PleaseTheIslandGod\n")
 
-if not "GOD_FAVOR" in world_db:
-    world_db["GOD_FAVOR"] = 0
-if not "FAVOR_STAGE" in world_db:
-    world_db["FAVOR_STAGE"] = 0
-if not "SLIPPERS" in world_db:
-    world_db["SLIPPERS"] = 0
-if not "PLANT_0" in world_db:
-    world_db["PLANT_0"] = 0
-if not "PLANT_1" in world_db:
-    world_db["PLANT_1"] = 0
-if not "ANIMAL_0" in world_db:
-    world_db["ANIMAL_0"] = 0
-if not "ANIMAL_1" in world_db:
-    world_db["ANIMAL_1"] = 0
-if not "TOOL_0" in world_db:
-    world_db["TOOL_0"] = 0
-if not "TOOL_1" in world_db:
-    world_db["TOOL_1"] = 0
-if not "LUMBER" in world_db:
-    world_db["LUMBER"] = 0
-if not "EMPATHY" in world_db:
-    world_db["EMPATHY"] = 0
+def set_zero_if_undefined(key):
+    if not key in world_db:
+        world_db[key] = 0
+
+set_zero_if_undefined("GOD_FAVOR")
+set_zero_if_undefined("FAVOR_STAGE")
+set_zero_if_undefined("EMPATHY")
+world_db["specials"] = ["SLIPPERS", "PLANT_0", "PLANT_1", "TOOL_0", "TOOL_1",
+    "LUMBER", "ANIMAL_0", "ANIMAL_1"]
+for key in world_db["specials"]:
+    set_zero_if_undefined(key)
+
 world_db["terrain_names"][":"] = "SOIL"
 world_db["terrain_names"]["|"] = "WALL"
 world_db["terrain_names"]["_"] = "ALTAR"
-world_db["specials"] = ["SLIPPERS", "PLANT_0", "PLANT_1", "TOOL_0", "TOOL_1",
-    "LUMBER", "ANIMAL_0", "ANIMAL_1"]
 io_db["worldstate_write_order"] += [["GOD_FAVOR", "world_int"]]
 io_db["worldstate_write_order"] += [[write_metamap_A, "func"]]
 io_db["worldstate_write_order"] += [[write_metamap_B, "func"]]
@@ -862,7 +727,17 @@ 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.make_world = make_world
 server.config.misc.decrement_lifepoints_func = decrement_lifepoints
 server.config.misc.calc_effort_func = calc_effort
+
+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