PlayStats.py

# Skout - Flagfootball Scouting Analysis Tool
#   Copyright (C) 2021  Xaver Klemenschits
#   See LICENSE.txt for details

class PlayStats:
    def incrementStrongSide(self, strongSideList, strongSideString):
        if(strongSideString == "right"):
            strongSideList[1] += 1
        elif(strongSideString == "left"):
            strongSideList[0] += 1
        else:
            print("Warning: Invalid StrongSide string: " + strongSideString)

    def switchStrongSide(self, routes):
        routes[0], routes[1], routes[2], routes[3] = routes[3], routes[2], routes[1], routes[0]

    def swapEntry(self, i, j, swappable):
        swappable[i], swappable[j] = swappable[j], swappable[i]

    def swapEntries(self, i, j):
        self.swapEntry(i, j, self.routesList)
        self.swapEntry(i, j, self.occurences)
        self.swapEntry(i, j, self.downStats)
        self.swapEntry(i, j, self.formations)
        self.swapEntry(i, j, self.strongSides)
        self.swapEntry(i, j, self.clipNumbers)
        self.swapEntry(i, j, self.progressions)
        self.swapEntry(i, j, self.intRecs)
        self.swapEntry(i, j, self.distances)
        self.swapEntry(i, j, self.notes)

    def __init__(self, playList):
        self.homeTeam = playList.homeTeam
        self.awayTeam = playList.awayTeam
        self.date = playList.date
        self.score = (playList.homeScore, playList.awayScore)
        playList.sort()
        self.routesList = []
        self.occurences = []
        self.downStats = []
        self.formations = []
        self.strongSides = []
        self.clipNumbers = []
        self.progressions = []
        self.intRecs = []
        self.distances = []
        self.notes = []

        # Calculate stats from the passed playList

        # just push 1st element
        self.routesList.append(playList.routes[0])
        if(playList.sides[0] == "left"):
            self.switchStrongSide(self.routesList[-1])
        self.occurences.append(1)
        downList = [0, 0, 0, 0, 0]
        downList[playList.downs[0]-1] = 1
        self.downStats.append(downList)
        self.formations.append([])
        self.formations[-1].append(playList.formations[0])
        strongside = [0, 0]
        self.incrementStrongSide(strongside, playList.sides[0])
        self.strongSides.append(strongside)
        self.clipNumbers.append([])
        self.clipNumbers[-1].append(playList.clipNumbers[0])
        self.progressions.append([])
        self.progressions[-1].append(playList.progression[0])
        self.intRecs.append([])
        self.intRecs[-1].append(playList.intRec[0])
        self.distances.append([])
        self.distances[-1].append(playList.distance[0])
        self.notes.append([])
        self.notes[-1].append(playList.notes[0])

        last = 0
        for i in range(1, len(playList.routes)):
            currentRoute = playList.routes[i]
            if(playList.sides[i] == "left"):
                self.switchStrongSide(currentRoute)

            # routes are equal, so add onto earlier play
            if(self.routesList[last] == currentRoute):
                self.occurences[last] += 1
                self.downStats[last][playList.downs[i]-1] += 1
                self.formations[last].append(playList.formations[i])
                self.incrementStrongSide(
                    self.strongSides[last], playList.sides[i])
                self.clipNumbers[last].append(playList.clipNumbers[i])
                self.progressions[last].append(playList.progression[i])
                self.intRecs[last].append(playList.intRec[i])
                self.distances[last].append(playList.distance[i])
                self.notes[last].append(playList.notes[i])

            # routes are not equal, add new play to list
            else:
                self.routesList.append(currentRoute)
                self.occurences.append(1)
                downList = [0, 0, 0, 0, 0]
                downList[playList.downs[i]-1] = 1
                self.downStats.append(downList)
                self.formations.append([])
                self.formations[-1].append(playList.formations[i])
                strongside = [0, 0]
                self.incrementStrongSide(strongside, playList.sides[i])
                self.strongSides.append(strongside)
                self.clipNumbers.append([])
                self.clipNumbers[-1].append(playList.clipNumbers[i])
                self.progressions.append([])
                self.progressions[-1].append(playList.progression[i])
                self.intRecs.append([])
                self.intRecs[-1].append(playList.intRec[i])
                self.distances.append([])
                self.distances[-1].append(playList.distance[i])
                self.notes.append([])
                self.notes[-1].append(playList.notes[i])
                last += 1

        # bubble sort results after number of occurences
        n = len(self.routesList)

        # traverse from front
        for i in range(n-1):
            #  last elements are already in place
            for j in range(n-i-1):
                if(self.occurences[j] < self.occurences[j+1]):
                    self.swapEntries(j, j+1)

    def print(self):
        for i in range(len(self.routesList)):
            outputStr = "Play " + str(i) + ":\n"
            outputStr += "Routes: " + str(self.routesList[i]) + "\n"
            outputStr += "Downcount: " + str(self.downStats[i]) + "\n"
            outputStr += "Formations: " + str(self.formations[i]) + "\n"
            outputStr += "Strongsides: " + str(self.strongSides[i]) + "\n"
            print(outputStr)

    def getFormations(self, index):
        if(len(self.formations[index]) == 0):
            return ""
        result = str(self.formations[index][0])
        for i in range(1, len(self.formations[index])):
            if(self.formations[index][i] != ""):
                if(result != ""):
                    result = result + ", "
                result = result + self.formations[index][i]

        return result