]>
git.immae.eu Git - perso/Denise/oms.git/blob - trace_courbe.py
4ccc6ebfb1efea5c058dfd4937cd20a69eca7816
2 # -*- coding: utf-8 -*-
3 from configuration
import *
4 from gestionOMS
import *
5 from gestion_unites
import *
7 import matplotlib
.pyplot
as plt
9 def cree_figure(conf
,l_jours
,l_poids
,liste_err
):
10 #warning("debut de cree_figure"+str(conf),liste_err)
12 liste_data_labels_p
,liste_data_labels_z
= renvoie_liste_labels(conf
,liste_data_choisie_p
,liste_data_choisie_z
,liste_err
)
14 erreur("bug avec liste data labels",liste_err
)
17 #warning("bla",liste_err)
20 jour_maxi
= max(l_jours
)# pas la peine d'aller très au delà du jour max
21 jour_maxi
= int(jour_maxi
* 1.1)+3 # on rajoute un peu
23 jour_maxi
= int(6*jours_dans_mois
) # si pas de données, arbitrairement on met ça
25 jour_maxi
= conf
["maxi"]+1
27 # On s'assure que c'est bien compris dans les bornes
28 #warning("bla"+str(jour_maxi),liste_err)
29 jour_maxi
= max(CONFIG
["jours_mini_courbe"],min(jour_maxi
,CONFIG
["jours_maxi_courbe"]))
30 #warning("bla"+str(jour_maxi),liste_err)
32 if conf
["unite"] == "":
33 conf
["unite"] = choix_unite(jour_maxi
)
35 l_jours_conv
= convertit_tableau(l_jours
,conf
["unite"],liste_err
)
36 # Attention, comme les jours commencent à partir de 0, faut enlever 1 pour avoir la borne...
37 age_maxi
= convertitunite(jour_maxi
-1,conf
["unite"],liste_err
)
40 titre
= "Courbe de poids OMS"
42 #warning("cree_figure : ça va jusque là ",liste_err)
43 #warning("bli"+str(conf),liste_err)
46 if conf
["typecourbe"] == "P":
48 liste_data_labels
= liste_data_labels_p
49 if conf
["sexe"] == "M":
50 fichier_oms
= f_poids_perc_garcon
51 titre
+= " (percentiles, garçon)"
52 elif conf
["sexe"] == "F":
53 fichier_oms
= f_poids_perc_fille
54 titre
+= " (percentiles, fille)"
56 fichier_oms
= f_poids_perc_mixte
57 titre
+= " (percentiles)"
58 elif conf
["typecourbe"] == "Z":
59 liste_data_labels
= liste_data_labels_z
60 if conf
["sexe"] == "G":
61 fichier_oms
= f_poids_z_garcon
62 titre
+= " (moyenne et écarts-types, garçon)"
63 elif conf
["sexe"] == "F":
64 fichier_oms
= f_poids_z_fille
65 titre
+= " (moyenne et écarts-types, fille)"
67 fichier_oms
= f_poids_z_mixte
68 titre
+= " (moyenne et écarts-types)"
70 warning("Type de courbe invalide"+conf
["typecourbe"],liste_err
)
72 # warning("cree_figure : ça va jusque là ",liste_err)
73 # warning("bli"+str(conf),liste_err)
76 # Si y'a un nom on met "courbe de machin"
78 titre
+= " de " +conf
["nom"]
81 # convertir les unités
83 #### On extrait les données des courbes, et on convertit les jours dans l'unité voulues
85 t
= lire_fichier_csv(fichier_oms
)
87 liste_err
[0].append("Impossible d'ouvrir le fichier "+fichier_oms
)
88 coljour
= convertit_tableau(extraire_colonne(t
,0,jour_maxi
),conf
["unite"],liste_err
)
92 fig
= plt
.figure(num
=None, figsize
=(conf
["largeur"], conf
["hauteur"]), dpi
=100, facecolor
=conf
["couleurs"]["fond"])
93 plt
.rcParams
['axes.facecolor'] = conf
["couleurs"]["fond"]
94 plt
.rcParams
['axes.edgecolor']= conf
["couleurs"]["cadretxt"]
95 plt
.rcParams
['xtick.color'] = conf
["couleurs"]["cadretxt"]
96 plt
.rcParams
['ytick.color'] = conf
["couleurs"]["cadretxt"]
97 plt
.rcParams
['grid.color'] = conf
["couleurs"]["grille"]
98 plt
.rcParams
['legend.edgecolor'] = conf
["couleurs"]["grille"]
100 #warning("bla"+str(liste_data_labels),liste_err)
101 for (i
,label
,couleur
) in liste_data_labels
:
102 plt
.plot(coljour
,extraire_colonne(t
,i
,jour_maxi
),label
=label
,color
=couleur
)
103 #warning("colonne"+str(i)+str(label)+str(couleur),liste_err)
107 # On extrait la valeur min et la valeur max des poids
108 (colonne_min
,_
,_
) = liste_data_labels
[-1]
109 (colonne_max
,_
,_
) = liste_data_labels
[0]
111 poids_min
= min(extraire_colonne(t
,colonne_min
,jour_maxi
))
112 poids_max
= max(extraire_colonne(t
,colonne_max
,jour_maxi
))
114 poids_min
= min(min(l_poids
),poids_min
)
115 poids_max
= max(max(l_poids
),poids_max
)
116 # On ajuste un peu ces min et max
117 # min : valeur min -1kg
118 poids_min
= max(0,poids_min
-1)
120 poids_max
= poids_max
* 1.05
122 #warning("Jusque là ça marche",liste_err)
125 plt
.plot(l_jours_conv
,l_poids
,label
=conf
["nom"],color
=conf
["couleur_cadretxt"],marker
='o')
127 plt
.xlabel("Âge en "+conf
["unite"],color
=conf
["couleurs"]["cadretxt"])
128 plt
.ylabel("Poids en kg",color
=conf
["couleurs"]["cadretxt"])
129 plt
.title(titre
,color
=conf
["couleurs"]["cadretxt"])
130 plt
.axis([0,age_maxi
,poids_min
,poids_max
])
132 if conf
['legende']=="oui":
133 legende
= plt
.legend(loc
=conf
['positionlegende'])
134 plt
.setp(legende
.get_texts(), color
=conf
["couleurs"]["cadretxt"])
135 plt
.grid(conf
["grille"]=="oui")
139 #warning("Jusque là ça marche",liste_err)