X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;f=client.py;h=9d0da0d534399be8f546258d8976036ac53c9eb6;hb=64bf1873a6686c1bc974321c13d3c9f7800db0d6;hp=308a15f1c3ef4df8b619668f11f362baecefa0b1;hpb=cea107a6abfd38546d6c41a392ba2c89c1331604;p=plomrogue2-experiments diff --git a/client.py b/client.py index 308a15f..9d0da0d 100755 --- a/client.py +++ b/client.py @@ -1,67 +1,124 @@ #!/usr/bin/env python3 - import urwid import plom_socket_io import socket import threading +from parser import ArgError, Parser +import game_common -class ArgumentError(Exception): - pass +def get_map_class(geometry): + return globals()['Map' + geometry] -class UrwidSetup: +class MapSquare(game_common.Map): - def __init__(self, socket): - """Build client urwid interface around socket communication. + def list_terrain_to_lines(self, terrain_as_list): + terrain = ''.join(terrain_as_list) + map_lines = [] + start_cut = 0 + while start_cut < len(terrain): + limit = start_cut + self.game.world.map_.size[1] + map_lines += [terrain[start_cut:limit]] + start_cut = limit + return "\n".join(map_lines) - Sets up all widgets for writing to the socket and representing data - from it. Sending via a self.EditToSocket widget is straightforward; - polling the socket for input from the server in parallel to the urwid - main loop not so much: - The urwid developers warn against sharing urwid resources among - threads, so having a socket polling thread for writing to an urwid - widget while other widgets are handled in other threads would be - dangerous. Urwid developers recommend using urwid's watch_pipe - mechanism instead: using a pipe from non-urwid threads into a single - urwid thread. We use self.recv_loop_thread to poll the socket, therein - write socket.recv output to an object that is then linked to by - self.server_output (which is known to the urwid thread), then use the - pipe to urwid to trigger it pulling new data from self.server_output to - handle via self.InputHandler. (We *could* pipe socket.recv output - directly, but then we get complicated buffering situations here as well - as in the urwid code that receives the pipe output. It's easier to just - tell the urwid code where it finds full new server messages to handle.) - """ - self.socket = socket - self.main_loop = urwid.MainLoop(self.setup_widgets()) - self.server_output = [] - input_handler = getattr(self.InputHandler(self.reply_widget, - self.map_widget, - self.turn_widget, - self.server_output), - 'handle_input') - self.urwid_pipe_write_fd = self.main_loop.watch_pipe(input_handler) - self.recv_loop_thread = threading.Thread(target=self.recv_loop) +class MapHex(game_common.Map): + + def list_terrain_to_lines(self, terrain_as_list): + new_terrain_list = [] + x = 0 + y = 0 + for c in terrain_as_list: + new_terrain_list += [c, ' '] + x += 1 + if x == self.size[1]: + new_terrain_list += ['\n'] + x = 0 + y += 1 + if y % 2 != 0: + new_terrain_list += [' '] + return ''.join(new_terrain_list) + - def setup_widgets(self): - """Return container widget with all widgets we want on our screen. +class World(game_common.World): - Sets up an urwid.Pile inside a returned urwid.Filler; top to bottom: - - an EditToSocketWidget, prefixing self.socket input with 'SEND: ' - - a 50-col wide urwid.Padding container for self.map_widget, which is - to print clipped map representations - - self.reply_widget, a urwid.Text widget printing self.socket replies + def __init__(self, *args, **kwargs): + """Extend original with local classes and empty default map. + + We need the empty default map because we draw the map widget + on any update, even before we actually receive map data. """ - edit_widget = self.EditToSocketWidget(self.socket, 'SEND: ') - self.map_widget = self.MapWidget('', wrap='clip') - self.turn_widget = self.TurnWidget('') - self.reply_widget = self.LogWidget('') + super().__init__(*args, **kwargs) + self.get_map_class = get_map_class + self.map_ = self.get_map_class('Hex')() + + +class Game(game_common.CommonCommandsMixin): + world = World() + log_text = '' + + def log(self, msg): + """Prefix msg plus newline to self.log_text.""" + self.log_text = msg + '\n' + self.log_text + + def symbol_for_type(self, type_): + symbol = '?' + if type_ == 'human': + symbol = '@' + elif type_ == 'monster': + symbol = 'm' + return symbol + + def cmd_LAST_PLAYER_TASK_RESULT(self, msg): + if msg != "success": + self.log_text = msg + '\n' + self.log_text + cmd_LAST_PLAYER_TASK_RESULT.argtypes = 'string' + + def cmd_TURN_FINISHED(self, n): + """Do nothing. (This may be extended later.)""" + pass + cmd_TURN_FINISHED.argtypes = 'int:nonneg' + + def cmd_NEW_TURN(self, n): + """Set self.turn to n, empty self.things.""" + self.world.turn = n + self.world.things = [] + cmd_NEW_TURN.argtypes = 'int:nonneg' + + def cmd_VISIBLE_MAP_LINE(self, y, terrain_line): + self.world.map_.set_line(y, terrain_line) + cmd_VISIBLE_MAP_LINE.argtypes = 'int:nonneg string' + + +class WidgetManager: + + def __init__(self, socket, game): + """Set up all urwid widgets we want on the screen.""" + self.game = game + edit_widget = self.EditToSocketWidget(socket, 'SEND: ') + self.map_widget = urwid.Text('', wrap='clip') + self.turn_widget = urwid.Text('') + self.log_widget = urwid.Text('') map_box = urwid.Padding(self.map_widget, width=50) widget_pile = urwid.Pile([edit_widget, map_box, self.turn_widget, - self.reply_widget]) - return urwid.Filler(widget_pile, valign='top') + self.log_widget]) + self.top = urwid.Filler(widget_pile, valign='top') + + def draw_map(self): + """Draw map view from .game.map_.terrain, .game.things.""" + terrain_as_list = list(self.game.world.map_.terrain[:]) + for t in self.game.world.things: + pos_i = self.game.world.map_.get_position_index(t.position) + terrain_as_list[pos_i] = self.game.symbol_for_type(t.type_) + return self.game.world.map_.list_terrain_to_lines(terrain_as_list) + + def update(self): + """Redraw all non-edit widgets.""" + self.turn_widget.set_text('TURN: ' + str(self.game.world.turn)) + self.log_widget.set_text(self.game.log_text) + self.map_widget.set_text(self.draw_map()) class EditToSocketWidget(urwid.Edit): """Extends urwid.Edit with socket to send input on 'enter' to.""" @@ -77,172 +134,71 @@ class UrwidSetup: plom_socket_io.send(self.socket, self.edit_text) self.edit_text = '' - class TurnWidget(urwid.Text): - """Displays turn number.""" - - def set_turn(self, turn_string): - turn_string = turn_string.strip() - if not turn_string.isdigit(): - raise ArgumentError('Argument must be non-negative integer.') - self.set_text('TURN: ' + turn_string) - - class LogWidget(urwid.Text): - """Displays client log, newest message on top.""" - - def add(self, text): - """Add text plus newline to (top of) log.""" - self.set_text(text + '\n' + self.text) - - class MapWidget(urwid.Text): - """Stores/updates/draws game map.""" - map_size = (5, 5) - terrain_map = ' ' * 25 - position = (0, 0) - things = [] - - class Thing: - def __init__(self, position, symbol): - self.position = position - self.symbol = symbol - - def draw_map(self): - """Draw map view from .map_size, .terrain_map, .position.""" - whole_map = [] - for c in self.terrain_map: - whole_map += [c] - for t in self.things: - pos_i = t.position[0] * (self.map_size[1] + 1) + t.position[1] - whole_map[pos_i] = t.symbol - self.set_text(''.join(whole_map)) - - def get_yx(self, yx_string): - - def get_axis_position_from_argument(axis, token): - if len(token) < 3 or token[:2] != axis + ':' or \ - not token[2:].isdigit(): - raise ArgumentError('Bad arg for ' + axis + ' position.') - return int(token[2:]) - - tokens = yx_string.split(',') - if len(tokens) != 2: - raise ArgumentError('wrong number of ","-separated arguments') - y = get_axis_position_from_argument('Y', tokens[0]) - x = get_axis_position_from_argument('X', tokens[1]) - return (y, x) - - def update_map_size(self, size_string): - """Set map size, redo self.terrain_map in new size, '?'-filled.""" - new_map_size = self.get_yx(size_string) - if 0 in new_map_size: - raise ArgumentError('size value for either axis must be >0') - self.map_size = new_map_size - self.terrain_map = '' - for y in range(self.map_size[0]): - self.terrain_map += '?' * self.map_size[1] + '\n' - self.draw_map() - - def update_terrain(self, terrain_map): - """Update self.terrain_map. Ensure size matching self.map_size.""" - lines = terrain_map.split('\n') - if len(lines) != self.map_size[0]: - raise ArgumentError('wrong map height') - for line in lines: - if len(line) != self.map_size[1]: - raise ArgumentError('wrong map width') - self.terrain_map = terrain_map - self.draw_map() - - def update_things(self, thing_description): - """Append thing of thing_description to self.things.""" - thing_types = {'human': '@', 'monster': 'M'} - tokens = thing_description.split() - if len(tokens) != 2: - raise ArgumentError('Wrong number of tokens.') - yx = self.get_yx(tokens[1]) - if yx[0] >= self.map_size[0] or yx[1] >= self.map_size[1]: - raise ArgumentError('Position outside of map size bounds.') - type_token = tokens[0] - prefix = 'TYPE:' - type_ = '?' - if len(type_token) <= len(prefix) or \ - type_token[:len(prefix)] != prefix: - raise ArgumentError('Invalid type token.') - type_ = type_token[len(prefix):] - if type_ not in thing_types: - raise ArgumentError('Unknown thing type.') - self.things += [self.Thing(yx, thing_types[type_])] - self.draw_map() - - def clear_things(self, _): - self.things = [] - - class InputHandler: - """Delivers data from other thread to widget via message_container. - - The class only exists to provide handle_input as a bound method, with - widget and message_container pre-set, as (bound) handle_input is used - as a callback in urwid's watch_pipe – which merely provides its - callback target with one parameter for a pipe to read data from an - urwid-external thread. + +class PlomRogueClient: + + def __init__(self, game, socket): + """Build client urwid interface around socket communication. + + Sets up all widgets for writing to the socket and representing data + from it. Sending via a WidgetManager.EditToSocket widget is + straightforward; polling the socket for input from the server in + parallel to the urwid main loop not so much: + + The urwid developers warn against sharing urwid resources among + threads, so having a socket polling thread for writing to an urwid + widget while other widgets are handled in other threads would be + dangerous. Urwid developers recommend using urwid's watch_pipe + mechanism instead: using a pipe from non-urwid threads into a single + urwid thread. We use self.recv_loop_thread to poll the socket, therein + write socket.recv output to an object that is then linked to by + self.server_output (which is known to the urwid thread), then use the + pipe to urwid to trigger it pulling new data from self.server_output to + handle via self.handle_input. (We *could* pipe socket.recv output + directly, but then we get complicated buffering situations here as well + as in the urwid code that receives the pipe output. It's easier to just + tell the urwid code where it finds full new server messages to handle.) """ + self.game = game + self.parser = Parser(self.game) + self.socket = socket + self.widget_manager = WidgetManager(self.socket, self.game) + self.server_output = [] + self.urwid_loop = urwid.MainLoop(self.widget_manager.top) + self.urwid_pipe_write_fd = self.urwid_loop.watch_pipe(self. + handle_input) + self.recv_loop_thread = threading.Thread(target=self.recv_loop) + + def handle_input(self, trigger): + """On input from recv_loop thread, parse and enact commands. - def __init__(self, log_widget, map_widget, turn_widget, - message_container): - self.log_widget = log_widget - self.map_widget = map_widget - self.turn_widget = turn_widget - self.message_container = message_container - - def handle_input(self, trigger): - """On input from other thread, either quit or write to widget text. - - Serves as a receiver to urwid's watch_pipe mechanism, with trigger - the data that a pipe defined by watch_pipe delivers. To avoid - buffering trouble, we don't care for that data beyond the fact that - its receival triggers this function: The sender is to write the - data it wants to deliver into the container referenced by - self.message_container, and just pipe the trigger to inform us - about this. - - If the message delivered is 'BYE', quits Urwid. - """ - - def mapdraw_command(prefix, func): - n = len(prefix) - if len(msg) > n and msg[:n] == prefix: - m = getattr(self.map_widget, func) - m(msg[n:]) - return True - return False - - def turndraw_command(prefix, func): - n = len(prefix) - if len(msg) > n and msg[:n] == prefix: - m = getattr(self.turn_widget, func) - m(msg[n:]) - return True - return False - - msg = self.message_container[0] - if msg == 'BYE': - raise urwid.ExitMainLoop() - return - found_command = False - try: - found_command = turndraw_command('NEW_TURN ', 'set_turn') or ( - mapdraw_command('NEW_TURN ', 'clear_things') or - mapdraw_command('TERRAIN\n', 'update_terrain') or - mapdraw_command('THING ', 'update_things') or - mapdraw_command('MAP_SIZE ', 'update_map_size')) - except ArgumentError as e: - self.log_widget.add('ARGUMENT ERROR: ' + msg + '\n' + str(e)) + Serves as a receiver to urwid's watch_pipe mechanism, with trigger the + data that a pipe defined by watch_pipe delivers. To avoid buffering + trouble, we don't care for that data beyond the fact that its receival + triggers this function: The sender is to write the data it wants to + deliver into the container referenced by self.server_output, and just + pipe the trigger to inform us about this. + + If the message delivered is 'BYE', quits Urwid. Otherwise tries to + parse it as a command, and enact it. In all cases but the 'BYE', calls + self.widget_manager.update. + """ + msg = self.server_output[0] + if msg == 'BYE': + raise urwid.ExitMainLoop() + try: + command = self.parser.parse(msg) + if command is None: + self.game.log('UNHANDLED INPUT: ' + msg) else: - if not found_command: - self.log_widget.add('UNHANDLED INPUT: ' + msg) - del self.message_container[0] + command() + except ArgError as e: + self.game.log('ARGUMENT ERROR: ' + msg + '\n' + str(e)) + self.widget_manager.update() + del self.server_output[0] def recv_loop(self): - """Loop to receive messages from socket and deliver them to urwid. + """Loop to receive messages from socket, deliver them to urwid thread. Waits for self.server_output to become empty (this signals that the input handler is finished / ready to receive new input), then writes @@ -251,19 +207,21 @@ class UrwidSetup: """ import os for msg in plom_socket_io.recv(self.socket): - while len(self.server_output) > 0: - pass + while len(self.server_output) > 0: # Wait until self.server_output + pass # is emptied by input handler. self.server_output += [msg] os.write(self.urwid_pipe_write_fd, b' ') def run(self): - """Run in parallel main and recv_loop thread.""" + """Run in parallel urwid_loop and recv_loop threads.""" self.recv_loop_thread.start() - self.main_loop.run() + self.urwid_loop.run() self.recv_loop_thread.join() -s = socket.create_connection(('127.0.0.1', 5000)) -u = UrwidSetup(s) -u.run() -s.close() +if __name__ == '__main__': + game = Game() + s = socket.create_connection(('127.0.0.1', 5000)) + p = PlomRogueClient(game, s) + p.run() + s.close()