changelog à jour (pour mise en prod)
[perso/Denise/oms.git] / calculs_extrapole.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 """
4 Created on Thu Feb 18 18:25:11 2021
5
6 @author: sekhmet
7 """
8
9 from gestion_erreurs import warning, debug
10 from configuration import CONFIG
11 from gestion_donnees import convertit_age_vers_texte
12 import gestion_unites as u
13 from math import ceil
14
15 #### Les fonctions qui permettent de faire les calculs additionnels une fois qu'on a l'extrapolation (ou pas)
16
17 def interpole_lineaire(dates, donnees, t, liste_err):
18 """ simple interpolation linéaire.
19 dates et donnees sont les dates et données de l'enfant.
20 t est le temps (jours) auquel on veut l'interpolation. On renvoie la donnée interpolée.
21 On interpole linéairement entre les deux données. Si ça
22 n'est pas possible parce que t n'est pas dans les bornes, on renvoie -1
23 """
24
25 if t<dates[0] or t>dates[-1]:
26 return -1 # en dehors des bornes
27
28 i=0 # on cherche l'intervalle : entre dates[i] et dates[i+1] on a la donnée t
29 while i<len(dates)-1 and dates[i+1]<t:
30 i+=1
31
32 m = (donnees[i+1] - donnees[i])/(dates[i+1] - dates[i]) # pente de la courbe
33 valeur = m * (t - dates[i]) + donnees[i]
34 return valeur
35
36
37 def interpole_lineaire_ordonnee(dates, donnees, val, liste_err):
38 """ interpolation linéaire des données.
39 dates et donnees sont les dates et données de l'enfant.
40 val est la valeur pour laquelle on veut trouver le temps.
41 On interpole linéairement entre les deux données. Si ça
42 n'est pas possible parce que t n'est pas dans les bornes, on renvoie -1
43
44 Comportement potentiellement foireux si jamais les données ne sont pas
45 croissantes (poids qui chute, etc)..."""
46
47 if val>max(donnees) or val<min(donnees):
48 return -1 # en dehors des maxi et mini
49
50 i = 0 # On cherche dans quel intervalles de données on se situe (donnnes[i]
51 # et donnees [i+1])
52 while i<len(donnees)-1 and donnees[i+1]<val:
53 i+=1
54
55 m = (dates[i+1] - dates[i])/(donnees[i+1] - donnees[i])
56 temps = m * (val - donnees[i]) + dates[i]
57
58 return ceil(temps)
59
60
61 ######################################
62
63
64
65
66
67 def calcule_donnee_extrapolee(dates_extrapole, donnees_extrapole, age_voulu, liste_err):
68 """ prend en argument les données extrapolées, l'âge voulu en jours.
69 Renvoie la donnée, ou -1 si ça n'a pas marché"""
70 age_voulu = int(age_voulu) # pour avoir les choses bien
71 debug("On veut la donnée à l'âge "+str(age_voulu), liste_err)
72 try:
73 i_date_voulue = dates_extrapole.index(age_voulu)
74 donnee_voulue = donnees_extrapole[i_date_voulue]
75
76 return donnee_voulue
77 #return formate_resultat_donnee(age_voulu, donnee_voulue, typedonnee, "", liste_err)
78 except:
79 warning("Impossible de calculer la donnée à l'âge "+age_voulu, liste_err)
80 return -1
81
82
83
84 def calcule_age_extrapole(dates_extrapole, donnees_extrapole, donnee_voulue, liste_err):
85 """ prend en argument les données extrapolées, la donnée voulue (dans l'unité adaptée)
86 Renvoie la donnée, ou -1 si ça n'a pas marché"""
87
88 debug("On veut savoir à quel âge on a "+str(donnee_voulue), liste_err)
89 try:
90 i=0
91 while i<len(donnees_extrapole) and donnees_extrapole[i]<donnee_voulue:
92 i+=1
93
94 return dates_extrapole[i]
95 #return formate_resultat_age(dates_extrapole[i], donnee_voulue, typedonnee, "", liste_err)
96 except:
97 warning("Impossible de calculer l'âge pour la donnée "+str(donnee_voulue), liste_err)
98 return -1
99
100 ######################################
101 ### Formatage du résultat "joli"
102
103 def met_s(chaine):
104 """ renvoie un s si pluriel, rien sinon"""
105 if int(chaine)>1:
106 return "s"
107 else:
108 return ""
109
110 def joliechaine_age(age):
111 """ prend en argument une chaîne du type 6a3m2j et renvoie une chaîne un peu
112 plus jolie à lire genre 6 ans, 3mois, 2 jours"""
113 retour = ""
114 nombre = ""
115 for char in age:
116 if char=="a":
117 retour+=nombre+" an"+met_s(nombre)+", "
118 nombre=""
119 elif char=="m":
120 retour+=nombre+" mois, "
121 nombre=""
122 elif char=="s":
123 retour+=nombre+" semaine"+met_s(nombre)+", "
124 nombre=""
125 elif char=="j":
126 retour+=nombre+" jour"+met_s(nombre)+", "
127 nombre=""
128 else:
129 nombre+=char
130 #print(retour)
131 return retour[:-2]
132
133 def formate_resultat_donnee(age, date, donnee, typedonnee, extra, liste_err):
134 """ Formate le tout en une zolie phrase
135 age date et donnee sont les données. date peut être None (sinon donnée de date)
136 typedonnee est le type de donnée (poids, etc)
137 extra est un truc additionnel à mettre entre parenthèses"""
138
139 donnee_arrondie = u.arrondit_donnee(donnee, typedonnee)
140 chaine = "À "+joliechaine_age(convertit_age_vers_texte(age))
141
142 if date != None:
143 chaine += ", le "+date.strftime("%d/%m/%Y")
144
145 if typedonnee == "poids":
146 chaine+= ", l'enfant pèsera "+str(donnee_arrondie)+" kg"
147 elif typedonnee == "taille":
148 chaine+= ", l'enfant mesurera "+str(donnee_arrondie)+" cm"
149 else: # phrase générique
150 chaine+= ", la donnée sera : "+str(donnee_arrondie)+" "+CONFIG["unites_typedonnees"][typedonnee]
151
152 if extra!="":
153 ajout=" ("+extra+")"
154 else:
155 ajout=""
156
157 chaine+=ajout+"."
158 return chaine
159
160 def formate_resultat_age(age, date, donnee, typedonnee, extra, liste_err):
161 """ formate les données en une zolie phrase
162 age, date et donnee sont les données. date est une donnée de date qui peut être vide.
163 typedonnee est le type de donnée (poids, etc)
164 extra est un truc additionnel à mettre entre parenthèses"""
165 age_joli = joliechaine_age(convertit_age_vers_texte(age))
166 if typedonnee=="poids":
167 chaine= "L'enfant atteindra un poids de "+str(donnee)+" kg à l'âge de "+age_joli
168 elif typedonnee=="taille":
169 chaine= "L'enfant atteindra la taille "+str(donnee)+ "cm à l'âge de "+age_joli
170 else:# phrase générique
171 chaine= "L'enfant atteindra la donnée "+typedonnee+" "+str(donnee)+" à l'âge de "+age_joli
172
173 if date != None:
174 chaine += ", le "+date.strftime("%d/%m/%Y")
175
176 if extra!="":
177 ajout=" ("+extra+")"
178 else:
179 ajout=""
180
181 chaine+=ajout+"."
182 return chaine
183
184 def formate_extrapole(nb_extra):
185 """ Renvoie une chaîne qui dit sur cb on a extrapolé"""
186 message="extrapolation à partir "
187 if nb_extra== 0:
188 message+="de l'ensemble des données"
189 elif nb_extra==1:
190 message+="de la donnée la plus récente"
191 else:
192 message+="des "+str(nb_extra)+" dernières données"
193 return message
194
195 def formate_interpole():
196 return "interpolation à partir des données existantes"