plomlompom.com Git - plomrogue/blob - plugins/server/TheCrawlingEater.py

   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.
   4
   5
   6 from server.config.world_data import world_db
   7
   8
   9 def command_help(str_int):
  10     val = integer_test(str_int, 0, 4)
  11     if None != val:
  12         log(str_int)
  13
  14
  15 def command_ai():
  16     if not (world_db["WORLD_ACTIVE"]
  17             and world_db["Things"][0]["T_LIFEPOINTS"] > 0):
  18         return
  19     world_db["ai"](world_db["Things"][0])
  20     world_db["turn_over"]()
  21
  22
  23 def play_drink():
  24     if not (action_exists("drink") and world_db["WORLD_ACTIVE"]
  25             and world_db["Things"][0]["T_LIFEPOINTS"] > 0):
  26         return
  27     pos = world_db["Things"][0]["pos"]
  28     if not (chr(world_db["MAP"][pos]) in "0-+"
  29             and world_db["wetmap"][pos] > ord("0")):
  30         log("NOTHING to drink here.")
  31         return
  32     elif world_db["Things"][0]["T_KIDNEY"] >= 32:
  33         log("You're too FULL to drink more.")
  34         return
  35     world_db["set_command"]("drink")
  36
  37
  38 def actor_drink(t):
  39     pos = t["pos"]
  40     if chr(world_db["MAP"][pos]) in "0-+" and \
  41                 world_db["wetmap"][pos] > ord("0") and t["T_KIDNEY"] < 32:
  42         if world_db["Things"][0] == t:
  43             log("You DRINK.")
  44         t["T_KIDNEY"] += 1
  45         world_db["wetmap"][pos] -= 1
  46
  47
  48 def play_pee():
  49     if not (action_exists("pee") and world_db["WORLD_ACTIVE"]
  50             and world_db["Things"][0]["T_LIFEPOINTS"] > 0):
  51         return
  52     if world_db["Things"][0]["T_BLADDER"] < 1:
  53         log("Nothing to drop from empty bladder.")
  54         return
  55     world_db["set_command"]("pee")
  56
  57
  58 def actor_pee(t):
  59     if t["T_BLADDER"] < 1:
  60         return
  61     if t == world_db["Things"][0]:
  62         log("You LOSE fluid.")
  63     if not world_db["test_air"](t):
  64         return
  65     t["T_BLADDER"] -= 1
  66     if chr(world_db["MAP"][t["pos"]]) not in "*&":
  67         world_db["wetmap"][t["pos"]] += 1
  68
  69
  70 def play_drop():
  71     if not (action_exists("drop") and world_db["WORLD_ACTIVE"]
  72             and world_db["Things"][0]["T_LIFEPOINTS"] > 0):
  73         return
  74     if world_db["Things"][0]["T_BOWEL"] < 1:
  75         log("Nothing to drop from empty bowel.")
  76         return
  77     world_db["set_command"]("drop")
  78
  79
  80 def actor_drop(t):
  81     if t["T_BOWEL"] < 1:
  82         return
  83     if t == world_db["Things"][0]:
  84         log("You DROP waste.")
  85     if not world_db["test_air"](t):
  86         return
  87     if world_db["MAP"][t["pos"]] == ord("+"):
  88         world_db["MAP"][t["pos"]] = ord("-")
  89     elif world_db["MAP"][t["pos"]] == ord("-"):
  90         world_db["MAP"][t["pos"]] = ord("0")
  91     elif chr(world_db["MAP"][t["pos"]]) not in "*&":
  92         world_db["MAP"][t["pos"]] += 1
  93     t["T_BOWEL"] -= 1
  94
  95
  96 def play_move(str_arg):
  97     if not (action_exists("move") and world_db["WORLD_ACTIVE"]
  98             and world_db["Things"][0]["T_LIFEPOINTS"] > 0):
  99         return
 100     from server.config.world_data import directions_db, symbols_passable
 101     t = world_db["Things"][0]
 102     if not str_arg in directions_db:
 103         print("Illegal move direction string.")
 104         return
 105     d = ord(directions_db[str_arg])
 106     from server.utils import mv_yx_in_dir_legal
 107     move_result = mv_yx_in_dir_legal(chr(d), t["T_POSY"], t["T_POSX"])
 108     if 1 == move_result[0]:
 109         pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
 110         hitted = [tid for tid in world_db["Things"]
 111                   if world_db["Things"][tid]["T_POSY"] == move_result[1]
 112                   if world_db["Things"][tid]["T_POSX"] == move_result[2]]
 113         if len(hitted) > 0:
 114             if t["T_STOMACH"] >= 32 and t["T_KIDNEY"] >= 32:
 115                 if t == world_db["Things"][0]:
 116                     log("You're too FULL to suck from another creature.")
 117                 return
 118             world_db["Things"][0]["T_ARGUMENT"] = d
 119             world_db["set_command"]("eat")
 120             return
 121         legal_targets = "34"
 122         if world_db["GRACE"] >= 8:
 123             legal_targets += "5"
 124         if chr(world_db["MAP"][pos]) in legal_targets:
 125             if t["T_STOMACH"] >= 32:
 126                 if t == world_db["Things"][0]:
 127                     log("You're too FULL to eat.")
 128                 return
 129             world_db["Things"][0]["T_ARGUMENT"] = d
 130             world_db["set_command"]("eat")
 131             return
 132         if chr(world_db["MAP"][pos]) in symbols_passable:
 133             world_db["Things"][0]["T_ARGUMENT"] = d
 134             world_db["set_command"]("move")
 135             return
 136     log("You CAN'T eat your way through there.")
 137
 138
 139 def suck_out_creature(t, tid):
 140     if t == None:
 141         t = world_db["Things"][tid]
 142     elif tid == None:
 143         tid = next(tid for tid in world_db["Things"]
 144                    if world_db["Things"][tid] == t)
 145     room_stomach = 32 - world_db["Things"][0]["T_STOMACH"]
 146     room_kidney = 32 - world_db["Things"][0]["T_KIDNEY"]
 147     if t["T_STOMACH"] > room_stomach:
 148         t["T_STOMACH"] -= room_stomach
 149         world_db["Things"][0]["T_STOMACH"] = 32
 150     else:
 151         world_db["Things"][0]["T_STOMACH"] + t["T_STOMACH"]
 152         t["T_STOMACH"] = 0
 153     if t["T_KIDNEY"] > room_stomach:
 154         t["T_KIDNEY"] -= room_stomach
 155         world_db["Things"][0]["T_KIDNEY"] = 32
 156     else:
 157         world_db["Things"][0]["T_KIDNEY"] + t["T_KIDNEY"]
 158         t["T_KIDNEY"] = 0
 159     hitted_name = world_db["ThingTypes"][t["T_TYPE"]]["TT_NAME"]
 160     log("You SUCK EVERYTHING from " + hitted_name + ", killing them.")
 161     world_db["die"](t, "FOO")
 162 world_db["suck_out_creature"] = suck_out_creature
 163
 164
 165 def actor_eat(t):
 166     from server.utils import mv_yx_in_dir_legal, rand
 167     from server.config.world_data import symbols_passable
 168     passable = False
 169     move_result = mv_yx_in_dir_legal(chr(t["T_ARGUMENT"]),
 170                                      t["T_POSY"], t["T_POSX"])
 171     if 1 == move_result[0]:
 172         pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
 173         hitted = [tid for tid in world_db["Things"]
 174                   if world_db["Things"][tid]["T_POSY"] == move_result[1]
 175                   if world_db["Things"][tid]["T_POSX"] == move_result[2]]
 176         if len(hitted):
 177             hit_id = hitted[0]
 178             hitted_tid = world_db["Things"][hit_id]["T_TYPE"]
 179             if t == world_db["Things"][0]:
 180                 if world_db["GRACE"] >= 16:
 181                     world_db["suck_out_creature"](None, hit_id)
 182                     return
 183                 hitted_name = world_db["ThingTypes"][hitted_tid]["TT_NAME"]
 184                 log("You SUCK from " + hitted_name + ".")
 185             elif 0 == hit_id:
 186                 if world_db["GRACE"] >= 16:
 187                     world_db["suck_out_creature"](t, None)
 188                     return
 189                 hitter_name = world_db["ThingTypes"][t["T_TYPE"]]["TT_NAME"]
 190                 log(hitter_name +" SUCKS from you.")
 191             hitted = world_db["Things"][hit_id]
 192             if t["T_STOMACH"] < 32:
 193                 t["T_STOMACH"] = t["T_STOMACH"] + 1
 194                 hitted["T_STOMACH"] -= 1
 195             if t["T_KIDNEY"] < 32:
 196                 t["T_KIDNEY"] = t["T_KIDNEY"] + 1
 197                 hitted["T_KIDNEY"] -= 1
 198             return
 199         passable = chr(world_db["MAP"][pos]) in symbols_passable
 200     if passable and t == world_db["Things"][0]:
 201         log("You try to EAT, but fail.")
 202     else:
 203         height = world_db["MAP"][pos] - ord("0")
 204         if t["T_STOMACH"] >= 32:
 205             return
 206         if height == 5 and not \
 207                 (t == world_db["Things"][0] and world_db["GRACE"] >= 8):
 208             return
 209         t["T_STOMACH"] += 1
 210         if t == world_db["Things"][0]:
 211             log("You EAT.")
 212         eaten = (height == 3 and 0 == int(rand.next() % 2)) or \
 213                 (height == 4 and 0 == int(rand.next() % 5)) or \
 214                 (height == 5 and 0 == int(rand.next() % 10))
 215         if eaten:
 216             world_db["MAP"][pos] = ord("0")
 217             if t["T_STOMACH"] > 32:
 218                 t["T_STOMACH"] = 32
 219
 220
 221 def actor_move(t):
 222     from server.build_fov_map import build_fov_map
 223     from server.utils import mv_yx_in_dir_legal, rand
 224     from server.config.world_data import symbols_passable
 225     passable = False
 226     move_result = mv_yx_in_dir_legal(chr(t["T_ARGUMENT"]),
 227                                      t["T_POSY"], t["T_POSX"])
 228     if 1 == move_result[0]:
 229         pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
 230         hitted = [tid for tid in world_db["Things"]
 231                   if world_db["Things"][tid]["T_POSY"] == move_result[1]
 232                   if world_db["Things"][tid]["T_POSX"] == move_result[2]]
 233         if len(hitted):
 234             hit_id = hitted[0]
 235             hitted_tid = world_db["Things"][hit_id]["T_TYPE"]
 236             if t == world_db["Things"][0]:
 237                 if world_db["GRACE"] >= 16:
 238                     world_db["suck_out_creature"](None, hit_id)
 239                     return
 240                 hitted_name = world_db["ThingTypes"][hitted_tid]["TT_NAME"]
 241                 log("You BUMP into " + hitted_name + ".")
 242             elif 0 == hit_id:
 243                 if world_db["GRACE"] >= 16:
 244                     world_db["suck_out_creature"](t, None)
 245                     return
 246                 hitter_name = world_db["ThingTypes"][t["T_TYPE"]]["TT_NAME"]
 247                 log(hitter_name +" BUMPS into you.")
 248             return
 249         passable = chr(world_db["MAP"][pos]) in symbols_passable
 250     if passable:
 251         t["T_POSY"] = move_result[1]
 252         t["T_POSX"] = move_result[2]
 253         t["pos"] = move_result[1] * world_db["MAP_LENGTH"] + move_result[2]
 254         world_db["soundmap"][t["pos"]] = ord("9")
 255         if t == world_db["Things"][0] and world_db["MAP"][t["pos"]] == ord("$"):
 256             world_db["MAP"][t["pos"]] = ord("0")
 257             if world_db["GRACE"] == 0:
 258                 log("You can now eat ALL walls.")
 259             elif world_db["GRACE"] == 8:
 260                 log("You now have the DEATH touch.")
 261             elif world_db["GRACE"] == 16:
 262                 log("You will now LEVITATE over holes.")
 263             elif world_db["GRACE"] == 24:
 264                 log("You are now READY to fly through the exit portal.")
 265             elif world_db["GRACE"] == 32:
 266                 log("You already have all the GRACE you can get.")
 267             if world_db["GRACE"] <= 24:
 268                 world_db["GRACE"] += 8
 269     elif t == world_db["Things"][0]:
 270         log("You try to MOVE there, but fail.")
 271
 272
 273 def test_hole(t):
 274     if t == world_db["Things"][0]:
 275         if world_db["GRACE"] >= 32 and world_db["MAP"][t["pos"]] == ord("&"):
 276             world_db["die"](t, "YOU WIN, CONGRATULATIONS.")
 277             return False
 278         if world_db["GRACE"] >= 24:
 279             return True
 280     if chr(world_db["MAP"][t["pos"]]) in "*&":
 281         world_db["die"](t, "You FALL in a hole, and die.")
 282         return False
 283     return True
 284 world_db["test_hole"] = test_hole
 285
 286
 287 def test_air(t):
 288     if world_db["terrain_fullness"](t["pos"]) > 5:
 289         world_db["die"](t, "You SUFFOCATE")
 290         return False
 291     return True
 292 world_db["test_air"] = test_air
 293
 294
 295 def die(t, message):
 296     t["T_LIFEPOINTS"] = 0
 297     if t == world_db["Things"][0]:
 298         t["fovmap"] = bytearray(b' ' * (world_db["MAP_LENGTH"] ** 2))
 299         t["T_MEMMAP"][t["pos"]] = ord("@")
 300         log(message)
 301     else:
 302         if world_db["MAP"][t["pos"]] != ord("$"):
 303             world_db["MAP"][t["pos"]] = ord("5")
 304         world_db["HUMILITY"] = t["T_KIDNEY"] + t["T_BLADDER"] + \
 305             (world_db["wetmap"][t["pos"]] - ord("0"))
 306         world_db["wetmap"][t["pos"]] = 0
 307         tid = next(tid for tid in world_db["Things"]
 308                    if world_db["Things"][tid] == t)
 309         del world_db["Things"][tid]
 310 world_db["die"] = die
 311
 312
 313 def make_map():
 314     from server.make_map import new_pos, is_neighbor
 315     from server.utils import rand
 316     world_db["MAP"] = bytearray(b'5' * (world_db["MAP_LENGTH"] ** 2))
 317     length = world_db["MAP_LENGTH"]
 318     add_half_width = (not (length % 2)) * int(length / 2)
 319     world_db["MAP"][int((length ** 2) / 2) + add_half_width] = ord("4")
 320     while (1):
 321         y, x, pos = new_pos()
 322         if "5" == chr(world_db["MAP"][pos]) and is_neighbor((y, x), "4"):
 323             if y == 0 or y == (length - 1) or x == 0 or x == (length - 1):
 324                 break
 325             world_db["MAP"][pos] = ord("4")
 326     n_ground = int((length ** 2) / 16)
 327     i_ground = 0
 328     while (i_ground <= n_ground):
 329         single_allowed = rand.next() % 32
 330         y, x, pos = new_pos()
 331         if "4" == chr(world_db["MAP"][pos]) \
 332                 and ((not single_allowed) or is_neighbor((y, x), "0")):
 333             world_db["MAP"][pos] = ord("0")
 334             i_ground += 1
 335     n_water = int((length ** 2) / 32)
 336     i_water = 0
 337     while (i_water <= n_water):
 338         y, x, pos = new_pos()
 339         if ord("0") == world_db["MAP"][pos] and \
 340                 ord("0") == world_db["wetmap"][pos]:
 341             world_db["wetmap"][pos] = ord("3")
 342             i_water += 1
 343     n_altars = 8
 344     i_altars = 0
 345     while (i_altars < n_altars):
 346         y, x, pos = new_pos()
 347         if ord("0") == world_db["MAP"][pos]:
 348             world_db["MAP"][pos] = ord("$")
 349             i_altars += 1
 350
 351
 352 def calc_effort(ta, t):
 353     from server.utils import mv_yx_in_dir_legal
 354     if ta["TA_NAME"] == "move":
 355         move_result = mv_yx_in_dir_legal(chr(t["T_ARGUMENT"]),
 356                                          t["T_POSY"], t["T_POSX"])
 357         if 1 == move_result[0]:
 358             pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
 359             if chr(world_db["MAP"][pos]) in "012":
 360                 narrowness = world_db["MAP"][pos] - ord("0")
 361                 return 2 ** narrowness
 362     return 1
 363 world_db["calc_effort"] = calc_effort
 364
 365
 366 def turn_over():
 367     from server.ai import ai
 368     from server.config.actions import action_db
 369     from server.update_map_memory import update_map_memory
 370     from server.io import try_worldstate_update
 371     from server.config.io import io_db
 372     from server.utils import rand
 373     from server.build_fov_map import build_fov_map
 374     while world_db["Things"][0]["T_LIFEPOINTS"]:
 375         for tid in [tid for tid in world_db["Things"]]:
 376             if not tid in world_db["Things"]:
 377                 continue
 378             t = world_db["Things"][tid]
 379             if t["T_LIFEPOINTS"]:
 380                 if not (world_db["test_air"](t) and world_db["test_hole"](t)):
 381                     continue
 382                 if not t["T_COMMAND"]:
 383                     update_map_memory(t)
 384                     build_fov_map(t)
 385                     if 0 == tid:
 386                         return
 387                     world_db["ai"](t)
 388                 if t["T_LIFEPOINTS"]:
 389                     t["T_PROGRESS"] += 1
 390                     taid = [a for a in world_db["ThingActions"]
 391                               if a == t["T_COMMAND"]][0]
 392                     ThingAction = world_db["ThingActions"][taid]
 393                     effort = world_db["calc_effort"](ThingAction, t)
 394                     if t["T_PROGRESS"] >= effort:
 395                         action = action_db["actor_" + ThingAction["TA_NAME"]]
 396                         action(t)
 397                         if t["T_LIFEPOINTS"] <= 0:
 398                             continue
 399                         t["T_COMMAND"] = 0
 400                         t["T_PROGRESS"] = 0
 401                     if t["T_BOWEL"] > 16:
 402                         if 0 == (rand.next() % (33 - t["T_BOWEL"])):
 403                             action_db["actor_drop"](t)
 404                     if t["T_BLADDER"] > 16:
 405                         if 0 == (rand.next() % (33 - t["T_BLADDER"])):
 406                             action_db["actor_pee"](t)
 407                     if 0 == world_db["TURN"] % 5:
 408                         t["T_STOMACH"] -= 1
 409                         t["T_BOWEL"] += 1
 410                         t["T_KIDNEY"] -= 1
 411                         t["T_BLADDER"] += 1
 412                     if t["T_STOMACH"] <= 0:
 413                         world_db["die"](t, "You DIE of hunger.")
 414                     elif t["T_KIDNEY"] <= 0:
 415                         world_db["die"](t, "You DIE of dehydration.")
 416         mapsize = world_db["MAP_LENGTH"] ** 2
 417         for pos in range(mapsize):
 418             wetness = world_db["wetmap"][pos] - ord("0")
 419             height = world_db["MAP"][pos] - ord("0")
 420             if world_db["MAP"][pos] == ord("-"):
 421                 height = -1
 422             elif world_db["MAP"][pos] == ord("+"):
 423                 height = -2
 424             elif world_db["MAP"][pos] == ord("$"):
 425                 height = -3
 426             if height == -2 and wetness > 1 \
 427                     and 0 == rand.next() % ((2 ** 11) / (2 ** wetness)):
 428                 world_db["MAP"][pos] = ord("*")
 429                 world_db["HUMIDITY"] += wetness
 430             if height == -1 and wetness > 1 \
 431                     and 0 == rand.next() % ((2 ** 10) / (2 ** wetness)):
 432                 world_db["MAP"][pos] = ord("+")
 433             if height == 0 and wetness > 1 \
 434                     and 0 == rand.next() % ((2 ** 9) / (2 ** wetness)):
 435                 world_db["MAP"][pos] = ord("-")
 436             if ((wetness > 0 and height > 0) or wetness > 1) \
 437                 and 0 == rand.next() % 5:
 438                 world_db["wetmap"][pos] -= 1
 439                 world_db["HUMIDITY"] += 1
 440         if world_db["HUMIDITY"] > 0:
 441             if world_db["HUMIDITY"] > 2 and 0 == rand.next() % 2:
 442                 world_db["NEW_SPAWN"] += 1
 443                 world_db["HUMIDITY"] -= 1
 444             if world_db["NEW_SPAWN"] >= 16:
 445                 world_db["NEW_SPAWN"] -= 16
 446                 from server.new_thing import new_Thing
 447                 while 1:
 448                     y = rand.next() % world_db["MAP_LENGTH"]
 449                     x = rand.next() % world_db["MAP_LENGTH"]
 450                     if chr(world_db["MAP"][y * world_db["MAP_LENGTH"] + x]) !=\
 451                         "5":
 452                         from server.utils import id_setter
 453                         tid = id_setter(-1, "Things")
 454                         world_db["Things"][tid] = new_Thing(
 455                             world_db["PLAYER_TYPE"], (y, x))
 456                         pos = y * world_db["MAP_LENGTH"] + x
 457                         break
 458             positions_to_wet = []
 459             for pos in range(mapsize):
 460                 if chr(world_db["MAP"][pos]) in "0-+" \
 461                         and world_db["wetmap"][pos] < ord("5"):
 462                     positions_to_wet += [pos]
 463             while world_db["HUMIDITY"] > 0 and len(positions_to_wet) > 0:
 464                 select = rand.next() % len(positions_to_wet)
 465                 pos = positions_to_wet[select]
 466                 world_db["wetmap"][pos] += 1
 467                 positions_to_wet.remove(pos)
 468                 world_db["HUMIDITY"] -= 1
 469         for pos in range(mapsize):
 470             if world_db["soundmap"][pos] > ord("0"):
 471                 world_db["soundmap"][pos] -= 1
 472         from server.utils import libpr
 473         libpr.init_score_map()
 474         def set_map_score(pos, score):
 475             test = libpr.set_map_score(pos, score)
 476             if test:
 477                 raise RuntimeError("No score map allocated for set_map_score().")
 478         [set_map_score(pos, 1) for pos in range(mapsize)
 479          if world_db["MAP"][pos] == ord("*")]
 480         for pos in range(mapsize):
 481             if world_db["MAP"][pos] == ord("*"):
 482                 if libpr.ready_neighbor_scores(pos):
 483                     raise RuntimeError("No score map allocated for " +
 484                                        "ready_neighbor_scores.()")
 485                 score = 0
 486                 dirs = "edcxsw"
 487                 for i in range(len(dirs)):
 488                     score += libpr.get_neighbor_score(i)
 489                 if score == 6:
 490                     world_db["MAP"][pos] = ord("&")
 491         libpr.free_score_map()
 492         world_db["TURN"] += 1
 493         io_db["worldstate_updateable"] = True
 494         try_worldstate_update()
 495 world_db["turn_over"] = turn_over
 496
 497
 498 def set_command(action):
 499     """Set player's T_COMMAND, then call turn_over()."""
 500     tid = [x for x in world_db["ThingActions"]
 501            if world_db["ThingActions"][x]["TA_NAME"] == action][0]
 502     world_db["Things"][0]["T_COMMAND"] = tid
 503     world_db["turn_over"]()
 504 world_db["set_command"] = set_command
 505
 506
 507 def play_wait():
 508     """Try "wait" as player's T_COMMAND."""
 509     if world_db["WORLD_ACTIVE"]:
 510         world_db["set_command"]("wait")
 511
 512
 513 def save_maps():
 514     length = world_db["MAP_LENGTH"]
 515     string = ""
 516     for i in range(length):
 517         line = world_db["wetmap"][i * length:(i * length) + length].decode()
 518         string = string + "WETMAP" + " "  + str(i) + " " + line + "\n"
 519     for i in range(length):
 520         line = world_db["soundmap"][i * length:(i * length) + length].decode()
 521         string = string + "SOUNDMAP" + " "  + str(i) + " " + line + "\n"
 522     return string
 523
 524
 525 def soundmapset(str_int, mapline):
 526     def valid_map_line(str_int, mapline):
 527         from server.utils import integer_test
 528         val = integer_test(str_int, 0, 255)
 529         if None != val:
 530             if val >= world_db["MAP_LENGTH"]:
 531                 print("Illegal value for map line number.")
 532             elif len(mapline) != world_db["MAP_LENGTH"]:
 533                 print("Map line length is unequal map width.")
 534             else:
 535                 return val
 536         return None
 537     val = valid_map_line(str_int, mapline)
 538     if None != val:
 539         length = world_db["MAP_LENGTH"]
 540         if not world_db["soundmap"]:
 541             m = bytearray(b' ' * (length ** 2))
 542         else:
 543             m = world_db["soundmap"]
 544         m[val * length:(val * length) + length] = mapline.encode()
 545         if not world_db["soundmap"]:
 546             world_db["soundmap"] = m
 547
 548
 549 def wetmapset(str_int, mapline):
 550     def valid_map_line(str_int, mapline):
 551         from server.utils import integer_test
 552         val = integer_test(str_int, 0, 255)
 553         if None != val:
 554             if val >= world_db["MAP_LENGTH"]:
 555                 print("Illegal value for map line number.")
 556             elif len(mapline) != world_db["MAP_LENGTH"]:
 557                 print("Map line length is unequal map width.")
 558             else:
 559                 return val
 560         return None
 561     val = valid_map_line(str_int, mapline)
 562     if None != val:
 563         length = world_db["MAP_LENGTH"]
 564         if not world_db["wetmap"]:
 565             m = bytearray(b' ' * (length ** 2))
 566         else:
 567             m = world_db["wetmap"]
 568         m[val * length:(val * length) + length] = mapline.encode()
 569         if not world_db["wetmap"]:
 570             world_db["wetmap"] = m
 571
 572
 573 def write_soundmap():
 574     from server.worldstate_write_helpers import write_map
 575     length = world_db["MAP_LENGTH"]
 576     return write_map(world_db["soundmap"], world_db["MAP_LENGTH"])
 577
 578
 579 def write_wetmap():
 580     from server.worldstate_write_helpers import write_map
 581     length = world_db["MAP_LENGTH"]
 582     visible_wetmap = bytearray(b' ' * (length ** 2))
 583     for i in range(length ** 2):
 584         if world_db["Things"][0]["fovmap"][i] == ord('v'):
 585             visible_wetmap[i] = world_db["wetmap"][i]
 586     return write_map(visible_wetmap, world_db["MAP_LENGTH"])
 587
 588
 589 def get_dir_to_target(t, target):
 590
 591     from server.utils import rand, libpr, c_pointer_to_bytearray
 592     from server.config.world_data import symbols_passable
 593
 594     def get_map_score(pos):
 595         result = libpr.get_map_score(pos)
 596         if result < 0:
 597             raise RuntimeError("No score map allocated for get_map_score().")
 598         return result
 599
 600     def zero_score_map_where_char_on_memdepthmap(c):
 601         map = c_pointer_to_bytearray(t["T_MEMDEPTHMAP"])
 602         if libpr.zero_score_map_where_char_on_memdepthmap(c, map):
 603             raise RuntimeError("No score map allocated for "
 604                                "zero_score_map_where_char_on_memdepthmap().")
 605
 606     def set_map_score(pos, score):
 607         test = libpr.set_map_score(pos, score)
 608         if test:
 609             raise RuntimeError("No score map allocated for set_map_score().")
 610
 611     def set_movement_cost_map():
 612         copy_memmap = t["T_MEMMAP"][:]
 613         copy_memmap.replace(b' ', b'4')
 614         memmap = c_pointer_to_bytearray(copy_memmap)
 615         if libpr.TCE_set_movement_cost_map(memmap):
 616             raise RuntimeError("No movement cost map allocated for "
 617                                "set_movement_cost_map().")
 618
 619     def animates_in_fov(maplength):
 620         return [Thing for Thing in world_db["Things"].values()
 621                 if Thing["T_LIFEPOINTS"] and 118 == t["fovmap"][Thing["pos"]]
 622                 and (not Thing == t)]
 623
 624     def seeing_thing():
 625         def exists(gen):
 626             try:
 627                 next(gen)
 628             except StopIteration:
 629                 return False
 630             return True
 631         mapsize = world_db["MAP_LENGTH"] ** 2
 632         if target == "food" and t["T_MEMMAP"]:
 633             return exists(pos for pos in range(mapsize)
 634                            if ord("2") < t["T_MEMMAP"][pos] < ord("5"))
 635         elif target == "fluid_certain" and t["fovmap"]:
 636             return exists(pos for pos in range(mapsize)
 637                            if t["fovmap"] == ord("v")
 638                            if world_db["MAP"][pos] == ord("0")
 639                            if world_db["wetmap"][pos] > ord("0"))
 640         elif target == "crack" and t["T_MEMMAP"]:
 641             return exists(pos for pos in range(mapsize)
 642                            if t["T_MEMMAP"][pos] == ord("-"))
 643         elif target == "fluid_potential" and t["T_MEMMAP"] and t["fovmap"]:
 644             return exists(pos for pos in range(mapsize)
 645                            if t["T_MEMMAP"][pos] == ord("0")
 646                            if t["fovmap"] != ord("v"))
 647         elif target == "space" and t["T_MEMMAP"] and t["fovmap"]:
 648             return exists(pos for pos in range(mapsize)
 649                           if ord("-") <= t["T_MEMMAP"][pos] <= ord("2")
 650                           if (t["fovmap"] != ord("v")
 651                               or world_db["terrain_fullness"](pos) < 5))
 652         elif target in {"hunt", "flee"} and t["fovmap"]:
 653             return exists(Thing for
 654                           Thing in animates_in_fov(world_db["MAP_LENGTH"])) \
 655                 or exists(pos for pos in range(mapsize)
 656                           if world_db["soundmap"][pos] > ord("0")
 657                           if t["fovmap"][pos] != ord("v"))
 658         return False
 659
 660     def init_score_map():
 661         mapsize = world_db["MAP_LENGTH"] ** 2
 662         test = libpr.TCE_init_score_map()
 663         [set_map_score(pos, 65535) for pos in range(mapsize)
 664          if chr(t["T_MEMMAP"][pos]) in "5*&"]
 665         set_movement_cost_map()
 666         if test:
 667             raise RuntimeError("Malloc error in init_score_map().")
 668         if target == "food" and t["T_MEMMAP"]:
 669             [set_map_score(pos, 0) for pos in range(mapsize)
 670              if ord("2") < t["T_MEMMAP"][pos] < ord("5")]
 671         elif target == "fluid_certain" and t["fovmap"]:
 672             [set_map_score(pos, 0) for pos in range(mapsize)
 673              if t["fovmap"] == ord("v")
 674              if world_db["MAP"][pos] == ord("0")
 675              if world_db["wetmap"][pos] > ord("0")]
 676         elif target == "crack" and t["T_MEMMAP"]:
 677             [set_map_score(pos, 0) for pos in range(mapsize)
 678              if t["T_MEMMAP"][pos] == ord("-")]
 679         elif target == "fluid_potential" and t["T_MEMMAP"] and t["fovmap"]:
 680             [set_map_score(pos, 0) for pos in range(mapsize)
 681              if t["T_MEMMAP"][pos] == ord("0")
 682              if t["fovmap"] != ord("v")]
 683         elif target == "space" and t["T_MEMMAP"] and t["fovmap"]:
 684             [set_map_score(pos, 0) for pos in range(mapsize)
 685              if ord("-") <= t["T_MEMMAP"][pos] <= ord("2")
 686              if (t["fovmap"] != ord("v")
 687                  or world_db["terrain_fullness"](pos) < 5)]
 688         elif target == "search":
 689             zero_score_map_where_char_on_memdepthmap(mem_depth_c[0])
 690         elif target in {"hunt", "flee"}:
 691             [set_map_score(Thing["pos"], 0) for
 692              Thing in animates_in_fov(world_db["MAP_LENGTH"])]
 693             [set_map_score(pos, 0) for pos in range(mapsize)
 694              if world_db["soundmap"][pos] > ord("0")
 695              if t["fovmap"][pos] != ord("v")]
 696
 697     def rand_target_dir(neighbors, cmp, dirs):
 698         candidates = []
 699         n_candidates = 0
 700         for i in range(len(dirs)):
 701             if cmp == neighbors[i]:
 702                 candidates.append(dirs[i])
 703                 n_candidates += 1
 704         return candidates[rand.next() % n_candidates] if n_candidates else 0
 705
 706     def get_neighbor_scores(dirs, eye_pos):
 707         scores = []
 708         if libpr.ready_neighbor_scores(eye_pos):
 709             raise RuntimeError("No score map allocated for " +
 710                                "ready_neighbor_scores.()")
 711         for i in range(len(dirs)):
 712             scores.append(libpr.get_neighbor_score(i))
 713         return scores
 714
 715     def get_dir_from_neighbors():
 716         import math
 717         dir_to_target = False
 718         dirs = "edcxsw"
 719         eye_pos = t["pos"]
 720         neighbors = get_neighbor_scores(dirs, eye_pos)
 721         minmax_start = 0 if "flee" == target else 65535 - 1
 722         minmax_neighbor = minmax_start
 723         for i in range(len(dirs)):
 724             if ("flee" == target and get_map_score(t["pos"]) < neighbors[i] and
 725                 minmax_neighbor < neighbors[i] and 65535 != neighbors[i]) \
 726                or ("flee" != target and minmax_neighbor > neighbors[i]):
 727                 minmax_neighbor = neighbors[i]
 728         if minmax_neighbor != minmax_start:
 729             dir_to_target = rand_target_dir(neighbors, minmax_neighbor, dirs)
 730         if "flee" == target:
 731             distance = get_map_score(t["pos"])
 732             fear_distance = 5
 733             attack_distance = 1
 734             if not dir_to_target:
 735                 if attack_distance >= distance:
 736                     dir_to_target = rand_target_dir(neighbors,
 737                                                     distance - 1, dirs)
 738             elif dir_to_target and fear_distance < distance:
 739                 dir_to_target = 0
 740         return dir_to_target, minmax_neighbor
 741
 742     dir_to_target = False
 743     mem_depth_c = b' '
 744     run_i = 9 + 1 if "search" == target else 1
 745     minmax_neighbor = 0
 746     while run_i and not dir_to_target and \
 747             ("search" == target or seeing_thing()):
 748         run_i -= 1
 749         init_score_map()
 750         mem_depth_c = b'9' if b' ' == mem_depth_c \
 751             else bytes([mem_depth_c[0] - 1])
 752         if libpr.TCE_dijkstra_map_with_movement_cost():
 753             raise RuntimeError("No score map allocated for dijkstra_map().")
 754         dir_to_target, minmax_neighbor = get_dir_from_neighbors()
 755         libpr.free_score_map()
 756         if dir_to_target and str == type(dir_to_target):
 757             action = "move"
 758             from server.utils import mv_yx_in_dir_legal
 759             move_result = mv_yx_in_dir_legal(dir_to_target, t["T_POSY"],
 760                                                             t["T_POSX"])
 761             if 1 != move_result[0]:
 762                 return False, 0
 763             pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
 764             hitted = [tid for tid in world_db["Things"]
 765                       if world_db["Things"][tid]["pos"] == pos]
 766             if world_db["MAP"][pos] > ord("2") or len(hitted) > 0:
 767                 action = "eat"
 768             t["T_COMMAND"] = [taid for taid in world_db["ThingActions"]
 769                               if world_db["ThingActions"][taid]["TA_NAME"]
 770                               == action][0]
 771             t["T_ARGUMENT"] = ord(dir_to_target)
 772     return dir_to_target, minmax_neighbor
 773 world_db["get_dir_to_target"] = get_dir_to_target
 774
 775
 776 def terrain_fullness(pos):
 777     wetness = world_db["wetmap"][pos] - ord("0")
 778     if chr(world_db["MAP"][pos]) in "-+":
 779         height = 0
 780     else:
 781         height = world_db["MAP"][pos] - ord("0")
 782     return wetness + height
 783 world_db["terrain_fullness"] = terrain_fullness
 784
 785
 786 def ai(t):
 787
 788     if t["T_LIFEPOINTS"] == 0:
 789         return
 790
 791     def standing_on_fluid(t):
 792         if world_db["MAP"][t["pos"]] == ord("0") and \
 793             world_db["wetmap"][t["pos"]] > ord("0"):
 794                 return True
 795         else:
 796             return False
 797
 798     def thing_action_id(name):
 799         return [taid for taid in world_db["ThingActions"]
 800                 if world_db["ThingActions"][taid]
 801                 ["TA_NAME"] == name][0]
 802
 803     t["T_COMMAND"] = thing_action_id("wait")
 804     needs = {
 805         "fix_cracks": 24,
 806         "flee": 24,
 807         "safe_pee": (world_db["terrain_fullness"](t["pos"]) * t["T_BLADDER"]) / 4,
 808         "safe_drop": (world_db["terrain_fullness"](t["pos"]) * t["T_BOWEL"]) / 4,
 809         "food": 33 - t["T_STOMACH"],
 810         "fluid_certain": 33 - t["T_KIDNEY"],
 811         "fluid_potential": 32 - t["T_KIDNEY"],
 812         "search": 1,
 813     }
 814     from operator import itemgetter
 815     needs = sorted(needs.items(), key=itemgetter(1,0))
 816     needs.reverse()
 817     for need in needs:
 818         if need[1] > 0:
 819             if need[0] == "fix_cracks":
 820                 if world_db["MAP"][t["pos"]] == ord("-") and \
 821                         t["T_BOWEL"] > 0 and \
 822                         world_db["terrain_fullness"](t["pos"]) <= 3:
 823                     t["T_COMMAND"] = thing_action_id("drop")
 824                     return
 825                 elif world_db["get_dir_to_target"](t, "crack")[0]:
 826                     return
 827             if need[0] in {"fluid_certain", "fluid_potential"}:
 828                 if standing_on_fluid(t):
 829                     t["T_COMMAND"] = thing_action_id("drink")
 830                     return
 831                 elif t["T_BLADDER"] > 0 and \
 832                          world_db["MAP"][t["pos"]] == ord("0"):
 833                     t["T_COMMAND"] = thing_action_id("pee")
 834                     return
 835             elif need[0] in {"safe_pee", "safe_drop"}:
 836                 action_name = need[0][len("safe_"):]
 837                 if world_db["terrain_fullness"](t["pos"]) <= 3:
 838                     t["T_COMMAND"] = thing_action_id(action_name)
 839                     return
 840                 test = world_db["get_dir_to_target"](t, "space")
 841                 if test[0]:
 842                     if test[1] < 5:
 843                         return
 844                     elif world_db["terrain_fullness"](t["pos"]) < 5:
 845                         t["T_COMMAND"] = thing_action_id(action_name)
 846                     return
 847                 if t["T_STOMACH"] < 32 and \
 848                         world_db["get_dir_to_target"](t, "food")[0]:
 849                     return
 850                 continue
 851             if need[0] in {"fluid_certain", "fluid_potential", "food"}:
 852                 if world_db["get_dir_to_target"](t, need[0])[0]:
 853                     return
 854                 elif world_db["get_dir_to_target"](t, "hunt")[0]:
 855                     return
 856                 elif need[0] != "food" and t["T_STOMACH"] < 32 and \
 857                         world_db["get_dir_to_target"](t, "food")[0]:
 858                     return
 859             elif world_db["get_dir_to_target"](t, need[0])[0]:
 860                 return
 861 world_db["ai"] = ai
 862
 863
 864 from server.config.io import io_db
 865 io_db["worldstate_write_order"] += [["T_STOMACH", "player_int"]]
 866 io_db["worldstate_write_order"] += [["T_KIDNEY", "player_int"]]
 867 io_db["worldstate_write_order"] += [["T_BOWEL", "player_int"]]
 868 io_db["worldstate_write_order"] += [["T_BLADDER", "player_int"]]
 869 io_db["worldstate_write_order"] += [[write_wetmap, "func"]]
 870 io_db["worldstate_write_order"] += [[write_soundmap, "func"]]
 871 io_db["worldstate_write_order"] += [["GRACE", "world_int"]]
 872 import server.config.world_data
 873 server.config.world_data.symbols_hide = "345"
 874 server.config.world_data.symbols_passable = "012-+*&$"
 875 server.config.world_data.thing_defaults["T_STOMACH"] = 16
 876 server.config.world_data.thing_defaults["T_BOWEL"] = 0
 877 server.config.world_data.thing_defaults["T_KIDNEY"] = 16
 878 server.config.world_data.thing_defaults["T_BLADDER"] = 0
 879 world_db["soundmap"] = bytearray(b"0" * world_db["MAP_LENGTH"] ** 2)
 880 world_db["wetmap"] = bytearray(b"0" * world_db["MAP_LENGTH"] ** 2)
 881 if not "NEW_SPAWN" in world_db:
 882     world_db["NEW_SPAWN"] = 0
 883 if not "HUMIDITY" in world_db:
 884     world_db["HUMIDITY"] = 0
 885 if not "GRACE" in world_db:
 886     world_db["GRACE"] = 0
 887 io_db["hook_save"] = save_maps
 888 import server.config.make_world_helpers
 889 server.config.make_world_helpers.make_map = make_map
 890 from server.config.commands import commands_db
 891 commands_db["THINGS_HERE"] = (2, True, lambda x, y: None)
 892 commands_db["HELP"] = (1, False, command_help)
 893 commands_db["ai"] = (0, False, command_ai)
 894 commands_db["move"] = (1, False, play_move)
 895 commands_db["eat"] = (1, False, play_move)
 896 commands_db["wait"] = (0, False, play_wait)
 897 commands_db["drop"] = (0, False, play_drop)
 898 commands_db["drink"] = (0, False, play_drink)
 899 commands_db["pee"] = (0, False, play_pee)
 900 commands_db["use"] = (1, False, lambda x: None)
 901 commands_db["pickup"] = (0, False, lambda: None)
 902 commands_db["GRACE"] = (1, False, setter(None, "GRACE", 0, 255))
 903 commands_db["NEW_SPAWN"] = (1, False, setter(None, "NEW_SPAWN", 0, 255))
 904 commands_db["HUMIDITY"] = (1, False, setter(None, "HUMIDITY", 0, 65535))
 905 commands_db["T_STOMACH"] = (1, False, setter("Thing", "T_STOMACH", 0, 255))
 906 commands_db["T_KIDNEY"] = (1, False, setter("Thing", "T_KIDNEY", 0, 255))
 907 commands_db["T_BOWEL"] = (1, False, setter("Thing", "T_BOWEL", 0, 255))
 908 commands_db["T_BLADDER"] = (1, False, setter("Thing", "T_BLADDER", 0, 255))
 909 commands_db["WETMAP"] = (2, False, wetmapset)
 910 commands_db["SOUNDMAP"] = (2, False, soundmapset)
 911 from server.actions import actor_wait
 912 import server.config.actions
 913 server.config.actions.action_db = {
 914     "actor_wait": actor_wait,
 915     "actor_move": actor_move,
 916     "actor_drop": actor_drop,
 917     "actor_drink": actor_drink,
 918     "actor_pee": actor_pee,
 919     "actor_eat": actor_eat,
 920 }
 921
 922 strong_write(io_db["file_out"], "PLUGIN TheCrawlingEater\n")