class Graph():
def __init__(self):
self._teams = {}
self._array = []
self._size = 0
self.algorithm = RatingAlgorithm()
def buildGraph(self, reader):
file = reader.readFile()
games = reader.parseFile(file)
if games:
for game in games:
scoreDiff1 = eval(game[1]) - eval(game[3])
scoreDiff2 = 0 - scoreDiff1
scoreDiff1 = self.algorithm.improveScores(scoreDiff1)
scoreDiff2 = self.algorithm.improveScores(scoreDiff2)
self.fillGraph(game[0], game[2], scoreDiff1, scoreDiff2)
print("Done")
def fillGraph(self, team1, team2, scoreDiff1, scoreDiff2):
if team1 not in self._teams.keys(): self.addTeam(team1)
self.addGame(team1, team2, scoreDiff1)
if team2 not in self._teams.keys(): self.addTeam(team2)
self.addGame(team2, team1, scoreDiff2)
def addTeam(self, team):
A = AdjList()
A.insertHead(team)
self._array.append(A)
self._teams[team] = self._size
self._size += 1
def addGame(self, team1, team2, scoreDiff):
arrayIndex = self.getTeam(team1)
self._array[arrayIndex].insert(team2, float(scoreDiff))
def getSize(self):
return self._size
def getTeam(self, name):
return self._teams[name]
def rateTeams(self):
print('Calculating Ratings...')
self._array = self.algorithm.rateTeams(self._array, self._teams)
print('Done')
def getTeams(self):
return self._teams
def getTeamNames(self):
return self._array
class Graph():
def __init__(self):
self._teams = {}
self._array = []
self._size = 0
self.algorithm = RatingAlgorithm()
def buildGraph(self, reader):
file = reader.readFile()
games = reader.parseFile(file)
if games:
for game in games:
scoreDiff1 = eval(game[1]) - eval(game[3])
scoreDiff2 = 0 - scoreDiff1
scoreDiff1 = self.algorithm.improveScores(scoreDiff1)
scoreDiff2 = self.algorithm.improveScores(scoreDiff2)
self.fillGraph(game[0],game[2],scoreDiff1,scoreDiff2)
print("Done")
def fillGraph(self,team1,team2,scoreDiff1,scoreDiff2):
if team1 not in self._teams.keys(): self.addTeam(team1)
self.addGame(team1,team2,scoreDiff1)
if team2 not in self._teams.keys(): self.addTeam(team2)
self.addGame(team2,team1,scoreDiff2)
def addTeam(self,team):
A = AdjList()
A.insertHead(team)
self._array.append(A)
self._teams[team] = self._size
self._size += 1
def addGame(self,team1,team2,scoreDiff):
arrayIndex = self.getTeam(team1)
self._array[arrayIndex].insert(team2,float(scoreDiff))
def getSize(self): return self._size
def getTeam(self,name): return self._teams[name]
def rateTeams(self):
print('Calculating Ratings...')
self._array = self.algorithm.rateTeams(self._array, self._teams)
print('Done')
def getTeams(self): return self._teams
def getTeamNames(self): return self._array
def __init__(self):
self._teams = {}
self._array = []
self._size = 0
self.algorithm = RatingAlgorithm()
def __init__(self):
self._teams = {}
self._array = []
self._size = 0
self.algorithm = RatingAlgorithm()