1 # This file is part of PlomRogue. PlomRogue is licensed under the GPL version 3
2 # or any later version. For details on its copyright, license, and warranties,
3 # see the file NOTICE in the root directory of the PlomRogue source package.
6 from server.config.world_data import world_db
10 if action_exists("drink") and world_db["WORLD_ACTIVE"]:
11 pos = world_db["Things"][0]["pos"]
12 if not (chr(world_db["MAP"][pos]) == "0"
13 and world_db["wetmap"][pos] > ord("0")):
14 log("NOTHING to drink here.")
16 elif world_db["Things"][0]["T_KIDNEY"] >= 32:
17 log("You're too FULL to drink more.")
19 world_db["set_command"]("drink")
23 pos = world_db["Things"][0]["pos"]
24 if chr(world_db["MAP"][pos]) == "0" and \
25 world_db["wetmap"][pos] > ord("0") and t["T_KIDNEY"] < 32:
28 world_db["wetmap"][pos] -= 1
29 if world_db["wetmap"][pos] == ord("0"):
30 world_db["MAP"][pos] = ord("0")
34 if action_exists("pee") and world_db["WORLD_ACTIVE"]:
35 if world_db["Things"][0]["T_BLADDER"] < 1:
36 log("Nothing to drop from empty bladder.")
38 world_db["set_command"]("pee")
42 if t["T_BLADDER"] < 1:
44 if t == world_db["Things"][0]:
45 log("You LOSE fluid.")
46 if not world_db["test_air"](t):
49 world_db["wetmap"][t["pos"]] += 1
53 if action_exists("drop") and world_db["WORLD_ACTIVE"]:
54 if world_db["Things"][0]["T_BOWEL"] < 1:
55 log("Nothing to drop from empty bowel.")
57 world_db["set_command"]("drop")
63 if t == world_db["Things"][0]:
64 log("You DROP waste.")
65 if not world_db["test_air"](t):
67 world_db["MAP"][t["pos"]] += 1
71 def play_move(str_arg):
72 """Try "move" as player's T_COMMAND, str_arg as T_ARGUMENT / direction."""
73 if action_exists("move") and world_db["WORLD_ACTIVE"]:
74 from server.config.world_data import directions_db, symbols_passable
75 t = world_db["Things"][0]
76 if not str_arg in directions_db:
77 print("Illegal move direction string.")
79 d = ord(directions_db[str_arg])
80 from server.utils import mv_yx_in_dir_legal
81 move_result = mv_yx_in_dir_legal(chr(d), t["T_POSY"], t["T_POSX"])
82 if 1 == move_result[0]:
83 pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
84 if chr(world_db["MAP"][pos]) in "34":
85 if t["T_STOMACH"] >= 32:
86 if t == world_db["Things"][0]:
87 log("You're too FULL to eat.")
89 world_db["Things"][0]["T_ARGUMENT"] = d
90 world_db["set_command"]("eat")
92 if chr(world_db["MAP"][pos]) in symbols_passable:
93 world_db["Things"][0]["T_ARGUMENT"] = d
94 world_db["set_command"]("move")
96 log("You CAN'T eat your way through there.")
100 from server.utils import mv_yx_in_dir_legal, rand
101 from server.config.world_data import symbols_passable
103 move_result = mv_yx_in_dir_legal(chr(t["T_ARGUMENT"]),
104 t["T_POSY"], t["T_POSX"])
105 if 1 == move_result[0]:
106 pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
107 #hitted = [tid for tid in world_db["Things"]
108 # if world_db["Things"][tid] != t
109 # if world_db["Things"][tid]["T_LIFEPOINTS"]
110 # if world_db["Things"][tid]["T_POSY"] == move_result[1]
111 # if world_db["Things"][tid]["T_POSX"] == move_result[2]]
114 # hitted_tid = world_db["Things"][hit_id]["T_TYPE"]
115 # if t == world_db["Things"][0]:
116 # hitted_name = world_db["ThingTypes"][hitted_tid]["TT_NAME"]
117 # log("You BUMP into " + hitted_name + ".")
119 # hitter_name = world_db["ThingTypes"][t["T_TYPE"]]["TT_NAME"]
120 # log(hitter_name +" BUMPS into you.")
122 passable = chr(world_db["MAP"][pos]) in symbols_passable
124 log("You try to EAT, but fail.")
126 height = world_db["MAP"][pos] - ord("0")
127 if t["T_STOMACH"] >= 32 or height == 5:
131 eaten = (height == 3 and 0 == int(rand.next() % 2)) or \
132 (height == 4 and 0 == int(rand.next() % 5))
134 world_db["MAP"][pos] = ord("0")
135 if t["T_STOMACH"] > 32:
140 from server.build_fov_map import build_fov_map
141 from server.utils import mv_yx_in_dir_legal, rand
142 from server.config.world_data import symbols_passable
144 move_result = mv_yx_in_dir_legal(chr(t["T_ARGUMENT"]),
145 t["T_POSY"], t["T_POSX"])
146 if 1 == move_result[0]:
147 pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
148 #hitted = [tid for tid in world_db["Things"]
149 # if world_db["Things"][tid] != t
150 # if world_db["Things"][tid]["T_LIFEPOINTS"]
151 # if world_db["Things"][tid]["T_POSY"] == move_result[1]
152 # if world_db["Things"][tid]["T_POSX"] == move_result[2]]
155 # hitted_tid = world_db["Things"][hit_id]["T_TYPE"]
156 # if t == world_db["Things"][0]:
157 # hitted_name = world_db["ThingTypes"][hitted_tid]["TT_NAME"]
158 # log("You BUMP into " + hitted_name + ".")
160 # hitter_name = world_db["ThingTypes"][t["T_TYPE"]]["TT_NAME"]
161 # log(hitter_name +" BUMPS into you.")
163 passable = chr(world_db["MAP"][pos]) in symbols_passable
165 t["T_POSY"] = move_result[1]
166 t["T_POSX"] = move_result[2]
167 t["pos"] = move_result[1] * world_db["MAP_LENGTH"] + move_result[2]
170 log("You try to MOVE there, but fail.")
174 if world_db["MAP"][t["pos"]] == ord("-"):
175 world_db["die"](t, "You FALL in a hole, and die.")
178 world_db["test_hole"] = test_hole
182 if world_db["terrain_fullness"](t["pos"]) > 5:
183 world_db["die"](t, "You SUFFOCATE")
186 world_db["test_air"] = test_air
190 t["T_LIFEPOINTS"] = 0
191 if t == world_db["Things"][0]:
192 t["fovmap"] = bytearray(b' ' * (world_db["MAP_LENGTH"] ** 2))
193 t["T_MEMMAP"][t["pos"]] = ord("@")
195 world_db["die"] = die
199 from server.make_map import new_pos, is_neighbor
200 from server.utils import rand
201 world_db["MAP"] = bytearray(b'5' * (world_db["MAP_LENGTH"] ** 2))
202 length = world_db["MAP_LENGTH"]
203 add_half_width = (not (length % 2)) * int(length / 2)
204 world_db["MAP"][int((length ** 2) / 2) + add_half_width] = ord("4")
206 y, x, pos = new_pos()
207 if "5" == chr(world_db["MAP"][pos]) and is_neighbor((y, x), "4"):
208 if y == 0 or y == (length - 1) or x == 0 or x == (length - 1):
210 world_db["MAP"][pos] = ord("4")
211 n_ground = int((length ** 2) / 16)
213 while (i_ground <= n_ground):
214 single_allowed = rand.next() % 32
215 y, x, pos = new_pos()
216 if "4" == chr(world_db["MAP"][pos]) \
217 and ((not single_allowed) or is_neighbor((y, x), "0")):
218 world_db["MAP"][pos] = ord("0")
220 n_water = int((length ** 2) / 64)
222 while (i_water <= n_water):
223 y, x, pos = new_pos()
224 if ord("0") == world_db["MAP"][pos] and \
225 ord("0") == world_db["wetmap"][pos]:
226 world_db["wetmap"][pos] = ord("3")
230 def calc_effort(ta, t):
231 from server.utils import mv_yx_in_dir_legal
232 if ta["TA_NAME"] == "move":
233 move_result = mv_yx_in_dir_legal(chr(t["T_ARGUMENT"]),
234 t["T_POSY"], t["T_POSX"])
235 if 1 == move_result[0]:
236 pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
237 narrowness = world_db["MAP"][pos] - ord("0")
238 return 2 ** narrowness
240 world_db["calc_effort"] = calc_effort
244 from server.ai import ai
245 from server.config.actions import action_db
246 from server.update_map_memory import update_map_memory
247 from server.io import try_worldstate_update
248 from server.config.io import io_db
249 from server.utils import rand
250 while world_db["Things"][0]["T_LIFEPOINTS"]:
251 for tid in [tid for tid in world_db["Things"]]:
252 if not tid in world_db["Things"]:
254 t = world_db["Things"][tid]
255 if t["T_LIFEPOINTS"]:
256 if not (world_db["test_air"](t) and world_db["test_hole"](t)):
258 if not t["T_COMMAND"]:
263 if t["T_LIFEPOINTS"]:
265 taid = [a for a in world_db["ThingActions"]
266 if a == t["T_COMMAND"]][0]
267 ThingAction = world_db["ThingActions"][taid]
268 effort = world_db["calc_effort"](ThingAction, t)
269 if t["T_PROGRESS"] >= effort:
270 action = action_db["actor_" + ThingAction["TA_NAME"]]
274 if t["T_BOWEL"] > 16:
275 if 0 == (rand.next() % (33 - t["T_BOWEL"])):
276 action_db["actor_drop"](t)
277 if t["T_BLADDER"] > 16:
278 if 0 == (rand.next() % (33 - t["T_BLADDER"])):
279 action_db["actor_pee"](t)
280 if 0 == world_db["TURN"] % 5:
285 if t["T_STOMACH"] <= 0:
286 world_db["die"](t, "You DIE of hunger.")
287 elif t["T_KIDNEY"] <= 0:
288 world_db["die"](t, "You DIE of dehydration.")
290 positions_to_wet = []
291 for pos in range(world_db["MAP_LENGTH"] ** 2):
292 if world_db["MAP"][pos] == ord("0") \
293 and world_db["wetmap"][pos] < ord("5"):
294 positions_to_wet += [pos]
295 i_positions_to_wet = len(positions_to_wet)
296 for pos in range(world_db["MAP_LENGTH"] ** 2):
297 wetness = world_db["wetmap"][pos] - ord("0")
298 height = world_db["MAP"][pos] - ord("0")
299 if height == 0 and wetness > 0 \
300 and 0 == rand.next() % ((2 ** 13) / (2 ** wetness)):
301 world_db["MAP"][pos] = ord("-")
302 if pos in positions_to_wet:
303 positions_to_wet.remove(pos)
304 i_positions_to_wet -= 1
305 if ((wetness > 0 and height != 0) or wetness > 1) \
306 and 0 == rand.next() % 5:
307 world_db["wetmap"][pos] -= 1
309 i_positions_to_wet -= 1
310 if i_positions_to_wet == 0:
314 select = rand.next() % len(positions_to_wet)
315 pos = positions_to_wet[select]
316 world_db["wetmap"][pos] += 1
317 positions_to_wet.remove(pos)
319 world_db["TURN"] += 1
320 io_db["worldstate_updateable"] = True
321 try_worldstate_update()
322 world_db["turn_over"] = turn_over
326 """Call ai() on player Thing, then turn_over()."""
327 from server.ai import ai
328 if world_db["WORLD_ACTIVE"]:
329 ai(world_db["Things"][0])
330 world_db["turn_over"]()
333 def set_command(action):
334 """Set player's T_COMMAND, then call turn_over()."""
335 tid = [x for x in world_db["ThingActions"]
336 if world_db["ThingActions"][x]["TA_NAME"] == action][0]
337 world_db["Things"][0]["T_COMMAND"] = tid
338 world_db["turn_over"]()
339 world_db["set_command"] = set_command
343 """Try "wait" as player's T_COMMAND."""
344 if world_db["WORLD_ACTIVE"]:
345 world_db["set_command"]("wait")
349 length = world_db["MAP_LENGTH"]
351 for i in range(length):
352 line = world_db["wetmap"][i * length:(i * length) + length].decode()
353 string = string + "WETMAP" + " " + str(i) + " " + line + "\n"
357 def wetmapset(str_int, mapline):
358 def valid_map_line(str_int, mapline):
359 from server.utils import integer_test
360 val = integer_test(str_int, 0, 255)
362 if val >= world_db["MAP_LENGTH"]:
363 print("Illegal value for map line number.")
364 elif len(mapline) != world_db["MAP_LENGTH"]:
365 print("Map line length is unequal map width.")
369 val = valid_map_line(str_int, mapline)
371 length = world_db["MAP_LENGTH"]
372 if not world_db["wetmap"]:
373 m = bytearray(b' ' * (length ** 2))
375 m = world_db["wetmap"]
376 m[val * length:(val * length) + length] = mapline.encode()
377 if not world_db["wetmap"]:
378 world_db["wetmap"] = m
381 from server.worldstate_write_helpers import write_map
382 length = world_db["MAP_LENGTH"]
383 visible_wetmap = bytearray(b' ' * (length ** 2))
384 for i in range(length ** 2):
385 if world_db["Things"][0]["fovmap"][i] == ord('v'):
386 visible_wetmap[i] = world_db["wetmap"][i]
387 return write_map(visible_wetmap, world_db["MAP_LENGTH"])
391 if world_db["WORLD_ACTIVE"]:
392 world_db["ai"](world_db["Things"][0])
393 world_db["turn_over"]()
396 def get_dir_to_target(t, target):
398 from server.utils import rand, libpr, c_pointer_to_bytearray
399 from server.config.world_data import symbols_passable
401 def zero_score_map_where_char_on_memdepthmap(c):
402 map = c_pointer_to_bytearray(t["T_MEMDEPTHMAP"])
403 if libpr.zero_score_map_where_char_on_memdepthmap(c, map):
404 raise RuntimeError("No score map allocated for "
405 "zero_score_map_where_char_on_memdepthmap().")
407 def set_map_score(pos, score):
408 test = libpr.set_map_score(pos, score)
410 raise RuntimeError("No score map allocated for set_map_score().")
412 def set_movement_cost_map():
413 memmap = c_pointer_to_bytearray(t["T_MEMMAP"])
414 if libpr.TCE_set_movement_cost_map(memmap):
415 raise RuntimeError("No movement cost map allocated for "
416 "set_movement_cost_map().")
422 except StopIteration:
425 mapsize = world_db["MAP_LENGTH"] ** 2
426 if target == "food" and t["T_MEMMAP"]:
427 return exists(pos for pos in range(mapsize)
428 if ord("2") < t["T_MEMMAP"][pos] < ord("5"))
429 elif target == "fluid_certain" and t["fovmap"]:
430 return exists(pos for pos in range(mapsize)
431 if t["fovmap"] == ord("v")
432 if world_db["MAP"][pos] == ord("0")
433 if world_db["wetmap"][pos] > ord("0"))
434 elif target == "fluid_potential" and t["T_MEMMAP"] and t["fovmap"]:
435 return exists(pos for pos in range(mapsize)
436 if t["T_MEMMAP"][pos] == ord("0")
437 if t["fovmap"] != ord("v"))
438 elif target == "space" and t["T_MEMMAP"] and t["fovmap"]:
439 return exists(pos for pos in range(mapsize)
440 if ord("0") <= t["T_MEMMAP"][pos] <= ord("2")
441 if (t["fovmap"] != ord("v")
442 or world_db["terrain_fullness"](pos) < 5))
445 def init_score_map():
446 test = libpr.init_score_map()
447 set_movement_cost_map()
448 mapsize = world_db["MAP_LENGTH"] ** 2
450 raise RuntimeError("Malloc error in init_score_map().")
451 if target == "food" and t["T_MEMMAP"]:
452 [set_map_score(pos, 0) for pos in range(mapsize)
453 if ord("2") < t["T_MEMMAP"][pos] < ord("5")]
454 elif target == "fluid_certain" and t["fovmap"]:
455 [set_map_score(pos, 0) for pos in range(mapsize)
456 if t["fovmap"] == ord("v")
457 if world_db["MAP"][pos] == ord("0")
458 if world_db["wetmap"][pos] > ord("0")]
459 elif target == "fluid_potential" and t["T_MEMMAP"] and t["fovmap"]:
460 [set_map_score(pos, 0) for pos in range(mapsize)
461 if t["T_MEMMAP"][pos] == ord("0")
462 if t["fovmap"] != ord("v")]
463 elif target == "space" and t["T_MEMMAP"] and t["fovmap"]:
464 [set_map_score(pos, 0) for pos in range(mapsize)
465 if ord("0") <= t["T_MEMMAP"][pos] <= ord("2")
466 if (t["fovmap"] != ord("v")
467 or world_db["terrain_fullness"](pos) < 5)]
468 elif target == "search":
469 zero_score_map_where_char_on_memdepthmap(mem_depth_c[0])
471 def rand_target_dir(neighbors, cmp, dirs):
474 for i in range(len(dirs)):
475 if cmp == neighbors[i]:
476 candidates.append(dirs[i])
478 return candidates[rand.next() % n_candidates] if n_candidates else 0
480 def get_neighbor_scores(dirs, eye_pos):
482 if libpr.ready_neighbor_scores(eye_pos):
483 raise RuntimeError("No score map allocated for " +
484 "ready_neighbor_scores.()")
485 for i in range(len(dirs)):
486 scores.append(libpr.get_neighbor_score(i))
489 def get_dir_from_neighbors():
491 dir_to_target = False
494 neighbors = get_neighbor_scores(dirs, eye_pos)
495 minmax_start = 65535 - 1
496 minmax_neighbor = minmax_start
497 for i in range(len(dirs)):
498 if minmax_neighbor > neighbors[i]:
499 minmax_neighbor = neighbors[i]
500 if minmax_neighbor != minmax_start:
501 dir_to_target = rand_target_dir(neighbors, minmax_neighbor, dirs)
502 return dir_to_target, minmax_neighbor
504 dir_to_target = False
506 run_i = 9 + 1 if "search" == target else 1
508 while run_i and not dir_to_target and \
509 ("search" == target or seeing_thing()):
512 mem_depth_c = b'9' if b' ' == mem_depth_c \
513 else bytes([mem_depth_c[0] - 1])
514 if libpr.TCE_dijkstra_map_with_movement_cost():
515 raise RuntimeError("No score map allocated for dijkstra_map().")
516 dir_to_target, minmax_neighbor = get_dir_from_neighbors()
517 libpr.free_score_map()
518 if dir_to_target and str == type(dir_to_target):
520 from server.utils import mv_yx_in_dir_legal
521 move_result = mv_yx_in_dir_legal(dir_to_target, t["T_POSY"],
523 if 1 != move_result[0]:
525 pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
526 if world_db["MAP"][pos] > ord("2"):
528 t["T_COMMAND"] = [taid for taid in world_db["ThingActions"]
529 if world_db["ThingActions"][taid]["TA_NAME"]
531 t["T_ARGUMENT"] = ord(dir_to_target)
532 return dir_to_target, minmax_neighbor
533 world_db["get_dir_to_target"] = get_dir_to_target
536 def terrain_fullness(pos):
537 return (world_db["MAP"][pos] - ord("0")) + \
538 (world_db["wetmap"][pos] - ord("0"))
539 world_db["terrain_fullness"] = terrain_fullness
544 if t["T_LIFEPOINTS"] == 0:
547 def standing_on_fluid(t):
548 if world_db["MAP"][t["pos"]] == ord("0") and \
549 world_db["wetmap"][t["pos"]] > ord("0"):
554 def thing_action_id(name):
555 return [taid for taid in world_db["ThingActions"]
556 if world_db["ThingActions"][taid]
557 ["TA_NAME"] == name][0]
559 t["T_COMMAND"] = thing_action_id("wait")
561 "safe_pee": (world_db["terrain_fullness"](t["pos"]) * t["T_BLADDER"]) / 4,
562 "safe_drop": (world_db["terrain_fullness"](t["pos"]) * t["T_BOWEL"]) / 4,
563 "food": 33 - t["T_STOMACH"],
564 "fluid_certain": 33 - t["T_KIDNEY"],
565 "fluid_potential": 32 - t["T_KIDNEY"],
568 from operator import itemgetter
569 needs = sorted(needs.items(), key=itemgetter(1,0))
573 if need[0] in {"fluid_certain", "fluid_potential"}:
574 if standing_on_fluid(t):
575 t["T_COMMAND"] = thing_action_id("drink")
577 elif t["T_BLADDER"] > 0 and \
578 world_db["MAP"][t["pos"]] == ord("0"):
579 t["T_COMMAND"] = thing_action_id("pee")
581 elif need[0] in {"safe_pee", "safe_drop"}:
582 action_name = need[0][len("safe_"):]
583 if world_db["terrain_fullness"](t["pos"]) < 4:
584 t["T_COMMAND"] = thing_action_id(action_name)
587 test = world_db["get_dir_to_target"](t, "space")
589 if (not test[1] < 5) and \
590 world_db["terrain_fullness"](t["pos"]) < 5:
591 t["T_COMMAND"] = thing_action_id(action_name)
593 if t["T_STOMACH"] < 32 and \
594 world_db["get_dir_to_target"](t, "food")[0]:
597 if world_db["get_dir_to_target"](t, need[0])[0]:
599 elif t["T_STOMACH"] < 32 and \
600 need[0] in {"fluid_certain", "fluid_potential"} and \
601 world_db["get_dir_to_target"](t, "food")[0]:
606 from server.config.io import io_db
607 io_db["worldstate_write_order"] += [["T_STOMACH", "player_int"]]
608 io_db["worldstate_write_order"] += [["T_KIDNEY", "player_int"]]
609 io_db["worldstate_write_order"] += [["T_BOWEL", "player_int"]]
610 io_db["worldstate_write_order"] += [["T_BLADDER", "player_int"]]
611 io_db["worldstate_write_order"] += [[write_wetmap, "func"]]
612 import server.config.world_data
613 server.config.world_data.symbols_hide = "345"
614 server.config.world_data.symbols_passable = "012-"
615 server.config.world_data.thing_defaults["T_STOMACH"] = 16
616 server.config.world_data.thing_defaults["T_BOWEL"] = 0
617 server.config.world_data.thing_defaults["T_KIDNEY"] = 16
618 server.config.world_data.thing_defaults["T_BLADDER"] = 0
619 world_db["wetmap"] = bytearray(b"0" * world_db["MAP_LENGTH"] ** 2)
620 io_db["hook_save"] = save_wetmap
621 import server.config.make_world_helpers
622 server.config.make_world_helpers.make_map = make_map
623 from server.config.commands import commands_db
624 commands_db["THINGS_HERE"] = (2, True, lambda x, y: None)
625 commands_db["ai"] = (0, False, command_ai)
626 commands_db["move"] = (1, False, play_move)
627 commands_db["eat"] = (1, False, play_move)
628 commands_db["wait"] = (0, False, play_wait)
629 commands_db["drop"] = (0, False, play_drop)
630 commands_db["drink"] = (0, False, play_drink)
631 commands_db["pee"] = (0, False, play_pee)
632 commands_db["use"] = (1, False, lambda x: None)
633 commands_db["pickup"] = (0, False, lambda: None)
634 commands_db["T_STOMACH"] = (1, False, setter("Thing", "T_STOMACH", 0, 255))
635 commands_db["T_KIDNEY"] = (1, False, setter("Thing", "T_KIDNEY", 0, 255))
636 commands_db["T_BOWEL"] = (1, False, setter("Thing", "T_BOWEL", 0, 255))
637 commands_db["T_BLADDER"] = (1, False, setter("Thing", "T_BLADDER", 0, 255))
638 commands_db["WETMAP"] = (2, False, wetmapset)
639 from server.actions import actor_wait
640 import server.config.actions
641 server.config.actions.action_db = {
642 "actor_wait": actor_wait,
643 "actor_move": actor_move,
644 "actor_drop": actor_drop,
645 "actor_drink": actor_drink,
646 "actor_pee": actor_pee,
647 "actor_eat": actor_eat,
650 strong_write(io_db["file_out"], "PLUGIN TheCrawlingEater\n")