diff options
Diffstat (limited to 'trace_courbe.py')
-rw-r--r-- | trace_courbe.py | 62 |
1 files changed, 47 insertions, 15 deletions
diff --git a/trace_courbe.py b/trace_courbe.py index d0f0b42..c86dbd9 100644 --- a/trace_courbe.py +++ b/trace_courbe.py | |||
@@ -3,12 +3,19 @@ | |||
3 | from configuration import * | 3 | from configuration import * |
4 | from gestionOMS import * | 4 | from gestionOMS import * |
5 | from gestion_unites import * | 5 | from gestion_unites import * |
6 | |||
6 | import matplotlib.pyplot as plt | 7 | import matplotlib.pyplot as plt |
7 | 8 | ||
8 | def cree_figure(conf,l_jours,l_poids,liste_err): | 9 | def cree_figure(conf,l_jours,l_poids,liste_err): |
9 | 10 | #warning("debut de cree_figure",liste_err) | |
10 | 11 | try: | |
11 | if conf["maxi"] == 0: | 12 | liste_data_labels_p,liste_data_labels_z = renvoie_liste_labels(conf,liste_data_choisie_p,liste_data_choisie_z,liste_err) |
13 | except: | ||
14 | erreur("bug avec liste data labels",liste_err) | ||
15 | return "" | ||
16 | |||
17 | #warning("bla",liste_err) | ||
18 | if conf["maxi"] ==0: | ||
12 | if l_jours != []: | 19 | if l_jours != []: |
13 | jour_maxi = max(l_jours)# pas la peine d'aller très au delà du jour max | 20 | jour_maxi = max(l_jours)# pas la peine d'aller très au delà du jour max |
14 | jour_maxi = int(jour_maxi* 1.1)+3 # on rajoute un peu | 21 | jour_maxi = int(jour_maxi* 1.1)+3 # on rajoute un peu |
@@ -26,9 +33,13 @@ def cree_figure(conf,l_jours,l_poids,liste_err): | |||
26 | # Attention, comme les jours commencent à partir de 0, faut enlever 1 pour avoir la borne... | 33 | # Attention, comme les jours commencent à partir de 0, faut enlever 1 pour avoir la borne... |
27 | age_maxi = convertitunite(jour_maxi-1,conf["unite"],liste_err) | 34 | age_maxi = convertitunite(jour_maxi-1,conf["unite"],liste_err) |
28 | 35 | ||
29 | 36 | ||
30 | titre = "Courbe de poids OMS" | 37 | titre = "Courbe de poids OMS" |
31 | 38 | ||
39 | # warning("cree_figure : ça va jusque là ",liste_err) | ||
40 | #warning("bli"+str(conf),liste_err) | ||
41 | #warninf("coucou") | ||
42 | |||
32 | if conf["typecourbe"] == "P": | 43 | if conf["typecourbe"] == "P": |
33 | # percentiles | 44 | # percentiles |
34 | liste_data_labels = liste_data_labels_p | 45 | liste_data_labels = liste_data_labels_p |
@@ -52,7 +63,13 @@ def cree_figure(conf,l_jours,l_poids,liste_err): | |||
52 | else: | 63 | else: |
53 | fichier_oms = f_poids_z_mixte | 64 | fichier_oms = f_poids_z_mixte |
54 | titre += " (moyenne et écarts-types)" | 65 | titre += " (moyenne et écarts-types)" |
55 | 66 | else: | |
67 | warning("Type de courbe invalide"+conf["typecourbe"],liste_err) | ||
68 | |||
69 | # warning("cree_figure : ça va jusque là ",liste_err) | ||
70 | # warning("bli"+str(conf),liste_err) | ||
71 | |||
72 | |||
56 | # Si y'a un nom on met "courbe de machin" | 73 | # Si y'a un nom on met "courbe de machin" |
57 | if conf["nom"] !="": | 74 | if conf["nom"] !="": |
58 | titre += " de " +conf["nom"] | 75 | titre += " de " +conf["nom"] |
@@ -67,16 +84,26 @@ def cree_figure(conf,l_jours,l_poids,liste_err): | |||
67 | liste_err[0].append("Impossible d'ouvrir le fichier "+fichier_oms) | 84 | liste_err[0].append("Impossible d'ouvrir le fichier "+fichier_oms) |
68 | coljour = convertit_tableau(extraire_colonne(t,0,jour_maxi),conf["unite"],liste_err) | 85 | coljour = convertit_tableau(extraire_colonne(t,0,jour_maxi),conf["unite"],liste_err) |
69 | 86 | ||
70 | 87 | ||
71 | #### La figure | 88 | #### La figure |
72 | fig = plt.figure(num=None, figsize=(conf["largeur"], conf["hauteur"]), dpi=100, facecolor='w', edgecolor='k') | 89 | fig = plt.figure(num=None, figsize=(conf["largeur"], conf["hauteur"]), dpi=100, facecolor=conf["couleur_fond"]) |
73 | 90 | plt.rcParams['axes.facecolor'] = conf["couleur_fond"] | |
91 | plt.rcParams['axes.edgecolor']= conf["couleur_cadretxt"] | ||
92 | plt.rcParams['xtick.color'] = conf["couleur_cadretxt"] | ||
93 | plt.rcParams['ytick.color'] = conf["couleur_cadretxt"] | ||
94 | plt.rcParams['grid.color'] = conf["couleur_grille"] | ||
95 | plt.rcParams['legend.edgecolor'] = conf["couleur_grille"] | ||
96 | |||
97 | #warning("bla"+str(liste_data_labels),liste_err) | ||
74 | for (i,label,couleur) in liste_data_labels: | 98 | for (i,label,couleur) in liste_data_labels: |
75 | plt.plot(coljour,extraire_colonne(t,i,jour_maxi),label=label,color=couleur) | 99 | plt.plot(coljour,extraire_colonne(t,i,jour_maxi),label=label,color=couleur) |
100 | #warning("colonne"+str(i)+str(label)+str(couleur),liste_err) | ||
101 | |||
102 | |||
76 | 103 | ||
77 | # On extrait la valeur min et la valeur max des poids | 104 | # On extrait la valeur min et la valeur max des poids |
78 | (colonne_min,_,_) = liste_data_labels[0] | 105 | (colonne_min,_,_) = liste_data_labels[-1] |
79 | (colonne_max,_,_) = liste_data_labels[-1] | 106 | (colonne_max,_,_) = liste_data_labels[0] |
80 | 107 | ||
81 | poids_min = min(extraire_colonne(t,colonne_min,jour_maxi)) | 108 | poids_min = min(extraire_colonne(t,colonne_min,jour_maxi)) |
82 | poids_max = max(extraire_colonne(t,colonne_max,jour_maxi)) | 109 | poids_max = max(extraire_colonne(t,colonne_max,jour_maxi)) |
@@ -89,17 +116,22 @@ def cree_figure(conf,l_jours,l_poids,liste_err): | |||
89 | #max : +5% | 116 | #max : +5% |
90 | poids_max = poids_max * 1.05 | 117 | poids_max = poids_max * 1.05 |
91 | 118 | ||
119 | #warning("Jusque là ça marche",liste_err) | ||
120 | |||
92 | if l_jours != []: | 121 | if l_jours != []: |
93 | plt.plot(l_jours_conv,l_poids,label=conf["nom"],color="black",marker='o') | 122 | plt.plot(l_jours_conv,l_poids,label=conf["nom"],color=conf["couleur_cadretxt"],marker='o') |
94 | 123 | ||
95 | plt.xlabel("Âge en "+conf["unite"]) | 124 | plt.xlabel("Âge en "+conf["unite"],color=conf["couleur_cadretxt"]) |
96 | plt.ylabel("Poids en kg") | 125 | plt.ylabel("Poids en kg",color=conf["couleur_cadretxt"]) |
97 | plt.title(titre) | 126 | plt.title(titre,color=conf["couleur_cadretxt"]) |
98 | plt.axis([0,age_maxi,poids_min,poids_max]) | 127 | plt.axis([0,age_maxi,poids_min,poids_max]) |
99 | 128 | ||
100 | if conf['legende']=="oui": | 129 | if conf['legende']=="oui": |
101 | plt.legend(loc=conf['positionlegende']) | 130 | legende = plt.legend(loc=conf['positionlegende']) |
131 | plt.setp(legende.get_texts(), color=conf["couleur_cadretxt"]) | ||
102 | plt.grid(conf["grille"]=="oui") | 132 | plt.grid(conf["grille"]=="oui") |
103 | 133 | ||
104 | fig.tight_layout() | 134 | fig.tight_layout() |
135 | |||
136 | #warning("Jusque là ça marche",liste_err) | ||
105 | return fig \ No newline at end of file | 137 | return fig \ No newline at end of file |