TCE: Add drinking, bladder.

[plomrogue] / plugins / client / TheCrawlingEater.py

diff --git a/plugins/client/TheCrawlingEater.py b/plugins/client/TheCrawlingEater.py
index e362409740c7f20f47b91400b40b63be8725c424..6996daf1ef4d25bfe611984c33cb2e4bc665bf25 100644 (file)
--- a/plugins/client/TheCrawlingEater.py
+++ b/plugins/client/TheCrawlingEater.py
@@ -3,6 +3,16 @@
 # see the file NOTICE in the root directory of the PlomRogue source package.
 
 
+def win_bladder(self):
+    winmap = []
+    curses.init_pair(79, curses.COLOR_WHITE, curses.COLOR_BLUE)
+    for i in range(world_data["bladder"]):
+        winmap += [("~", curses.color_pair(79))]
+    winmap_size = [1, len(winmap)]
+    offset = [0, 0]
+    return offset, winmap_size, winmap
+
+
 def win_stomach(self):
     winmap = []
     curses.init_pair(80, curses.COLOR_YELLOW, curses.COLOR_RED)
@@ -33,13 +43,16 @@ def win_map(self):
     curses.init_pair(2, curses.COLOR_BLUE, curses.COLOR_BLACK)
     curses.init_pair(3, curses.COLOR_RED, curses.COLOR_BLACK)
     curses.init_pair(4, curses.COLOR_YELLOW, curses.COLOR_BLACK)
+    curses.init_pair(5, curses.COLOR_MAGENTA, curses.COLOR_BLACK)
+    curses.init_pair(6, curses.COLOR_BLACK, curses.COLOR_BLUE)
     col_unknown = curses.color_pair(1)
     col_mem_obstacle = curses.color_pair(2)
     col_mem = curses.color_pair(2)
-    col_ground = curses.color_pair(3)
-    col_stone = curses.color_pair(4)
+    col_stone = curses.color_pair(1)
+    col_dirt = curses.color_pair(4)
     col_earth = curses.color_pair(3)
-    col_player = curses.color_pair(1)
+    col_player = curses.color_pair(5)
+    col_water = curses.color_pair(6)
     for y in range(world_data["map_size"]):
         for x in range(world_data["map_size"]):
             pos = y * world_data["map_size"] + x
@@ -61,13 +74,20 @@ def win_map(self):
                 winmap += [(char, attribute), bonus]
             else:
                 attribute = col_stone
+                bonus = " "
                 if char == ".":
-                    attribute = col_ground
-                if char == "#":
+                    attribute = col_dirt
+                elif char == ":":
                     attribute = col_earth
-                elif char == "@":
+                elif char == "%":
+                    attribute = col_earth
+                elif char == "#":
+                    attribute = col_dirt
+                elif char == "~":
+                    attribute = col_water
+                elif char == "o":
                     attribute = col_player
-                bonus = (" ", attribute)
+                    bonus = (char, attribute)
                 winmap += [(char, attribute), bonus]
         if y % 2 == 0:
             winmap += "  "
@@ -75,14 +95,21 @@ def win_map(self):
 
 from client.config.world_data import world_data
 world_data["stomach"] = 0
+world_data["bladder"] = 0
 from client.config.io import io
 io["worldstate_read_order"] += [["stomach", "int"]]
+io["worldstate_read_order"] += [["bladder", "int"]]
 from client.config.windows import windows_config
 from client.windows import win_log
 windows_config[:] = [
     {"config": [0, -34], "func": win_map, "title": "The Crawling Eater"},
     {"config": [1, 33], "func": win_stomach, "title": "stomach"},
-    {"config": [-2, 33], "func": win_log, "title": "log"}
+    {"config": [1, 33], "func": win_bladder, "title": "bladder"},
+    {"config": [-4, 33], "func": win_log, "title": "log"}
 ]
 from client.window_management import set_windows
 set_windows()
+from client.commands import command_sender
+from client.config.commands import commands
+commands["S"] = (command_sender("drop"),)
+commands["D"] = (command_sender("drink"),)