FindYourLeague/main.py

import numpy as np
import skfuzzy as fuzz
from skfuzzy import control as ctrl
import matplotlib.pyplot as plt
from read_data import get_leagues_data
import pandas as pd

class LeagueController:
    clubs = None
    goals = None
    avgValue = None
    players = None
    teamsChL = None
    teamsEuL = None
    teamsCfL = None
    score = None

    rule1 = None
    rule2 = None
    rule3 = None
    rule4 = None
    rule5 = None

    league_ctrl = None
    league_system = None

    def __init__(self):
        # Universes for variables
        self.clubs = ctrl.Antecedent(np.arange(2, 31, 1), 'clubs')
        self.goals = ctrl.Antecedent(np.arange(0.99, 5.01, 0.01), 'goals')
        self.avgValue = ctrl.Antecedent(np.arange(0.01, 500.01, 0.01), 'avgValue')
        self.players = ctrl.Antecedent(np.arange(0, 101, 1), 'players')
        self.teamsChL = ctrl.Antecedent(np.arange(0, 6, 1), 'teamsChL')
        self.teamsEuL = ctrl.Antecedent(np.arange(0, 6, 1), 'teamsEuL')
        self.teamsCfL = ctrl.Antecedent(np.arange(0, 6, 1), 'teamsCfL')
        # Result score universum
        self.score = ctrl.Consequent(np.arange(0, 101, 1), 'score')

        # Membership Functions
        # trimf - triangular function
        # trapmf - trapezoidal function
        self.clubs['few'] = fuzz.trapmf(self.clubs.universe, [2, 2, 5, 8])
        self.clubs['some'] = fuzz.trapmf(self.clubs.universe, [3, 6, 9, 12])
        self.clubs['several'] = fuzz.trapmf(self.clubs.universe, [8, 11, 15, 18])
        self.clubs['many'] = fuzz.trapmf(self.clubs.universe, [15, 18, 30, 30])    

        self.goals['few'] = fuzz.trapmf(self.goals.universe, [1.0, 1.0, 2.0, 3.0])
        self.goals['moderate'] = fuzz.trimf(self.goals.universe, [2.5, 3.0, 3.5])
        self.goals['many'] = fuzz.trapmf(self.goals.universe, [3.0, 3.5, 5.0, 5.0])
        
        self.avgValue['vpoor'] = fuzz.trapmf(self.avgValue.universe, [0.01, 0.01, 1.0, 2.0])
        self.avgValue['poor'] = fuzz.trimf(self.avgValue.universe, [1.0, 5.0, 15.0])
        self.avgValue['mpoor'] = fuzz.trimf(self.avgValue.universe, [7.5, 20.0, 50.0])
        self.avgValue['moderate'] = fuzz.trimf(self.avgValue.universe, [20.0, 50.0, 100.0])
        self.avgValue['mrich'] = fuzz.trimf(self.avgValue.universe, [50.0, 100.0, 175.0])
        self.avgValue['rich'] = fuzz.trimf(self.avgValue.universe, [150.0, 200.0, 275.0])
        self.avgValue['vrich'] = fuzz.trapmf(self.avgValue.universe, [200.0, 300.0, 500.0, 500.0])

        self.players['none'] = fuzz.trimf(self.players.universe, [0, 0, 1])
        self.players['few'] = fuzz.trapmf(self.players.universe, [1, 5, 10, 15])
        self.players['moderate'] = fuzz.trapmf(self.players.universe, [10, 15, 20, 25])
        self.players['many'] = fuzz.trapmf(self.players.universe, [20, 25, 100, 100])

        self.teamsChL['min'] = fuzz.trapmf(self.teamsChL.universe, [0, 0, 1, 2])
        self.teamsChL['mid'] = fuzz.trapmf(self.teamsChL.universe, [1, 2, 3, 4])
        self.teamsChL['max'] = fuzz.trapmf(self.teamsChL.universe, [3, 4, 5, 5])

        self.teamsEuL['min'] = fuzz.trimf(self.teamsEuL.universe, [0, 0, 1])
        self.teamsEuL['mid'] = fuzz.trimf(self.teamsEuL.universe, [0, 1, 2])
        self.teamsEuL['max'] = fuzz.trapmf(self.teamsEuL.universe, [1, 2, 5, 5])

        self.teamsCfL['min'] = fuzz.trapmf(self.teamsCfL.universe, [0, 0, 1, 2])
        self.teamsCfL['mid'] = fuzz.trimf(self.teamsCfL.universe, [1, 2, 3])
        self.teamsCfL['max'] = fuzz.trapmf(self.teamsCfL.universe, [2, 3, 5, 5])

        self.score['tragic'] = fuzz.trapmf(self.score.universe, [0, 0, 15, 25])
        self.score['bad'] = fuzz.trapmf(self.score.universe, [15, 25, 35, 45])
        self.score['moderate'] = fuzz.trapmf(self.score.universe, [35, 45, 55, 65])
        self.score['good'] = fuzz.trapmf(self.score.universe, [55, 65, 75, 85])
        self.score['best'] = fuzz.trapmf(self.score.universe, [75, 85, 100, 100])

        # FuzzyRules
        self.rule1 = ctrl.Rule((self.goals['few'] | self.avgValue['vpoor'] | self.avgValue['poor']) & self.players['none'] & self.teamsCfL['max'] | self.teamsChL['min'], self.score['tragic'])
        self.rule2 = ctrl.Rule((self.goals['few'] | self.goals['moderate']) & (self.avgValue['mpoor'] | self.avgValue['poor']) & (self.players['none'] | self.players['few']) & self.teamsCfL['max'] | self.teamsEuL['mid'], self.score['bad'])
        self.rule3 = ctrl.Rule((self.avgValue['mpoor'] | self.avgValue['moderate'] | self.avgValue['mrich']) & (self.players['few'] | self.players['moderate']) & (self.teamsEuL['mid'] | self.teamsEuL['max']) & (self.teamsChL['min'] | self.teamsChL['mid']), self.score['moderate'])
        self.rule4 = ctrl.Rule((self.clubs['many'] | self.clubs['several']) & (self.goals['many'] | self.goals['moderate']) & (self.avgValue['mrich'] | self.avgValue['rich']) & (self.players['many'] | self.players['moderate']) | self.teamsChL['mid'] | self.teamsEuL['mid'] | self.teamsCfL['mid'], self.score['good'])
        self.rule5 = ctrl.Rule((self.clubs['many'] | self.goals['many'] | self.avgValue['rich'] | self.avgValue['vrich']) & (self.players['many']) & (self.teamsChL['max']), self.score['best'])

        self.league_ctrl = ctrl.ControlSystem([self.rule1, self.rule2, self.rule3, self.rule4, self.rule5])
        self.league_system = ctrl.ControlSystemSimulation(self.league_ctrl)

    def getScore(self, values):
        self.league_system.input['clubs'] = int(values['clubs'])
        self.league_system.input['goals'] = float(values['goalsPerMatch'])
        self.league_system.input['avgValue'] = float(values['avgMarketValueInMln'])
        self.league_system.input['players'] = int(values['famousPlayers'])
        self.league_system.input['teamsChL'] = int(values['teamsInChampionsLeague'])
        self.league_system.input['teamsEuL'] = int(values['teamsInEuropaLeague'])
        self.league_system.input['teamsCfL'] = int(values['teamsInConferenceLeague'])

        self.league_system.compute()

        return self.league_system.output['score']

        #score.view(sim=league_system)

    def displayMembershipFunctionPlot(self, function_id):
        if function_id == 0:
            self.score.view()
        elif function_id == 1:
            self.clubs.view()
        elif function_id == 2:
            self.goals.view()
        elif function_id == 3:
            self.avgValue.view()
        elif function_id == 4:
            self.players.view()
        elif function_id == 5:
            self.teamsChL.view()
        elif function_id == 6:
            self.teamsEuL.view()
        elif function_id == 7:
            self.teamsCfL.view()

    def displayRulePlot(self, rule_id):
        if rule_id == 1:
            self.rule1.view()
        elif rule_id == 2:
            self.rule2.view()
        elif rule_id == 3:
            self.rule3.view()
        elif rule_id == 4:
            self.rule4.view()
        elif rule_id == 5:
            self.rule5.view()

    def displayControllerResultPlot(self):
        self.score.view(sim=self.league_system)



if __name__ == "__main__":
    league_name_col = []
    league_score_col = []
    # initFuzzyMembershipFunctions()
    controller = LeagueController()
    data_source = None
    function_rule_result = None
    function_id = None
    rule_id = None
    view_plot = None
    data_save = None
    while not(data_source == '1' or data_source == '2'):
        data_source = input('Wpisz 1 by wczytać dane z pliku lub 2 by wpisać dane z konsoli:\n')
    if data_source == '1':
        leagues_data = get_leagues_data('leagues.csv')
        for league in leagues_data:
            #activateFuzzyMembershipFunctions(league)
            #result = defineRules()
            result = controller.getScore(league)
            print(f"{league['leagueName']} score is: {result}")
            league_name_col.append(league['leagueName'])
            league_score_col.append(result)
        data_save = input('Wpisz 1 by zapisać dane do pliku lub dowolny klawisz żeby kontynuować:\n')
        if data_save == '1':
            file_name = input('Podaj nazwę pliku:\n')
            file_name += ".csv"
            d = {'League':league_name_col,'Score':league_score_col}
            df = pd.DataFrame(d)
            df.to_csv(file_name, index=False)
            print('Wyniki zostały zapisane w pliku csv w folderze programu')
    else:
        league = {'leagueName': input('Podaj nazwę ligi:\n'), 'country': input('Podaj kraj ligi:\n'),
                  'clubs': input('Podaj liczbę klubów grających w lidze:\n'),
                  'goalsPerMatch': input('Podaj średnią liczbę goli na mecz:\n'),
                  'avgMarketValueInMln': input('Podaj średnią wartośc klubu w lidze (w milionach):\n'),
                  'teamsInChampionsLeague': input('Podaj liczbę klubów grających w Lidze Mistrzów:\n'),
                  'teamsInEuropaLeague': input('Podaj liczbę klubów grających w Lidze Europy:\n'),
                  'teamsInConferenceLeague': input('Podaj liczbę klubów grających w Lidze Konferencji:\n'),
                  'famousPlayers': input('Podaj liczbę sławnych piłkarzy:\n')}
        # activateFuzzyMembershipFunctions(league)
        # result = defineRules()
        result = controller.getScore(league)
        print(f"{league['leagueName']} score is: {result}")
    view_plot = input('Wpisz 1 jeżeli chcesz obejrzeć wykresy funkcji i reguł dla ostatnio sprawdzanej ligi lub dowolny klawisz żeby kontynuować:\n')
    if view_plot == '1':
        while (function_rule_result != '0'):
            #function_rule_result = input('Wybierz rodzaj wykresu:\n1 - wykresy funkcji przynależności\n2 - wykresy reguł\n3 - wykres wyniku kontrolera\n0 - skończ wizualizację\n')
            function_rule_result = input('Wybierz rodzaj wykresu:\n1 - wykresy funkcji przynależności\n2 - wykres wyniku kontrolera\n0 - skończ wizualizację\n')
            if function_rule_result == "1":
                while (function_id != '8'):
                    function_id = input('Wybierz funkcje przynależności:\n0 - funkcje punktów atrakcyjności ligi\n1 - funkcje klubów w lidze\n2 - funkcje średniej ilości golów\n3 - funkcje średniej wartości rynkowej\n4 - funkcje ilości słynnych zawodników\n5 - funkcje drużyn w lidze mistrzów\n6 - funkcję drużyn w lidze europy\n7 - funkcję drużyn w lidze konfederacji\n8 - powrót do poprzedniego menu\n')
                    if function_id == "0" or function_id == "1" or function_id == "2" or function_id == "3" or function_id == "4" or function_id == "5" or function_id == "6" or function_id == "7":
                        controller.displayMembershipFunctionPlot(int(function_id))
            # elif function_rule_result == "2":
            #     while (rule_id != '6'):
            #         rule_id = input('Wybierz regułę:\n1 - reguła dla tragicznych lig\n2 - reguła dla słabych lig\n3 - reguła dla przeciętnych lig\n4 - reguła dla dobrych lig\n5 - reguła dla najlepszych lig\n6 - powrót do poprzedniego menu\n')
            #         if rule_id == "1" or rule_id == "2" or rule_id == "3" or rule_id == "4" or rule_id == "5":
            #             controller.displayRulePlot(int(rule_id))
            elif function_rule_result == "2":
                controller.displayControllerResultPlot()
    #plt.show()