home · contact · privacy
7DRL: De-/increment God's mood with lifepoints, and proliferation.
[plomrogue] / roguelike-server
1 #!/usr/bin/python3
2
3 # This file is part of PlomRogue. PlomRogue is licensed under the GPL version 3
4 # or any later version. For details on its copyright, license, and warranties,
5 # see the file NOTICE in the root directory of the PlomRogue source package.
6
7
8 import argparse
9 import errno
10 import os
11 import shlex
12 import shutil
13 import time
14 import ctypes
15
16
17 class RandomnessIO:
18     """"Interface to libplomrogue's pseudo-randomness generator."""
19
20     def set_seed(self, seed):
21         libpr.seed_rrand(1, seed)
22
23     def get_seed(self):
24         return libpr.seed_rrand(0, 0)
25
26     def next(self):
27         return libpr.rrand()
28
29     seed = property(get_seed, set_seed)
30
31
32 def prep_library():
33     """Prepare ctypes library at ./libplomrogue.so"""
34     libpath = ("./libplomrogue.so")
35     if not os.access(libpath, os.F_OK):
36         raise SystemExit("No library " + libpath + ", run ./redo first?")
37     libpr = ctypes.cdll.LoadLibrary(libpath)
38     libpr.seed_rrand.restype = ctypes.c_uint32
39     return libpr
40
41
42 def c_pointer_to_bytearray(ba):
43     """Return C char * pointer to ba."""
44     type = ctypes.c_char * len(ba)
45     return type.from_buffer(ba)
46
47
48 def strong_write(file, string):
49     """Apply write(string), then flush()."""
50     file.write(string)
51     file.flush()
52
53
54 def setup_server_io():
55     """Fill IO files DB with proper file( path)s. Write process IO test string.
56
57     Ensure IO files directory at server/. Remove any old input file if found.
58     Set up new input file for reading, and new output file for writing. Start
59     output file with process hash line of format PID + " " + floated UNIX time
60     (io_db["teststring"]). Raise SystemExit if file is found at path of either
61     record or save file plus io_db["tmp_suffix"].
62     """
63     def detect_atomic_leftover(path, tmp_suffix):
64         path_tmp = path + tmp_suffix
65         msg = "Found file '" + path_tmp + "' that may be a leftover from an " \
66               "aborted previous attempt to write '" + path + "'. Aborting " \
67              "until matter is resolved by removing it from its current path."
68         if os.access(path_tmp, os.F_OK):
69             raise SystemExit(msg)
70     io_db["teststring"] = str(os.getpid()) + " " + str(time.time())
71     io_db["save_wait"] = 0
72     io_db["verbose"] = False
73     io_db["record_chunk"] = ""
74     os.makedirs(io_db["path_server"], exist_ok=True)
75     io_db["file_out"] = open(io_db["path_out"], "w")
76     strong_write(io_db["file_out"], io_db["teststring"] + "\n")
77     if os.access(io_db["path_in"], os.F_OK):
78         os.remove(io_db["path_in"])
79     io_db["file_in"] = open(io_db["path_in"], "w")
80     io_db["file_in"].close()
81     io_db["file_in"] = open(io_db["path_in"], "r")
82     detect_atomic_leftover(io_db["path_save"], io_db["tmp_suffix"])
83     detect_atomic_leftover(io_db["path_record"], io_db["tmp_suffix"])
84
85
86 def cleanup_server_io():
87     """Close and (if io_db["kicked_by_rival"] false) remove files in io_db."""
88     def helper(file_key, path_key):
89         if file_key in io_db:
90             io_db[file_key].close()
91         if not io_db["kicked_by_rival"] \
92            and os.access(io_db[path_key], os.F_OK):
93             os.remove(io_db[path_key])
94     helper("file_in", "path_in")
95     helper("file_out", "path_out")
96     helper("file_worldstate", "path_worldstate")
97     if "file_record" in io_db:
98         io_db["file_record"].close()
99
100
101 def obey(command, prefix, replay=False, do_record=False):
102     """Call function from commands_db mapped to command's first token.
103
104     Tokenize command string with shlex.split(comments=True). If replay is set,
105     a non-meta command from the commands_db merely triggers obey() on the next
106     command from the records file. If not, non-meta commands set
107     io_db["worldstate_updateable"] to world_db["WORLD_EXISTS"], and, if
108     do_record is set, are recorded to io_db["record_chunk"], and save_world()
109     is called (and io_db["record_chunk"] written) if 15 seconds have passed
110     since the last time it was called. The prefix string is inserted into the
111     server's input message between its beginning 'input ' and ':'. All activity
112     is preceded by a server_test() call.
113     """
114     server_test()
115     if io_db["verbose"]:
116         print("input " + prefix + ": " + command)
117     try:
118         tokens = shlex.split(command, comments=True)
119     except ValueError as err:
120         print("Can't tokenize command string: " + str(err) + ".")
121         return
122     if len(tokens) > 0 and tokens[0] in commands_db \
123        and len(tokens) == commands_db[tokens[0]][0] + 1:
124         if commands_db[tokens[0]][1]:
125             commands_db[tokens[0]][2](*tokens[1:])
126         elif replay:
127             print("Due to replay mode, reading command as 'go on in record'.")
128             line = io_db["file_record"].readline()
129             if len(line) > 0:
130                 obey(line.rstrip(), io_db["file_record"].prefix
131                      + str(io_db["file_record"].line_n))
132                 io_db["file_record"].line_n = io_db["file_record"].line_n + 1
133             else:
134                 print("Reached end of record file.")
135         else:
136             commands_db[tokens[0]][2](*tokens[1:])
137             if do_record:
138                 io_db["record_chunk"] += command + "\n"
139                 if time.time() > io_db["save_wait"] + 15:
140                     atomic_write(io_db["path_record"], io_db["record_chunk"],
141                                  do_append=True)
142                     save_world()
143                     io_db["record_chunk"] = ""
144                     io_db["save_wait"] = time.time()
145             io_db["worldstate_updateable"] = world_db["WORLD_ACTIVE"]
146     elif 0 != len(tokens):
147         print("Invalid command/argument, or bad number of tokens.")
148
149
150 def atomic_write(path, text, do_append=False, delete=True):
151     """Atomic write of text to file at path, appended if do_append is set."""
152     path_tmp = path + io_db["tmp_suffix"]
153     mode = "w"
154     if do_append:
155         mode = "a"
156         if os.access(path, os.F_OK):
157             shutil.copyfile(path, path_tmp)
158     file = open(path_tmp, mode)
159     strong_write(file, text)
160     file.close()
161     if delete and os.access(path, os.F_OK):
162         os.remove(path)
163     os.rename(path_tmp, path)
164
165
166 def save_world():
167     """Save all commands needed to reconstruct current world state."""
168
169     def quote(string):
170         string = string.replace("\u005C", '\u005C\u005C')
171         return '"' + string.replace('"', '\u005C"') + '"'
172
173     def mapsetter(key):
174         def helper(id):
175             string = ""
176             if world_db["Things"][id][key]:
177                 map = world_db["Things"][id][key]
178                 length = world_db["MAP_LENGTH"]
179                 for i in range(length):
180                     line = map[i * length:(i * length) + length].decode()
181                     string = string + key + " " + str(i) + " " + quote(line) \
182                              + "\n"
183             return string
184         return helper
185
186     def memthing(id):
187         string = ""
188         for memthing in world_db["Things"][id]["T_MEMTHING"]:
189             string = string + "T_MEMTHING " + str(memthing[0]) + " " + \
190                      str(memthing[1]) + " " + str(memthing[2]) + "\n"
191         return string
192
193     def helper(category, id_string, special_keys={}):
194         string = ""
195         for id in world_db[category]:
196             string = string + id_string + " " + str(id) + "\n"
197             for key in world_db[category][id]:
198                 if not key in special_keys:
199                     x = world_db[category][id][key]
200                     argument = quote(x) if str == type(x) else str(x)
201                     string = string + key + " " + argument + "\n"
202                 elif special_keys[key]:
203                     string = string + special_keys[key](id)
204         return string
205
206     string = ""
207     for key in world_db:
208         if dict != type(world_db[key]) and key != "MAP" and \
209            key != "WORLD_ACTIVE" and key != "SEED_MAP":
210             string = string + key + " " + str(world_db[key]) + "\n"
211     string = string + "SEED_MAP " + str(world_db["SEED_MAP"]) + "\n"
212     string = string + helper("ThingActions", "TA_ID")
213     string = string + helper("ThingTypes", "TT_ID", {"TT_CORPSE_ID": False})
214     for id in world_db["ThingTypes"]:
215         string = string + "TT_ID " + str(id) + "\n" + "TT_CORPSE_ID " + \
216                  str(world_db["ThingTypes"][id]["TT_CORPSE_ID"]) + "\n"
217     string = string + helper("Things", "T_ID",
218                              {"T_CARRIES": False, "carried": False,
219                               "T_MEMMAP": mapsetter("T_MEMMAP"),
220                               "T_MEMTHING": memthing, "fovmap": False,
221                               "T_MEMDEPTHMAP": mapsetter("T_MEMDEPTHMAP")})
222     for id in world_db["Things"]:
223         if [] != world_db["Things"][id]["T_CARRIES"]:
224             string = string + "T_ID " + str(id) + "\n"
225             for carried_id in world_db["Things"][id]["T_CARRIES"]:
226                 string = string + "T_CARRIES " + str(carried_id) + "\n"
227     string = string + "SEED_RANDOMNESS " + str(rand.seed) + "\n" + \
228              "WORLD_ACTIVE " + str(world_db["WORLD_ACTIVE"])
229     atomic_write(io_db["path_save"], string)
230
231
232 def obey_lines_in_file(path, name, do_record=False):
233     """Call obey() on each line of path's file, use name in input prefix."""
234     file = open(path, "r")
235     line_n = 1
236     for line in file.readlines():
237         obey(line.rstrip(), name + "file line " + str(line_n),
238              do_record=do_record)
239         line_n = line_n + 1
240     file.close()
241
242
243 def parse_command_line_arguments():
244     """Return settings values read from command line arguments."""
245     parser = argparse.ArgumentParser()
246     parser.add_argument('-s', nargs='?', type=int, dest='replay', const=1,
247                         action='store')
248     parser.add_argument('-l', nargs="?", const="save", dest='savefile',
249                         action="store")
250     parser.add_argument('-v', dest='verbose', action='store_true')
251     opts, unknown = parser.parse_known_args()
252     return opts
253
254
255 def server_test():
256     """Ensure valid server out file belonging to current process.
257
258     This is done by comparing io_db["teststring"] to what's found at the start
259     of the current file at io_db["path_out"]. On failure, set
260     io_db["kicked_by_rival"] and raise SystemExit.
261     """
262     if not os.access(io_db["path_out"], os.F_OK):
263         raise SystemExit("Server output file has disappeared.")
264     file = open(io_db["path_out"], "r")
265     test = file.readline().rstrip("\n")
266     file.close()
267     if test != io_db["teststring"]:
268         io_db["kicked_by_rival"] = True
269         msg = "Server test string in server output file does not match. This" \
270               " indicates that the current server process has been " \
271               "superseded by another one."
272         raise SystemExit(msg)
273
274
275 def read_command():
276     """Return next newline-delimited command from server in file.
277
278     Keep building return string until a newline is encountered. Pause between
279     unsuccessful reads, and after too much waiting, run server_test().
280     """
281     wait_on_fail = 0.03333
282     max_wait = 5
283     now = time.time()
284     command = ""
285     while True:
286         add = io_db["file_in"].readline()
287         if len(add) > 0:
288             command = command + add
289             if len(command) > 0 and "\n" == command[-1]:
290                 command = command[:-1]
291                 break
292         else:
293             time.sleep(wait_on_fail)
294             if now + max_wait < time.time():
295                 server_test()
296                 now = time.time()
297     return command
298
299
300 def try_worldstate_update():
301     """Write worldstate file if io_db["worldstate_updateable"] is set."""
302     if io_db["worldstate_updateable"]:
303
304         def draw_visible_Things(map, run):
305             for id in world_db["Things"]:
306                 type = world_db["Things"][id]["T_TYPE"]
307                 consumable = world_db["ThingTypes"][type]["TT_CONSUMABLE"]
308                 alive = world_db["ThingTypes"][type]["TT_LIFEPOINTS"]
309                 if (0 == run and not consumable and not alive) \
310                    or (1 == run and consumable and not alive) \
311                    or (2 == run and alive):
312                     y = world_db["Things"][id]["T_POSY"]
313                     x = world_db["Things"][id]["T_POSX"]
314                     fovflag = world_db["Things"][0]["fovmap"][(y * length) + x]
315                     if 'v' == chr(fovflag):
316                         c = world_db["ThingTypes"][type]["TT_SYMBOL"]
317                         map[(y * length) + x] = ord(c)
318
319         def write_map(string, map):
320             for i in range(length):
321                 line = map[i * length:(i * length) + length].decode()
322                 string = string + line + "\n"
323             return string
324
325         inventory = ""
326         if [] == world_db["Things"][0]["T_CARRIES"]:
327             inventory = "(none)\n"
328         else:
329             for id in world_db["Things"][0]["T_CARRIES"]:
330                 type_id = world_db["Things"][id]["T_TYPE"]
331                 name = world_db["ThingTypes"][type_id]["TT_NAME"]
332                 inventory = inventory + name + "\n"
333         # # 7DRL additions:  GOD_MOOD, GOD_FAVOR
334         string = str(world_db["TURN"]) + "\n" + \
335                  str(world_db["GOD_MOOD"]) + "\n" + \
336                  str(world_db["GOD_FAVOR"]) + "\n" + \
337                  str(world_db["Things"][0]["T_LIFEPOINTS"]) + "\n" + \
338                  str(world_db["Things"][0]["T_SATIATION"]) + "\n" + \
339                  inventory + "%\n" + \
340                  str(world_db["Things"][0]["T_POSY"]) + "\n" + \
341                  str(world_db["Things"][0]["T_POSX"]) + "\n" + \
342                  str(world_db["MAP_LENGTH"]) + "\n"
343         length = world_db["MAP_LENGTH"]
344         fov = bytearray(b' ' * (length ** 2))
345         for pos in range(length ** 2):
346             if 'v' == chr(world_db["Things"][0]["fovmap"][pos]):
347                 fov[pos] = world_db["MAP"][pos]
348         for i in range(3):
349             draw_visible_Things(fov, i)
350         string = write_map(string, fov)
351         mem = world_db["Things"][0]["T_MEMMAP"][:]
352         for i in range(2):
353             for mt in world_db["Things"][0]["T_MEMTHING"]:
354                 consumable = world_db["ThingTypes"][mt[0]]["TT_CONSUMABLE"]
355                 if (i == 0 and not consumable) or (i == 1 and consumable):
356                     c = world_db["ThingTypes"][mt[0]]["TT_SYMBOL"]
357                     mem[(mt[1] * length) + mt[2]] = ord(c)
358         string = write_map(string, mem)
359         atomic_write(io_db["path_worldstate"], string, delete=False)
360         strong_write(io_db["file_out"], "WORLD_UPDATED\n")
361         io_db["worldstate_updateable"] = False
362
363
364 def replay_game():
365     """Replay game from record file.
366
367     Use opts.replay as breakpoint turn to which to replay automatically before
368     switching to manual input by non-meta commands in server input file
369     triggering further reads of record file. Ensure opts.replay is at least 1.
370     Run try_worldstate_update() before each interactive obey()/read_command().
371     """
372     if opts.replay < 1:
373         opts.replay = 1
374     print("Replay mode. Auto-replaying up to turn " + str(opts.replay) +
375           " (if so late a turn is to be found).")
376     if not os.access(io_db["path_record"], os.F_OK):
377         raise SystemExit("No record file found to replay.")
378     io_db["file_record"] = open(io_db["path_record"], "r")
379     io_db["file_record"].prefix = "record file line "
380     io_db["file_record"].line_n = 1
381     while world_db["TURN"] < opts.replay:
382         line = io_db["file_record"].readline()
383         if "" == line:
384             break
385         obey(line.rstrip(), io_db["file_record"].prefix
386              + str(io_db["file_record"].line_n))
387         io_db["file_record"].line_n = io_db["file_record"].line_n + 1
388     while True:
389         try_worldstate_update()
390         obey(read_command(), "in file", replay=True)
391
392
393 def play_game():
394     """Play game by server input file commands. Before, load save file found.
395
396     If no save file is found, a new world is generated from the commands in the
397     world config plus a 'MAKE WORLD [current Unix timestamp]'. Record this
398     command and all that follow via the server input file. Run
399     try_worldstate_update() before each interactive obey()/read_command().
400     """
401     if os.access(io_db["path_save"], os.F_OK):
402         obey_lines_in_file(io_db["path_save"], "save")
403     else:
404         if not os.access(io_db["path_worldconf"], os.F_OK):
405             msg = "No world config file from which to start a new world."
406             raise SystemExit(msg)
407         obey_lines_in_file(io_db["path_worldconf"], "world config ",
408                            do_record=True)
409         obey("MAKE_WORLD " + str(int(time.time())), "in file", do_record=True)
410     while True:
411         try_worldstate_update()
412         obey(read_command(), "in file", do_record=True)
413
414
415 def remake_map():
416     """(Re-)make island map.
417
418     Let "~" represent water, "." land, "X" trees: Build island shape randomly,
419     start with one land cell in the middle, then go into cycle of repeatedly
420     selecting a random sea cell and transforming it into land if it is neighbor
421     to land. The cycle ends when a land cell is due to be created at the map's
422     border. Then put some trees on the map (TODO: more precise algorithm desc).
423     """
424     def is_neighbor(coordinates, type):
425         y = coordinates[0]
426         x = coordinates[1]
427         length = world_db["MAP_LENGTH"]
428         ind = y % 2
429         diag_west = x + (ind > 0)
430         diag_east = x + (ind < (length - 1))
431         pos = (y * length) + x
432         if (y > 0 and diag_east
433             and type == chr(world_db["MAP"][pos - length + ind])) \
434            or (x < (length - 1)
435                and type == chr(world_db["MAP"][pos + 1])) \
436            or (y < (length - 1) and diag_east
437                and type == chr(world_db["MAP"][pos + length + ind])) \
438            or (y > 0 and diag_west
439                and type == chr(world_db["MAP"][pos - length - (not ind)])) \
440            or (x > 0
441                and type == chr(world_db["MAP"][pos - 1])) \
442            or (y < (length - 1) and diag_west
443                and type == chr(world_db["MAP"][pos + length - (not ind)])):
444             return True
445         return False
446     store_seed = rand.seed
447     rand.seed = world_db["SEED_MAP"]
448     world_db["MAP"] = bytearray(b'~' * (world_db["MAP_LENGTH"] ** 2))
449     length = world_db["MAP_LENGTH"]
450     add_half_width = (not (length % 2)) * int(length / 2)
451     world_db["MAP"][int((length ** 2) / 2) + add_half_width] = ord(".")
452     while (1):
453         y = rand.next() % length
454         x = rand.next() % length
455         pos = (y * length) + x
456         if "~" == chr(world_db["MAP"][pos]) and is_neighbor((y, x), "."):
457             if y == 0 or y == (length - 1) or x == 0 or x == (length - 1):
458                 break
459             world_db["MAP"][pos] = ord(".")
460     n_trees = int((length ** 2) / 16)
461     i_trees = 0
462     while (i_trees <= n_trees):
463         single_allowed = rand.next() % 32
464         y = rand.next() % length
465         x = rand.next() % length
466         pos = (y * length) + x
467         if "." == chr(world_db["MAP"][pos]) \
468           and ((not single_allowed) or is_neighbor((y, x), "X")):
469             world_db["MAP"][pos] = ord("X")
470             i_trees += 1
471     rand.seed = store_seed
472     # This all-too-precise replica of the original C code misses iter_limit().
473
474
475 def update_map_memory(t, age_map=True):
476     """Update t's T_MEMMAP with what's in its FOV now,age its T_MEMMEPTHMAP."""
477     def age_some_memdepthmap_on_nonfov_cells():
478         # OUTSOURCED FOR PERFORMANCE REASONS TO libplomrogue.so:
479         # ord_v = ord("v")
480         # ord_0 = ord("0")
481         # ord_9 = ord("9")
482         # for pos in [pos for pos in range(world_db["MAP_LENGTH"] ** 2)
483         #             if not ord_v == t["fovmap"][pos]
484         #             if ord_0 <= t["T_MEMDEPTHMAP"][pos]
485         #             if ord_9 > t["T_MEMDEPTHMAP"][pos]
486         #             if not rand.next() % (2 **
487         #                                   (t["T_MEMDEPTHMAP"][pos] - 48))]:
488         #     t["T_MEMDEPTHMAP"][pos] += 1
489         memdepthmap = c_pointer_to_bytearray(t["T_MEMDEPTHMAP"])
490         fovmap = c_pointer_to_bytearray(t["fovmap"])
491         libpr.age_some_memdepthmap_on_nonfov_cells(memdepthmap, fovmap)
492     if not t["T_MEMMAP"]:
493         t["T_MEMMAP"] = bytearray(b' ' * (world_db["MAP_LENGTH"] ** 2))
494     if not t["T_MEMDEPTHMAP"]:
495         t["T_MEMDEPTHMAP"] = bytearray(b' ' * (world_db["MAP_LENGTH"] ** 2))
496     ord_v = ord("v")
497     ord_0 = ord("0")
498     ord_space = ord(" ")
499     for pos in [pos for pos in range(world_db["MAP_LENGTH"] ** 2)
500                 if ord_v == t["fovmap"][pos]]:
501         t["T_MEMDEPTHMAP"][pos] = ord_0
502         if ord_space == t["T_MEMMAP"][pos]:
503             t["T_MEMMAP"][pos] = world_db["MAP"][pos]
504     if age_map:
505         age_some_memdepthmap_on_nonfov_cells()
506     for mt in [mt for mt in t["T_MEMTHING"]
507                if ord_v == t["fovmap"][(mt[1] * world_db["MAP_LENGTH"])
508                                        + mt[2]]]:
509         t["T_MEMTHING"].remove(mt)
510     for id in [id for id in world_db["Things"]
511                if not world_db["Things"][id]["carried"]]:
512         type = world_db["Things"][id]["T_TYPE"]
513         if not world_db["ThingTypes"][type]["TT_LIFEPOINTS"]:
514             y = world_db["Things"][id]["T_POSY"]
515             x = world_db["Things"][id]["T_POSX"]
516             if ord_v == t["fovmap"][(y * world_db["MAP_LENGTH"]) + x]:
517                 t["T_MEMTHING"].append((type, y, x))
518
519
520 def set_world_inactive():
521     """Set world_db["WORLD_ACTIVE"] to 0 and remove worldstate file."""
522     server_test()
523     if os.access(io_db["path_worldstate"], os.F_OK):
524         os.remove(io_db["path_worldstate"])
525     world_db["WORLD_ACTIVE"] = 0
526
527
528 def integer_test(val_string, min, max=None):
529     """Return val_string if possible integer >= min and <= max, else None."""
530     try:
531         val = int(val_string)
532         if val < min or (max is not None and val > max):
533             raise ValueError
534         return val
535     except ValueError:
536         msg = "Ignoring: Please use integer >= " + str(min)
537         if max is not None:
538             msg += " and <= " + str(max)
539         msg += "."
540         print(msg)
541         return None
542
543
544 def setter(category, key, min, max=None):
545     """Build setter for world_db([category + "s"][id])[key] to >=min/<=max."""
546     if category is None:
547         def f(val_string):
548             val = integer_test(val_string, min, max)
549             if None != val:
550                 world_db[key] = val
551     else:
552         if category == "Thing":
553             id_store = command_tid
554             decorator = test_Thing_id
555         elif category == "ThingType":
556             id_store = command_ttid
557             decorator = test_ThingType_id
558         elif category == "ThingAction":
559             id_store = command_taid
560             decorator = test_ThingAction_id
561
562         @decorator
563         def f(val_string):
564             val = integer_test(val_string, min, max)
565             if None != val:
566                 world_db[category + "s"][id_store.id][key] = val
567     return f
568
569
570 def build_fov_map(t):
571     """Build Thing's FOV map."""
572     t["fovmap"] = bytearray(b'v' * (world_db["MAP_LENGTH"] ** 2))
573     fovmap = c_pointer_to_bytearray(t["fovmap"])
574     map = c_pointer_to_bytearray(world_db["MAP"])
575     if libpr.build_fov_map(t["T_POSY"], t["T_POSX"], fovmap, map):
576         raise RuntimeError("Malloc error in build_fov_Map().")
577
578
579 def decrement_lifepoints(t):
580     """Decrement t's lifepoints by 1, and if to zero, corpse it.
581
582     If t is the player avatar, only blank its fovmap, so that the client may
583     still display memory data. On non-player things, erase fovmap and memory.
584     """
585     # # 7DRL: also decrements God's mood; deaths heavily so
586     t["T_LIFEPOINTS"] -= 1
587     world_db["GOD_MOOD"] -= 1  # #
588     if 0 == t["T_LIFEPOINTS"]:
589         world_db["GOD_MOOD"] -= 9  # #
590         t["T_TYPE"] = world_db["ThingTypes"][t["T_TYPE"]]["TT_CORPSE_ID"]
591         if world_db["Things"][0] == t:
592             t["fovmap"] = bytearray(b' ' * (world_db["MAP_LENGTH"] ** 2))
593             strong_write(io_db["file_out"], "LOG You die.\n")
594         else:
595             t["fovmap"] = False
596             t["T_MEMMAP"] = False
597             t["T_MEMDEPTHMAP"] = False
598             t["T_MEMTHING"] = []
599
600
601 def mv_yx_in_dir_legal(dir, y, x):
602     """Wrapper around libpr.mv_yx_in_dir_legal to simplify its use."""
603     dir_c = dir.encode("ascii")[0]
604     test = libpr.mv_yx_in_dir_legal_wrap(dir_c, y, x)
605     if -1 == test:
606         raise RuntimeError("Too much wrapping in mv_yx_in_dir_legal_wrap()!")
607     return (test, libpr.result_y(), libpr.result_x())
608
609
610 def actor_wait(t):
611     """Make t do nothing (but loudly, if player avatar)."""
612     if t == world_db["Things"][0]:
613         strong_write(io_db["file_out"], "LOG You wait.\n")
614
615
616 def actor_move(t):
617     """If passable, move/collide(=attack) thing into T_ARGUMENT's direction."""
618     passable = False
619     move_result = mv_yx_in_dir_legal(chr(t["T_ARGUMENT"]),
620                                      t["T_POSY"], t["T_POSX"])
621     if 1 == move_result[0]:
622         pos = (move_result[1] * world_db["MAP_LENGTH"]) + move_result[2]
623         passable = "." == chr(world_db["MAP"][pos])
624         hitted = [id for id in world_db["Things"]
625                   if world_db["Things"][id] != t
626                   if world_db["Things"][id]["T_LIFEPOINTS"]
627                   if world_db["Things"][id]["T_POSY"] == move_result[1]
628                   if world_db["Things"][id]["T_POSX"] == move_result[2]]
629         if len(hitted):
630             hit_id = hitted[0]
631             if t == world_db["Things"][0]:
632                 hitted_type = world_db["Things"][hit_id]["T_TYPE"]
633                 hitted_name = world_db["ThingTypes"][hitted_type]["TT_NAME"]
634                 strong_write(io_db["file_out"], "LOG You wound " + hitted +
635                                                 ".\n")
636             elif 0 == hit_id:
637                 hitter_name = world_db["ThingTypes"][t["T_TYPE"]]["TT_NAME"]
638                 strong_write(io_db["file_out"], "LOG " + hitter_name +
639                                                 " wounds you.\n")
640             decrement_lifepoints(world_db["Things"][hit_id])
641             return
642     dir = [dir for dir in directions_db
643            if directions_db[dir] == chr(t["T_ARGUMENT"])][0]
644     if passable:
645         t["T_POSY"] = move_result[1]
646         t["T_POSX"] = move_result[2]
647         for id in t["T_CARRIES"]:
648             world_db["Things"][id]["T_POSY"] = move_result[1]
649             world_db["Things"][id]["T_POSX"] = move_result[2]
650         build_fov_map(t)
651         if t == world_db["Things"][0]:
652             strong_write(io_db["file_out"], "LOG You move " + dir + ".\n")
653     elif t == world_db["Things"][0]:
654         strong_write(io_db["file_out"], "LOG You fail to move " + dir + ".\n")
655
656
657 def actor_pick_up(t):
658     """Make t pick up (topmost?) Thing from ground into inventory."""
659     # Topmostness is actually not defined so far. Picks Thing with highest ID.
660     used_slots = len(t["T_CARRIES"]) # #
661     if used_slots < world_db["ThingTypes"][t["T_TYPE"]]["TT_STORAGE"]: # #
662         ids = [id for id in world_db["Things"] if world_db["Things"][id] != t
663                if not world_db["Things"][id]["carried"]
664                if world_db["Things"][id]["T_POSY"] == t["T_POSY"]
665                if world_db["Things"][id]["T_POSX"] == t["T_POSX"]]
666         if len(ids):
667             highest_id = 0
668             for id in ids:
669                 if id > highest_id:
670                     highest_id = id
671             world_db["Things"][highest_id]["carried"] = True
672             t["T_CARRIES"].append(highest_id)
673             if t == world_db["Things"][0]:
674                 strong_write(io_db["file_out"], "LOG You pick up an object.\n")
675         elif t == world_db["Things"][0]:
676             err = "You try to pick up an object, but there is none."
677             strong_write(io_db["file_out"], "LOG " + err + "\n")
678     elif t == world_db["Things"][0]: # #
679         strong_write(io_db["file_out"], "LOG Can't pick up object: " + # #
680                                         "No storage room to carry more.\n") # #
681
682
683 def actor_drop(t):
684     """Make t rop Thing from inventory to ground indexed by T_ARGUMENT."""
685     # TODO: Handle case where T_ARGUMENT matches nothing.
686     if len(t["T_CARRIES"]):
687         id = t["T_CARRIES"][t["T_ARGUMENT"]]
688         t["T_CARRIES"].remove(id)
689         world_db["Things"][id]["carried"] = False
690         if t == world_db["Things"][0]:
691             strong_write(io_db["file_out"], "LOG You drop an object.\n")
692     elif t == world_db["Things"][0]:
693         err = "You try to drop an object, but you own none."
694         strong_write(io_db["file_out"], "LOG " + err + "\n")
695
696
697 def actor_use(t):
698     """Make t use (for now: consume) T_ARGUMENT-indexed Thing in inventory."""
699     # TODO: Handle case where T_ARGUMENT matches nothing.
700     if len(t["T_CARRIES"]):
701         id = t["T_CARRIES"][t["T_ARGUMENT"]]
702         type = world_db["Things"][id]["T_TYPE"]
703         if world_db["ThingTypes"][type]["TT_CONSUMABLE"]:
704             t["T_CARRIES"].remove(id)
705             del world_db["Things"][id]
706             t["T_SATIATION"] += world_db["ThingTypes"][type]["TT_CONSUMABLE"]
707             if t == world_db["Things"][0]:
708                 strong_write(io_db["file_out"],
709                              "LOG You consume this object.\n")
710         elif t == world_db["Things"][0]:
711             strong_write(io_db["file_out"],
712                          "LOG You try to use this object, but fail.\n")
713     elif t == world_db["Things"][0]:
714         strong_write(io_db["file_out"],
715                      "LOG You try to use an object, but you own none.\n")
716
717
718 def thingproliferation(t):
719     """To chance of 1/TT_PROLIFERATE, create t offspring in neighbor cell.
720
721     Naturally only works with TT_PROLIFERATE > 0. The neighbor cell must be
722     passable and not be inhabited by a Thing of the same type, or, if Thing is
723     animate, any other animate Thing. If there are several map cell candidates,
724     one is selected randomly.
725     """
726     # # 7DRL: success increments God's mood
727     def test_cell(t, y, x):
728         if "." == chr(world_db["MAP"][(y * world_db["MAP_LENGTH"]) + x]):
729             for id in [id for id in world_db["Things"]
730                        if y == world_db["Things"][id]["T_POSY"]
731                        if x == world_db["Things"][id]["T_POSX"]
732                        if (t["T_TYPE"] == world_db["Things"][id]["T_TYPE"])
733                        or (t["T_LIFEPOINTS"] and
734                            world_db["Things"][id]["T_LIFEPOINTS"])]:
735                 return False
736             return True
737         return False
738     prolscore = world_db["ThingTypes"][t["T_TYPE"]]["TT_PROLIFERATE"]
739     if prolscore and (1 == prolscore or 1 == (rand.next() % prolscore)):
740         candidates = []
741         for dir in [directions_db[key] for key in directions_db]:
742             mv_result = mv_yx_in_dir_legal(dir, t["T_POSY"], t["T_POSX"])
743             if mv_result[0] and test_cell(t, mv_result[1], mv_result[2]):
744                 candidates.append((mv_result[1], mv_result[2]))
745         if len(candidates):
746             i = rand.next() % len(candidates)
747             id = id_setter(-1, "Things")
748             newT = new_Thing(t["T_TYPE"], (candidates[i][0], candidates[i][1]))
749             world_db["Things"][id] = newT
750             world_db["GOD_MOOD"] += 1  # #
751
752
753 def try_healing(t):
754     """Grow t's HP to a 1/32 chance if < HP max, satiation > 0, and waiting.
755
756     On success, decrease satiation score by 32.
757     """
758     # # 7DRL: Successful heals increment God's mood.
759     if t["T_SATIATION"] > 0 \
760        and t["T_LIFEPOINTS"] < \
761            world_db["ThingTypes"][t["T_TYPE"]]["TT_LIFEPOINTS"] \
762        and 0 == (rand.next() % 31) \
763        and t["T_COMMAND"] == [id for id in world_db["ThingActions"]
764                               if world_db["ThingActions"][id]["TA_NAME"] ==
765                                  "wait"][0]:
766         t["T_LIFEPOINTS"] += 1
767         world_db["GOD_MOOD"] += 1  # #
768         t["T_SATIATION"] -= 32
769         if t == world_db["Things"][0]:
770             strong_write(io_db["file_out"], "LOG You heal.\n")
771
772
773 def hunger(t):
774     """Decrement t's satiation,dependent on it trigger lifepoint dec chance."""
775     if t["T_SATIATION"] > -32768:
776         t["T_SATIATION"] -= 1
777     testbase = t["T_SATIATION"] if t["T_SATIATION"] >= 0 else -t["T_SATIATION"]
778     if not world_db["ThingTypes"][t["T_TYPE"]]["TT_LIFEPOINTS"]:
779         raise RuntimeError("A thing that should not hunger is hungering.")
780     stomach = int(32767 / world_db["ThingTypes"][t["T_TYPE"]]["TT_LIFEPOINTS"])
781     if int(int(testbase / stomach) / ((rand.next() % stomach) + 1)):
782         if t == world_db["Things"][0]:
783             strong_write(io_db["file_out"], "LOG You suffer from hunger.\n")
784         decrement_lifepoints(t)
785
786
787 def get_dir_to_target(t, filter):
788     """Try to set T_COMMAND/T_ARGUMENT for move to "filter"-determined target.
789
790     The path-wise nearest target is chosen, via the shortest available path.
791     Target must not be t. On succcess, return positive value, else False.
792     Filters:
793     "a": Thing in FOV is below a certain distance, animate, but of ThingType
794          that is not t's, and starts out weaker than t is; build path as
795          avoiding things of t's ThingType
796     "f": neighbor cell (not inhabited by any animate Thing) further away from
797          animate Thing not further than x steps away and in FOV and of a
798          ThingType that is not t's, and starts out stronger or as strong as t
799          is currently; or (cornered), if no such flight cell, but Thing of
800          above criteria is too near,1 a cell closer to it, or, if less near,
801          just wait
802     "c": Thing in memorized map is consumable
803     "s": memory map cell with greatest-reachable degree of unexploredness
804     """
805
806     def zero_score_map_where_char_on_memdepthmap(c):
807         # OUTSOURCED FOR PERFORMANCE REASONS TO libplomrogue.so:
808         # for i in [i for i in range(world_db["MAP_LENGTH"] ** 2)
809         #           if t["T_MEMDEPTHMAP"][i] == mem_depth_c[0]]:
810         #     set_map_score(i, 0)
811         map = c_pointer_to_bytearray(t["T_MEMDEPTHMAP"])
812         if libpr.zero_score_map_where_char_on_memdepthmap(c, map):
813             raise RuntimeError("No score map allocated for "
814                                "zero_score_map_where_char_on_memdepthmap().")
815
816     def set_map_score(pos, score):
817         test = libpr.set_map_score(pos, score)
818         if test:
819             raise RuntimeError("No score map allocated for set_map_score().")
820
821     def get_map_score(pos):
822         result = libpr.get_map_score(pos)
823         if result < 0:
824             raise RuntimeError("No score map allocated for get_map_score().")
825         return result
826
827     def seeing_thing():
828         if t["fovmap"] and ("a" == filter or "f" == filter):
829             for id in world_db["Things"]:
830                 Thing = world_db["Things"][id]
831                 if Thing != t and Thing["T_LIFEPOINTS"] and \
832                    t["T_TYPE"] != Thing["T_TYPE"] and \
833                    'v' == chr(t["fovmap"][(Thing["T_POSY"]
834                                           * world_db["MAP_LENGTH"])
835                                           + Thing["T_POSX"]]):
836                     ThingType = world_db["ThingTypes"][Thing["T_TYPE"]]
837                     if ("f" == filter and ThingType["TT_LIFEPOINTS"] >=
838                                           t["T_LIFEPOINTS"]) \
839                        or ("a" == filter and ThingType["TT_LIFEPOINTS"] <
840                                              t["T_LIFEPOINTS"]):
841                         return True
842         elif t["T_MEMMAP"] and "c" == filter:
843             for mt in t["T_MEMTHING"]:
844                 if ' ' != chr(t["T_MEMMAP"][(mt[1] * world_db["MAP_LENGTH"])
845                                          + mt[2]]) \
846                    and world_db["ThingTypes"][mt[0]]["TT_CONSUMABLE"]:
847                     return True
848         return False
849
850     def set_cells_passable_on_memmap_to_65534_on_scoremap():
851         # OUTSOURCED FOR PERFORMANCE REASONS TO libplomrogue.so:
852         # ord_dot = ord(".")
853         # memmap = t["T_MEMMAP"]
854         # for i in [i for i in range(world_db["MAP_LENGTH"] ** 2)
855         #            if ord_dot == memmap[i]]:
856         #     set_map_score(i, 65534) # i.e. 65535-1
857         map = c_pointer_to_bytearray(t["T_MEMMAP"])
858         if libpr.set_cells_passable_on_memmap_to_65534_on_scoremap(map):
859             raise RuntimeError("No score map allocated for "
860                         "set_cells_passable_on_memmap_to_65534_on_scoremap().")
861
862     def init_score_map():
863         test = libpr.init_score_map()
864         if test:
865             raise RuntimeError("Malloc error in init_score_map().")
866         ord_v = ord("v")
867         ord_blank = ord(" ")
868         set_cells_passable_on_memmap_to_65534_on_scoremap()
869         if "a" == filter:
870             for id in world_db["Things"]:
871                 Thing = world_db["Things"][id]
872                 pos = Thing["T_POSY"] * world_db["MAP_LENGTH"] \
873                       + Thing["T_POSX"]
874                 if t != Thing and Thing["T_LIFEPOINTS"] and \
875                    t["T_TYPE"] != Thing["T_TYPE"] and \
876                    ord_v == t["fovmap"][pos] and \
877                    t["T_LIFEPOINTS"] > \
878                    world_db["ThingTypes"][Thing["T_TYPE"]]["TT_LIFEPOINTS"]:
879                     set_map_score(pos, 0)
880                 elif t["T_TYPE"] == Thing["T_TYPE"]:
881                     set_map_score(pos, 65535)
882         elif "f" == filter:
883             for id in [id for id in world_db["Things"]
884                        if world_db["Things"][id]["T_LIFEPOINTS"]]:
885                 Thing = world_db["Things"][id]
886                 pos = Thing["T_POSY"] * world_db["MAP_LENGTH"] \
887                       + Thing["T_POSX"]
888                 if t["T_TYPE"] != Thing["T_TYPE"] and \
889                    ord_v == t["fovmap"][pos] and \
890                    t["T_LIFEPOINTS"] <= \
891                    world_db["ThingTypes"][Thing["T_TYPE"]]["TT_LIFEPOINTS"]:
892                     set_map_score(pos, 0)
893         elif "c" == filter:
894             for mt in [mt for mt in t["T_MEMTHING"]
895                        if ord_blank != t["T_MEMMAP"][mt[1]
896                                                     * world_db["MAP_LENGTH"]
897                                                     + mt[2]]
898                        if world_db["ThingTypes"][mt[0]]["TT_CONSUMABLE"]]:
899                 set_map_score(mt[1] * world_db["MAP_LENGTH"] + mt[2], 0)
900         elif "s" == filter:
901             zero_score_map_where_char_on_memdepthmap(mem_depth_c[0])
902
903     def rand_target_dir(neighbors, cmp, dirs):
904         candidates = []
905         n_candidates = 0
906         for i in range(len(dirs)):
907             if cmp == neighbors[i]:
908                 candidates.append(dirs[i])
909                 n_candidates += 1
910         return candidates[rand.next() % n_candidates] if n_candidates else 0
911
912     def get_neighbor_scores(dirs, eye_pos):
913         scores = []
914         if libpr.ready_neighbor_scores(eye_pos):
915             raise RuntimeError("No score map allocated for " +
916                                "ready_neighbor_scores.()")
917         for i in range(len(dirs)):
918             scores.append(libpr.get_neighbor_score(i))
919         return scores
920
921     def get_dir_from_neighbors():
922         dir_to_target = False
923         dirs = "edcxsw"
924         eye_pos = t["T_POSY"] * world_db["MAP_LENGTH"] + t["T_POSX"]
925         neighbors = get_neighbor_scores(dirs, eye_pos)
926         if "f" == filter:
927             inhabited = [world_db["Things"][id]["T_POSY"]
928                          * world_db["MAP_LENGTH"]
929                          + world_db["Things"][id]["T_POSX"]
930                          for id in world_db["Things"]
931                          if world_db["Things"][id]["T_LIFEPOINTS"]]
932             for i in range(len(dirs)):
933                 mv_yx_in_dir_legal(dirs[i], t["T_POSY"], t["T_POSX"])
934                 pos_cmp = libpr.result_y() * world_db["MAP_LENGTH"] \
935                           + libpr.result_x()
936                 for pos in [pos for pos in inhabited if pos == pos_cmp]:
937                     neighbors[i] = 65535
938                     break
939         minmax_start = 0 if "f" == filter else 65535 - 1
940         minmax_neighbor = minmax_start
941         for i in range(len(dirs)):
942             if ("f" == filter and get_map_score(eye_pos) < neighbors[i] and
943                 minmax_neighbor < neighbors[i] and 65535 != neighbors[i]) \
944                or ("f" != filter and minmax_neighbor > neighbors[i]):
945                 minmax_neighbor = neighbors[i]
946         if minmax_neighbor != minmax_start:
947             dir_to_target = rand_target_dir(neighbors, minmax_neighbor, dirs)
948         if "f" == filter:
949             if not dir_to_target:
950                 if 1 == get_map_score(eye_pos):
951                     dir_to_target = rand_target_dir(neighbors, 0, dirs)
952                 elif 3 >= get_map_score(eye_pos):
953                     t["T_COMMAND"] = [id for id in world_db["ThingActions"]
954                                       if
955                                       world_db["ThingActions"][id]["TA_NAME"]
956                                          == "wait"][0]
957                     return 1
958             elif dir_to_target and 3 < get_map_score(eye_pos):
959                 dir_to_target = 0
960         elif "a" == filter and 10 <= get_map_score(eye_pos):
961             dir_to_target = 0
962         return dir_to_target
963
964     dir_to_target = False
965     mem_depth_c = b' '
966     run_i = 9 + 1 if "s" == filter else 1
967     while run_i and not dir_to_target and ("s" == filter or seeing_thing()):
968         run_i -= 1
969         init_score_map()
970         mem_depth_c = b'9' if b' ' == mem_depth_c \
971                            else bytes([mem_depth_c[0] - 1])
972         if libpr.dijkstra_map():
973             raise RuntimeError("No score map allocated for dijkstra_map().")
974         dir_to_target = get_dir_from_neighbors()
975         libpr.free_score_map()
976         if dir_to_target and str == type(dir_to_target):
977             t["T_COMMAND"] = [id for id in world_db["ThingActions"]
978                               if world_db["ThingActions"][id]["TA_NAME"]
979                                  == "move"][0]
980             t["T_ARGUMENT"] = ord(dir_to_target)
981     return dir_to_target
982
983
984 def standing_on_consumable(t):
985     """Return True/False whether t is standing on a consumable."""
986     for id in [id for id in world_db["Things"] if world_db["Things"][id] != t
987                if world_db["Things"][id]["T_POSY"] == t["T_POSY"]
988                if world_db["Things"][id]["T_POSX"] == t["T_POSX"]
989                if world_db["ThingTypes"][world_db["Things"][id]["T_TYPE"]]
990                           ["TT_CONSUMABLE"]]:
991         return True
992     return False
993
994
995 def get_inventory_slot_to_consume(t):
996     """Return slot Id of strongest consumable in t's inventory, else -1."""
997     cmp_consumability = 0
998     selection = -1
999     i = 0
1000     for id in t["T_CARRIES"]:
1001         type = world_db["Things"][id]["T_TYPE"]
1002         if world_db["ThingTypes"][type]["TT_CONSUMABLE"] > cmp_consumability:
1003             cmp_consumability = world_db["ThingTypes"][type]["TT_CONSUMABLE"]
1004             selection = i
1005         i += 1
1006     return selection
1007
1008
1009 def ai(t):
1010     """Determine next command/argment for actor t via AI algorithms.
1011
1012     AI will look for, and move towards, enemies (animate Things not of their
1013     own ThingType); if they see none, they will consume consumables in their
1014     inventory; if there are none, they will pick up what they stand on if they
1015     stand on consumables; if they stand on none, they will move towards the
1016     next consumable they see or remember on the map; if they see or remember
1017     none, they will explore parts of the map unseen since ever or for at least
1018     one turn; if there is nothing to explore, they will simply wait.
1019     """
1020     # # 7DRL add: Don't pick up or search things when inventory is full.
1021     t["T_COMMAND"] = [id for id in world_db["ThingActions"]
1022                       if world_db["ThingActions"][id]["TA_NAME"] == "wait"][0]
1023     if not get_dir_to_target(t, "f"):
1024         sel = get_inventory_slot_to_consume(t)
1025         if -1 != sel:
1026             t["T_COMMAND"] = [id for id in world_db["ThingActions"]
1027                               if world_db["ThingActions"][id]["TA_NAME"]
1028                                  == "use"][0]
1029             t["T_ARGUMENT"] = sel
1030         elif standing_on_consumable(t) \
1031              and (len(t["T_CARRIES"]) < # #
1032                  world_db["ThingTypes"][t["T_TYPE"]]["TT_STORAGE"]): # #
1033             t["T_COMMAND"] = [id for id in world_db["ThingActions"]
1034                               if world_db["ThingActions"][id]["TA_NAME"]
1035                                  == "pick_up"][0]
1036         elif (not
1037               (len(t["T_CARRIES"]) < # #
1038                 world_db["ThingTypes"][t["T_TYPE"]]["TT_STORAGE"] # #
1039                and get_dir_to_target(t, "c"))) and \
1040              (not get_dir_to_target(t, "a")):
1041             get_dir_to_target(t, "s")
1042
1043
1044 def turn_over():
1045     """Run game world and its inhabitants until new player input expected."""
1046     id = 0
1047     whilebreaker = False
1048     while world_db["Things"][0]["T_LIFEPOINTS"]:
1049         for id in [id for id in world_db["Things"]]:  # Only what's from start!
1050             if not id in world_db["Things"] or \
1051                world_db["Things"][id]["carried"]:   # May have been consumed or
1052                 continue                            # picked up during turn …
1053             Thing = world_db["Things"][id]
1054             if Thing["T_LIFEPOINTS"]:
1055                 if not Thing["T_COMMAND"]:
1056                     update_map_memory(Thing)
1057                     if 0 == id:
1058                         whilebreaker = True
1059                         break
1060                     ai(Thing)
1061                 try_healing(Thing)
1062                 Thing["T_PROGRESS"] += 1
1063                 taid = [a for a in world_db["ThingActions"]
1064                           if a == Thing["T_COMMAND"]][0]
1065                 ThingAction = world_db["ThingActions"][taid]
1066                 if Thing["T_PROGRESS"] == ThingAction["TA_EFFORT"]:
1067                     eval("actor_" + ThingAction["TA_NAME"])(Thing)
1068                     Thing["T_COMMAND"] = 0
1069                     Thing["T_PROGRESS"] = 0
1070                 hunger(Thing)
1071             thingproliferation(Thing)
1072         if whilebreaker:
1073             break
1074         world_db["TURN"] += 1
1075
1076
1077 def new_Thing(type, pos=(0, 0)):
1078     """Return Thing of type T_TYPE, with fovmap if alive and world active."""
1079     thing = {
1080         "T_LIFEPOINTS": world_db["ThingTypes"][type]["TT_LIFEPOINTS"],
1081         "T_ARGUMENT": 0,
1082         "T_PROGRESS": 0,
1083         "T_SATIATION": 0,
1084         "T_COMMAND": 0,
1085         "T_TYPE": type,
1086         "T_POSY": pos[0],
1087         "T_POSX": pos[1],
1088         "T_CARRIES": [],
1089         "carried": False,
1090         "T_MEMTHING": [],
1091         "T_MEMMAP": False,
1092         "T_MEMDEPTHMAP": False,
1093         "fovmap": False
1094     }
1095     if world_db["WORLD_ACTIVE"] and thing["T_LIFEPOINTS"]:
1096         build_fov_map(thing)
1097     return thing
1098
1099
1100 def id_setter(id, category, id_store=False, start_at_1=False):
1101     """Set ID of object of category to manipulate ID unused? Create new one.
1102
1103     The ID is stored as id_store.id (if id_store is set). If the integer of the
1104     input is valid (if start_at_1, >= 0, else >= -1), but <0 or (if start_at_1)
1105     <1, calculate new ID: lowest unused ID >=0 or (if start_at_1) >= 1. None is
1106     always returned when no new object is created, otherwise the new object's
1107     ID.
1108     """
1109     min = 0 if start_at_1 else -1
1110     if str == type(id):
1111         id = integer_test(id, min)
1112     if None != id:
1113         if id in world_db[category]:
1114             if id_store:
1115                 id_store.id = id
1116             return None
1117         else:
1118             if (start_at_1 and 0 == id) \
1119                or ((not start_at_1) and (id < 0)):
1120                 id = 0 if start_at_1 else -1
1121                 while 1:
1122                     id = id + 1
1123                     if id not in world_db[category]:
1124                         break
1125             if id_store:
1126                 id_store.id = id
1127     return id
1128
1129
1130 def command_ping():
1131     """Send PONG line to server output file."""
1132     strong_write(io_db["file_out"], "PONG\n")
1133
1134
1135 def command_quit():
1136     """Abort server process."""
1137     save_world()
1138     atomic_write(io_db["path_record"], io_db["record_chunk"], do_append=True)
1139     raise SystemExit("received QUIT command")
1140
1141
1142 def command_thingshere(str_y, str_x):
1143     """Write to out file list of Things known to player at coordinate y, x."""
1144     if world_db["WORLD_ACTIVE"]:
1145         y = integer_test(str_y, 0, 255)
1146         x = integer_test(str_x, 0, 255)
1147         length = world_db["MAP_LENGTH"]
1148         if None != y and None != x and y < length and x < length:
1149             pos = (y * world_db["MAP_LENGTH"]) + x
1150             strong_write(io_db["file_out"], "THINGS_HERE START\n")
1151             if "v" == chr(world_db["Things"][0]["fovmap"][pos]):
1152                 for id in world_db["Things"]:
1153                     if y == world_db["Things"][id]["T_POSY"] \
1154                        and x == world_db["Things"][id]["T_POSX"] \
1155                        and not world_db["Things"][id]["carried"]:
1156                         type = world_db["Things"][id]["T_TYPE"]
1157                         name = world_db["ThingTypes"][type]["TT_NAME"]
1158                         strong_write(io_db["file_out"], name + "\n")
1159             else:
1160                 for mt in world_db["Things"][0]["T_MEMTHING"]:
1161                     if y == mt[1] and x == mt[2]:
1162                         name = world_db["ThingTypes"][mt[0]]["TT_NAME"]
1163                         strong_write(io_db["file_out"], name + "\n")
1164             strong_write(io_db["file_out"], "THINGS_HERE END\n")
1165         else:
1166             print("Ignoring: Invalid map coordinates.")
1167     else:
1168         print("Ignoring: Command only works on existing worlds.")
1169
1170
1171 def play_commander(action, args=False):
1172     """Setter for player's T_COMMAND and T_ARGUMENT, then calling turn_over().
1173
1174     T_ARGUMENT is set to direction char if action=="wait",or 8-bit int if args.
1175     """
1176
1177     def set_command():
1178         id = [x for x in world_db["ThingActions"]
1179                 if world_db["ThingActions"][x]["TA_NAME"] == action][0]
1180         world_db["Things"][0]["T_COMMAND"] = id
1181         turn_over()
1182
1183     def set_command_and_argument_int(str_arg):
1184         val = integer_test(str_arg, 0, 255)
1185         if None != val:
1186             world_db["Things"][0]["T_ARGUMENT"] = val
1187             set_command()
1188
1189     def set_command_and_argument_movestring(str_arg):
1190         if str_arg in directions_db:
1191             world_db["Things"][0]["T_ARGUMENT"] = ord(directions_db[str_arg])
1192             set_command()
1193         else:
1194             print("Ignoring: Argument must be valid direction string.")
1195
1196     if action == "move":
1197         return set_command_and_argument_movestring
1198     elif args:
1199         return set_command_and_argument_int
1200     else:
1201         return set_command
1202
1203
1204 def command_seedrandomness(seed_string):
1205     """Set rand seed to int(seed_string)."""
1206     val = integer_test(seed_string, 0, 4294967295)
1207     if None != val:
1208         rand.seed = val
1209
1210
1211 def command_seedmap(seed_string):
1212     """Set world_db["SEED_MAP"] to int(seed_string), then (re-)make map."""
1213     setter(None, "SEED_MAP", 0, 4294967295)(seed_string)
1214     remake_map()
1215
1216
1217 def command_makeworld(seed_string):
1218     """(Re-)build game world, i.e. map, things, to a new turn 1 from seed.
1219
1220     Seed rand with seed, fill it into world_db["SEED_MAP"]. Do more only with a
1221     "wait" ThingAction and world["PLAYER_TYPE"] matching ThingType of
1222     TT_START_NUMBER > 0. Then, world_db["Things"] emptied, call remake_map()
1223     and set world_db["WORLD_ACTIVE"], world_db["TURN"] to 1. Build new Things
1224     according to ThingTypes' TT_START_NUMBERS, with Thing of ID 0 to ThingType
1225     of ID = world["PLAYER_TYPE"]. Place Things randomly, and actors not on each
1226     other. Init player's memory map. Write "NEW_WORLD" line to out file.
1227     """
1228
1229     def free_pos():
1230         i = 0
1231         while 1:
1232             err = "Space to put thing on too hard to find. Map too small?"
1233             while 1:
1234                 y = rand.next() % world_db["MAP_LENGTH"]
1235                 x = rand.next() % world_db["MAP_LENGTH"]
1236                 if "." == chr(world_db["MAP"][y * world_db["MAP_LENGTH"] + x]):
1237                     break
1238                 i += 1
1239                 if i == 65535:
1240                     raise SystemExit(err)
1241             # Replica of C code, wrongly ignores animatedness of new Thing.
1242             pos_clear = (0 == len([id for id in world_db["Things"]
1243                                    if world_db["Things"][id]["T_LIFEPOINTS"]
1244                                    if world_db["Things"][id]["T_POSY"] == y
1245                                    if world_db["Things"][id]["T_POSX"] == x]))
1246             if pos_clear:
1247                 break
1248         return (y, x)
1249
1250     val = integer_test(seed_string, 0, 4294967295)
1251     if None == val:
1252         return
1253     rand.seed = val
1254     world_db["SEED_MAP"] = val
1255     player_will_be_generated = False
1256     playertype = world_db["PLAYER_TYPE"]
1257     for ThingType in world_db["ThingTypes"]:
1258         if playertype == ThingType:
1259             if 0 < world_db["ThingTypes"][ThingType]["TT_START_NUMBER"]:
1260                 player_will_be_generated = True
1261             break
1262     if not player_will_be_generated:
1263         print("Ignoring beyond SEED_MAP: " +
1264               "No player type with start number >0 defined.")
1265         return
1266     wait_action = False
1267     for ThingAction in world_db["ThingActions"]:
1268         if "wait" == world_db["ThingActions"][ThingAction]["TA_NAME"]:
1269             wait_action = True
1270     if not wait_action:
1271         print("Ignoring beyond SEED_MAP: " +
1272               "No thing action with name 'wait' defined.")
1273         return
1274     world_db["Things"] = {}
1275     remake_map()
1276     world_db["WORLD_ACTIVE"] = 1
1277     world_db["TURN"] = 1
1278     for i in range(world_db["ThingTypes"][playertype]["TT_START_NUMBER"]):
1279         id = id_setter(-1, "Things")
1280         world_db["Things"][id] = new_Thing(playertype, free_pos())
1281     update_map_memory(world_db["Things"][0])
1282     for type in world_db["ThingTypes"]:
1283         for i in range(world_db["ThingTypes"][type]["TT_START_NUMBER"]):
1284             if type != playertype:
1285                 id = id_setter(-1, "Things")
1286                 world_db["Things"][id] = new_Thing(type, free_pos())
1287     strong_write(io_db["file_out"], "NEW_WORLD\n")
1288
1289
1290 def command_maplength(maplength_string):
1291     """Redefine map length. Invalidate map, therefore lose all things on it."""
1292     val = integer_test(maplength_string, 1, 256)
1293     if None != val:
1294         world_db["MAP_LENGTH"] = val
1295         set_world_inactive()
1296         world_db["Things"] = {}
1297         libpr.set_maplength(val)
1298
1299
1300 def command_worldactive(worldactive_string):
1301     """Toggle world_db["WORLD_ACTIVE"] if possible.
1302
1303     An active world can always be set inactive. An inactive world can only be
1304     set active with a "wait" ThingAction, and a player Thing (of ID 0). On
1305     activation, rebuild all Things' FOVs, and the player's map memory.
1306     """
1307     # In original version, map existence was also tested (unnecessarily?).
1308     val = integer_test(worldactive_string, 0, 1)
1309     if val:
1310         if 0 != world_db["WORLD_ACTIVE"]:
1311             if 0 == val:
1312                 set_world_inactive()
1313             else:
1314                 print("World already active.")
1315         elif 0 == world_db["WORLD_ACTIVE"]:
1316             wait_exists = False
1317             for ThingAction in world_db["ThingActions"]:
1318                 if "wait" == world_db["ThingActions"][ThingAction]["TA_NAME"]:
1319                     wait_exists = True
1320                     break
1321             player_exists = False
1322             for Thing in world_db["Things"]:
1323                 if 0 == Thing:
1324                     player_exists = True
1325                     break
1326             if wait_exists and player_exists:
1327                 for id in world_db["Things"]:
1328                     if world_db["Things"][id]["T_LIFEPOINTS"]:
1329                         build_fov_map(world_db["Things"][id])
1330                         if 0 == id:
1331                             update_map_memory(world_db["Things"][id], False)
1332                 world_db["WORLD_ACTIVE"] = 1
1333
1334
1335 def test_for_id_maker(object, category):
1336     """Return decorator testing for object having "id" attribute."""
1337     def decorator(f):
1338         def helper(*args):
1339             if hasattr(object, "id"):
1340                 f(*args)
1341             else:
1342                 print("Ignoring: No " + category +
1343                       " defined to manipulate yet.")
1344         return helper
1345     return decorator
1346
1347
1348 def command_tid(id_string):
1349     """Set ID of Thing to manipulate. ID unused? Create new one.
1350
1351     Default new Thing's type to the first available ThingType, others: zero.
1352     """
1353     id = id_setter(id_string, "Things", command_tid)
1354     if None != id:
1355         if world_db["ThingTypes"] == {}:
1356             print("Ignoring: No ThingType to settle new Thing in.")
1357             return
1358         type = list(world_db["ThingTypes"].keys())[0]
1359         world_db["Things"][id] = new_Thing(type)
1360
1361
1362 test_Thing_id = test_for_id_maker(command_tid, "Thing")
1363
1364
1365 @test_Thing_id
1366 def command_tcommand(str_int):
1367     """Set T_COMMAND of selected Thing."""
1368     val = integer_test(str_int, 0)
1369     if None != val:
1370         if 0 == val or val in world_db["ThingActions"]:
1371             world_db["Things"][command_tid.id]["T_COMMAND"] = val
1372         else:
1373             print("Ignoring: ThingAction ID belongs to no known ThingAction.")
1374
1375
1376 @test_Thing_id
1377 def command_ttype(str_int):
1378     """Set T_TYPE of selected Thing."""
1379     val = integer_test(str_int, 0)
1380     if None != val:
1381         if val in world_db["ThingTypes"]:
1382             world_db["Things"][command_tid.id]["T_TYPE"] = val
1383         else:
1384             print("Ignoring: ThingType ID belongs to no known ThingType.")
1385
1386
1387 @test_Thing_id
1388 def command_tcarries(str_int):
1389     """Append int(str_int) to T_CARRIES of selected Thing.
1390
1391     The ID int(str_int) must not be of the selected Thing, and must belong to a
1392     Thing with unset "carried" flag. Its "carried" flag will be set on owning.
1393     """
1394     val = integer_test(str_int, 0)
1395     if None != val:
1396         if val == command_tid.id:
1397             print("Ignoring: Thing cannot carry itself.")
1398         elif val in world_db["Things"] \
1399              and not world_db["Things"][val]["carried"]:
1400             world_db["Things"][command_tid.id]["T_CARRIES"].append(val)
1401             world_db["Things"][val]["carried"] = True
1402         else:
1403             print("Ignoring: Thing not available for carrying.")
1404     # Note that the whole carrying structure is different from the C version:
1405     # Carried-ness is marked by a "carried" flag, not by Things containing
1406     # Things internally.
1407
1408
1409 @test_Thing_id
1410 def command_tmemthing(str_t, str_y, str_x):
1411     """Add (int(str_t), int(str_y), int(str_x)) to selected Thing's T_MEMTHING.
1412
1413     The type must fit to an existing ThingType, and the position into the map.
1414     """
1415     type = integer_test(str_t, 0)
1416     posy = integer_test(str_y, 0, 255)
1417     posx = integer_test(str_x, 0, 255)
1418     if None != type and None != posy and None != posx:
1419         if type not in world_db["ThingTypes"] \
1420            or posy >= world_db["MAP_LENGTH"] or posx >= world_db["MAP_LENGTH"]:
1421             print("Ignoring: Illegal value for thing type or position.")
1422         else:
1423             memthing = (type, posy, posx)
1424             world_db["Things"][command_tid.id]["T_MEMTHING"].append(memthing)
1425
1426
1427 def setter_map(maptype):
1428     """Set selected Thing's map of maptype's int(str_int)-th line to mapline.
1429
1430     If Thing has no map of maptype yet, initialize it with ' ' bytes first.
1431     """
1432     @test_Thing_id
1433     def helper(str_int, mapline):
1434         val = integer_test(str_int, 0, 255)
1435         if None != val:
1436             if val >= world_db["MAP_LENGTH"]:
1437                 print("Illegal value for map line number.")
1438             elif len(mapline) != world_db["MAP_LENGTH"]:
1439                 print("Map line length is unequal map width.")
1440             else:
1441                 length = world_db["MAP_LENGTH"]
1442                 map = None
1443                 if not world_db["Things"][command_tid.id][maptype]:
1444                     map = bytearray(b' ' * (length ** 2))
1445                 else:
1446                     map = world_db["Things"][command_tid.id][maptype]
1447                 map[val * length:(val * length) + length] = mapline.encode()
1448                 world_db["Things"][command_tid.id][maptype] = map
1449     return helper
1450
1451
1452 def setter_tpos(axis):
1453     """Generate setter for T_POSX or  T_POSY of selected Thing.
1454
1455     If world is active, rebuilds animate things' fovmap, player's memory map.
1456     """
1457     @test_Thing_id
1458     def helper(str_int):
1459         val = integer_test(str_int, 0, 255)
1460         if None != val:
1461             if val < world_db["MAP_LENGTH"]:
1462                 world_db["Things"][command_tid.id]["T_POS" + axis] = val
1463                 if world_db["WORLD_ACTIVE"] \
1464                    and world_db["Things"][command_tid.id]["T_LIFEPOINTS"]:
1465                     build_fov_map(world_db["Things"][command_tid.id])
1466                     if 0 == command_tid.id:
1467                         update_map_memory(world_db["Things"][command_tid.id])
1468             else:
1469                 print("Ignoring: Position is outside of map.")
1470     return helper
1471
1472
1473 def command_ttid(id_string):
1474     """Set ID of ThingType to manipulate. ID unused? Create new one.
1475
1476     Default new ThingType's TT_SYMBOL to "?", TT_CORPSE_ID to self, others: 0.
1477     """
1478     id = id_setter(id_string, "ThingTypes", command_ttid)
1479     if None != id:
1480         world_db["ThingTypes"][id] = {
1481             "TT_NAME": "(none)",
1482             "TT_CONSUMABLE": 0,
1483             "TT_LIFEPOINTS": 0,
1484             "TT_PROLIFERATE": 0,
1485             "TT_START_NUMBER": 0,
1486             "TT_STORAGE": 0, # #
1487             "TT_SYMBOL": "?",
1488             "TT_CORPSE_ID": id
1489         }
1490
1491
1492 test_ThingType_id = test_for_id_maker(command_ttid, "ThingType")
1493
1494
1495 @test_ThingType_id
1496 def command_ttname(name):
1497     """Set TT_NAME of selected ThingType."""
1498     world_db["ThingTypes"][command_ttid.id]["TT_NAME"] = name
1499
1500
1501 @test_ThingType_id
1502 def command_ttsymbol(char):
1503     """Set TT_SYMBOL of selected ThingType. """
1504     if 1 == len(char):
1505         world_db["ThingTypes"][command_ttid.id]["TT_SYMBOL"] = char
1506     else:
1507         print("Ignoring: Argument must be single character.")
1508
1509
1510 @test_ThingType_id
1511 def command_ttcorpseid(str_int):
1512     """Set TT_CORPSE_ID of selected ThingType."""
1513     val = integer_test(str_int, 0)
1514     if None != val:
1515         if val in world_db["ThingTypes"]:
1516             world_db["ThingTypes"][command_ttid.id]["TT_CORPSE_ID"] = val
1517         else:
1518             print("Ignoring: Corpse ID belongs to no known ThignType.")
1519
1520
1521 def command_taid(id_string):
1522     """Set ID of ThingAction to manipulate. ID unused? Create new one.
1523
1524     Default new ThingAction's TA_EFFORT to 1, its TA_NAME to "wait".
1525     """
1526     id = id_setter(id_string, "ThingActions", command_taid, True)
1527     if None != id:
1528         world_db["ThingActions"][id] = {
1529             "TA_EFFORT": 1,
1530             "TA_NAME": "wait"
1531         }
1532
1533
1534 test_ThingAction_id = test_for_id_maker(command_taid, "ThingAction")
1535
1536
1537 @test_ThingAction_id
1538 def command_taname(name):
1539     """Set TA_NAME of selected ThingAction.
1540
1541     The name must match a valid thing action function. If after the name
1542     setting no ThingAction with name "wait" remains, call set_world_inactive().
1543     """
1544     if name == "wait" or name == "move" or name == "use" or name == "drop" \
1545        or name == "pick_up":
1546         world_db["ThingActions"][command_taid.id]["TA_NAME"] = name
1547         if 1 == world_db["WORLD_ACTIVE"]:
1548             wait_defined = False
1549             for id in world_db["ThingActions"]:
1550                 if "wait" == world_db["ThingActions"][id]["TA_NAME"]:
1551                     wait_defined = True
1552                     break
1553             if not wait_defined:
1554                 set_world_inactive()
1555     else:
1556         print("Ignoring: Invalid action name.")
1557     # In contrast to the original,naming won't map a function to a ThingAction.
1558
1559
1560 def command_ai():
1561     """Call ai() on player Thing, then turn_over()."""
1562     ai(world_db["Things"][0])
1563     turn_over()
1564
1565
1566 """Commands database.
1567
1568 Map command start tokens to ([0]) number of expected command arguments, ([1])
1569 the command's meta-ness (i.e. is it to be written to the record file, is it to
1570 be ignored in replay mode if read from server input file), and ([2]) a function
1571 to be called on it.
1572 """
1573 commands_db = {
1574     "QUIT": (0, True, command_quit),
1575     "PING": (0, True, command_ping),
1576     "THINGS_HERE": (2, True, command_thingshere),
1577     "MAKE_WORLD": (1, False, command_makeworld),
1578     "SEED_MAP": (1, False, command_seedmap),
1579     "SEED_RANDOMNESS": (1, False, command_seedrandomness),
1580     "TURN": (1, False, setter(None, "TURN", 0, 65535)),
1581     "GOD_MOOD": (1, False, setter(None, "GOD_MOOD", -32768, 32767)),  # #
1582     "GOD_FAVOR": (1, False, setter(None, "GOD_FAVOR", -32768, 32767)),  # #
1583     "PLAYER_TYPE": (1, False, setter(None, "PLAYER_TYPE", 0)),
1584     "MAP_LENGTH": (1, False, command_maplength),
1585     "WORLD_ACTIVE": (1, False, command_worldactive),
1586     "TA_ID": (1, False, command_taid),
1587     "TA_EFFORT": (1, False, setter("ThingAction", "TA_EFFORT", 0, 255)),
1588     "TA_NAME": (1, False, command_taname),
1589     "TT_ID": (1, False, command_ttid),
1590     "TT_NAME": (1, False, command_ttname),
1591     "TT_SYMBOL": (1, False, command_ttsymbol),
1592     "TT_CORPSE_ID": (1, False, command_ttcorpseid),
1593     "TT_CONSUMABLE": (1, False, setter("ThingType", "TT_CONSUMABLE",
1594                                        0, 65535)),
1595     "TT_START_NUMBER": (1, False, setter("ThingType", "TT_START_NUMBER",
1596                                          0, 255)),
1597     "TT_PROLIFERATE": (1, False, setter("ThingType", "TT_PROLIFERATE",
1598                                         0, 255)),
1599     "TT_LIFEPOINTS": (1, False, setter("ThingType", "TT_LIFEPOINTS", 0, 255)),
1600     "TT_STORAGE": (1, False, setter("ThingType", "TT_STORAGE", 0, 255)), # #
1601     "T_ID": (1, False, command_tid),
1602     "T_ARGUMENT": (1, False, setter("Thing", "T_ARGUMENT", 0, 255)),
1603     "T_PROGRESS": (1, False, setter("Thing", "T_PROGRESS", 0, 255)),
1604     "T_LIFEPOINTS": (1, False, setter("Thing", "T_LIFEPOINTS", 0, 255)),
1605     "T_SATIATION": (1, False, setter("Thing", "T_SATIATION", -32768, 32767)),
1606     "T_COMMAND": (1, False, command_tcommand),
1607     "T_TYPE": (1, False, command_ttype),
1608     "T_CARRIES": (1, False, command_tcarries),
1609     "T_MEMMAP": (2, False, setter_map("T_MEMMAP")),
1610     "T_MEMDEPTHMAP": (2, False, setter_map("T_MEMDEPTHMAP")),
1611     "T_MEMTHING": (3, False, command_tmemthing),
1612     "T_POSY": (1, False, setter_tpos("Y")),
1613     "T_POSX": (1, False, setter_tpos("X")),
1614     "wait": (0, False, play_commander("wait")),
1615     "move": (1, False, play_commander("move")),
1616     "pick_up": (0, False, play_commander("pick_up")),
1617     "drop": (1, False, play_commander("drop", True)),
1618     "use": (1, False, play_commander("use", True)),
1619     "ai": (0, False, command_ai)
1620 }
1621
1622
1623 """World state database. With sane default values. (Randomness is in rand.)"""
1624 world_db = {
1625     "TURN": 0,
1626     "MAP_LENGTH": 64,
1627     "SEED_MAP": 0,
1628     "PLAYER_TYPE": 0,
1629     "WORLD_ACTIVE": 0,
1630     "GOD_MOOD": 0,  # #
1631     "GOD_FAVOR": 0,  # #
1632     "ThingActions": {},
1633     "ThingTypes": {},
1634     "Things": {}
1635 }
1636
1637 """Mapping of direction names to internal direction chars."""
1638 directions_db = {"east": "d", "south-east": "c", "south-west": "x",
1639                  "west": "s", "north-west": "w", "north-east": "e"}
1640
1641 """File IO database."""
1642 io_db = {
1643     "path_save": "save",
1644     "path_record": "record_save",
1645     "path_worldconf": "confserver/world",
1646     "path_server": "server/",
1647     "path_in": "server/in",
1648     "path_out": "server/out",
1649     "path_worldstate": "server/worldstate",
1650     "tmp_suffix": "_tmp",
1651     "kicked_by_rival": False,
1652     "worldstate_updateable": False
1653 }
1654
1655
1656 try:
1657     libpr = prep_library()
1658     rand = RandomnessIO()
1659     opts = parse_command_line_arguments()
1660     if opts.savefile:
1661         io_db["path_save"] = opts.savefile
1662         io_db["path_record"] = "record_" + opts.savefile
1663     setup_server_io()
1664     if opts.verbose:
1665         io_db["verbose"] = True
1666     if None != opts.replay:
1667         replay_game()
1668     else:
1669         play_game()
1670 except SystemExit as exit:
1671     print("ABORTING: " + exit.args[0])
1672 except:
1673     print("SOMETHING WENT WRONG IN UNEXPECTED WAYS")
1674     raise
1675 finally:
1676     cleanup_server_io()