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
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)
13 # system python modules or whatever they are called
20 [game, graphics] = [None, None]
22 def make_player(name, colour):
25 if name[1:] == "human":
26 return HumanPlayer(name, colour)
27 s = name[1:].split(":")
41 elif colour == "white":
44 return NetworkPlayer(colour, Network((ip, port)), None)
45 if s[0] == "internal":
48 internal_agents = inspect.getmembers(sys.modules[__name__], inspect.isclass)
49 internal_agents = [x for x in internal_agents if issubclass(x[1], InternalAgent)]
50 internal_agents.remove(('InternalAgent', InternalAgent))
53 sys.stderr.write(sys.argv[0] + " : '@internal' should be followed by ':' and an agent name\n")
54 sys.stderr.write(sys.argv[0] + " : Choices are: " + str(map(lambda e : e[0], internal_agents)) + "\n")
57 for a in internal_agents:
59 return a[1](name, colour)
61 sys.stderr.write(sys.argv[0] + " : Can't find an internal agent matching \"" + s[1] + "\"\n")
62 sys.stderr.write(sys.argv[0] + " : Choices are: " + str(map(lambda e : e[0], internal_agents)) + "\n")
66 return FifoPlayer(s[1], colour)
68 return FifoPlayer(str(os.getpid())+"."+colour, colour)
71 return ExternalAgent(name, colour)
75 # The main function! It does the main stuff!
78 # Apparently python will silently treat things as local unless you do this
79 # Anyone who says "You should never use a global variable" can die in a fire
87 global graphics_enabled
88 global always_reveal_states
101 # Get the important warnings out of the way
102 if platform.system() == "Windows":
103 sys.stderr.write(sys.argv[0] + " : Warning - You are using " + platform.system() + "\n")
104 if platform.release() == "Vista":
105 sys.stderr.write(sys.argv[0] + " : God help you.\n")
110 while i < len(argv)-1:
117 # Option parsing goes here
118 if arg[1] == '-' and arg[2:] == "classical":
120 elif arg[1] == '-' and arg[2:] == "quantum":
122 elif arg[1] == '-' and arg[2:] == "reveal":
123 always_reveal_states = True
124 elif (arg[1] == '-' and arg[2:] == "graphics"):
125 graphics_enabled = True
126 elif (arg[1] == '-' and arg[2:] == "no-graphics"):
127 graphics_enabled = False
128 elif (arg[1] == '-' and arg[2:].split("=")[0] == "file"):
129 # Load game from file
130 if len(arg[2:].split("=")) == 1:
133 f = arg[2:].split("=")[1]
134 if f[0:7] == "http://":
135 src_file = HttpReplay(f)
137 src_file = FileReplay(f.split(":")[0])
139 if len(f.split(":")) == 2:
140 max_moves = int(f.split(":")[1])
142 elif (arg[1] == '-' and arg[2:].split("=")[0] == "server"):
143 #debug("Server: " + str(arg[2:]))
144 if len(arg[2:].split("=")) <= 1:
147 server_addr = arg[2:].split("=")[1]
149 elif (arg[1] == '-' and arg[2:].split("=")[0] == "log"):
151 if len(arg[2:].split("=")) == 1:
152 log_files.append(LogFile(sys.stdout,""))
154 f = arg[2:].split("=")[1]
156 log_files.append(LogFile(sys.stdout, ""))
158 log_files.append(ShortLog(f[1:]))
160 log_files.append(LogFile(open(f, "w", 0), f))
161 elif (arg[1] == '-' and arg[2:].split("=")[0] == "delay"):
163 if len(arg[2:].split("=")) == 1:
166 turn_delay = float(arg[2:].split("=")[1])
168 elif (arg[1] == '-' and arg[2:].split("=")[0] == "timeout"):
170 if len(arg[2:].split("=")) == 1:
173 agent_timeout = float(arg[2:].split("=")[1])
174 elif (arg[1] == '-' and arg[2:].split("=")[0] == "blackout"):
176 if len(arg[2:].split("=")) == 1:
179 sleep_timeout = float(arg[2:].split("=")[1])
180 elif (arg[1] == '-' and arg[2:] == "retry-illegal"):
181 retry_illegal = not retry_illegal
182 elif (arg[1] == '-' and arg[2:] == "help"):
184 os.system("less data/help.txt") # The best help function
189 #debug("server_addr = " + str(server_addr))
191 if server_addr != None:
192 if server_addr == True:
193 return dedicated_server()
196 sys.stderr.write("Only a single player may be provided when --server is used\n")
198 if len(players) == 1:
199 return client(server_addr, players[0])
201 return client(server_addr)
203 for i in xrange(len(players)):
204 p = make_player(players[i], colour)
205 if not isinstance(p, Player):
206 sys.stderr.write(sys.argv[0] + " : Fatal error creating " + colour + " player\n")
209 if colour == "white":
211 elif colour == "black":
214 sys.stderr.write(sys.argv[0] + " : Too many players (max 2)\n")
218 # Construct a GameThread! Make it global! Damn the consequences!
221 # Hack to stop ReplayThread from exiting
222 #if len(players) == 0:
223 # players = [HumanPlayer("dummy", "white"), HumanPlayer("dummy", "black")]
225 # Normally the ReplayThread exits if there are no players
226 # TODO: Decide which behaviour to use, and fix it
227 end = (len(players) == 0)
229 players = [Player("dummy", "white"), Player("dummy", "black")]
230 elif len(players) != 2:
231 sys.stderr.write(sys.argv[0] + " : Usage " + sys.argv[0] + " white black\n")
233 sys.stderr.write(sys.argv[0] + " : (You won't get a GUI, because --file was used, and the author is lazy)\n")
235 game = ReplayThread(players, src_file, end=end, max_moves=max_moves)
238 board.max_moves = max_moves
239 game = GameThread(board, players)
240 game.retry_illegal = retry_illegal
246 if graphics_enabled == True:
248 graphics = GraphicsThread(game.board, grid_sz = [64,64]) # Construct a GraphicsThread!
250 graphics.sleep_timeout = sleep_timeout
254 sys.stderr.write(sys.argv[0] + " : Got exception trying to initialise graphics\n"+str(e.message)+"\nDisabled graphics\n")
255 graphics_enabled = False
257 # If there are no players listed, display a nice pretty menu
258 if len(players) != 2:
261 server_addr = graphics.SelectServer()
262 if server_addr != None:
263 pygame.quit() # Time to say goodbye
264 if server_addr == True:
265 return dedicated_server()
267 return client(server_addr)
269 players = graphics.SelectPlayers(players)
271 sys.stderr.write(sys.argv[0] + " : Usage " + sys.argv[0] + " white black\n")
274 # If there are still no players, quit
275 if players == None or len(players) != 2:
276 sys.stderr.write(sys.argv[0] + " : Graphics window closed before players chosen\n")
281 if isinstance(p, NetworkPlayer):
282 for i in range(len(old)):
283 if old[i] == p or isinstance(old[i], NetworkPlayer):
285 players[i] = NetworkPlayer(old[i].colour, p.network, old[i])
289 if isinstance(p, NetworkPlayer):
291 if not p.network.connected:
292 if not p.network.server:
297 # If using windows, select won't work; use horrible TimeoutPlayer hack
298 if agent_timeout > 0:
299 if platform.system() == "Windows":
300 for i in range(len(players)):
301 if isinstance(players[i], ExternalAgent) or isinstance(players[i], InternalAgent):
302 players[i] = TimeoutPlayer(players[i], agent_timeout)
306 # InternalAgents get wrapped to an ExternalAgent when there is a timeout
307 # This is not confusing at all.
308 for i in range(len(players)):
309 if isinstance(players[i], InternalAgent):
310 players[i] = ExternalWrapper(players[i])
318 log_init(game.board, players)
322 game.start() # This runs in a new thread
328 error = game.error + graphics.error
337 if src_file != None and src_file != sys.stdin:
340 sys.stdout.write(game.final_result + "\n")
351 # This is how python does a main() function...
352 if __name__ == "__main__":
355 retcode = main(sys.argv)
356 except KeyboardInterrupt:
357 sys.stderr.write(sys.argv[0] + " : Got KeyboardInterrupt. Stopping everything\n")
358 if isinstance(graphics, StoppableThread):
360 graphics.run() # Will clean up graphics because it is stopped, not run it (a bit dodgy)
362 if isinstance(game, StoppableThread):
367 #except Exception, e:
368 # sys.stderr.write(sys.argv[0] + " : " + e.message + "\n")