home · contact · privacy
Server: Fix AI bug (searching too costly consumable when not player).
[plomrogue] / server / utils.py
1 import ctypes
2
3
4 def mv_yx_in_dir_legal(dir, y, x):
5     """Wrapper around libpr.mv_yx_in_dir_legal to simplify its use."""
6     dir_c = dir.encode("ascii")[0]
7     test = libpr.mv_yx_in_dir_legal_wrap(dir_c, y, x)
8     if -1 == test:
9         raise RuntimeError("Too much wrapping in mv_yx_in_dir_legal_wrap()!")
10     return (test, libpr.result_y(), libpr.result_x())
11
12
13 class RandomnessIO:
14     """"Interface to libplomrogue's pseudo-randomness generator."""
15
16     def set_seed(self, seed):
17         libpr.seed_rrand(1, seed)
18
19     def get_seed(self):
20         return libpr.seed_rrand(0, 0)
21
22     def next(self):
23         return libpr.rrand()
24
25     seed = property(get_seed, set_seed)
26
27
28 def integer_test(val_string, min, max=None):
29     """Return val_string if integer >= min & (if max set) <= max, else None."""
30     try:
31         val = int(val_string)
32         if val < min or (max is not None and val > max):
33             raise ValueError
34         return val
35     except ValueError:
36         msg = "Ignoring: Please use integer >= " + str(min)
37         if max is not None:
38             msg += " and <= " + str(max)
39         msg += "."
40         print(msg)
41         return None
42
43
44 def id_setter(id, category, id_store=False, start_at_1=False):
45     """Set ID of object of category to manipulate. ID unused? Create new one.
46
47     The ID is stored as id_store.id (if id_store is set). If the integer of the
48     input is valid (if start_at_1, >= 0, else >= -1), but <0 or (if start_at_1)
49     <1, calculate new ID: lowest unused ID >=0 or (if start_at_1) >= 1. None is
50     always returned when no new object is created, else the new object's ID.
51     """
52     from server.config.world_data import world_db
53     min = 0 if start_at_1 else -1
54     if str == type(id):
55         id = integer_test(id, min)
56     if None != id:
57         if id in world_db[category]:
58             if id_store:
59                 id_store.id = id
60             return None
61         else:
62             if (start_at_1 and 0 == id) \
63                or ((not start_at_1) and (id < 0)):
64                 id = 0 if start_at_1 else -1
65                 while 1:
66                     id = id + 1
67                     if id not in world_db[category]:
68                         break
69             if id_store:
70                 id_store.id = id
71     return id
72
73
74 def prep_library():
75     """Prepare ctypes library at ./libplomrogue.so"""
76     import os
77     libpath = ("./libplomrogue.so")
78     if not os.access(libpath, os.F_OK):
79         raise SystemExit("No library " + libpath + ", run ./build.sh first?")
80     libpr = ctypes.cdll.LoadLibrary(libpath)
81     libpr.seed_rrand.restype = ctypes.c_uint32
82     return libpr
83
84
85 def c_pointer_to_bytearray(ba):
86     """Return C char * pointer to ba."""
87     type = ctypes.c_char * len(ba)
88     return type.from_buffer(ba)
89
90
91 def parse_command_line_arguments():
92     """Return settings values read from command line arguments."""
93     import argparse
94     parser = argparse.ArgumentParser()
95     parser.add_argument('-s', nargs='?', type=int, dest='replay', const=1,
96                         action='store')
97     parser.add_argument('-l', nargs="?", const="save", dest='savefile',
98                         action="store")
99     parser.add_argument('-w', type=str, default="confserver/world",
100                         dest='worldconf', action='store')
101     parser.add_argument('-v', dest='verbose', action='store_true')
102     opts, unknown = parser.parse_known_args()
103     return opts
104
105
106 libpr = prep_library()
107 rand = RandomnessIO()
108 opts = parse_command_line_arguments()