plomlompom.com Git - plomrogue2-experiments/blob - new/plomrogue/misc.py

   1 #!/usr/bin/env python3
   2 import socketserver
   3 import threading
   4 import queue
   5 import sys
   6 import parser
   7
   8
   9 class GameError(Exception):
  10     pass
  11
  12
  13 # Avoid "Address already in use" errors.
  14 socketserver.TCPServer.allow_reuse_address = True
  15
  16
  17 class Server(socketserver.ThreadingTCPServer):
  18     """Bind together threaded IO handling server and message queue."""
  19
  20     def __init__(self, queue, port, *args, **kwargs):
  21         super().__init__(('localhost', port), IO_Handler, *args, **kwargs)
  22         self.queue_out = queue
  23         self.daemon_threads = True  # Else, server's threads have daemon=False.
  24
  25
  26 class IO_Handler(socketserver.BaseRequestHandler):
  27
  28     def handle(self):
  29         """Move messages between network socket and game IO loop via queues.
  30
  31         On start (a new connection from client to server), sets up a
  32         new queue, sends it via self.server.queue_out to the game IO
  33         loop thread, and from then on receives messages to send back
  34         from the game IO loop via that new queue.
  35
  36         At the same time, loops over socket's recv to get messages
  37         from the outside into the game IO loop by way of
  38         self.server.queue_out into the game IO. Ends connection once a
  39         'QUIT' message is received from socket, and then also calls
  40         for a kill of its own queue.
  41
  42         All messages to the game IO loop are tuples, with the first
  43         element a meta command ('ADD_QUEUE' for queue creation,
  44         'KILL_QUEUE' for queue deletion, and 'COMMAND' for everything
  45         else), the second element a UUID that uniquely identifies the
  46         thread (so that the game IO loop knows whom to send replies
  47         back to), and optionally a third element for further
  48         instructions.
  49
  50         """
  51
  52         def send_queue_messages(plom_socket, queue_in, thread_alive):
  53             """Send messages via socket from queue_in while thread_alive[0]."""
  54             while thread_alive[0]:
  55                 try:
  56                     msg = queue_in.get(timeout=1)
  57                 except queue.Empty:
  58                     continue
  59                 plom_socket.send(msg, True)
  60
  61         import uuid
  62         import plom_socket
  63         plom_socket = plom_socket.PlomSocket(self.request)
  64         print('CONNECTION FROM:', str(self.client_address))
  65         connection_id = uuid.uuid4()
  66         queue_in = queue.Queue()
  67         self.server.queue_out.put(('ADD_QUEUE', connection_id, queue_in))
  68         thread_alive = [True]
  69         t = threading.Thread(target=send_queue_messages,
  70                              args=(plom_socket, queue_in, thread_alive))
  71         t.start()
  72         for message in plom_socket.recv():
  73             if message is None:
  74                 plom_socket.send('BAD MESSAGE', True)
  75             elif 'QUIT' == message:
  76                 plom_socket.send('BYE', True)
  77                 break
  78             else:
  79                 self.server.queue_out.put(('COMMAND', connection_id, message))
  80         self.server.queue_out.put(('KILL_QUEUE', connection_id))
  81         thread_alive[0] = False
  82         print('CONNECTION CLOSED FROM:', str(self.client_address))
  83         plom_socket.socket.close()
  84
  85
  86 class GameIO():
  87
  88     def __init__(self, game_file_name, game):
  89         self.game_file_name = game_file_name
  90         self.queues_out = {}
  91         self.parser = parser.Parser(game)
  92
  93     def loop(self, q):
  94         """Handle commands coming through queue q, send results back.
  95
  96         Commands from q are expected to be tuples, with the first element
  97         either 'ADD_QUEUE', 'COMMAND', or 'KILL_QUEUE', the second element
  98         a UUID, and an optional third element of arbitrary type. The UUID
  99         identifies a receiver for replies.
 100
 101         An 'ADD_QUEUE' command should contain as third element a queue
 102         through which to send messages back to the sender of the
 103         command. A 'KILL_QUEUE' command removes the queue for that
 104         receiver from the list of queues through which to send replies.
 105
 106         A 'COMMAND' command is specified in greater detail by a string
 107         that is the tuple's third element. The game_command_handler takes
 108         care of processing this and sending out replies.
 109
 110         """
 111         while True:
 112             x = q.get()
 113             command_type = x[0]
 114             connection_id = x[1]
 115             content = None if len(x) == 2 else x[2]
 116             if command_type == 'ADD_QUEUE':
 117                 self.queues_out[connection_id] = content
 118             elif command_type == 'KILL_QUEUE':
 119                 del self.queues_out[connection_id]
 120             elif command_type == 'COMMAND':
 121                 self.handle_input(content, connection_id)
 122
 123     def run_loop_with_server(self):
 124         """Run connection of server talking to clients and game IO loop.
 125
 126         We have the TCP server (an instance of Server) and we have the
 127         game IO loop, a thread running self.loop. Both communicate with
 128         each other via a queue.Queue. While the TCP server may spawn
 129         parallel threads to many clients, the IO loop works sequentially
 130         through game commands received from the TCP server's threads (=
 131         client connections to the TCP server). A processed command may
 132         trigger messages to the commanding client or to all clients,
 133         delivered from the IO loop to the TCP server via the queue.
 134
 135         """
 136         q = queue.Queue()
 137         c = threading.Thread(target=self.loop, daemon=True, args=(q,))
 138         c.start()
 139         server = Server(q, 5000)
 140         try:
 141             server.serve_forever()
 142         except KeyboardInterrupt:
 143             pass
 144         finally:
 145             print('Killing server')
 146             server.server_close()
 147
 148     def handle_input(self, input_, connection_id=None, store=True):
 149         """Process input_ to command grammar, call command handler if found."""
 150         from inspect import signature
 151
 152         def answer(connection_id, msg):
 153             if connection_id:
 154                 self.send(msg, connection_id)
 155             else:
 156                 print(msg)
 157
 158         try:
 159             command, args = self.parser.parse(input_)
 160             if command is None:
 161                 answer(connection_id, 'UNHANDLED_INPUT')
 162             else:
 163                 if 'connection_id' in list(signature(command).parameters):
 164                     command(*args, connection_id=connection_id)
 165                 else:
 166                     command(*args)
 167                     if store and not hasattr(command, 'dont_save'):
 168                         with open(self.game_file_name, 'a') as f:
 169                             f.write(input_ + '\n')
 170         except parser.ArgError as e:
 171             answer(connection_id, 'ARGUMENT_ERROR ' + quote(str(e)))
 172         except GameError as e:
 173             answer(connection_id, 'GAME_ERROR ' + quote(str(e)))
 174
 175     def send(self, msg, connection_id=None):
 176         """Send message msg to server's client(s) via self.queues_out.
 177
 178         If a specific client is identified by connection_id, only
 179         sends msg to that one. Else, sends it to all clients
 180         identified in self.queues_out.
 181
 182         """
 183         if connection_id:
 184             self.queues_out[connection_id].put(msg)
 185         else:
 186             for connection_id in self.queues_out:
 187                 self.queues_out[connection_id].put(msg)
 188
 189
 190 class MapBase:
 191
 192     def __init__(self, size=(0, 0)):
 193         self.size = size
 194         self.terrain = '?'*self.size_i
 195
 196     @property
 197     def size_i(self):
 198         return self.size[0] * self.size[1]
 199
 200     def set_line(self, y, line):
 201         height_map = self.size[0]
 202         width_map = self.size[1]
 203         if y >= height_map:
 204             raise ArgError('too large row number %s' % y)
 205         width_line = len(line)
 206         if width_line > width_map:
 207             raise ArgError('too large map line width %s' % width_line)
 208         self.terrain = self.terrain[:y * width_map] + line +\
 209                        self.terrain[(y + 1) * width_map:]
 210
 211     def get_position_index(self, yx):
 212         return yx[0] * self.size[1] + yx[1]
 213
 214
 215 class Map(MapBase):
 216
 217     def __getitem__(self, yx):
 218         return self.terrain[self.get_position_index(yx)]
 219
 220     def __setitem__(self, yx, c):
 221         pos_i = self.get_position_index(yx)
 222         if type(c) == str:
 223             self.terrain = self.terrain[:pos_i] + c + self.terrain[pos_i + 1:]
 224         else:
 225             self.terrain[pos_i] = c
 226
 227     def __iter__(self):
 228         """Iterate over YX position coordinates."""
 229         for y in range(self.size[0]):
 230             for x in range(self.size[1]):
 231                 yield [y, x]
 232
 233     def lines(self):
 234         width = self.size[1]
 235         for y in range(self.size[0]):
 236             yield (y, self.terrain[y * width:(y + 1) * width])
 237
 238     def get_fov_map(self, yx):
 239         return self.fov_map_type(self, yx)
 240
 241     def get_directions(self):
 242         directions = []
 243         for name in dir(self):
 244             if name[:5] == 'move_':
 245                 directions += [name[5:]]
 246         return directions
 247
 248     def get_neighbors(self, pos):
 249         neighbors = {}
 250         if not hasattr(self, 'neighbors_to'):
 251             self.neighbors_to = {}
 252         if pos in self.neighbors_to:
 253             return self.neighbors_to[pos]
 254         for direction in self.get_directions():
 255             neighbors[direction] = None
 256             try:
 257                 neighbors[direction] = self.move(pos, direction)
 258             except GameError:
 259                 pass
 260         self.neighbors_to[pos] = neighbors
 261         return neighbors
 262
 263     def new_from_shape(self, init_char):
 264         import copy
 265         new_map = copy.deepcopy(self)
 266         for pos in new_map:
 267             new_map[pos] = init_char
 268         return new_map
 269
 270     def move(self, start_pos, direction):
 271         mover = getattr(self, 'move_' + direction)
 272         new_pos = mover(start_pos)
 273         if new_pos[0] < 0 or new_pos[1] < 0 or \
 274                 new_pos[0] >= self.size[0] or new_pos[1] >= self.size[1]:
 275             raise GameError('would move outside map bounds')
 276         return new_pos
 277
 278     def move_LEFT(self, start_pos):
 279         return [start_pos[0], start_pos[1] - 1]
 280
 281     def move_RIGHT(self, start_pos):
 282         return [start_pos[0], start_pos[1] + 1]
 283
 284
 285
 286 class MapHex(Map):
 287
 288     def __init__(self, *args, **kwargs):
 289         super().__init__(*args, **kwargs)
 290         self.fov_map_type = FovMapHex
 291
 292     def move_UPLEFT(self, start_pos):
 293         if start_pos[0] % 2 == 1:
 294             return [start_pos[0] - 1, start_pos[1] - 1]
 295         else:
 296             return [start_pos[0] - 1, start_pos[1]]
 297
 298     def move_UPRIGHT(self, start_pos):
 299         if start_pos[0] % 2 == 1:
 300             return [start_pos[0] - 1, start_pos[1]]
 301         else:
 302             return [start_pos[0] - 1, start_pos[1] + 1]
 303
 304     def move_DOWNLEFT(self, start_pos):
 305         if start_pos[0] % 2 == 1:
 306              return [start_pos[0] + 1, start_pos[1] - 1]
 307         else:
 308                return [start_pos[0] + 1, start_pos[1]]
 309
 310     def move_DOWNRIGHT(self, start_pos):
 311         if start_pos[0] % 2 == 1:
 312             return [start_pos[0] + 1, start_pos[1]]
 313         else:
 314             return [start_pos[0] + 1, start_pos[1] + 1]
 315
 316
 317
 318 class FovMap:
 319
 320     def __init__(self, source_map, yx):
 321         self.source_map = source_map
 322         self.size = self.source_map.size
 323         self.terrain = '?' * self.size_i
 324         self[yx] = '.'
 325         self.shadow_cones = []
 326         self.circle_out(yx, self.shadow_process_hex)
 327
 328     def shadow_process_hex(self, yx, distance_to_center, dir_i, dir_progress):
 329         # Possible optimization: If no shadow_cones yet and self[yx] == '.',
 330         # skip all.
 331         CIRCLE = 360  # Since we'll float anyways, number is actually arbitrary.
 332
 333         def correct_arm(arm):
 334             if arm < 0:
 335                 arm += CIRCLE
 336             return arm
 337
 338         def in_shadow_cone(new_cone):
 339             for old_cone in self.shadow_cones:
 340                 if old_cone[0] >= new_cone[0] and \
 341                     new_cone[1] >= old_cone[1]:
 342                     #print('DEBUG shadowed by:', old_cone)
 343                     return True
 344                 # We might want to also shade hexes whose middle arm is inside a
 345                 # shadow cone for a darker FOV. Note that we then could not for
 346                 # optimization purposes rely anymore on the assumption that a
 347                 # shaded hex cannot add growth to existing shadow cones.
 348             return False
 349
 350         def merge_cone(new_cone):
 351             import math
 352             for old_cone in self.shadow_cones:
 353                 if new_cone[0] > old_cone[0] and \
 354                     (new_cone[1] < old_cone[0] or
 355                      math.isclose(new_cone[1], old_cone[0])):
 356                     #print('DEBUG merging to', old_cone)
 357                     old_cone[0] = new_cone[0]
 358                     #print('DEBUG merged cone:', old_cone)
 359                     return True
 360                 if new_cone[1] < old_cone[1] and \
 361                     (new_cone[0] > old_cone[1] or
 362                      math.isclose(new_cone[0], old_cone[1])):
 363                     #print('DEBUG merging to', old_cone)
 364                     old_cone[1] = new_cone[1]
 365                     #print('DEBUG merged cone:', old_cone)
 366                     return True
 367             return False
 368
 369         def eval_cone(cone):
 370             #print('DEBUG CONE', cone, '(', step_size, distance_to_center, number_steps, ')')
 371             if in_shadow_cone(cone):
 372                 return
 373             self[yx] = '.'
 374             if self.source_map[yx] != '.':
 375                 #print('DEBUG throws shadow', cone)
 376                 unmerged = True
 377                 while merge_cone(cone):
 378                     unmerged = False
 379                 if unmerged:
 380                     self.shadow_cones += [cone]
 381
 382         #print('DEBUG', yx)
 383         step_size = (CIRCLE/len(self.circle_out_directions)) / distance_to_center
 384         number_steps = dir_i * distance_to_center + dir_progress
 385         left_arm = correct_arm(-(step_size/2) - step_size*number_steps)
 386         right_arm = correct_arm(left_arm - step_size)
 387         # Optimization potential: left cone could be derived from previous
 388         # right cone. Better even: Precalculate all cones.
 389         if right_arm > left_arm:
 390             eval_cone([left_arm, 0])
 391             eval_cone([CIRCLE, right_arm])
 392         else:
 393             eval_cone([left_arm, right_arm])
 394
 395     def basic_circle_out_move(self, pos, direction):
 396         """Move position pos into direction. Return whether still in map."""
 397         mover = getattr(self, 'move_' + direction)
 398         pos[:] = mover(pos)
 399         if pos[0] < 0 or pos[1] < 0 or \
 400             pos[0] >= self.size[0] or pos[1] >= self.size[1]:
 401             return False
 402         return True
 403
 404     def circle_out(self, yx, f):
 405         # Optimization potential: Precalculate movement positions. (How to check
 406         # circle_in_map then?)
 407         # Optimization potential: Precalculate what hexes are shaded by what hex
 408         # and skip evaluation of already shaded hexes. (This only works if hex
 409         # shading implies they completely lie in existing shades; otherwise we
 410         # would lose shade growth through hexes at shade borders.)
 411
 412         # TODO: Start circling only in earliest obstacle distance.
 413         circle_in_map = True
 414         distance = 1
 415         yx = yx[:]
 416         #print('DEBUG CIRCLE_OUT', yx)
 417         while circle_in_map:
 418             circle_in_map = False
 419             self.basic_circle_out_move(yx, 'RIGHT')
 420             for dir_i in range(len(self.circle_out_directions)):
 421                 for dir_progress in range(distance):
 422                     direction = self.circle_out_directions[dir_i]
 423                     if self.circle_out_move(yx, direction):
 424                         f(yx, distance, dir_i, dir_progress)
 425                         circle_in_map = True
 426             distance += 1
 427
 428
 429
 430 class FovMapHex(FovMap, MapHex):
 431     circle_out_directions = ('DOWNLEFT', 'LEFT', 'UPLEFT',
 432                              'UPRIGHT', 'RIGHT', 'DOWNRIGHT')
 433
 434     def circle_out_move(self, yx, direction):
 435         return self.basic_circle_out_move(yx, direction)
 436
 437
 438
 439 class ThingBase:
 440
 441     def __init__(self, world, id_, type_='?', position=[0,0]):
 442         self.world = world
 443         self.id_ = id_
 444         self.type_ = type_
 445         self.position = position
 446
 447
 448 class Thing(ThingBase):
 449
 450     def __init__(self, *args, **kwargs):
 451         super().__init__(*args, **kwargs)
 452         self.set_task('WAIT')
 453         self._last_task_result = None
 454         self._stencil = None
 455
 456     def move_towards_target(self, target):
 457         dijkstra_map = type(self.world.map_)(self.world.map_.size)
 458         n_max = 256
 459         dijkstra_map.terrain = [n_max for i in range(dijkstra_map.size_i)]
 460         dijkstra_map[target] = 0
 461         shrunk = True
 462         visible_map = self.get_visible_map()
 463         while shrunk:
 464             shrunk = False
 465             for pos in dijkstra_map:
 466                 if visible_map[pos] != '.':
 467                     continue
 468                 neighbors = dijkstra_map.get_neighbors(tuple(pos))
 469                 for direction in neighbors:
 470                     yx = neighbors[direction]
 471                     if yx is not None and dijkstra_map[yx] < dijkstra_map[pos] - 1:
 472                         dijkstra_map[pos] = dijkstra_map[yx] + 1
 473                         shrunk = True
 474         #with open('log', 'a') as f:
 475         #    f.write('---------------------------------\n')
 476         #    for y, line in dijkstra_map.lines():
 477         #        for val in line:
 478         #            if val < 10:
 479         #                f.write(str(val))
 480         #            elif val == 256:
 481         #                f.write('x')
 482         #            else:
 483         #                f.write('~')
 484         #        f.write('\n')
 485         neighbors = dijkstra_map.get_neighbors(tuple(self.position))
 486         n = n_max
 487         #print('DEBUG', self.position, neighbors)
 488         #dirs = dijkstra_map.get_directions()
 489         #print('DEBUG dirs', dirs)
 490         #print('DEBUG neighbors', neighbors)
 491         #debug_scores = []
 492         #for pos in neighbors:
 493         #    if pos is None:
 494         #        debug_scores += [9000]
 495         #    else:
 496         #        debug_scores += [dijkstra_map[pos]]
 497         #print('DEBUG debug_scores', debug_scores)
 498         target_direction = None
 499         for direction in neighbors:
 500             yx = neighbors[direction]
 501             if yx is not None:
 502                 n_new = dijkstra_map[yx]
 503                 if n_new < n:
 504                     n = n_new
 505                     target_direction = direction
 506         #print('DEBUG result', direction)
 507         if target_direction:
 508             self.set_task('MOVE', (target_direction,))
 509
 510     def decide_task(self):
 511         # TODO: Check if monster can follow player too well (even when they should lose them)
 512         visible_things = self.get_visible_things()
 513         target = None
 514         for t in visible_things:
 515             if t.type_ == 'human':
 516                 target = t.position
 517                 break
 518         if target is not None:
 519             try:
 520                 self.move_towards_target(target)
 521                 return
 522             except GameError:
 523                 pass
 524         self.set_task('WAIT')
 525
 526     def set_task(self, task_name, args=()):
 527         task_class = self.world.game.tasks[task_name]
 528         self.task = task_class(self, args)
 529         self.task.check()  # will throw GameError if necessary
 530
 531     def proceed(self, is_AI=True):
 532         """Further the thing in its tasks.
 533
 534         Decrements .task.todo; if it thus falls to <= 0, enacts method
 535         whose name is 'task_' + self.task.name and sets .task =
 536         None. If is_AI, calls .decide_task to decide a self.task.
 537
 538         Before doing anything, ensures an empty map visibility stencil
 539         and checks that task is still possible, and aborts it
 540         otherwise (for AI things, decides a new task).
 541
 542         """
 543         self._stencil = None
 544         try:
 545             self.task.check()
 546         except GameError as e:
 547             self.task = None
 548             self._last_task_result = e
 549             if is_AI:
 550                 try:
 551                     self.decide_task()
 552                 except GameError:
 553                     self.set_task('WAIT')
 554             return
 555         self.task.todo -= 1
 556         if self.task.todo <= 0:
 557             self._last_task_result = self.task.do()
 558             self.task = None
 559         if is_AI and self.task is None:
 560             try:
 561                 self.decide_task()
 562             except GameError:
 563                 self.set_task('WAIT')
 564
 565     def get_stencil(self):
 566         if self._stencil is not None:
 567             return self._stencil
 568         self._stencil = self.world.map_.get_fov_map(self.position)
 569         return self._stencil
 570
 571     def get_visible_map(self):
 572         stencil = self.get_stencil()
 573         m = self.world.map_.new_from_shape(' ')
 574         for pos in m:
 575             if stencil[pos] == '.':
 576                 m[pos] = self.world.map_[pos]
 577         return m
 578
 579     def get_visible_things(self):
 580         stencil = self.get_stencil()
 581         visible_things = []
 582         for thing in self.world.things:
 583             if stencil[thing.position] == '.':
 584                 visible_things += [thing]
 585         return visible_things
 586
 587
 588
 589 class Task:
 590     argtypes = ''
 591
 592     def __init__(self, thing, args=()):
 593         self.thing = thing
 594         self.args = args
 595         self.todo = 3
 596
 597     @property
 598     def name(self):
 599         prefix = 'Task_'
 600         class_name = self.__class__.__name__
 601         return class_name[len(prefix):]
 602
 603     def check(self):
 604         pass
 605
 606     def get_args_string(self):
 607         stringed_args = []
 608         for arg in self.args:
 609             if type(arg) == str:
 610                 stringed_args += [quote(arg)]
 611             else:
 612                 raise GameError('stringifying arg type not implemented')
 613         return ' '.join(stringed_args)
 614
 615
 616
 617 class Task_WAIT(Task):
 618
 619     def do(self):
 620         return 'success'
 621
 622
 623
 624 class Task_MOVE(Task):
 625     argtypes = 'string:direction'
 626
 627     def check(self):
 628         test_pos = self.thing.world.map_.move(self.thing.position, self.args[0])
 629         if self.thing.world.map_[test_pos] != '.':
 630             raise GameError('%s would move into illegal terrain' % self.thing.id_)
 631         for t in self.thing.world.things:
 632             if t.position == test_pos:
 633                 raise GameError('%s would move into other thing' % self.thing.id_)
 634
 635     def do(self):
 636         self.thing.position = self.thing.world.map_.move(self.thing.position,
 637                                                          self.args[0])
 638
 639
 640
 641 class WorldBase:
 642
 643     def __init__(self, game):
 644         self.turn = 0
 645         self.things = []
 646         self.game = game
 647
 648     def get_thing(self, id_, create_unfound=True):
 649         for thing in self.things:
 650             if id_ == thing.id_:
 651                 return thing
 652         if create_unfound:
 653             t = self.game.thing_type(self, id_)
 654             self.things += [t]
 655             return t
 656         return None
 657
 658
 659 class World(WorldBase):
 660
 661     def __init__(self, *args, **kwargs):
 662         super().__init__(*args, **kwargs)
 663         self.player_id = 0
 664
 665     def new_map(self, yx):
 666         self.map_ = self.game.map_type(yx)
 667
 668     def proceed_to_next_player_turn(self):
 669         """Run game world turns until player can decide their next step.
 670
 671         Iterates through all non-player things, on each step
 672         furthering them in their tasks (and letting them decide new
 673         ones if they finish). The iteration order is: first all things
 674         that come after the player in the world things list, then
 675         (after incrementing the world turn) all that come before the
 676         player; then the player's .proceed() is run, and if it does
 677         not finish his task, the loop starts at the beginning. Once
 678         the player's task is finished, the loop breaks.
 679         """
 680         while True:
 681             player = self.get_player()
 682             player_i = self.things.index(player)
 683             for thing in self.things[player_i+1:]:
 684                 thing.proceed()
 685             self.turn += 1
 686             for thing in self.things[:player_i]:
 687                 thing.proceed()
 688             player.proceed(is_AI=False)
 689             if player.task is None:
 690                 break
 691
 692     def get_player(self):
 693         return self.get_thing(self.player_id)
 694
 695     def make_new(self, yx, seed):
 696         import random
 697         random.seed(seed)
 698         self.turn = 0
 699         self.new_map(yx)
 700         for pos in self.map_:
 701             if 0 in pos or (yx[0] - 1) == pos[0] or (yx[1] - 1) == pos[1]:
 702                 self.map_[pos] = '#'
 703                 continue
 704             self.map_[pos] = random.choice(('.', '.', '.', '.', 'x'))
 705         player = self.game.thing_type(self, 0)
 706         player.type_ = 'human'
 707         player.position = [random.randint(0, yx[0] -1),
 708                            random.randint(0, yx[1] - 1)]
 709         npc = self.game.thing_type(self, 1)
 710         npc.type_ = 'monster'
 711         npc.position = [random.randint(0, yx[0] -1),
 712                         random.randint(0, yx[1] -1)]
 713         self.things = [player, npc]
 714         return 'success'
 715
 716
 717
 718 def cmd_GEN_WORLD(self, yx, seed):
 719     self.world.make_new(yx, seed)
 720 cmd_GEN_WORLD.argtypes = 'yx_tuple:pos string'
 721
 722 def cmd_GET_GAMESTATE(self, connection_id):
 723     """Send game state to caller."""
 724     self.send_gamestate(connection_id)
 725
 726 def cmd_MAP(self, yx):
 727     """Create new map of size yx and only '?' cells."""
 728     self.world.new_map(yx)
 729 cmd_MAP.argtypes = 'yx_tuple:pos'
 730
 731 def cmd_THING_TYPE(self, i, type_):
 732     t = self.world.get_thing(i)
 733     t.type_ = type_
 734 cmd_THING_TYPE.argtypes = 'int:nonneg string'
 735
 736 def cmd_THING_POS(self, i, yx):
 737     t = self.world.get_thing(i)
 738     t.position = list(yx)
 739 cmd_THING_POS.argtypes = 'int:nonneg yx_tuple:nonneg'
 740
 741 def cmd_TERRAIN_LINE(self, y, terrain_line):
 742     self.world.map_.set_line(y, terrain_line)
 743 cmd_TERRAIN_LINE.argtypes = 'int:nonneg string'
 744
 745 def cmd_PLAYER_ID(self, id_):
 746     # TODO: test whether valid thing ID
 747     self.world.player_id = id_
 748 cmd_PLAYER_ID.argtypes = 'int:nonneg'
 749
 750 def cmd_TURN(self, n):
 751     self.world.turn = n
 752 cmd_TURN.argtypes = 'int:nonneg'
 753
 754 def cmd_SWITCH_PLAYER(self):
 755     player = self.world.get_player()
 756     player.set_task('WAIT')
 757     thing_ids = [t.id_ for t in self.world.things]
 758     player_index = thing_ids.index(player.id_)
 759     if player_index == len(thing_ids) - 1:
 760         self.world.player_id = thing_ids[0]
 761     else:
 762         self.world.player_id = thing_ids[player_index + 1]
 763     self.proceed()
 764
 765 def cmd_SAVE(self):
 766
 767     def write(f, msg):
 768         f.write(msg + '\n')
 769
 770     save_file_name = self.io.game_file_name + '.save'
 771     with open(save_file_name, 'w') as f:
 772         write(f, 'TURN %s' % self.world.turn)
 773         write(f, 'MAP ' + stringify_yx(self.world.map_.size))
 774         for y, line in self.world.map_.lines():
 775             write(f, 'TERRAIN_LINE %5s %s' % (y, quote(line)))
 776         for thing in self.world.things:
 777             write(f, 'THING_TYPE %s %s' % (thing.id_, thing.type_))
 778             write(f, 'THING_POS %s %s' % (thing.id_,
 779                                           stringify_yx(thing.position)))
 780             task = thing.task
 781             if task is not None:
 782                 task_args = task.get_args_string()
 783                 write(f, 'SET_TASK:%s %s %s %s' % (task.name, thing.id_,
 784                                                    task.todo, task_args))
 785         write(f, 'PLAYER_ID %s' % self.world.player_id)
 786 cmd_SAVE.dont_save = True
 787
 788
 789 class Game:
 790
 791     def __init__(self, game_file_name):
 792         self.io = GameIO(game_file_name, self)
 793         self.map_type = MapHex
 794         self.tasks = {'WAIT': Task_WAIT, 'MOVE': Task_MOVE}
 795         self.commands = {'GEN_WORLD': cmd_GEN_WORLD,
 796                          'GET_GAMESTATE': cmd_GET_GAMESTATE,
 797                          'MAP': cmd_MAP,
 798                          'THING_TYPE': cmd_THING_TYPE,
 799                          'THING_POS': cmd_THING_POS,
 800                          'TERRAIN_LINE': cmd_TERRAIN_LINE,
 801                          'PLAYER_ID': cmd_PLAYER_ID,
 802                          'TURN': cmd_TURN,
 803                          'SWITCH_PLAYER': cmd_SWITCH_PLAYER,
 804                          'SAVE': cmd_SAVE}
 805         self.world_type = World
 806         self.world = self.world_type(self)
 807         self.thing_type = Thing
 808
 809     def get_string_options(self, string_option_type):
 810         if string_option_type == 'direction':
 811             return self.world.map_.get_directions()
 812         return None
 813
 814     def send_gamestate(self, connection_id=None):
 815         """Send out game state data relevant to clients."""
 816
 817         self.io.send('TURN ' + str(self.world.turn))
 818         self.io.send('MAP ' + stringify_yx(self.world.map_.size))
 819         visible_map = self.world.get_player().get_visible_map()
 820         for y, line in visible_map.lines():
 821             self.io.send('VISIBLE_MAP_LINE %5s %s' % (y, quote(line)))
 822         visible_things = self.world.get_player().get_visible_things()
 823         for thing in visible_things:
 824             self.io.send('THING_TYPE %s %s' % (thing.id_, thing.type_))
 825             self.io.send('THING_POS %s %s' % (thing.id_,
 826                                               stringify_yx(thing.position)))
 827         player = self.world.get_player()
 828         self.io.send('PLAYER_POS %s' % (stringify_yx(player.position)))
 829         self.io.send('GAME_STATE_COMPLETE')
 830
 831     def proceed(self):
 832         """Send turn finish signal, run game world, send new world data.
 833
 834         First sends 'TURN_FINISHED' message, then runs game world
 835         until new player input is needed, then sends game state.
 836         """
 837         self.io.send('TURN_FINISHED ' + str(self.world.turn))
 838         self.world.proceed_to_next_player_turn()
 839         msg = str(self.world.get_player()._last_task_result)
 840         self.io.send('LAST_PLAYER_TASK_RESULT ' + quote(msg))
 841         self.send_gamestate()
 842
 843     def get_command(self, command_name):
 844         from functools import partial
 845
 846         def cmd_TASK_colon(task_name, game, *args):
 847             game.world.get_player().set_task(task_name, args)
 848             game.proceed()
 849
 850         def cmd_SET_TASK_colon(task_name, game, thing_id, todo, *args):
 851             t = game.world.get_thing(thing_id, False)
 852             if t is None:
 853                 raiseArgError('No such Thing.')
 854             task_class = game.tasks[task_name]
 855             t.task = task_class(t, args)
 856             t.task.todo = todo
 857
 858         def task_prefixed(command_name, task_prefix, task_command,
 859                           argtypes_prefix=None):
 860             if command_name[:len(task_prefix)] == task_prefix:
 861                 task_name = command_name[len(task_prefix):]
 862                 if task_name in self.tasks:
 863                     f = partial(task_command, task_name, self)
 864                     task = self.tasks[task_name]
 865                     if argtypes_prefix:
 866                         f.argtypes = argtypes_prefix + ' ' + task.argtypes
 867                     else:
 868                         f.argtypes = task.argtypes
 869                     return f
 870             return None
 871
 872         command = task_prefixed(command_name, 'TASK:', cmd_TASK_colon)
 873         if command:
 874             return command
 875         command = task_prefixed(command_name, 'SET_TASK:', cmd_SET_TASK_colon,
 876                                 'int:nonneg int:nonneg ')
 877         if command:
 878             return command
 879         if command_name in self.commands:
 880             f = partial(self.commands[command_name], self)
 881             if hasattr(self.commands[command_name], 'argtypes'):
 882                 f.argtypes = self.commands[command_name].argtypes
 883             return f
 884         return None
 885
 886
 887
 888 def quote(string):
 889     """Quote & escape string so client interprets it as single token."""
 890     quoted = []
 891     quoted += ['"']
 892     for c in string:
 893         if c in {'"', '\\'}:
 894             quoted += ['\\']
 895         quoted += [c]
 896     quoted += ['"']
 897     return ''.join(quoted)
 898
 899
 900 def stringify_yx(tuple_):
 901     """Transform tuple (y,x) into string 'Y:'+str(y)+',X:'+str(x)."""
 902     return 'Y:' + str(tuple_[0]) + ',X:' + str(tuple_[1])
 903
 904
 905
 906 if __name__ == "__main__":
 907     import sys
 908     import os
 909     if len(sys.argv) != 2:
 910         print('wrong number of arguments, expected one (game file)')
 911         exit(1)
 912     game_file_name = sys.argv[1]
 913     game = Game(game_file_name)
 914     if os.path.exists(game_file_name):
 915         if not os.path.isfile(game_file_name):
 916             print('game file name does not refer to a valid game file')
 917         else:
 918             with open(game_file_name, 'r') as f:
 919                 lines = f.readlines()
 920             for i in range(len(lines)):
 921                 line = lines[i]
 922                 print("FILE INPUT LINE %5s: %s" % (i, line), end='')
 923                 game.io.handle_input(line, store=False)
 924     else:
 925         game.io.handle_input('GEN_WORLD Y:16,X:16 bar')
 926     game.io.run_loop_with_server()