home · contact · privacy
TCE: Use wetness map separate from terrain map.
[plomrogue] / plugins / server / TheCrawlingEater.py
index e23f00f2cb893e4350d413a37b864b4fd70dd86b..7a634165d8e83249426a9a0a3e0f708e2e88e5cb 100644 (file)
@@ -6,32 +6,78 @@
 from server.config.world_data import world_db
 
 
+def play_drink():
+    if action_exists("drink") and world_db["WORLD_ACTIVE"]:
+        if not chr(world_db["MAP"][world_db["Things"][0]["pos"]]) == "~":
+            log("NOTHING to drink here.")
+            return
+        elif world_db["Things"][0]["T_BLADDER"] >= 32:
+            log("You're too FULL to drink more.")
+            return
+        world_db["set_command"]("drink")
+
+
+def actor_drink(t):
+    pos = world_db["Things"][0]["pos"]
+    if chr(world_db["MAP"][pos]) == "~" and t["T_BLADDER"] < 32:
+        log("You DRINK.")
+        t["T_BLADDER"] += 1
+        world_db["wetmap"][pos] -= 1
+        if world_db["wetmap"][pos] == 48:
+            world_db["MAP"][pos] = ord("_")
+
+
+def play_pee():
+    if action_exists("pee") and world_db["WORLD_ACTIVE"]:
+        if world_db["Things"][0]["T_BLADDER"] < 1:
+            log("Nothing to drop from empty bladder.")
+            return
+        world_db["set_command"]("pee")
+
+
+def actor_pee(t):
+    if t["T_BLADDER"] < 1:
+        return
+    if t == world_db["Things"][0]:
+        log("You LOSE fluid.")
+    t["T_BLADDER"] -= 1
+    terrain = world_db["MAP"][t["pos"]]
+    world_db["wetmap"][t["pos"]] += 1
+    if terrain == ord("_"):
+        world_db["MAP"][t["pos"]] = ord("~")
+    elif world_db["wetmap"][t["pos"]] > 51:
+        t["T_LIFEPOINTS"] = 0
+        if t == world_db["Things"][0]:
+            t["fovmap"] = bytearray(b' ' * (world_db["MAP_LENGTH"] ** 2))
+            log("You DROWN.")
+
+
 def play_drop():
     if action_exists("drop") and world_db["WORLD_ACTIVE"]:
-        if world_db["Things"][0]["T_STOMACH"] < 1:
-            log("Nothing to drop from empty stomach.")
+        if world_db["Things"][0]["T_BOWEL"] < 1:
+            log("Nothing to drop from empty bowel.")
             return
         world_db["set_command"]("drop")
 
 
 def actor_drop(t):
-    if t["T_STOMACH"] < 1:
+    if t["T_BOWEL"] < 1:
         return
     if t == world_db["Things"][0]:
-        log("You DEFECATE.")
+        log("You DROP waste.")
     terrain = world_db["MAP"][t["pos"]]
-    t["T_STOMACH"] -= 1
-    if chr(terrain) == "_":
+    t["T_BOWEL"] -= 1
+    if chr(terrain) in "_~":
         world_db["MAP"][t["pos"]] = ord(".")
-    elif chr(terrain) == ".":
+    elif terrain == ord("."):
         world_db["MAP"][t["pos"]] = ord(":")
-    elif chr(terrain) == ":":
+    elif terrain == ord(":"):
         world_db["MAP"][t["pos"]] = ord("%")
-    elif chr(terrain) == "%":
+    elif terrain == ord("%"):
         world_db["MAP"][t["pos"]] = ord("#")
-    elif chr(terrain) == "#":
+    elif terrain == ord("#"):
         world_db["MAP"][t["pos"]] = ord("X")
-    elif chr(terrain) == "X":
+    elif terrain == ord("X"):
         t["T_LIFEPOINTS"] = 0
         if t == world_db["Things"][0]:
             t["fovmap"] = bytearray(b' ' * (world_db["MAP_LENGTH"] ** 2))
@@ -51,8 +97,11 @@ def play_move(str_arg):
         move_result = mv_yx_in_dir_legal(chr(d), t["T_POSY"], t["T_POSX"])
         if 1 == move_result[0]:
             pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
-            if ord("%") == world_db["MAP"][pos] or \
-                    ord("#") == world_db["MAP"][pos]:
+            if chr(world_db["MAP"][pos]) in "%#":
+                if t["T_BOWEL"] >= 32:
+                    if t == world_db["Things"][0]:
+                        log("You're too FULL to eat.")
+                    return
                 world_db["Things"][0]["T_ARGUMENT"] = d
                 world_db["set_command"]("move")
                 return
@@ -96,19 +145,19 @@ def actor_move(t):
         t["pos"] = move_result[1] * world_db["MAP_LENGTH"] + move_result[2]
         build_fov_map(t)
     else:
-        if t["T_STOMACH"] >= 32:
-            if t == world_db["Things"][0]:
-                log("You're too FULL to eat.")
-        elif ord("%") == world_db["MAP"][pos] and 0 == int(rand.next() % 2):
-            log("You EAT.")
-            world_db["MAP"][pos] = ord("_")
-            t["T_STOMACH"] += 3
-        elif ord("#") == world_db["MAP"][pos] and 0 == int(rand.next() % 5):
-            log("You EAT.")
+        if t["T_BOWEL"] >= 32 or chr(world_db["MAP"][pos]) == "X":
+            return
+        elif chr(world_db["MAP"][pos]) == "%" and 0 == int(rand.next() % 2):
+            t["T_BOWEL"] += 3
+        elif chr(world_db["MAP"][pos]) in "#BEH" and 0 == int(rand.next() % 5):
+            t["T_BOWEL"] += 4
+        log("You EAT.")
+        if world_db["wetmap"][pos] == 48:
             world_db["MAP"][pos] = ord("_")
-            t["T_STOMACH"] += 4
-        if t["T_STOMACH"] > 32:
-            t["T_STOMACH"] = 32
+        else:
+            world_db["MAP"][pos] = ord("~")
+        if t["T_BOWEL"] > 32:
+            t["T_BOWEL"] = 32
 
 
 def make_map():
@@ -124,15 +173,25 @@ def make_map():
             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):
+    n_ground = int((length ** 2) / 16)
+    i_ground = 0
+    while (i_ground <= n_ground):
         single_allowed = rand.next() % 32
         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_trees += 1
+            i_ground += 1
+    n_water = int((length ** 2) / 64)
+    i_water = 0
+    while (i_water <= n_water):
+        single_allowed = rand.next() % 32
+        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("~")
+            world_db["wetmap"][pos] = 51
+            i_water += 1
 
 
 def calc_effort(ta, t):
@@ -180,9 +239,12 @@ def turn_over():
                         action(Thing)
                         Thing["T_COMMAND"] = 0
                         Thing["T_PROGRESS"] = 0
-                    if Thing["T_STOMACH"] > 16:
-                        if 0 == (rand.next() % (33 - Thing["T_STOMACH"])):
+                    if Thing["T_BOWEL"] > 16:
+                        if 0 == (rand.next() % (33 - Thing["T_BOWEL"])):
                             action_db["actor_drop"](Thing)
+                    if Thing["T_BLADDER"] > 16:
+                        if 0 == (rand.next() % (33 - Thing["T_BLADDER"])):
+                            action_db["actor_pee"](Thing)
         world_db["TURN"] += 1
         io_db["worldstate_updateable"] = True
         try_worldstate_update()
@@ -212,12 +274,60 @@ def play_wait():
         world_db["set_command"]("wait")
 
 
+def save_wetmap():
+    length = world_db["MAP_LENGTH"]
+    string = ""
+    for i in range(length):
+        line = world_db["wetmap"][i * length:(i * length) + length].decode()
+        string = string + "WETMAP" + " "  + str(i) + " " + line + "\n"
+    return string
+
+
+def wetmapset(str_int, mapline):
+    def valid_map_line(str_int, mapline):
+        from server.utils import integer_test
+        val = integer_test(str_int, 0, 255)
+        if None != val:
+            if val >= world_db["MAP_LENGTH"]:
+                print("Illegal value for map line number.")
+            elif len(mapline) != world_db["MAP_LENGTH"]:
+                print("Map line length is unequal map width.")
+            else:
+                return val
+        return None
+    val = valid_map_line(str_int, mapline)
+    if None != val:
+        length = world_db["MAP_LENGTH"]
+        if not world_db["wetmap"]:
+            m = bytearray(b' ' * (length ** 2))
+        else:
+            m = world_db["wetmap"]
+        m[val * length:(val * length) + length] = mapline.encode()
+        if not world_db["wetmap"]:
+            world_db["wetmap"] = m
+
+
+def write_wetmap():
+    from server.worldstate_write_helpers import write_map
+    length = world_db["MAP_LENGTH"]
+    visible_wetmap = bytearray(b' ' * (length ** 2))
+    for i in range(length ** 2):
+        if world_db["Things"][0]["fovmap"][i] == ord('v'):
+            visible_wetmap[i] = world_db["wetmap"][i]
+    return write_map(visible_wetmap, world_db["MAP_LENGTH"])
+
+
 from server.config.io import io_db
-io_db["worldstate_write_order"] += [["T_STOMACH", "player_int"]]
+io_db["worldstate_write_order"] += [["T_BOWEL", "player_int"]]
+io_db["worldstate_write_order"] += [["T_BLADDER", "player_int"]]
+io_db["worldstate_write_order"] += [[write_wetmap, "func"]]
 import server.config.world_data
-server.config.world_data.symbols_hide = "%#X"
-server.config.world_data.symbols_passable = "_.:"
-server.config.world_data.thing_defaults["T_STOMACH"] = 0
+server.config.world_data.symbols_hide = "%#X" + "ABC" + "DEF" + "GHI"
+server.config.world_data.symbols_passable = "_.:" + "~JK" + "LMN" + "OPQ"
+server.config.world_data.thing_defaults["T_BOWEL"] = 0
+server.config.world_data.thing_defaults["T_BLADDER"] = 0
+world_db["wetmap"] = bytearray(b"0" * world_db["MAP_LENGTH"] ** 2)
+io_db["hook_save"] = save_wetmap
 import server.config.make_world_helpers
 server.config.make_world_helpers.make_map = make_map
 from server.config.commands import commands_db
@@ -226,15 +336,21 @@ commands_db["ai"] = (0, False, command_ai)
 commands_db["move"] = (1, False, play_move)
 commands_db["wait"] = (0, False, play_wait)
 commands_db["drop"] = (0, False, play_drop)
+commands_db["drink"] = (0, False, play_drink)
+commands_db["pee"] = (0, False, play_pee)
 commands_db["use"] = (1, False, lambda x: None)
 commands_db["pickup"] = (0, False, lambda: None)
-commands_db["T_STOMACH"] = (1, False, setter("Thing", "T_STOMACH", 0, 255))
+commands_db["T_BOWEL"] = (1, False, setter("Thing", "T_BOWEL", 0, 255))
+commands_db["T_BLADDER"] = (1, False, setter("Thing", "T_BLADDER", 0, 255))
+commands_db["WETMAP"] = (2, False, wetmapset)
 from server.actions import actor_wait
 import server.config.actions
 server.config.actions.action_db = {
     "actor_wait": actor_wait,
     "actor_move": actor_move,
-    "actor_drop": actor_drop
+    "actor_drop": actor_drop,
+    "actor_drink": actor_drink,
+    "actor_pee": actor_pee,
 }
 
 strong_write(io_db["file_out"], "PLUGIN TheCrawlingEater\n")