import matplotlib.pyplot as plt; plt.rcdefaults()
import numpy as np
import matplotlib.pyplot as plt
from pylab import *


TailleGrilleX = 15
TailleGrilleY = 15
GrilleInitiale = 0
HistoriqueGrille = list()
StratsResultats = list()
ListeStrategies = list()


"""def recup_mat(matrice, param):
    """ array*str-> array
    Récupère la matrice avec seulement le paramètre (état,strat,années de prison) voulu afin de pouvoir avoir les stats plus facilement"""
   
    
    matR = np.random.randint(0,1,TailleGrilleX,TailleGrilleY) 
    
    for ligne in range (0,TailleGrilleX):  #int ligne
        for colonne in range (0, TailleGrilleY): #int colonne
            matR[ligne][colonne]=Grille [ligne][colonne][param]
     
    return matR"""
    
def bar_chart():
    """array->graph
    Retourne le diagramme en baton qui met en évidence le nombre d'années de prison en fonction de la stratégie. 
    Il est mis à jour à chaque fos que tous les éléments de la grille ont fait leur 8 parties locales"""
    
    matStrat=recup_mat('strategie')
    matGain= recup_mat('annees_de_prison')
   
    #initialisation des paramètres
    #list gain
    gain=[]
    #list strat
    strat=[]
    
    for i in range(5):
        
        gain.append(0)
        strat.append(0)
        
    
        
       
    
    for i in range(0,len(ListeStrategies)):
        strat[i]=StratsResultats[i]['nb_utilisations']
        gain[i]=StratsResultats[i]['total_ans_prisons']
   
    
                    
            
            
                 
    Strat=('1','2', '3','4' ,'5')
    
    x_pos = np.arange(len(Strat))
    
    plt.subplot(221)
    
    plt.bar(x_pos, gain, align='center', color='b' )
   
    plt.xlabel("Stratégies")
    plt.ylabel("Nombre d'années de prison")
    
    plt.subplot(222)
    plt.bar(x_pos, strat, align='center' , color='r')
    plt.xlabel("Stratégies")
    plt.ylabel("NB individus ayant adopté stratégie")
    
    
    plt.xticks(x_pos,Strat)

    plt.show()