import curses
+import math
import os
import signal
import time
def draw_screen():
+ def healthy_addch(y, x, char, attr=0):
+ """Workaround for <http://stackoverflow.com/questions/7063128/>."""
+ if y == screen_size[0] - 1 and x == screen_size[1] - 1:
+ char_before = stdscr.inch(y, x - 1)
+ stdscr.addch(y, x - 1, char, attr)
+ stdscr.insstr(y, x - 1, " ")
+ stdscr.addch(y, x - 1, char_before)
+ else:
+ stdscr.addch(y, x, char, attr)
+
def draw_window_border_lines():
for win in windows:
for k in range(2):
end = win["start"][k] + win["size"][k]
end = end if end < screen_size[k] else screen_size[k]
if k:
- [stdscr.addch(j, i, '-') for i in range(start, end)]
+ [healthy_addch(j, i, '-') for i in range(start, end)]
else:
- [stdscr.addch(i, j, '|') for i in range(start, end)]
+ [healthy_addch(i, j, '|') for i in range(start, end)]
def draw_window_border_corners():
for win in windows:
right = win["start"][1] + win["size"][1]
if (up >= 0 and up < screen_size[0]):
if (left >= 0 and left < screen_size[1]):
- stdscr.addch(up, left, '+')
+ healthy_addch(up, left, '+')
if (right >= 0 and right < screen_size[1]):
- stdscr.addch(up, right, '+')
+ healthy_addch(up, right, '+')
if (down >= 0 and down < screen_size[0]):
if (left >= 0 and left < screen_size[1]):
- stdscr.addch(down, left, '+')
+ healthy_addch(down, left, '+')
if (right >= 0 and right < screen_size[1]):
- stdscr.addch(down, right, '+')
+ healthy_addch(down, right, '+')
def draw_window_contents():
def draw_winmap():
stop = [0, 0]
for i in range(2):
stop[i] = win["size"][i] + offset[i]
- stop[i] = stop[i] if stop[i] < size[i] else size[i]
+ if stop[i] >= winmap_size[i]:
+ stop[i] = winmap_size[i]
for y in range(offset[0], stop[0]):
for x in range(offset[1], stop[1]):
- cell = winmap[y * size[1] + x]
+ cell = winmap[y * winmap_size[1] + x]
+ attr = 0
+ if len(cell) > 1:
+ attr = cell[1]
+ cell = cell[0]
y_in_screen = win["start"][0] + (y - offset[0])
x_in_screen = win["start"][1] + (x - offset[1])
if (y_in_screen < screen_size[0]
and x_in_screen < screen_size[1]):
- stdscr.addch(y_in_screen, x_in_screen, cell)
+ healthy_addch(y_in_screen, x_in_screen, cell, attr)
def draw_scroll_hints():
def draw_scroll_string(n_lines_outside):
hint = ' ' + str(n_lines_outside + 1) + ' more ' + unit + ' '
for j in range(win["size"][ni] - non_hint_space):
pos_2 = win["start"][ni] + hint_offset + j
x, y = (pos_2, pos_1) if ni else (pos_1, pos_2)
- stdscr.addch(y, x, hint[j], curses.A_REVERSE)
+ healthy_addch(y, x, hint[j], curses.A_REVERSE)
def draw_scroll_arrows(ar1, ar2):
for j in range(win["size"][ni]):
pos_2 = win["start"][ni] + j
x, y = (pos_2, pos_1) if ni else (pos_1, pos_2)
- stdscr.addch(y, x, ar1 if ni else ar2, curses.A_REVERSE)
+ healthy_addch(y, x, ar1 if ni else ar2, curses.A_REVERSE)
for i in range(2):
ni = int(i == 0)
unit = 'rows' if ni else 'columns'
pos_1 = win["start"][i]
draw_scroll_arrows('^', '<')
draw_scroll_string(offset[i])
- if (size[i] > offset[i] + win["size"][i]):
+ if (winmap_size[i] > offset[i] + win["size"][i]):
pos_1 = win["start"][i] + win["size"][i] - 1
draw_scroll_arrows('v', '>')
- draw_scroll_string(size[i] - offset[i] - win["size"][i])
+ draw_scroll_string(winmap_size[i] - offset[i]
+ - win["size"][i])
for win in windows:
- offset, size, winmap = win["func"]()
+ offset, winmap_size, winmap = win["func"]()
draw_winmap()
draw_scroll_hints()
- stdscr.clear()
+ stdscr.erase()
draw_window_border_lines()
draw_window_border_corners()
draw_window_contents()
stdscr.refresh()
-def ping_test(server_answered):
- half_wait_time = 5
- if server_answered:
- ping_test.sent = False
- elif ping_test.wait_start + half_wait_time < time.time():
- if not ping_test.sent:
- io["file_out"].write("PING\n")
- io["file_out"].flush()
- ping_test.sent = True
- ping_test.wait_start = time.time()
- elif ping_test.sent:
- raise SystemExit("Server not answering anymore.")
-ping_test.wait_start = 0
-
-
-def read_into_message_queue(string):
- if string == "":
- return
- new_open_end = False
- if string[-1] is not "\n":
- new_open_end = True
- new_messages = string.splitlines()
- if message_queue["open_end"]:
- message_queue["messages"][-1] = message_queue["messages"][-1] + new_messages[0]
- del new_messages[0]
- message_queue["messages"] = message_queue["messages"] + new_messages
- if new_open_end:
- message_queue["open_end"] = True
+def read_worldstate():
+ if not os.access(io["path_worldstate"], os.F_OK):
+ msg = "No world state file found at " + io["path_worldstate"] + "."
+ raise SystemExit(msg)
+ read_anew = False
+ worldstate_file = open(io["path_worldstate"], "r")
+ turn_string = worldstate_file.readline()
+ if int(turn_string) != world_data["turn"]:
+ read_anew = True
+ if not read_anew: # In rare cases, world may change, but not turn number.
+ mtime = os.stat(io["path_worldstate"])
+ if mtime != read_worldstate.last_checked_mtime:
+ read_worldstate.last_checked_mtime = mtime
+ read_anew = True
+ if read_anew:
+ cursed_main.redraw = True
+ world_data["turn"] = int(turn_string)
+ world_data["lifepoints"] = int(worldstate_file.readline())
+ world_data["satiation"] = int(worldstate_file.readline())
+ world_data["inventory"] = []
+ while True:
+ line = worldstate_file.readline().replace("\n", "")
+ if line == '%':
+ break
+ world_data["inventory"] += [line]
+ world_data["avatar_position"][0] = int(worldstate_file.readline())
+ world_data["avatar_position"][1] = int(worldstate_file.readline())
+ if not world_data["look_mode"]:
+ world_data["map_center"][0] = world_data["avatar_position"][0]
+ world_data["map_center"][1] = world_data["avatar_position"][1]
+ world_data["map_size"] = int(worldstate_file.readline())
+ world_data["fov_map"] = ""
+ for i in range(world_data["map_size"]):
+ line = worldstate_file.readline().replace("\n", "")
+ world_data["fov_map"] += line
+ world_data["mem_map"] = ""
+ for i in range(world_data["map_size"]):
+ line = worldstate_file.readline().replace("\n", "")
+ world_data["mem_map"] += line
+ worldstate_file.close()
+read_worldstate.last_checked_mtime = -1
+
+
+def read_message_queue():
+ while (len(message_queue["messages"]) > 1
+ or (len(message_queue["messages"]) == 0
+ and not message_queue["open_end"])):
+ message = message_queue["messages"].pop(0)
+ if message[0:4] == "LOG ":
+ world_data["log"] += [message[4:]]
+ cursed_main.redraw = True
def cursed_main(stdscr):
+
+ def ping_test():
+ half_wait_time = 5
+ if len(new_data_from_server) > 0:
+ ping_test.sent = False
+ elif ping_test.wait_start + half_wait_time < time.time():
+ if not ping_test.sent:
+ io["file_out"].write("PING\n")
+ io["file_out"].flush()
+ ping_test.sent = True
+ ping_test.wait_start = time.time()
+ elif ping_test.sent:
+ raise SystemExit("Server not answering anymore.")
+ ping_test.wait_start = 0
+
+ def read_into_message_queue():
+ if new_data_from_server == "":
+ return
+ new_open_end = False
+ if new_data_from_server[-1] is not "\n":
+ new_open_end = True
+ new_messages = new_data_from_server.splitlines()
+ if message_queue["open_end"]:
+ message_queue["messages"][-1] += new_messages[0]
+ del new_messages[0]
+ message_queue["messages"] += new_messages
+ if new_open_end:
+ message_queue["open_end"] = True
+
curses.noecho()
curses.curs_set(False)
# stdscr.keypad(True)
draw_screen()
cursed_main.redraw = False
char = stdscr.getch()
- if (char >= 0):
- if chr(char) in commands:
- commands[chr(char)]()
- cursed_main.redraw = True
+ if char >= 0:
+ char = chr(char)
+ if char in commands:
+ if len(commands[char]) == 1 or not world_data["look_mode"]:
+ commands[char][0]()
+ cursed_main.redraw = True
+ else:
+ commands[char][1]()
+ cursed_main.redraw = True
new_data_from_server = io["file_in"].read()
- ping_test(len(new_data_from_server) > 0)
- read_into_message_queue(new_data_from_server)
+ ping_test()
+ read_into_message_queue()
+ read_worldstate()
+ read_message_queue()
+
+
+def win_foo():
+ winmap = [('.', 0), ('o', 0), ('.', 0), ('o', 0), ('O', 0), ('o', 0),
+ ('.', 0), ('o', 0), ('.', 0), ('x', 0), ('y', 0), ('x', 0)]
+ winmap_size = [4, 3]
+ offset = [0, 0]
+ return offset, winmap_size, winmap
+
+
+def win_map():
+ win_size = next(win["size"] for win in windows if win["func"] == win_map)
+ offset = [0, 0]
+ for i in range(2):
+ if world_data["map_center"][i] * (i + 1) > win_size[i] / 2:
+ if world_data["map_center"][i] * (i + 1) \
+ < world_data["map_size"] * (i + 1) - win_size[i] / 2:
+ offset[i] = world_data["map_center"][i] * (i + 1) \
+ - int(win_size[i] / 2)
+ else:
+ offset[i] = world_data["map_size"] * (i + 1) - win_size[i] + i
+ winmap_size = [world_data["map_size"], world_data["map_size"] * 2 + 1]
+ winmap = []
+ curses.init_pair(1, curses.COLOR_WHITE, curses.COLOR_BLUE)
+ curses.init_pair(2, curses.COLOR_BLUE, curses.COLOR_BLACK)
+ for y in range(world_data["map_size"]):
+ for x in range(world_data["map_size"]):
+ char = world_data["fov_map"][y * world_data["map_size"] + x]
+ if world_data["look_mode"] and y == world_data["map_center"][0] \
+ and x == world_data["map_center"][1]:
+ if char == " ":
+ char = \
+ world_data["mem_map"][y * world_data["map_size"] + x]
+ winmap += [(char, curses.A_REVERSE), (" ", curses.A_REVERSE)]
+ continue
+ if char == " ":
+ char = world_data["mem_map"][y * world_data["map_size"] + x]
+ attribute = curses.color_pair(1) if char == " " \
+ else curses.color_pair(2)
+ winmap += [(char, attribute), (" ", attribute)]
+ else:
+ winmap += char + " "
+ if y % 2 == 0:
+ winmap += " "
+ return offset, winmap_size, winmap
+
+
+def win_inventory():
+ winmap = ""
+ winmap_size = [0, 0]
+ for line in world_data["inventory"]:
+ winmap_size[1] = winmap_size[1] if len(line) <= winmap_size[1] \
+ else len(line)
+ for line in world_data["inventory"]:
+ padding_size = winmap_size[1] - len(line)
+ winmap += line + (" " * padding_size)
+ winmap_size[0] = winmap_size[0] + 1
+ offset = [0, 0]
+ return offset, winmap_size, winmap
+
+
+def win_info():
+ winmap = "T: " + str(world_data["turn"]) \
+ + " H: " + str(world_data["lifepoints"]) \
+ + " S: " + str(world_data["satiation"])
+ winmap_size = [1, len(winmap)]
+ offset = [0, 0]
+ return offset, winmap_size, winmap
-def foo():
- winmap = ['.', 'o', '.', 'o', 'O', 'o', '.', 'o', '.', 'x', 'y', 'x']
- size = [4, 3]
+def win_log():
+ win_size = next(win["size"] for win in windows if win["func"] == win_log)
offset = [0, 0]
- return offset, size, winmap
+ winmap = ""
+ number_of_lines = 0
+ for line in world_data["log"]:
+ number_of_lines += math.ceil(len(line) / win_size[1])
+ padding_size = win_size[1] - (len(line) % win_size[1])
+ winmap += line + (padding_size * " ")
+ if number_of_lines < win_size[0]:
+ winmap = (" " * win_size[1] * (win_size[0] - number_of_lines)) + winmap
+ number_of_lines = win_size[0]
+ elif number_of_lines > win_size[0]:
+ offset[0] = number_of_lines - win_size[0]
+ winmap_size = [number_of_lines, win_size[1]]
+ return offset, winmap_size, winmap
def command_quit():
- io["file_out"].write("QUIT\n")
- io["file_out"].flush()
+ command_sender("QUIT")()
raise SystemExit("Received QUIT command, forwarded to server, leaving.")
+def command_toggle_look_mode():
+ if not world_data["look_mode"]:
+ world_data["look_mode"] = True
+ else:
+ world_data["look_mode"] = False
+ world_data["map_center"] = world_data["avatar_position"]
+
+
+def command_sender(string):
+ def command_send():
+ io["file_out"].write(string + "\n")
+ io["file_out"].flush()
+ return command_send
+
+
+def command_looker(string):
+ def command_look():
+ if string == "west" \
+ and world_data["map_center"][1] > 0:
+ world_data["map_center"][1] -= 1
+ elif string == "east" \
+ and world_data["map_center"][1] < world_data["map_size"] - 1:
+ world_data["map_center"][1] += 1
+ else:
+ y_unevenness = world_data["map_center"][0] % 2
+ y_evenness = int(not(y_unevenness))
+ if string[6:] == "west" and \
+ world_data["map_center"][1] > -y_unevenness:
+ if string[:5] == "north" and world_data["map_center"][0] > 0:
+ world_data["map_center"][0] -= 1
+ world_data["map_center"][1] -= y_evenness
+ if string[:5] == "south" and world_data["map_center"][0] \
+ < world_data["map_size"] - 1:
+ world_data["map_center"][0] += 1
+ world_data["map_center"][1] -= y_evenness
+ elif string[6:] == "east" and world_data["map_center"][1] \
+ < world_data["map_size"] - y_unevenness:
+ if string[:5] == "north" and world_data["map_center"][0] > 0:
+ world_data["map_center"][0] -= 1
+ world_data["map_center"][1] += y_unevenness
+ if string[:5] == "south" and world_data["map_center"][0] \
+ < world_data["map_size"] - 1:
+ world_data["map_center"][0] += 1
+ world_data["map_center"][1] += y_unevenness
+ return command_look
+
+
windows = [
- {"config": [1, 33], "func": foo},
- {"config": [-7, 33], "func": foo},
- {"config": [4, 16], "func": foo},
- {"config": [4, 16], "func": foo},
- {"config": [0, -34], "func": foo}
+ {"config": [1, 33], "func": win_info},
+ {"config": [-7, 33], "func": win_log},
+ {"config": [4, 16], "func": win_inventory},
+ {"config": [4, 16], "func": win_foo},
+ {"config": [0, -34], "func": win_map}
]
io = {
"path_out": "server/in",
- "path_in": "server/out"
+ "path_in": "server/out",
+ "path_worldstate": "server/worldstate"
}
commands = {
- "Q": command_quit
+ "Q": (command_quit,),
+ "c": (command_sender("move south-east"), command_looker("south-east")),
+ "d": (command_sender("move east"), command_looker("east")),
+ "e": (command_sender("move north-east"), command_looker("north-east")),
+ "l": (command_toggle_look_mode,),
+ "s": (command_sender("move west"), command_looker("west")),
+ "w": (command_sender("move north-west"), command_looker("north-west")),
+ "x": (command_sender("move south-west"), command_looker("south-west")),
}
message_queue = {
"open_end": False,
"messages": []
}
+world_data = {
+ "avatar_position": [-1, -1],
+ "fov_map": "",
+ "inventory": [],
+ "lifepoints": -1,
+ "look_mode": False,
+ "log": [],
+ "map_center": [-1, -1],
+ "map_size": 0,
+ "mem_map": "",
+ "satiation": -1,
+ "turn": -1
+}
sep_size = 1 # Width of inter-window borders and title bars.
stdscr = None
screen_size = [0,0]