Salome HOME
Python 3 compatibility improvement (UTF-8) and data interface changes
[modules/adao.git] / src / daComposant / daAlgorithms / LinearityTest.py
1 # -*- coding: utf-8 -*-
2 #
3 # Copyright (C) 2008-2017 EDF R&D
4 #
5 # This library is free software; you can redistribute it and/or
6 # modify it under the terms of the GNU Lesser General Public
7 # License as published by the Free Software Foundation; either
8 # version 2.1 of the License.
9 #
10 # This library is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13 # Lesser General Public License for more details.
14 #
15 # You should have received a copy of the GNU Lesser General Public
16 # License along with this library; if not, write to the Free Software
17 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
18 #
19 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #
21 # Author: Jean-Philippe Argaud, jean-philippe.argaud@edf.fr, EDF R&D
22
23 import sys, logging
24 from daCore import BasicObjects, PlatformInfo
25 import numpy, math
26 mpr = PlatformInfo.PlatformInfo().MachinePrecision()
27 if sys.version_info.major > 2:
28     unicode = str
29
30 # ==============================================================================
31 class ElementaryAlgorithm(BasicObjects.Algorithm):
32     def __init__(self):
33         BasicObjects.Algorithm.__init__(self, "LINEARITYTEST")
34         self.defineRequiredParameter(
35             name     = "ResiduFormula",
36             default  = "CenteredDL",
37             typecast = str,
38             message  = "Formule de résidu utilisée",
39             listval  = ["CenteredDL", "Taylor", "NominalTaylor", "NominalTaylorRMS"],
40             )
41         self.defineRequiredParameter(
42             name     = "EpsilonMinimumExponent",
43             default  = -8,
44             typecast = int,
45             message  = "Exposant minimal en puissance de 10 pour le multiplicateur d'incrément",
46             minval   = -20,
47             maxval   = 0,
48             )
49         self.defineRequiredParameter(
50             name     = "InitialDirection",
51             default  = [],
52             typecast = list,
53             message  = "Direction initiale de la dérivée directionnelle autour du point nominal",
54             )
55         self.defineRequiredParameter(
56             name     = "AmplitudeOfInitialDirection",
57             default  = 1.,
58             typecast = float,
59             message  = "Amplitude de la direction initiale de la dérivée directionnelle autour du point nominal",
60             )
61         self.defineRequiredParameter(
62             name     = "AmplitudeOfTangentPerturbation",
63             default  = 1.e-2,
64             typecast = float,
65             message  = "Amplitude de la perturbation pour le calcul de la forme tangente",
66             minval   = 1.e-10,
67             maxval   = 1.,
68             )
69         self.defineRequiredParameter(
70             name     = "SetSeed",
71             typecast = numpy.random.seed,
72             message  = "Graine fixée pour le générateur aléatoire",
73             )
74         self.defineRequiredParameter(
75             name     = "ResultTitle",
76             default  = "",
77             typecast = str,
78             message  = "Titre du tableau et de la figure",
79             )
80         self.defineRequiredParameter(
81             name     = "StoreSupplementaryCalculations",
82             default  = [],
83             typecast = tuple,
84             message  = "Liste de calculs supplémentaires à stocker et/ou effectuer",
85             listval  = ["CurrentState", "Residu", "SimulatedObservationAtCurrentState"]
86             )
87
88     def run(self, Xb=None, Y=None, U=None, HO=None, EM=None, CM=None, R=None, B=None, Q=None, Parameters=None):
89         self._pre_run(Parameters)
90         #
91         def RMS(V1, V2):
92             import math
93             return math.sqrt( ((numpy.ravel(V2) - numpy.ravel(V1))**2).sum() / float(numpy.ravel(V1).size) )
94         #
95         # Operateurs
96         # ----------
97         Hm = HO["Direct"].appliedTo
98         if self._parameters["ResiduFormula"] in ["Taylor", "NominalTaylor", "NominalTaylorRMS"]:
99             Ht = HO["Tangent"].appliedInXTo
100         #
101         # Construction des perturbations
102         # ------------------------------
103         Perturbations = [ 10**i for i in range(self._parameters["EpsilonMinimumExponent"],1) ]
104         Perturbations.reverse()
105         #
106         # Calcul du point courant
107         # -----------------------
108         Xn      = numpy.asmatrix(numpy.ravel( Xb )).T
109         FX      = numpy.asmatrix(numpy.ravel( Hm( Xn ) )).T
110         NormeX  = numpy.linalg.norm( Xn )
111         NormeFX = numpy.linalg.norm( FX )
112         if "CurrentState" in self._parameters["StoreSupplementaryCalculations"]:
113             self.StoredVariables["CurrentState"].store( numpy.ravel(Xn) )
114         if "SimulatedObservationAtCurrentState" in self._parameters["StoreSupplementaryCalculations"]:
115             self.StoredVariables["SimulatedObservationAtCurrentState"].store( numpy.ravel(FX) )
116         #
117         # Fabrication de la direction de l'increment dX
118         # ---------------------------------------------
119         if len(self._parameters["InitialDirection"]) == 0:
120             dX0 = []
121             for v in Xn.A1:
122                 if abs(v) > 1.e-8:
123                     dX0.append( numpy.random.normal(0.,abs(v)) )
124                 else:
125                     dX0.append( numpy.random.normal(0.,Xn.mean()) )
126         else:
127             dX0 = numpy.ravel( self._parameters["InitialDirection"] )
128         #
129         dX0 = float(self._parameters["AmplitudeOfInitialDirection"]) * numpy.matrix( dX0 ).T
130         #
131         # Calcul du gradient au point courant X pour l'increment dX
132         # ---------------------------------------------------------
133         if self._parameters["ResiduFormula"] in ["Taylor", "NominalTaylor", "NominalTaylorRMS"]:
134             dX1      = float(self._parameters["AmplitudeOfTangentPerturbation"]) * dX0
135             GradFxdX = Ht( (Xn, dX1) )
136             GradFxdX = numpy.asmatrix(numpy.ravel( GradFxdX )).T
137             GradFxdX = float(1./self._parameters["AmplitudeOfTangentPerturbation"]) * GradFxdX
138         #
139         # Entete des resultats
140         # --------------------
141         __marge =  12*u" "
142         __precision = u"""
143             Remarque : les nombres inferieurs a %.0e (environ) representent un zero
144                        a la precision machine.\n"""%mpr
145         if self._parameters["ResiduFormula"] == "CenteredDL":
146             __entete = u"  i   Alpha     ||X||      ||F(X)||   |   R(Alpha)  log10( R )  "
147             __msgdoc = u"""
148             On observe le residu provenant de la difference centree des valeurs de F
149             au point nominal et aux points perturbes, normalisee par la valeur au
150             point nominal :
151
152                          || F(X+Alpha*dX) + F(X-Alpha*dX) - 2*F(X) ||
153               R(Alpha) = --------------------------------------------
154                                          || F(X) ||
155
156             S'il reste constamment tres faible par rapport a 1, l'hypothese de linearite
157             de F est verifiee.
158
159             Si le residu varie, ou qu'il est de l'ordre de 1 ou plus, et qu'il n'est
160             faible qu'a partir d'un certain ordre d'increment, l'hypothese de linearite
161             de F n'est pas verifiee.
162
163             Si le residu decroit et que la decroissance se fait en Alpha**2 selon Alpha,
164             cela signifie que le gradient est calculable jusqu'a la precision d'arret
165             de la decroissance quadratique.
166
167             On prend dX0 = Normal(0,X) et dX = Alpha*dX0. F est le code de calcul.
168             """ + __precision
169         if self._parameters["ResiduFormula"] == "Taylor":
170             __entete = u"  i   Alpha     ||X||      ||F(X)||   |   R(Alpha)  log10( R )  "
171             __msgdoc = u"""
172             On observe le residu issu du developpement de Taylor de la fonction F,
173             normalisee par la valeur au point nominal :
174
175                          || F(X+Alpha*dX) - F(X) - Alpha * GradientF_X(dX) ||
176               R(Alpha) = ----------------------------------------------------
177                                          || F(X) ||
178
179             S'il reste constamment tres faible par rapport a 1, l'hypothese de linearite
180             de F est verifiee.
181
182             Si le residu varie, ou qu'il est de l'ordre de 1 ou plus, et qu'il n'est
183             faible qu'a partir d'un certain ordre d'increment, l'hypothese de linearite
184             de F n'est pas verifiee.
185
186             Si le residu decroit et que la decroissance se fait en Alpha**2 selon Alpha,
187             cela signifie que le gradient est bien calcule jusqu'a la precision d'arret
188             de la decroissance quadratique.
189
190             On prend dX0 = Normal(0,X) et dX = Alpha*dX0. F est le code de calcul.
191             """ + __precision
192         if self._parameters["ResiduFormula"] == "NominalTaylor":
193             __entete = u"  i   Alpha     ||X||      ||F(X)||   |   R(Alpha)   |R-1| en %  "
194             __msgdoc = u"""
195             On observe le residu obtenu a partir de deux approximations d'ordre 1 de F(X),
196             normalisees par la valeur au point nominal :
197
198               R(Alpha) = max(
199                 || F(X+Alpha*dX) - Alpha * F(dX) || / || F(X) ||,
200                 || F(X-Alpha*dX) + Alpha * F(dX) || / || F(X) ||,
201               )
202
203             S'il reste constamment egal a 1 a moins de 2 ou 3 pourcents pres (c'est-a-dire
204             que |R-1| reste egal a 2 ou 3 pourcents), c'est que l'hypothese de linearite
205             de F est verifiee.
206
207             S'il est egal a 1 sur une partie seulement du domaine de variation de
208             l'increment Alpha, c'est sur cette partie que l'hypothese de linearite de F
209             est verifiee.
210
211             On prend dX0 = Normal(0,X) et dX = Alpha*dX0. F est le code de calcul.
212             """ + __precision
213         if self._parameters["ResiduFormula"] == "NominalTaylorRMS":
214             __entete = u"  i   Alpha     ||X||      ||F(X)||   |   R(Alpha)    |R| en %  "
215             __msgdoc = u"""
216             On observe le residu obtenu a partir de deux approximations d'ordre 1 de F(X),
217             normalisees par la valeur au point nominal :
218
219               R(Alpha) = max(
220                 RMS( F(X), F(X+Alpha*dX) - Alpha * F(dX) ) / || F(X) ||,
221                 RMS( F(X), F(X-Alpha*dX) + Alpha * F(dX) ) / || F(X) ||,
222               )
223
224             S'il reste constamment egal a 0 a moins de 1 ou 2 pourcents pres, c'est
225             que l'hypothese de linearite de F est verifiee.
226
227             S'il est egal a 0 sur une partie seulement du domaine de variation de
228             l'increment Alpha, c'est sur cette partie que l'hypothese de linearite de F
229             est verifiee.
230
231             On prend dX0 = Normal(0,X) et dX = Alpha*dX0. F est le code de calcul.
232             """ + __precision
233         #
234         if len(self._parameters["ResultTitle"]) > 0:
235             __rt = unicode(self._parameters["ResultTitle"])
236             msgs  = u"\n"
237             msgs += __marge + "====" + "="*len(__rt) + "====\n"
238             msgs += __marge + "    " + __rt + "\n"
239             msgs += __marge + "====" + "="*len(__rt) + "====\n"
240         else:
241             msgs  = u""
242         msgs += __msgdoc
243         #
244         __nbtirets = len(__entete)
245         msgs += "\n" + __marge + "-"*__nbtirets
246         msgs += "\n" + __marge + __entete
247         msgs += "\n" + __marge + "-"*__nbtirets
248         #
249         # Boucle sur les perturbations
250         # ----------------------------
251         for i,amplitude in enumerate(Perturbations):
252             dX      = amplitude * dX0
253             #
254             if self._parameters["ResiduFormula"] == "CenteredDL":
255                 if "CurrentState" in self._parameters["StoreSupplementaryCalculations"]:
256                     self.StoredVariables["CurrentState"].store( numpy.ravel(Xn + dX) )
257                     self.StoredVariables["CurrentState"].store( numpy.ravel(Xn - dX) )
258                 #
259                 FX_plus_dX  = numpy.asmatrix(numpy.ravel( Hm( Xn + dX ) )).T
260                 FX_moins_dX = numpy.asmatrix(numpy.ravel( Hm( Xn - dX ) )).T
261                 #
262                 if "SimulatedObservationAtCurrentState" in self._parameters["StoreSupplementaryCalculations"]:
263                     self.StoredVariables["SimulatedObservationAtCurrentState"].store( numpy.ravel(FX_plus_dX) )
264                     self.StoredVariables["SimulatedObservationAtCurrentState"].store( numpy.ravel(FX_moins_dX) )
265                 #
266                 Residu = numpy.linalg.norm( FX_plus_dX + FX_moins_dX - 2 * FX ) / NormeFX
267                 #
268                 self.StoredVariables["Residu"].store( Residu )
269                 msg = "  %2i  %5.0e   %9.3e   %9.3e   |   %9.3e   %4.0f"%(i,amplitude,NormeX,NormeFX,Residu,math.log10(max(1.e-99,Residu)))
270                 msgs += "\n" + __marge + msg
271             #
272             if self._parameters["ResiduFormula"] == "Taylor":
273                 if "CurrentState" in self._parameters["StoreSupplementaryCalculations"]:
274                     self.StoredVariables["CurrentState"].store( numpy.ravel(Xn + dX) )
275                 #
276                 FX_plus_dX  = numpy.asmatrix(numpy.ravel( Hm( Xn + dX ) )).T
277                 #
278                 if "SimulatedObservationAtCurrentState" in self._parameters["StoreSupplementaryCalculations"]:
279                     self.StoredVariables["SimulatedObservationAtCurrentState"].store( numpy.ravel(FX_plus_dX) )
280                 #
281                 Residu = numpy.linalg.norm( FX_plus_dX - FX - amplitude * GradFxdX ) / NormeFX
282                 #
283                 self.StoredVariables["Residu"].store( Residu )
284                 msg = "  %2i  %5.0e   %9.3e   %9.3e   |   %9.3e   %4.0f"%(i,amplitude,NormeX,NormeFX,Residu,math.log10(max(1.e-99,Residu)))
285                 msgs += "\n" + __marge + msg
286             #
287             if self._parameters["ResiduFormula"] == "NominalTaylor":
288                 if "CurrentState" in self._parameters["StoreSupplementaryCalculations"]:
289                     self.StoredVariables["CurrentState"].store( numpy.ravel(Xn + dX) )
290                     self.StoredVariables["CurrentState"].store( numpy.ravel(Xn - dX) )
291                     self.StoredVariables["CurrentState"].store( numpy.ravel(dX) )
292                 #
293                 FX_plus_dX  = numpy.asmatrix(numpy.ravel( Hm( Xn + dX ) )).T
294                 FX_moins_dX = numpy.asmatrix(numpy.ravel( Hm( Xn - dX ) )).T
295                 FdX         = numpy.asmatrix(numpy.ravel( Hm( dX ) )).T
296                 #
297                 if "SimulatedObservationAtCurrentState" in self._parameters["StoreSupplementaryCalculations"]:
298                     self.StoredVariables["SimulatedObservationAtCurrentState"].store( numpy.ravel(FX_plus_dX) )
299                     self.StoredVariables["SimulatedObservationAtCurrentState"].store( numpy.ravel(FX_moins_dX) )
300                     self.StoredVariables["SimulatedObservationAtCurrentState"].store( numpy.ravel(FdX) )
301                 #
302                 Residu = max(
303                     numpy.linalg.norm( FX_plus_dX  - amplitude * FdX ) / NormeFX,
304                     numpy.linalg.norm( FX_moins_dX + amplitude * FdX ) / NormeFX,
305                     )
306                 #
307                 self.StoredVariables["Residu"].store( Residu )
308                 msg = "  %2i  %5.0e   %9.3e   %9.3e   |   %9.3e   %5i %s"%(i,amplitude,NormeX,NormeFX,Residu,100.*abs(Residu-1.),"%")
309                 msgs += "\n" + __marge + msg
310             #
311             if self._parameters["ResiduFormula"] == "NominalTaylorRMS":
312                 if "CurrentState" in self._parameters["StoreSupplementaryCalculations"]:
313                     self.StoredVariables["CurrentState"].store( numpy.ravel(Xn + dX) )
314                     self.StoredVariables["CurrentState"].store( numpy.ravel(Xn - dX) )
315                     self.StoredVariables["CurrentState"].store( numpy.ravel(dX) )
316                 #
317                 FX_plus_dX  = numpy.asmatrix(numpy.ravel( Hm( Xn + dX ) )).T
318                 FX_moins_dX = numpy.asmatrix(numpy.ravel( Hm( Xn - dX ) )).T
319                 FdX         = numpy.asmatrix(numpy.ravel( Hm( dX ) )).T
320                 #
321                 if "SimulatedObservationAtCurrentState" in self._parameters["StoreSupplementaryCalculations"]:
322                     self.StoredVariables["SimulatedObservationAtCurrentState"].store( numpy.ravel(FX_plus_dX) )
323                     self.StoredVariables["SimulatedObservationAtCurrentState"].store( numpy.ravel(FX_moins_dX) )
324                     self.StoredVariables["SimulatedObservationAtCurrentState"].store( numpy.ravel(FdX) )
325                 #
326                 Residu = max(
327                     RMS( FX, FX_plus_dX   - amplitude * FdX ) / NormeFX,
328                     RMS( FX, FX_moins_dX  + amplitude * FdX ) / NormeFX,
329                     )
330                 #
331                 self.StoredVariables["Residu"].store( Residu )
332                 msg = "  %2i  %5.0e   %9.3e   %9.3e   |   %9.3e   %5i %s"%(i,amplitude,NormeX,NormeFX,Residu,100.*Residu,"%")
333                 msgs += "\n" + __marge + msg
334         #
335         msgs += "\n" + __marge + "-"*__nbtirets
336         msgs += "\n"
337         #
338         # Sorties eventuelles
339         # -------------------
340         print("\nResults of linearity check by \"%s\" formula:"%self._parameters["ResiduFormula"])
341         print(msgs)
342         #
343         self._post_run(HO)
344         return 0
345
346 # ==============================================================================
347 if __name__ == "__main__":
348     print('\n AUTODIAGNOSTIC \n')