qchess/src/main.py

   1 #!/usr/bin/python -u
   2
   3 # Do you know what the -u does? It unbuffers stdin and stdout
   4 # I can't remember why, but last year things broke without that
   5
   6 """
   7         UCC::Progcomp 2013 Quantum Chess game
   8         @author Sam Moore [SZM] "matches"
   9         @copyright The University Computer Club, Incorporated
  10                 (ie: You can copy it for not for profit purposes)
  11 """
  12
  13 # system python modules or whatever they are called
  14 import sys
  15 import os
  16 import time
  17
  18 turn_delay = 0.5
  19 sleep_timeout = None
  20 [game, graphics] = [None, None]
  21
  22 def make_player(name, colour):
  23         if name[0] == '@':
  24                 if name[1:] == "human":
  25                         return HumanPlayer(name, colour)
  26                 s = name[1:].split(":")
  27                 if s[0] == "network":
  28                         ip = None
  29                         port = 4562
  30                         #print str(s)
  31                         if len(s) > 1:
  32                                 if s[1] != "":
  33                                         ip = s[1]
  34                         if len(s) > 2:
  35                                 port = int(s[2])
  36
  37                         if ip == None:
  38                                 if colour == "black":
  39                                         port += 1
  40                         elif colour == "white":
  41                                 port += 1
  42
  43                         return NetworkPlayer(colour, Network((ip, port)), None)
  44                 if s[0] == "internal":
  45
  46                         import inspect
  47                         internal_agents = inspect.getmembers(sys.modules[__name__], inspect.isclass)
  48                         internal_agents = [x for x in internal_agents if issubclass(x[1], InternalAgent)]
  49                         internal_agents.remove(('InternalAgent', InternalAgent))
  50
  51                         if len(s) != 2:
  52                                 sys.stderr.write(sys.argv[0] + " : '@internal' should be followed by ':' and an agent name\n")
  53                                 sys.stderr.write(sys.argv[0] + " : Choices are: " + str(map(lambda e : e[0], internal_agents)) + "\n")
  54                                 return None
  55
  56                         for a in internal_agents:
  57                                 if s[1] == a[0]:
  58                                         return a[1](name, colour)
  59
  60                         sys.stderr.write(sys.argv[0] + " : Can't find an internal agent matching \"" + s[1] + "\"\n")
  61                         sys.stderr.write(sys.argv[0] + " : Choices are: " + str(map(lambda e : e[0], internal_agents)) + "\n")
  62                         return None
  63                 if s[0] == "fifo":
  64                         if len(s) > 1:
  65                                 return FifoPlayer(s[1], colour)
  66                         else:
  67                                 return FifoPlayer(str(os.getpid())+"."+colour, colour)
  68
  69         else:
  70                 return ExternalAgent(name, colour)
  71
  72
  73
  74 # The main function! It does the main stuff!
  75 def main(argv):
  76
  77         # Apparently python will silently treat things as local unless you do this
  78         # Anyone who says "You should never use a global variable" can die in a fire
  79         global game
  80         global graphics
  81
  82         global turn_delay
  83         global agent_timeout
  84         global log_files
  85         global src_file
  86         global graphics_enabled
  87         global always_reveal_states
  88         global sleep_timeout
  89
  90
  91         server_addr = None
  92
  93         max_moves = None
  94         src_file = None
  95
  96         style = "quantum"
  97         colour = "white"
  98
  99         # Get the important warnings out of the way
 100         if platform.system() == "Windows":
 101                 sys.stderr.write(sys.argv[0] + " : Warning - You are using " + platform.system() + "\n")
 102                 if platform.release() == "Vista":
 103                         sys.stderr.write(sys.argv[0] + " : God help you.\n")
 104
 105
 106         players = []
 107         i = 0
 108         while i < len(argv)-1:
 109                 i += 1
 110                 arg = argv[i]
 111                 if arg[0] != '-':
 112                         p = make_player(arg, colour)
 113                         if not isinstance(p, Player):
 114                                 sys.stderr.write(sys.argv[0] + " : Fatal error creating " + colour + " player\n")
 115                                 return 100
 116                         players.append(p)
 117                         if colour == "white":
 118                                 colour = "black"
 119                         elif colour == "black":
 120                                 pass
 121                         else:
 122                                 sys.stderr.write(sys.argv[0] + " : Too many players (max 2)\n")
 123                         continue
 124
 125                 # Option parsing goes here
 126                 if arg[1] == '-' and arg[2:] == "classical":
 127                         style = "classical"
 128                 elif arg[1] == '-' and arg[2:] == "quantum":
 129                         style = "quantum"
 130                 elif arg[1] == '-' and arg[2:] == "reveal":
 131                         always_reveal_states = True
 132                 elif (arg[1] == '-' and arg[2:] == "graphics"):
 133                         graphics_enabled = True
 134                 elif (arg[1] == '-' and arg[2:] == "no-graphics"):
 135                         graphics_enabled = False
 136                 elif (arg[1] == '-' and arg[2:].split("=")[0] == "file"):
 137                         # Load game from file
 138                         if len(arg[2:].split("=")) == 1:
 139                                 src_file = sys.stdin
 140                         else:
 141                                 f = arg[2:].split("=")[1]
 142                                 if f[0:7] == "http://":
 143                                         src_file = HttpReplay(f)
 144                                 else:
 145                                         src_file = FileReplay(f.split(":")[0])
 146
 147                                         if len(f.split(":")) == 2:
 148                                                 max_moves = int(f.split(":")[1])
 149
 150                 elif (arg[1] == '-' and arg[2:].split("=")[0] == "server"):
 151                         #debug("Server: " + str(arg[2:]))
 152                         if len(arg[2:].split("=")) <= 1:
 153                                 server_addr = True
 154                         else:
 155                                 server_addr = arg[2:].split("=")[1]
 156
 157                 elif (arg[1] == '-' and arg[2:].split("=")[0] == "log"):
 158                         # Log file
 159                         if len(arg[2:].split("=")) == 1:
 160                                 log_files.append(LogFile(sys.stdout,""))
 161                         else:
 162                                 f = arg[2:].split("=")[1]
 163                                 if f == "":
 164                                         log_files.append(LogFile(sys.stdout, ""))
 165                                 elif f[0] == '@':
 166                                         log_files.append(ShortLog(f[1:]))
 167                                 else:
 168                                         log_files.append(LogFile(open(f, "w", 0), f))
 169                 elif (arg[1] == '-' and arg[2:].split("=")[0] == "delay"):
 170                         # Delay
 171                         if len(arg[2:].split("=")) == 1:
 172                                 turn_delay = 0
 173                         else:
 174                                 turn_delay = float(arg[2:].split("=")[1])
 175
 176                 elif (arg[1] == '-' and arg[2:].split("=")[0] == "timeout"):
 177                         # Timeout
 178                         if len(arg[2:].split("=")) == 1:
 179                                 agent_timeout = -1
 180                         else:
 181                                 agent_timeout = float(arg[2:].split("=")[1])
 182                 elif (arg[1] == '-' and arg[2:].split("=")[0] == "blackout"):
 183                         # Screen saver delay
 184                         if len(arg[2:].split("=")) == 1:
 185                                 sleep_timeout = -1
 186                         else:
 187                                 sleep_timeout = float(arg[2:].split("=")[1])
 188
 189                 elif (arg[1] == '-' and arg[2:] == "help"):
 190                         # Help
 191                         os.system("less data/help.txt") # The best help function
 192                         return 0
 193
 194         # Dedicated server?
 195
 196         #debug("server_addr = " + str(server_addr))
 197
 198         if server_addr != None:
 199                 if server_addr == True:
 200                         return dedicated_server()
 201                 else:
 202                         if len(players) > 1:
 203                                 sys.stderr.write("Only a single player may be provided when --server is used\n")
 204                                 return 1
 205                         if len(players) == 1:
 206                                 return client(server_addr, players[0].name)
 207                         else:
 208                                 return client(server_addr)
 209
 210
 211         # Create the board
 212
 213         # Construct a GameThread! Make it global! Damn the consequences!
 214
 215         if src_file != None:
 216                 # Hack to stop ReplayThread from exiting
 217                 #if len(players) == 0:
 218                 #       players = [HumanPlayer("dummy", "white"), HumanPlayer("dummy", "black")]
 219
 220                 # Normally the ReplayThread exits if there are no players
 221                 # TODO: Decide which behaviour to use, and fix it
 222                 end = (len(players) == 0)
 223                 if end:
 224                         players = [Player("dummy", "white"), Player("dummy", "black")]
 225                 elif len(players) != 2:
 226                         sys.stderr.write(sys.argv[0] + " : Usage " + sys.argv[0] + " white black\n")
 227                         if graphics_enabled:
 228                                 sys.stderr.write(sys.argv[0] + " : (You won't get a GUI, because --file was used, and the author is lazy)\n")
 229                         return 44
 230                 game = ReplayThread(players, src_file, end=end, max_moves=max_moves)
 231         else:
 232                 board = Board(style)
 233                 board.max_moves = max_moves
 234                 game = GameThread(board, players)
 235
 236
 237
 238
 239         # Initialise GUI
 240         if graphics_enabled == True:
 241                 try:
 242                         graphics = GraphicsThread(game.board, grid_sz = [64,64]) # Construct a GraphicsThread!
 243
 244                         graphics.sleep_timeout = sleep_timeout
 245
 246                 except Exception,e:
 247                         graphics = None
 248                         sys.stderr.write(sys.argv[0] + " : Got exception trying to initialise graphics\n"+str(e.message)+"\nDisabled graphics\n")
 249                         graphics_enabled = False
 250
 251         # If there are no players listed, display a nice pretty menu
 252         if len(players) != 2:
 253                 if graphics != None:
 254
 255                         server_addr = graphics.SelectServer()
 256                         if server_addr != None:
 257                                 pygame.quit() # Time to say goodbye
 258                                 if server_addr == True:
 259                                         return dedicated_server()
 260                                 else:
 261                                         return client(server_addr)
 262
 263                         players = graphics.SelectPlayers(players)
 264                 else:
 265                         sys.stderr.write(sys.argv[0] + " : Usage " + sys.argv[0] + " white black\n")
 266                         return 44
 267
 268         # If there are still no players, quit
 269         if players == None or len(players) != 2:
 270                 sys.stderr.write(sys.argv[0] + " : Graphics window closed before players chosen\n")
 271                 return 45
 272
 273         old = players[:]
 274         for p in old:
 275                 if isinstance(p, NetworkPlayer):
 276                         for i in range(len(old)):
 277                                 if old[i] == p or isinstance(old[i], NetworkPlayer):
 278                                         continue
 279                                 players[i] = NetworkPlayer(old[i].colour, p.network, old[i])
 280
 281         for p in players:
 282                 #debug(str(p))
 283                 if isinstance(p, NetworkPlayer):
 284                         p.board = game.board
 285                         if not p.network.connected:
 286                                 if not p.network.server:
 287                                         time.sleep(0.2)
 288                                 p.network.connect()
 289
 290
 291         # If using windows, select won't work; use horrible TimeoutPlayer hack
 292         if agent_timeout > 0:
 293                 if platform.system() == "Windows":
 294                         for i in range(len(players)):
 295                                 if isinstance(players[i], ExternalAgent) or isinstance(players[i], InternalAgent):
 296                                         players[i] = TimeoutPlayer(players[i], agent_timeout)
 297
 298                 else:
 299                         warned = False
 300                         # InternalAgents get wrapped to an ExternalAgent when there is a timeout
 301                         # This is not confusing at all.
 302                         for i in range(len(players)):
 303                                 if isinstance(players[i], InternalAgent):
 304                                                 players[i] = ExternalWrapper(players[i])
 305
 306
 307
 308
 309
 310
 311
 312         log_init(game.board, players)
 313
 314
 315         if graphics != None:
 316                 game.start() # This runs in a new thread
 317                 graphics.run()
 318                 if game.is_alive():
 319                         game.join()
 320
 321
 322                 error = game.error + graphics.error
 323         else:
 324                 game.run()
 325                 error = game.error
 326
 327
 328         for l in log_files:
 329                 l.close()
 330
 331         if src_file != None and src_file != sys.stdin:
 332                 src_file.close()
 333
 334         sys.stdout.write(game.final_result + "\n")
 335
 336         return error
 337
 338
 339
 340
 341
 342
 343
 344
 345 # This is how python does a main() function...
 346 if __name__ == "__main__":
 347         retcode = 0
 348         try:
 349                 retcode = main(sys.argv)
 350         except KeyboardInterrupt:
 351                 sys.stderr.write(sys.argv[0] + " : Got KeyboardInterrupt. Stopping everything\n")
 352                 if isinstance(graphics, StoppableThread):
 353                         graphics.stop()
 354                         graphics.run() # Will clean up graphics because it is stopped, not run it (a bit dodgy)
 355
 356                 if isinstance(game, StoppableThread):
 357                         game.stop()
 358                         if game.is_alive():
 359                                 game.join()
 360                 retcode = 102
 361         #except Exception, e:
 362         #       sys.stderr.write(sys.argv[0] + " : " + e.message + "\n")
 363         #       retcode = 103
 364
 365         try:
 366                 sys.stdout.close()
 367         except:
 368                 pass
 369         try:
 370                 sys.stderr.close()
 371         except:
 372                 pass
 373         sys.exit(retcode)
 374
 375