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