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
19 [game, graphics] = [None, None]
21 def make_player(name, colour):
23 if name[1:] == "human":
24 return HumanPlayer(name, colour)
25 s = name[1:].split(":")
30 return NetworkReceiver(colour, address)
31 if s[0] == "internal":
34 internal_agents = inspect.getmembers(sys.modules[__name__], inspect.isclass)
35 internal_agents = [x for x in internal_agents if issubclass(x[1], InternalAgent)]
36 internal_agents.remove(('InternalAgent', InternalAgent))
39 sys.stderr.write(sys.argv[0] + " : '@internal' should be followed by ':' and an agent name\n")
40 sys.stderr.write(sys.argv[0] + " : Choices are: " + str(map(lambda e : e[0], internal_agents)) + "\n")
43 for a in internal_agents:
45 return a[1](name, colour)
47 sys.stderr.write(sys.argv[0] + " : Can't find an internal agent matching \"" + s[1] + "\"\n")
48 sys.stderr.write(sys.argv[0] + " : Choices are: " + str(map(lambda e : e[0], internal_agents)) + "\n")
53 return ExternalAgent(name, colour)
57 # The main function! It does the main stuff!
60 # Apparently python will silently treat things as local unless you do this
61 # Anyone who says "You should never use a global variable" can die in a fire
69 global graphics_enabled
70 global always_reveal_states
78 # Get the important warnings out of the way
79 if platform.system() == "Windows":
80 sys.stderr.write(sys.argv[0] + " : Warning - You are using " + platform.system() + "\n")
81 if platform.release() == "Vista":
82 sys.stderr.write(sys.argv[0] + " : God help you.\n")
87 while i < len(argv)-1:
91 p = make_player(arg, colour)
92 if not isinstance(p, Player):
93 sys.stderr.write(sys.argv[0] + " : Fatal error creating " + colour + " player\n")
98 elif colour == "black":
101 sys.stderr.write(sys.argv[0] + " : Too many players (max 2)\n")
104 # Option parsing goes here
105 if arg[1] == '-' and arg[2:] == "classical":
107 elif arg[1] == '-' and arg[2:] == "quantum":
109 elif arg[1] == '-' and arg[2:] == "reveal":
110 always_reveal_states = True
111 elif (arg[1] == '-' and arg[2:] == "graphics"):
112 graphics_enabled = not graphics_enabled
113 elif (arg[1] == '-' and arg[2:].split("=")[0] == "file"):
114 # Load game from file
115 if len(arg[2:].split("=")) == 1:
118 src_file = open(arg[2:].split("=")[1].split(":")[0])
120 if len(arg[2:].split(":")) == 2:
121 max_lines = int(arg[2:].split(":")[1])
123 elif (arg[1] == '-' and arg[2:].split("=")[0] == "log"):
125 if len(arg[2:].split("=")) == 1:
126 log_file = sys.stdout
128 log_file = open(arg[2:].split("=")[1], "w")
129 elif (arg[1] == '-' and arg[2:].split("=")[0] == "delay"):
131 if len(arg[2:].split("=")) == 1:
134 turn_delay = float(arg[2:].split("=")[1])
136 elif (arg[1] == '-' and arg[2:].split("=")[0] == "timeout"):
138 if len(arg[2:].split("=")) == 1:
141 agent_timeout = float(arg[2:].split("=")[1])
143 elif (arg[1] == '-' and arg[2:] == "help"):
145 os.system("less data/help.txt") # The best help function
151 # Construct a GameThread! Make it global! Damn the consequences!
154 # Hack to stop ReplayThread from exiting
155 #if len(players) == 0:
156 # players = [HumanPlayer("dummy", "white"), HumanPlayer("dummy", "black")]
158 # Normally the ReplayThread exits if there are no players
159 # TODO: Decide which behaviour to use, and fix it
160 end = (len(players) == 0)
162 players = [Player("dummy", "white"), Player("dummy", "black")]
163 elif len(players) != 2:
164 sys.stderr.write(sys.argv[0] + " : Usage " + sys.argv[0] + " white black\n")
166 sys.stderr.write(sys.argv[0] + " : (You won't get a GUI, because --file was used, and the author is lazy)\n")
168 game = ReplayThread(players, src_file, end=end, max_lines=max_lines)
171 game = GameThread(board, players)
176 if graphics_enabled == True:
178 graphics = GraphicsThread(game.board, grid_sz = [64,64]) # Construct a GraphicsThread!
182 sys.stderr.write(sys.argv[0] + " : Got exception trying to initialise graphics\n"+str(e.message)+"\nDisabled graphics\n")
183 graphics_enabled = False
185 # If there are no players listed, display a nice pretty menu
186 if len(players) != 2:
188 players = graphics.SelectPlayers(players)
190 sys.stderr.write(sys.argv[0] + " : Usage " + sys.argv[0] + " white black\n")
193 # If there are still no players, quit
194 if players == None or len(players) != 2:
195 sys.stderr.write(sys.argv[0] + " : Graphics window closed before players chosen\n")
199 # Wrap NetworkSender players around original players if necessary
200 for i in range(len(players)):
201 if isinstance(players[i], NetworkReceiver):
202 players[i].board = board # Network players need direct access to the board
203 for j in range(len(players)):
206 if isinstance(players[j], NetworkSender) or isinstance(players[j], NetworkReceiver):
208 players[j] = NetworkSender(players[j], players[i].address)
209 players[j].board = board
211 # Connect the networked players
213 if isinstance(p, NetworkSender) or isinstance(p, NetworkReceiver):
215 graphics.board.display_grid(graphics.window, graphics.grid_sz)
216 graphics.message("Connecting to " + p.colour + " player...")
220 # If using windows, select won't work; use horrible TimeoutPlayer hack
221 if agent_timeout > 0:
222 if platform.system() == "Windows":
223 for i in range(len(players)):
224 if isinstance(players[i], ExternalAgent) or isinstance(players[i], InternalAgent):
225 players[i] = TimeoutPlayer(players[i], agent_timeout)
229 # InternalAgents get wrapped to an ExternalAgent when there is a timeout
230 # This is not confusing at all.
231 for i in range(len(players)):
232 if isinstance(players[i], InternalAgent):
233 players[i] = ExternalWrapper(players[i])
241 log_init(game.board, players)
245 game.start() # This runs in a new thread
251 error = game.error + graphics.error
257 if log_file != None and log_file != sys.stdout:
258 log_file.write("# EOF\n")
261 if src_file != None and src_file != sys.stdin:
266 # This is how python does a main() function...
267 if __name__ == "__main__":
269 sys.exit(main(sys.argv))
270 except KeyboardInterrupt:
271 sys.stderr.write(sys.argv[0] + " : Got KeyboardInterrupt. Stopping everything\n")
272 if isinstance(graphics, StoppableThread):
274 graphics.run() # Will clean up graphics because it is stopped, not run it (a bit dodgy)
276 if isinstance(game, StoppableThread):