]>
git.immae.eu Git - perso/Denise/oms.git/blob - trace_courbe.py
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
):
11 liste_data_labels_p
,liste_data_labels_z
= renvoie_liste_labels(conf
,liste_data_choisie_p
,liste_data_choisie_z
,liste_err
)
13 erreur("bug avec liste data labels",liste_err
)
18 jour_maxi
= max(l_jours
)# pas la peine d'aller très au delà du jour max
19 jour_maxi
= int(jour_maxi
* 1.1)+3 # on rajoute un peu
21 jour_maxi
= CONFIG
["jours_defaut_donneesvides"] # si pas de données, arbitrairement on met ça
23 jour_maxi
= conf
["maxi"]+1
25 # On s'assure que c'est bien compris dans les bornes
26 jour_maxi
= max(CONFIG
["jours_mini_courbe"],min(jour_maxi
,CONFIG
["jours_maxi_courbe"]))
28 if conf
["unite"] == "":
29 conf
["unite"] = choix_unite(jour_maxi
)
31 l_jours_conv
= convertit_tableau(l_jours
,conf
["unite"],liste_err
)
32 # Attention, comme les jours commencent à partir de 0, faut enlever 1 pour avoir la borne...
33 age_maxi
= convertitunite(jour_maxi
-1,conf
["unite"],liste_err
)
36 titre
= "Courbe de poids OMS"
39 if conf
["typecourbe"] == "P":
41 liste_data_labels
= liste_data_labels_p
42 if conf
["sexe"] == "M":
43 fichier_oms
= f_poids_perc_garcon
44 titre
+= " (percentiles, garçon)"
45 elif conf
["sexe"] == "F":
46 fichier_oms
= f_poids_perc_fille
47 titre
+= " (percentiles, fille)"
49 fichier_oms
= f_poids_perc_mixte
50 titre
+= " (percentiles)"
51 elif conf
["typecourbe"] == "Z":
52 liste_data_labels
= liste_data_labels_z
53 if conf
["sexe"] == "G":
54 fichier_oms
= f_poids_z_garcon
55 titre
+= " (moyenne et écarts-types, garçon)"
56 elif conf
["sexe"] == "F":
57 fichier_oms
= f_poids_z_fille
58 titre
+= " (moyenne et écarts-types, fille)"
60 fichier_oms
= f_poids_z_mixte
61 titre
+= " (moyenne et écarts-types)"
63 warning("Type de courbe invalide"+conf
["typecourbe"],liste_err
)
65 # warning("cree_figure : ça va jusque là ",liste_err)
66 # warning("bli"+str(conf),liste_err)
69 # Si y'a un nom on met "courbe de machin"
71 titre
+= " de " +conf
["nom"]
74 # convertir les unités
76 #### On extrait les données des courbes, et on convertit les jours dans l'unité voulues
78 t
= lire_fichier_csv(fichier_oms
)
80 liste_err
[0].append("Impossible d'ouvrir le fichier "+fichier_oms
)
81 coljour
= convertit_tableau(extraire_colonne(t
,0,jour_maxi
),conf
["unite"],liste_err
)
85 fig
= plt
.figure(num
=None, figsize
=(conf
["largeur"], conf
["hauteur"]), dpi
=100, facecolor
=conf
["couleurs"]["fond"])
86 plt
.rcParams
['axes.facecolor'] = conf
["couleurs"]["fond"]
87 plt
.rcParams
['axes.edgecolor']= conf
["couleurs"]["cadretxt"]
88 plt
.rcParams
['xtick.color'] = conf
["couleurs"]["cadretxt"]
89 plt
.rcParams
['ytick.color'] = conf
["couleurs"]["cadretxt"]
90 plt
.rcParams
['grid.color'] = conf
["couleurs"]["grille"]
91 plt
.rcParams
['legend.edgecolor'] = conf
["couleurs"]["grille"]
93 #warning("bla"+str(liste_data_labels),liste_err)
94 for (i
,label
,couleur
) in liste_data_labels
:
95 plt
.plot(coljour
,extraire_colonne(t
,i
,jour_maxi
),label
=label
,color
=couleur
)
96 #warning("colonne"+str(i)+str(label)+str(couleur),liste_err)
100 # On extrait la valeur min et la valeur max des poids
101 (colonne_min
,_
,_
) = liste_data_labels
[-1]
102 (colonne_max
,_
,_
) = liste_data_labels
[0]
104 poids_min
= min(extraire_colonne(t
,colonne_min
,jour_maxi
))
105 poids_max
= max(extraire_colonne(t
,colonne_max
,jour_maxi
))
107 poids_min
= min(min(l_poids
),poids_min
)
108 poids_max
= max(max(l_poids
),poids_max
)
109 # On ajuste un peu ces min et max
110 # min : valeur min -1kg
111 poids_min
= max(0,poids_min
-1)
113 poids_max
= poids_max
* 1.05
115 #warning("Jusque là ça marche",liste_err)
118 plt
.plot(l_jours_conv
,l_poids
,label
=conf
["nom"],color
=conf
["couleur_cadretxt"],marker
='o')
120 plt
.xlabel("Âge en "+conf
["unite"],color
=conf
["couleurs"]["cadretxt"])
121 plt
.ylabel("Poids en kg",color
=conf
["couleurs"]["cadretxt"])
122 plt
.title(titre
,color
=conf
["couleurs"]["cadretxt"])
123 plt
.axis([0,age_maxi
,poids_min
,poids_max
])
125 if conf
['legende']=="oui":
126 legende
= plt
.legend(loc
=conf
['positionlegende'])
127 plt
.setp(legende
.get_texts(), color
=conf
["couleurs"]["cadretxt"])
128 plt
.grid(conf
["grille"]=="oui")
132 #warning("Jusque là ça marche",liste_err)