rc.py

from trueskill import Rating, quality_1vs1, rate_1vs1, TrueSkill, rate


def printWelcomeMessage():
	print ""
	print "                |\_______________ (_____\ \______________"
	print "      HH========#H###############H#######################"
	print "                ' ~'''''''''''`##(_))#H|''''''''Y########"
	print "                                  ))    \#H\       `'Y###"
	print "                                  \"\"      }#H)\n"

def printHelp():
	print ""
	print "## LIST OF CLASSES AND COMMANDS ##:\n"
	print "change_map \033[4mMAP\033[0m -- Change current map -- Alternative: 'm'"
	print "scoreboard -- Show scoreboard (not available in match)"
	print "exit -- Write changes on ratings and exit the program -- Alternative: 'e'"
	print "help -- Display help -- Alternative: 'h'"
	print "start_match -- Start match in current map -- Alternative: 's'"
	print "      \033[4m1\033[0m -- for each time team1 win a round"
	print "      \033[4m2\033[0m -- for each time team2 win a round"
	print "      \033[4m0\033[0m -- for finishing the map\n"



dict = {}
map_list = ["backlot", "bog", "crossfire", "wetwork"]
with open("ratings2.txt","rw") as f:
	# Read ratings and std deviations to put in dictionary
	for line in f:
		if line.replace(" ", "") == "":
			line = f.readline
			continue
		list = line.split(" ")
		# Create rating of player based on parameters
		dict[str(list[0])] = Rating( float(list[1]), float(list[2]) )
    	line = f.readline()

	team1 = []
	team2 = []
	
	printWelcomeMessage()
	print "Welcome to cod4ceng, type 'help' to reach command list."
	print "Create teams by using command 't' or 'create_teams'\n"

	current_map = ""
	teams_done = False
	map_done = False

	while(1):
		command = raw_input("(c4c) ")
		command_split = command.split(" ")
		
		if command == 't' or command == "create_teams":
			teams_done = False
			size1 = input("team1_size: ")	# Team1
			for i in range (0,size1):
				name = raw_input("(" + str(i+1) + "): ")
				team1.append(name)
			size2 = input("team2_size: ")	# Team2
			for i in range (0,size2):
				name = raw_input("(" + str(i+1) + "): ")
				team2.append(name)
			if size1 > 0 and size2 > 0:
				teams_done = True	

			# Create team ratings to pass in rate function
			team1_ratings = []
			for playerName in team1:
				team1_ratings.append(dict[playerName])
			team2_ratings = []
			for playerName in team2:
				team2_ratings.append(dict[playerName])


		elif command_split[0] == 'change_map' or command_split[0] == 'm':
			map_done = False
			if command_split[1].lower() not in map_list:
				print "Map is not in map list, current map is set to null"
				continue
			current_map = command_split[1]
			map_done = True
			print "Map changed to " + current_map 
		

		elif command == 'start_match' or command == "s":
			if not teams_done:
				print "Match can not start, teams are not ready yet"
				continue
			if not map_done:
				print "Match can not start, map is not choosen yet"
				continue
			print "Match started at " + current_map + ", enter results (0 to finish map)"
			while(1):
				res = raw_input("(c4c" + "-" + current_map + ") ")
				if res == '1': 		# Team1 won
					team1_ratings, team2_ratings = rate([team1_ratings, team2_ratings])
				elif res == '2': 	# Team2 won
					team2_ratings, team1_ratings = rate([team2_ratings, team1_ratings]) 	
				elif res == '0':	# Finish match
					print "Match is finished, waiting for command"
					break
				elif res == 'scoreboard':
					print "Scoreboard can't be shown while a match is ongoing"
		

		elif command == 'help' or command == "h":
			printHelp()

		elif command == 'exit' or command == "e":
			print "Writing rating changes to file"
			break

		else:
			if not command.replace(" ", "") == "":
				print "Undefined command:" + command+ ".  Try \"help\"."	

	# Update rating values in dictionary
	index = 0
	for playerName in team1:
		dict[playerName] = team1_ratings[index]
		index = index+1
	index = 0
	for playerName in team2:
		dict[playerName] = team2_ratings[index]
		index = index+1

	for key in dict:
		print key + ": " + str(dict[key])


with open("ratings2.txt","w") as f2:
	
	for key in dict:
		line = key + " {1:.2f} {1:.2f}\n".format(dict[key].mu, dict[key].sigma)
		f2.write(line)