e87259ec7324f618e4fd6055b19fd9a52a358e9c
[progcomp2012.git] / judge / simulator / simulate.py
1 #!/usr/bin/python -u
2
3 '''
4  simulate.py - simulation script for the 2012 UCC Programming Competition
5         NOTE: This is not the manager program for a stratego game
6         It merely calls the manager program as appropriate, and records results
7         Plays exactly ONE round, but does not overwrite previously played rounds
8         eg: run once to generate round1.results, twice to generate round2.results etc
9         Also generates total.scores based on results from every round.
10
11         Now (sortof) generates .html files to display results in a prettiful manner.
12         
13         THIS FILE IS TERRIBLE
14
15  author Sam Moore (matches) [SZM]
16  website http://matches.ucc.asn.au/stratego
17  email progcomp@ucc.asn.au or matches@ucc.asn.au
18  git git.ucc.asn.au/progcomp2012.git
19 '''
20
21 import os
22 import sys
23 from time import time
24
25 #Global variables/arguments
26
27 baseDirectory = "../.." #Base directory for results, logs, agents
28 nGames = 2 #Number of games played by each agent against each opponent. Half will be played as RED, half as BLUE. If nGames <= 1, then no games will be played (useful for dry run?)
29 nRounds = 1
30
31 timeoutValue = 2
32
33 if len(sys.argv) >= 2:
34         nRounds = int(sys.argv[1])
35 if len(sys.argv) >= 3:
36         nGames = int(sys.argv[2])
37         if nGames % 2 != 0:
38                 print "Warning: nGames should be even. "+str(nGames)+" specified, but only " + str(int(nGames/2) * 2)+" will be played!"
39 if len(sys.argv) >= 4:
40         baseDirectory = sys.argv[3]
41 if len(sys.argv) >= 6:
42         print "Useage: " +sys.argv[0] + " [nRounds=1] [nGames=10] [baseDirectory=\""+baseDirectory+"\"] [managerPath=baseDirectory+\"/judge/manager/stratego\"]"
43         sys.exit(1)
44
45 resultsDirectory = baseDirectory+"/web/results/" #Where results will go (results are in the form of text files of agent names and scores)
46 logDirectory = baseDirectory+"/web/log/" #Where log files go (direct output of manager program)
47 agentsDirectory = baseDirectory+"/agents/" #Where agents are found (each agent has its own subdirectory within this directory)
48 managerPath = baseDirectory+"/judge/manager/stratego" #Path to the executable that plays the games
49 if len(sys.argv) >= 5:
50         managerPath = sys.argv[5] 
51
52
53 #Score dictionary - Tuple is of the form: (end score, other score, other result) where end is the player on whose turn the result occurs, other is the other player, other result indicates what to record the outcome as for the other player.
54 scores = {"VICTORY":(0.3,0.1, "DEFEAT"), "DEFEAT":(0.1,0.3, "VICTORY"), "SURRENDER":(0,0.3, "VICTORY"), "DRAW":(0.2,0.2, "DRAW"), "DRAW_DEFAULT":(0.1,0.1, "DRAW_DEFAULT"), "ILLEGAL":(-0.1,0.2, "DEFAULT"), "DEFAULT":(0.2,-0.1, "ILLEGAL"), "BOTH_ILLEGAL":(-0.1,-0.1, "BOTH_ILLEGAL"), "INTERNAL_ERROR":(0,0, "INTERNAL_ERROR"), "BAD_SETUP":(0,0,"BAD_SETUP")}
55
56
57 #Verbose - print lots of useless stuff about what you are doing (kind of like matches talking on irc...)
58 verbose = True
59
60
61
62 #Check the manager program exists TODO: And is executable!
63 if os.path.exists(managerPath) == False:
64         print "Manager program at \""+managerPath+"\" doesn't exist!"
65         sys.exit(1)
66
67 #Make necessary directories
68
69 if os.path.exists(resultsDirectory) == False:
70         os.mkdir(resultsDirectory) #Make the results directory if it didn't exist
71
72         
73 #Identify the round number by reading from the "info" file in the results directory, if it doesn't exist then start at round 1.
74 if os.path.exists(resultsDirectory+"info") == False:
75         totalRounds = 1
76 else:
77         info = open(resultsDirectory+"info", "r")
78         totalRounds = int(info.readline().strip())
79         info.close()
80         os.remove(resultsDirectory+"info")
81
82 info = open(resultsDirectory+"info", "w")
83 info.write(str(totalRounds + nRounds) + "\n")
84 info.close()
85         
86
87
88 if os.path.exists(logDirectory) == False:
89         os.mkdir(logDirectory) #Make the log directory if it didn't exist
90
91
92 startTime = time() #Record time at which simulation starts
93
94 if verbose:
95         if nRounds > 1:
96                 print "Simulating " + str(nRounds) + " rounds (" + str(totalRounds) + " to " + str(totalRounds + nRounds-1) + ")"
97         else:
98                 print "Simulating one round."
99         print ""
100         print "Identifying possible agents in \""+agentsDirectory+"\""
101
102 #Get all agent names from agentsDirectory
103 agentNames = os.listdir(agentsDirectory) 
104 agents = []
105 for name in agentNames:
106         if verbose:
107                 sys.stdout.write("Scan \""+name+"\"... ")
108         if os.path.isdir(agentsDirectory+name) == False: #Remove non-directories
109                 if verbose:
110                         sys.stdout.write(" Invalid! (Not a directory)\n")
111                 continue
112
113         if os.path.exists(agentsDirectory+name+"/info") == False: #Try and find the special "info" file in each directory; ignore if it doesn't exist   
114                 if verbose:
115                         sys.stdout.write(" Invalid! (No \"info\" file found)\n")
116                 continue
117
118         infoFile = open(agentsDirectory+name+"/info", "r")
119         agentExecutable = agentsDirectory+name+"/"+(infoFile.readline().strip())
120         author = infoFile.readline().strip()
121         language = infoFile.readline().strip()
122         description = ""
123         while True:
124                 line = infoFile.readline()
125                 if len(line) > 0:
126                         description += line
127                 else:
128                         break
129         infoFile.close()
130         
131         if os.path.exists(agentExecutable.split(" ")[0]) == False:
132                 if verbose:
133                         sys.stdout.write(" Invalid! (Path: \""+agentExecutable+"\" does not exist!)\n")
134                 continue
135
136
137         if verbose:
138                 sys.stdout.write(" Valid! (Path: \""+agentExecutable+"\")\n")
139
140         #Convert array of valid names into array of dictionaries containing information about each agent
141         #I'm starting to like python...
142         agents.append({"name":name, "path":agentExecutable, "author":author, "language":language, "description":description, "score":[0], "VICTORY":[], "DEFEAT":[], "DRAW":[], "ILLEGAL":[], "DEFAULT":[], "INTERNAL_ERROR":[], "SURRENDER":[], "DRAW_DEFAULT":[], "BOTH_ILLEGAL":[], "BAD_SETUP":[], "ALL":[], "totalScore":0, "Wins":0, "Losses":0, "Draws":0, "Illegal":0, "Errors":0})
143
144 if len(agents) == 0:
145         print "Couldn't find any agents! Check paths (Edit this script) or generate \"info\" files for agents."
146         sys.exit(0)
147 if verbose:
148         print "Total: " + str(len(agents)) + " valid agents found (From "+str(len(agentNames))+" possibilities)"
149         print ""
150
151 #Prepare the pretty .html files if they don't exist
152 if verbose:
153         print "Preparing .html results files..."
154
155
156 if os.path.exists(resultsDirectory + "index.html") == True:
157         os.remove(resultsDirectory + "index.html") #Delete the file
158 totalFile = open(resultsDirectory + "index.html", "w")
159 totalFile.write("<html>\n<head>\n <title> Round in progress... </title>\n</head>\n<body>\n")
160 if nRounds > 1:
161         totalFile.write("<h1> Rounds " + str(totalRounds) + " to " + str(totalRounds + nRounds-1) + " in progress...</h1>\n")
162 else:
163         totalFile.write("<h1> Round " + str(totalRounds) + " in progress...</h1>\n")
164 totalFile.write("<p> Please wait for the rounds to finish. You can view the current progress by watching the <a href = \"../log\"/>Log Files</a> </p>")
165 if totalRounds > 1:
166         totalFile.write("<h2> Round Summaries </h2>\n")
167         totalFile.write("<table border=\"0\" cellpadding=\"10\">\n")
168         for i in range(1, totalRounds):
169                 totalFile.write("<tr> <td> <a href=round"+str(i)+".html>Round " + str(i) + "</a> </td> </tr>\n")
170         totalFile.write("</table>\n")
171
172 totalFile.write("</body>\n<!-- Total Results file autogenerated by \"" + sys.argv[0] + "\" at time " + str(time()) + " -->\n</html>\n\n")
173 totalFile.close()
174
175
176 for agent in agents:
177         if os.path.exists(resultsDirectory+agent["name"] + ".html") == False:
178                 agentFile = open(resultsDirectory+agent["name"] + ".html", "w")
179                 agentFile.write("<html>\n<head>\n <title> " + agent["name"] + " overview</title>\n</head>\n<body>\n<h1> Overview for " + agent["name"]+" </h1>\n")
180                 agentFile.write("<table border=\"0\" cellpadding=\"10\">\n")
181                 agentFile.write("<tr> <th> Name </th> <th> Author </th> <th> Language </th> </tr>\n")
182                 agentFile.write("<tr> <td> "+agent["name"]+" </td> <td> "+agent["author"]+" </td> <td> "+agent["language"]+" </td> </tr>\n")
183                 agentFile.write("</table>\n");
184
185                 agentFile.write("<p> <b>Description</b> </p>\n")
186                 agentFile.write("<p> " + agent["description"] + " </p>\n")
187                 agentFile.close()
188
189         os.rename(resultsDirectory+agent["name"] + ".html", "tmpfile")
190         
191         oldFile = open("tmpfile", "r")
192         agentFile = open(resultsDirectory+agent["name"] + ".html", "w")
193         line = oldFile.readline()
194         while line != "":
195                 #if verbose:
196                 #       print "Interpreting line \"" + line.strip() + "\""
197                 if line.strip() == "</body>" or line.strip() == "<!--end-->":
198                         break
199                 elif line == "<h3> Round Overview </h3>\n":
200                         agentFile.write(line)
201                         line = oldFile.readline()
202                         agentFile.write(line)
203                         line = oldFile.readline()
204                         if line == "<tr> <th> Score </th> <th> Wins </th> <th> Losses </th> <th> Draws </th> <th> Illegal </th> <th> Errors </th></tr>\n":
205                                 #sys.stdout.write("Adding scores... " + line + "\n")
206                                 agentFile.write(line)
207                                 line = oldFile.readline()
208                                 values = line.split(' ')
209                                 agent["totalScore"] += int(values[2].strip())
210                                 agent["Wins"] += int(values[5].strip())
211                                 agent["Losses"] += int(values[8].strip())
212                                 agent["Draws"] += int(values[11].strip())
213                                 agent["Illegal"] += int(values[14].strip())
214                                 agent["Errors"] += int(values[17].strip())
215                 agentFile.write(line)
216                 line = oldFile.readline()
217
218         if verbose:
219                 print "Prepared results file \"" + resultsDirectory+agent["name"] + ".html\"."
220         oldFile.close()
221         agentFile.close()
222         os.remove("tmpfile")
223
224 if verbose:
225         print ""
226
227 #Do each round...
228 totalGames = nGames/2 * len(agents) * (len(agents)-1)
229 for roundNumber in range(totalRounds, totalRounds + nRounds):
230
231         if os.path.exists(logDirectory + "round"+str(roundNumber)) == False:
232                 os.mkdir(logDirectory + "round"+str(roundNumber)) #Check there is a directory for this round's logs
233
234         for agent in agents:
235                 agent.update({"name":agent["name"], "path":agent["path"],  "score":[0], "VICTORY":[], "DEFEAT":[], "DRAW":[], "ILLEGAL":[], "DEFAULT":[], "INTERNAL_ERROR":[], "SURRENDER":[], "DRAW_DEFAULT":[], "BOTH_ILLEGAL":[], "BAD_SETUP":[], "ALL":[]})
236
237         
238         print "Commencing ROUND " + str(roundNumber) + " combat!"
239         print "Total: " + str(totalGames) + " games to be played. This could take a while..."
240
241
242
243         managerErrors = 0
244         #This double for loop simulates a round robin, with each agent getting the chance to play as both red and blue against every other agent.
245         gameNumber = 0
246         for red in agents:  #for each agent playing as red,
247                 for blue in agents: #against each other agent, playing as blue
248                         if red == blue:
249                                 continue #Exclude battles against self
250                         
251                         
252                         for i in range(1, nGames/2 + 1):
253                                 gameNumber += 1
254                                 gameID = str(roundNumber) + "." + str(gameNumber)
255                                 #Play a game and read the result. Note the game is logged to a file based on the agent's names
256                                 if verbose:
257                                         sys.stdout.write("Agents: \""+red["name"]+"\" and \""+blue["name"]+"\" playing game (ID: " + gameID + ") ... ")
258                                 logFile = logDirectory + "round"+str(roundNumber) + "/"+red["name"]+".vs."+blue["name"]+"."+str(gameID)
259                                 errorLog = [logDirectory + "error/" + red["name"] + "."+str(gameID), logDirectory + "error/" + blue["name"] + "."+str(gameID)]
260                                 #Run the game, outputting to logFile; stderr of (both) AI programs is directed to logFile.stderr
261                                 outline = os.popen(managerPath + " -m 1000 -o " + logFile + " -T " + str(timeoutValue) + " \"" + red["path"] + "\" \"" + blue["path"] + "\" 2>> " + logFile+".stderr", "r").read()
262                                 
263                                 #os.system("mv tmp.mp4 " + logFile + ".mp4")
264                                 
265                                 #If there were no errors, get rid of the stderr file
266                                 if os.stat(logFile+".stderr").st_size <= 0:
267                                         os.remove(logFile+".stderr")
268                                 results = outline.split(' ')
269                         
270                                 if len(results) != 6:
271                                         if verbose:
272                                                 sys.stdout.write("Garbage output! \"" + outline + "\"\n")
273                                         red["INTERNAL_ERROR"].append((blue["name"], gameID, scores["INTERNAL_ERROR"][0]))
274                                         blue["INTERNAL_ERROR"].append((red["name"], gameID, scores["INTERNAL_ERROR"][0]))
275                                         red["ALL"].append((blue["name"], gameID, scores["INTERNAL_ERROR"][0], "INTERNAL_ERROR", "RED"))
276                                         blue["ALL"].append((red["name"], gameID, scores["INTERNAL_ERROR"][0], "INTERNAL_ERROR", "BLUE"))
277                                         managerErrors += 1
278                                 else:
279
280                                         if results[1] == "RED":
281                                                 endColour = red
282                                                 otherColour = blue
283                                                 endStr = "RED"
284                                                 otherStr = "BLUE"
285                                         elif results[1] == "BLUE":
286                                                 endColour = blue
287                                                 otherColour = red
288                                                 endStr = "BLUE"
289                                                 otherStr = "RED"
290
291
292                                         if results[1] == "BOTH":
293                                                 red["INTERNAL_ERROR"].append((blue["name"], gameID, scores["INTERNAL_ERROR"][0]))
294                                                 blue["INTERNAL_ERROR"].append((red["name"], gameID, scores["INTERNAL_ERROR"][0]))
295                                                 red["ALL"].append((blue["name"], gameID, scores["INTERNAL_ERROR"][0], "INTERNAL_ERROR", "RED"))
296                                                 blue["ALL"].append((red["name"], gameID, scores["INTERNAL_ERROR"][0], "INTERNAL_ERROR", "BLUE"))
297                                                 managerErrors += 1
298                                         else:
299                                                 endColour["score"].insert(0,endColour["score"][0] + scores[results[2]][0])
300                                                 endColour[results[2]].append((otherColour["name"], gameID, scores[results[2]][0]))
301                                                 endColour["ALL"].append((otherColour["name"], gameID, scores[results[2]][0], results[2], endStr))
302                                                 otherColour["score"].insert(0, otherColour["score"][0] + scores[results[2]][1])
303                                                 otherColour[scores[results[2]][2]].append((endColour["name"], gameID, scores[results[2]][1]))
304                                                 otherColour["ALL"].append((endColour["name"], gameID, scores[results[2]][1], scores[results[2]][2], otherStr))
305                                                 #Write scores to raw text files
306                                                 for agent in [endColour, otherColour]:
307                                                         scoreFile = open(resultsDirectory + agent["name"] + ".scores", "a")
308                                                         scoreFile.write(str(agent["totalScore"] + agent["score"][0]) + "\n")
309                                                         scoreFile.close()
310                                                 
311                                                 
312
313                                         
314                                         if verbose:
315                                                 sys.stdout.write(" Result \"")
316                                                 for ii in range(1, len(results)):
317                                                         sys.stdout.write(results[ii].strip())
318                                                         if ii < (len(results) - 1):
319                                                                 sys.stdout.write(" ")
320                                                 sys.stdout.write("\"\n")
321                 
322         if verbose:
323                 print "Completed combat. Total of " + str(gameNumber) + " games played. "
324         if managerErrors != 0:
325                 print "WARNING: Registered "+str(managerErrors)+" errors. Check the manager program."
326
327         if verbose:
328                 print "" 
329         #We should now have complete score values.
330         if verbose:     
331                 print "RESULTS FOR ROUND " + str(roundNumber)
332
333         #totalFile = open(resultsDirectory+"total.scores", "w") #Recreate the file
334                 #for agent in agents:   
335                 #totalFile.write(agent["name"] + " " + str(agent["totalScore"]) +"\n") #Write the total scores in descending order
336                 #if verbose:
337                 #               print "Agent: " + str(agent)
338         
339
340         if verbose:
341                 print "Updating pretty .html files... "
342
343         for agent in agents:
344                 agentFile = open(resultsDirectory + agent["name"]+".html", "a")
345                 agentFile.write("<h2> Round " + str(roundNumber) + "</h2>\n")
346                 agentFile.write("<h3> Round Overview </h3>\n")
347                 agentFile.write("<table border=\"0\" cellpadding=\"10\">\n")
348                 agentFile.write("<tr> <th> Score </th> <th> Wins </th> <th> Losses </th> <th> Draws </th> <th> Illegal </th> <th> Errors </th></tr>\n")
349                 agentFile.write("<tr> <td> "+str(agent["score"][0])+" </td> <td> "+str(len(agent["VICTORY"]) + len(agent["DEFAULT"]))+" </td> <td> "+str(len(agent["DEFEAT"]) + len(agent["SURRENDER"]))+" </td> <td> "+str(len(agent["DRAW"]) + len(agent["DRAW_DEFAULT"]))+" </td> <td> "+str(len(agent["ILLEGAL"]) + len(agent["BOTH_ILLEGAL"]) + len(agent["BAD_SETUP"]))+" </td> <td> " +str(len(agent["INTERNAL_ERROR"]))+" </td> </tr>\n")
350
351                 agentFile.write("</table>\n")
352                 agentFile.write("<p> <a href=round"+str(roundNumber)+".html>Round "+str(roundNumber) + " Scoreboard</a></p>\n")
353
354                 agentFile.write("<h3> Detailed </h3>\n")
355                 agentFile.write("<table border=\"0\" cellpadding=\"10\">\n")
356                 agentFile.write("<tr> <th> Game ID </th> <th> Opponent </th> <th> Played as </th> <th> Outcome </th> <th> Score </th> <th> Accumulated Score </th> </tr> </th>\n")
357                 
358                 
359
360                 for index in range(0, len(agent["ALL"])):
361                         if agent["ALL"][index][4] == "RED":
362                                 logFile = "../log/round"+str(roundNumber) + "/"+agent["name"]+".vs."+agent["ALL"][index][0]+"."+str(agent["ALL"][index][1])
363                         else:
364                                 logFile = "../log/round"+str(roundNumber) + "/"+agent["ALL"][index][0]+".vs."+agent["name"]+"."+str(agent["ALL"][index][1])
365                         agentFile.write("<tr> <td> <a href="+logFile+">" + str(agent["ALL"][index][1]) + " </a> </td> <td> <a href="+agent["ALL"][index][0]+".html>"+agent["ALL"][index][0] + " </a> </td> <td> " + agent["ALL"][index][4] + " </td> <td> " + agent["ALL"][index][3] + " </td> <td> " + str(agent["ALL"][index][2]) + "</td> <td> " + str(agent["score"][len(agent["score"])-index -2]) + " </td> </tr> </th>\n")
366                 agentFile.write("</table>\n")
367                 
368                 agent["totalScore"] += agent["score"][0]
369                 agent["Wins"] += len(agent["VICTORY"]) + len(agent["DEFAULT"])
370                 agent["Losses"] += len(agent["DEFEAT"]) + len(agent["SURRENDER"])
371                 agent["Draws"] += len(agent["DRAW"]) + len(agent["DRAW_DEFAULT"])
372                 agent["Illegal"] += len(agent["ILLEGAL"]) + len(agent["BOTH_ILLEGAL"]) + len(agent["BAD_SETUP"])
373                 agent["Errors"] += len(agent["INTERNAL_ERROR"])
374
375                 agentFile.write("<h3> Accumulated Results </h3>\n")
376                 agentFile.write("<table border=\"0\" cellpadding=\"10\">\n")
377                 agentFile.write("<tr> <th> Score </th> <th> Wins </th> <th> Losses </th> <th> Draws </th> <th> Illegal </th> <th> Errors </th></tr>\n")
378                 agentFile.write("<tr> <td> "+str(agent["totalScore"])+" </td> <td> "+str(agent["Wins"])+" </td> <td> "+str(agent["Losses"])+" </td> <td> "+str(agent["Draws"])+" </td> <td> "+str(agent["Illegal"])+" </td> <td> " +str(agent["Errors"])+" </td> </tr>\n")
379
380                 agentFile.write("</table>\n")
381
382
383                 
384
385
386                 agentFile.close()       
387
388         #Update round file
389         roundFile = open(resultsDirectory + "round"+str(roundNumber)+".html", "w")
390         roundFile.write("<html>\n<head>\n <title> Round " +str(roundNumber)+ " Overview </title>\n</head>\n<body>\n")
391         roundFile.write("<h1> Round " +str(roundNumber)+ " Overview </h1>\n")
392         roundFile.write("<table border=\"0\" cellpadding=\"10\">\n")
393         roundFile.write("<tr> <th> Name </th> <th> Score </th> <th> Total Score </th> </tr>\n")
394         agents.sort(key = lambda e : e["score"][0], reverse=True)
395         for agent in agents:
396                 roundFile.write("<tr> <td> <a href="+agent["name"]+".html>"+agent["name"] + " </a> </td> <td> " + str(agent["score"][0]) + " </td> <td> " + str(agent["totalScore"]) + " </td> </tr>\n")
397         roundFile.write("</table>\n")
398
399         command = "cp scores.plt " + resultsDirectory + "scores.plt;"
400         os.system(command)
401
402         scorePlot = open(resultsDirectory + "scores.plt", "a")
403         scorePlot.write("plot ")
404         for i in range(0, len(agents)):
405                 if i > 0:
406                         scorePlot.write(", ")
407                 scorePlot.write("\""+agents[i]["name"]+".scores\" using ($0+1):1 with linespoints title \""+agents[i]["name"]+"\"")
408
409         scorePlot.write("\nexit\n")
410         scorePlot.close()
411
412         command = "d=$(pwd); cd " + resultsDirectory + ";"
413         command += "gnuplot scores.plt;"
414         command += "rm -f scores.plt;"
415         command += "mv scores.png round"+str(roundNumber)+".png;"
416         command += "cd $d;"
417         os.system(command)
418
419         roundFile.write("<h2> Accumulated Scores - up to Round " + str(roundNumber)+" </h2>\n")
420         roundFile.write("<img src=\"round"+str(roundNumber)+".png\" alt = \"round"+str(roundNumber)+".png\" title = \"round"+str(roundNumber)+".png\" width = \"640\" height = \"480\"/>\n")
421
422         roundFile.write("<p> <a href=index.html>Current Scoreboard</a></p>\n")
423         roundFile.write("</body>\n<!-- Results file for Round " + str(roundNumber) + " autogenerated by \"" + sys.argv[0] + "\" at time " + str(time()) + " -->\n</html>\n\n")
424         roundFile.close()
425
426
427         
428         
429
430 if verbose:
431         print "Finalising .html files... "
432 for agent in agents:
433         agentFile = open(resultsDirectory + agent["name"]+".html", "a")
434         agentFile.write("<!--end-->\n")
435         #Write a graph
436         #Comment out if you don't have gnuplot
437
438         command  = "rm -f " + agent["name"] + ".png;"
439         command += "cp template.plt " + resultsDirectory + agent["name"] + ".plt;"
440         command += "d=$(pwd); cd " + resultsDirectory + ";"
441         command += "sed -i \"s:\[NAME\]:"+agent["name"]+":g\" " +resultsDirectory + agent["name"]+".plt;"
442         command += "gnuplot " + resultsDirectory + agent["name"]+".plt;"
443         command += "rm -f " + resultsDirectory + agent["name"] + ".plt;"
444         command += "cd $d;"
445         os.system(command)
446         agentFile.write("<!--end-->\n")
447         agentFile.write("<h3> Score Graph </h3>\n")
448         agentFile.write("<img src=\""+agent["name"]+".png\" alt=\""+agent["name"]+".png\" title=\""+agent["name"]+".png\" width=\"640\" height=\"480\"/>\n")
449
450         #Link to main file
451         agentFile.write("<p> <a href=\"index.html\"/>Total Statistics</a> </p>\n")
452
453         agentFile.write("</body>\n<!-- Results file for \"" + agent["name"] + "\" autogenerated by \"" + sys.argv[0] + "\" at time " + str(time()) + " -->\n</html>\n\n")
454         agentFile.close()
455
456 if os.path.exists(resultsDirectory + "index.html") == True:
457         os.remove(resultsDirectory + "index.html") #Delete the file
458
459 totalFile = open(resultsDirectory + "index.html", "w")
460 totalFile.write("<html>\n<head>\n <title> Total Overview </title>\n</head>\n<body>\n")
461 totalFile.write("<h1> Total Overview </h1>\n")
462 totalFile.write("<table border=\"0\" cellpadding=\"10\">\n")
463 totalFile.write("<tr> <th> Name </th> <th> Total Score </th> </tr>\n")
464 agents.sort(key = lambda e : e["totalScore"], reverse=True)
465 for agent in agents:
466         totalFile.write("<tr> <td> <a href="+agent["name"]+".html>"+agent["name"] + " </a> </td> <td> " + str(agent["totalScore"]) + " </td> </tr>\n")
467 totalFile.write("</table>\n")
468
469 totalFile.write("<h2> Score Graph </h2>\n")
470
471
472 command = "d=$(pwd);"
473 command += "cd " + resultsDirectory + ";"
474 command += "rm -f scores.png;"
475 command += "cp round"+str(roundNumber)+".png scores.png;"
476 command += "cd $d;"
477 os.system(command)
478
479 totalFile.write("<img src=\"scores.png\" alt=\"scores.png\" title=\"scores.png\" width=\"640\" height=\"480\"/>\n")
480
481
482 totalFile.write("<h2> Round Summaries </h2>\n")
483 totalFile.write("<table border=\"0\" cellpadding=\"10\">\n")
484 for i in range(1, totalRounds+1):
485         totalFile.write("<tr> <td> <a href=round"+str(i)+".html>Round " + str(i) + "</a> </td> </tr>\n")
486 totalFile.write("</table>\n")
487
488 totalFile.write("</body>\n<!-- Total Results file autogenerated by \"" + sys.argv[0] + "\" at time " + str(time()) + " -->\n</html>\n\n")
489 totalFile.close()
490
491 #Write results to a raw text file as well
492 textResults = open(resultsDirectory + "total.txt", "w")
493 for agent in agents:
494         textResults.write(agent["name"] + " " + str(agent["totalScore"]) + "\n")
495 textResults.close()
496         
497 if verbose:
498         print "Done!"
499
500 endTime = time()
501 print "Completed simulating " + str(nRounds) + " rounds in " + str(endTime - startTime) + " seconds."
502 sys.exit(0)

UCC git Repository :: git.ucc.asn.au