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 [email protected] or [email protected]
  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":(3,1, "DEFEAT"), "DEFEAT":(1,3, "VICTORY"), "SURRENDER":(0,3, "VICTORY"), "DRAW":(2,2, "DRAW"), "DRAW_DEFAULT":(1,1, "DRAW_DEFAULT"), "ILLEGAL":(-1,2, "DEFAULT"), "DEFAULT":(2,-1, "ILLEGAL"), "BOTH_ILLEGAL":(-1,-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 + " -o " + logFile + " -T " + str(timeoutValue) + " \"" + red["path"] + "\" \"" + blue["path"] + "\" 2>> " + logFile+".stderr", "r").read()
 262                                 #os.system("mv tmp.mp4 " + logFile + ".mp4")
 263
 264                                 #If there were no errors, get rid of the stderr file
 265                                 if os.stat(logFile+".stderr").st_size <= 0:
 266                                         os.remove(logFile+".stderr")
 267                                 results = outline.split(' ')
 268
 269                                 if len(results) != 6:
 270                                         if verbose:
 271                                                 sys.stdout.write("Garbage output! \"" + outline + "\"\n")
 272                                         red["INTERNAL_ERROR"].append((blue["name"], gameID, scores["INTERNAL_ERROR"][0]))
 273                                         blue["INTERNAL_ERROR"].append((red["name"], gameID, scores["INTERNAL_ERROR"][0]))
 274                                         red["ALL"].append((blue["name"], gameID, scores["INTERNAL_ERROR"][0], "INTERNAL_ERROR", "RED"))
 275                                         blue["ALL"].append((red["name"], gameID, scores["INTERNAL_ERROR"][0], "INTERNAL_ERROR", "BLUE"))
 276                                         managerErrors += 1
 277                                 else:
 278
 279                                         if results[1] == "RED":
 280                                                 endColour = red
 281                                                 otherColour = blue
 282                                                 endStr = "RED"
 283                                                 otherStr = "BLUE"
 284                                         elif results[1] == "BLUE":
 285                                                 endColour = blue
 286                                                 otherColour = red
 287                                                 endStr = "BLUE"
 288                                                 otherStr = "RED"
 289
 290
 291                                         if results[1] == "BOTH":
 292                                                 red["INTERNAL_ERROR"].append((blue["name"], gameID, scores["INTERNAL_ERROR"][0]))
 293                                                 blue["INTERNAL_ERROR"].append((red["name"], gameID, scores["INTERNAL_ERROR"][0]))
 294                                                 red["ALL"].append((blue["name"], gameID, scores["INTERNAL_ERROR"][0], "INTERNAL_ERROR", "RED"))
 295                                                 blue["ALL"].append((red["name"], gameID, scores["INTERNAL_ERROR"][0], "INTERNAL_ERROR", "BLUE"))
 296                                                 managerErrors += 1
 297                                         else:
 298                                                 endColour["score"].insert(0,endColour["score"][0] + scores[results[2]][0])
 299                                                 endColour[results[2]].append((otherColour["name"], gameID, scores[results[2]][0]))
 300                                                 endColour["ALL"].append((otherColour["name"], gameID, scores[results[2]][0], results[2], endStr))
 301                                                 otherColour["score"].insert(0, otherColour["score"][0] + scores[results[2]][1])
 302                                                 otherColour[scores[results[2]][2]].append((endColour["name"], gameID, scores[results[2]][1]))
 303                                                 otherColour["ALL"].append((endColour["name"], gameID, scores[results[2]][1], scores[results[2]][2], otherStr))
 304                                                 #Write scores to raw text files
 305                                                 for agent in [endColour, otherColour]:
 306                                                         scoreFile = open(resultsDirectory + agent["name"] + ".scores", "a")
 307                                                         scoreFile.write(str(agent["totalScore"] + agent["score"][0]) + "\n")
 308                                                         scoreFile.close()
 309
 310
 311
 312
 313                                         if verbose:
 314                                                 sys.stdout.write(" Result \"")
 315                                                 for ii in range(1, len(results)):
 316                                                         sys.stdout.write(results[ii].strip())
 317                                                         if ii < (len(results) - 1):
 318                                                                 sys.stdout.write(" ")
 319                                                 sys.stdout.write("\"\n")
 320
 321         if verbose:
 322                 print "Completed combat. Total of " + str(gameNumber) + " games played. "
 323         if managerErrors != 0:
 324                 print "WARNING: Registered "+str(managerErrors)+" errors. Check the manager program."
 325
 326         if verbose:
 327                 print ""
 328         #We should now have complete score values.
 329         if verbose:
 330                 print "RESULTS FOR ROUND " + str(roundNumber)
 331
 332         #totalFile = open(resultsDirectory+"total.scores", "w") #Recreate the file
 333                 #for agent in agents:
 334                 #totalFile.write(agent["name"] + " " + str(agent["totalScore"]) +"\n") #Write the total scores in descending order
 335                 #if verbose:
 336                 #               print "Agent: " + str(agent)
 337
 338
 339         if verbose:
 340                 print "Updating pretty .html files... "
 341
 342         for agent in agents:
 343                 agentFile = open(resultsDirectory + agent["name"]+".html", "a")
 344                 agentFile.write("<h2> Round " + str(roundNumber) + "</h2>\n")
 345                 agentFile.write("<h3> Round Overview </h3>\n")
 346                 agentFile.write("<table border=\"0\" cellpadding=\"10\">\n")
 347                 agentFile.write("<tr> <th> Score </th> <th> Wins </th> <th> Losses </th> <th> Draws </th> <th> Illegal </th> <th> Errors </th></tr>\n")
 348                 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")
 349
 350                 agentFile.write("</table>\n")
 351                 agentFile.write("<p> <a href=round"+str(roundNumber)+".html>Round "+str(roundNumber) + " Scoreboard</a></p>\n")
 352
 353                 agentFile.write("<h3> Detailed </h3>\n")
 354                 agentFile.write("<table border=\"0\" cellpadding=\"10\">\n")
 355                 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")
 356
 357
 358
 359                 for index in range(0, len(agent["ALL"])):
 360                         if agent["ALL"][index][4] == "RED":
 361                                 logFile = "../log/round"+str(roundNumber) + "/"+agent["name"]+".vs."+agent["ALL"][index][0]+"."+str(agent["ALL"][index][1])
 362                         else:
 363                                 logFile = "../log/round"+str(roundNumber) + "/"+agent["ALL"][index][0]+".vs."+agent["name"]+"."+str(agent["ALL"][index][1])
 364                         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")
 365                 agentFile.write("</table>\n")
 366
 367                 agent["totalScore"] += agent["score"][0]
 368                 agent["Wins"] += len(agent["VICTORY"]) + len(agent["DEFAULT"])
 369                 agent["Losses"] += len(agent["DEFEAT"]) + len(agent["SURRENDER"])
 370                 agent["Draws"] += len(agent["DRAW"]) + len(agent["DRAW_DEFAULT"])
 371                 agent["Illegal"] += len(agent["ILLEGAL"]) + len(agent["BOTH_ILLEGAL"]) + len(agent["BAD_SETUP"])
 372                 agent["Errors"] += len(agent["INTERNAL_ERROR"])
 373
 374                 agentFile.write("<h3> Accumulated Results </h3>\n")
 375                 agentFile.write("<table border=\"0\" cellpadding=\"10\">\n")
 376                 agentFile.write("<tr> <th> Score </th> <th> Wins </th> <th> Losses </th> <th> Draws </th> <th> Illegal </th> <th> Errors </th></tr>\n")
 377                 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")
 378
 379                 agentFile.write("</table>\n")
 380
 381
 382
 383
 384
 385                 agentFile.close()
 386
 387         #Update round file
 388         roundFile = open(resultsDirectory + "round"+str(roundNumber)+".html", "w")
 389         roundFile.write("<html>\n<head>\n <title> Round " +str(roundNumber)+ " Overview </title>\n</head>\n<body>\n")
 390         roundFile.write("<h1> Round " +str(roundNumber)+ " Overview </h1>\n")
 391         roundFile.write("<table border=\"0\" cellpadding=\"10\">\n")
 392         roundFile.write("<tr> <th> Name </th> <th> Score </th> <th> Total Score </th> </tr>\n")
 393         agents.sort(key = lambda e : e["score"][0], reverse=True)
 394         for agent in agents:
 395                 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")
 396         roundFile.write("</table>\n")
 397
 398         command = "cp scores.plt " + resultsDirectory + "scores.plt;"
 399         os.system(command)
 400
 401         scorePlot = open(resultsDirectory + "scores.plt", "a")
 402         scorePlot.write("plot ")
 403         for i in range(0, len(agents)):
 404                 if i > 0:
 405                         scorePlot.write(", ")
 406                 scorePlot.write("\""+agents[i]["name"]+".scores\" using ($0+1):1 with linespoints title \""+agents[i]["name"]+"\"")
 407
 408         scorePlot.write("\nexit\n")
 409         scorePlot.close()
 410
 411         command = "d=$(pwd); cd " + resultsDirectory + ";"
 412         command += "gnuplot scores.plt;"
 413         command += "rm -f scores.plt;"
 414         command += "mv scores.png round"+str(roundNumber)+".png;"
 415         command += "cd $d;"
 416         os.system(command)
 417
 418         roundFile.write("<h2> Accumulated Scores - up to Round " + str(roundNumber)+" </h2>\n")
 419         roundFile.write("<img src=\"round"+str(roundNumber)+".png\" alt = \"round"+str(roundNumber)+".png\" title = \"round"+str(roundNumber)+".png\" width = \"640\" height = \"480\"/>\n")
 420
 421         roundFile.write("<p> <a href=index.html>Current Scoreboard</a></p>\n")
 422         roundFile.write("</body>\n<!-- Results file for Round " + str(roundNumber) + " autogenerated by \"" + sys.argv[0] + "\" at time " + str(time()) + " -->\n</html>\n\n")
 423         roundFile.close()
 424
 425
 426
 427
 428
 429 if verbose:
 430         print "Finalising .html files... "
 431 for agent in agents:
 432         agentFile = open(resultsDirectory + agent["name"]+".html", "a")
 433         agentFile.write("<!--end-->\n")
 434         #Write a graph
 435         #Comment out if you don't have gnuplot
 436
 437         command  = "rm -f " + agent["name"] + ".png;"
 438         command += "cp template.plt " + resultsDirectory + agent["name"] + ".plt;"
 439         command += "d=$(pwd); cd " + resultsDirectory + ";"
 440         command += "sed -i \"s:\[NAME\]:"+agent["name"]+":g\" " +resultsDirectory + agent["name"]+".plt;"
 441         command += "gnuplot " + resultsDirectory + agent["name"]+".plt;"
 442         command += "rm -f " + resultsDirectory + agent["name"] + ".plt;"
 443         command += "cd $d;"
 444         os.system(command)
 445         agentFile.write("<!--end-->\n")
 446         agentFile.write("<h3> Score Graph </h3>\n")
 447         agentFile.write("<img src=\""+agent["name"]+".png\" alt=\""+agent["name"]+".png\" title=\""+agent["name"]+".png\" width=\"640\" height=\"480\"/>\n")
 448
 449         #Link to main file
 450         agentFile.write("<p> <a href=\"index.html\"/>Total Statistics</a> </p>\n")
 451
 452         agentFile.write("</body>\n<!-- Results file for \"" + agent["name"] + "\" autogenerated by \"" + sys.argv[0] + "\" at time " + str(time()) + " -->\n</html>\n\n")
 453         agentFile.close()
 454
 455 if os.path.exists(resultsDirectory + "index.html") == True:
 456         os.remove(resultsDirectory + "index.html") #Delete the file
 457
 458 totalFile = open(resultsDirectory + "index.html", "w")
 459 totalFile.write("<html>\n<head>\n <title> Total Overview </title>\n</head>\n<body>\n")
 460 totalFile.write("<h1> Total Overview </h1>\n")
 461 totalFile.write("<table border=\"0\" cellpadding=\"10\">\n")
 462 totalFile.write("<tr> <th> Name </th> <th> Total Score </th> </tr>\n")
 463 agents.sort(key = lambda e : e["totalScore"], reverse=True)
 464 for agent in agents:
 465         totalFile.write("<tr> <td> <a href="+agent["name"]+".html>"+agent["name"] + " </a> </td> <td> " + str(agent["totalScore"]) + " </td> </tr>\n")
 466 totalFile.write("</table>\n")
 467
 468 totalFile.write("<h2> Score Graph </h2>\n")
 469
 470
 471 command = "d=$(pwd);"
 472 command += "cd " + resultsDirectory + ";"
 473 command += "rm -f scores.png;"
 474 command += "cp round"+str(roundNumber)+".png scores.png;"
 475 command += "cd $d;"
 476 os.system(command)
 477
 478 totalFile.write("<img src=\"scores.png\" alt=\"scores.png\" title=\"scores.png\" width=\"640\" height=\"480\"/>\n")
 479
 480
 481 totalFile.write("<h2> Round Summaries </h2>\n")
 482 totalFile.write("<table border=\"0\" cellpadding=\"10\">\n")
 483 for i in range(1, totalRounds+1):
 484         totalFile.write("<tr> <td> <a href=round"+str(i)+".html>Round " + str(i) + "</a> </td> </tr>\n")
 485 totalFile.write("</table>\n")
 486
 487 totalFile.write("</body>\n<!-- Total Results file autogenerated by \"" + sys.argv[0] + "\" at time " + str(time()) + " -->\n</html>\n\n")
 488 totalFile.close()
 489
 490 #Write results to a raw text file as well
 491 textResults = open(resultsDirectory + "total.txt", "w")
 492 for agent in agents:
 493         textResults.write(agent["name"] + " " + str(agent["totalScore"]) + "\n")
 494 textResults.close()
 495
 496 if verbose:
 497         print "Done!"
 498
 499 endTime = time()
 500 print "Completed simulating " + str(nRounds) + " rounds in " + str(endTime - startTime) + " seconds."
 501 sys.exit(0)