1 #-*-coding:iso-8859-1-*-
3 # Copyright (C) 2008-2012 EDF R&D
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.
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.
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
19 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 # Author: Jean-Philippe Argaud, jean-philippe.argaud@edf.fr, EDF R&D
24 Définit les versions approximées des opérateurs tangents et adjoints.
26 __author__ = "Jean-Philippe ARGAUD"
28 import os, numpy, time
30 # logging.getLogger().setLevel(logging.DEBUG)
32 # ==============================================================================
33 class FDApproximation:
35 Cette classe sert d'interface pour définir les opérateurs approximés. A la
36 création d'un objet, en fournissant une fonction "Function", on obtient un
37 objet qui dispose de 3 méthodes "DirectOperator", "TangentOperator" et
38 "AdjointOperator". On contrôle l'approximation DF avec l'incrément
39 multiplicatif "increment" valant par défaut 1%, ou avec l'incrément fixe
40 "dX" qui sera multiplié par "increment" (donc en %), et on effectue de DF
41 centrées si le booléen "centeredDF" est vrai.
43 def __init__(self, Function = None, centeredDF = False, increment = 0.01, dX = None):
44 self.DirectOperator = Function
45 self.__centeredDF = bool(centeredDF)
46 if float(increment) <> 0.:
47 self.__increment = float(increment)
49 self.__increment = 0.01
53 self.__dX = numpy.asmatrix(numpy.ravel( dX )).T
55 # ---------------------------------------------------------
56 def TangentMatrix(self, X ):
58 Calcul de l'opérateur tangent comme la Jacobienne par différences finies,
59 c'est-à-dire le gradient de H en X. On utilise des différences finies
60 directionnelles autour du point X. X est un numpy.matrix.
62 Différences finies centrées :
63 1/ Pour chaque composante i de X, on ajoute et on enlève la perturbation
64 dX[i] à la composante X[i], pour composer X_plus_dXi et X_moins_dXi, et
65 on calcule les réponses HX_plus_dXi = H( X_plus_dXi ) et HX_moins_dXi =
67 2/ On effectue les différences (HX_plus_dXi-HX_moins_dXi) et on divise par
69 3/ Chaque résultat, par composante, devient une colonne de la Jacobienne
71 Différences finies non centrées :
72 1/ Pour chaque composante i de X, on ajoute la perturbation dX[i] à la
73 composante X[i] pour composer X_plus_dXi, et on calcule la réponse
74 HX_plus_dXi = H( X_plus_dXi )
75 2/ On calcule la valeur centrale HX = H(X)
76 3/ On effectue les différences (HX_plus_dXi-HX) et on divise par
78 4/ Chaque résultat, par composante, devient une colonne de la Jacobienne
81 logging.debug(" == Calcul de la Jacobienne")
82 logging.debug(" Incrément de............: %s*X"%float(self.__increment))
83 logging.debug(" Approximation centrée...: %s"%(self.__centeredDF))
85 _X = numpy.asmatrix(numpy.ravel( X )).T
88 _dX = self.__increment * _X
90 _dX = numpy.asmatrix(numpy.ravel( self.__dX )).T
95 _dX = numpy.where( _dX == 0., float(self.__increment), _dX )
97 _dX = numpy.where( _dX == 0., moyenne, _dX )
101 # Boucle de calcul des colonnes de la Jacobienne
102 # ----------------------------------------------
104 for i in range( len(_dX) ):
105 X_plus_dXi = numpy.array( _X.A1, dtype=float )
106 X_plus_dXi[i] = _X[i] + _dX[i]
107 X_moins_dXi = numpy.array( _X.A1, dtype=float )
108 X_moins_dXi[i] = _X[i] - _dX[i]
110 HX_plus_dXi = self.DirectOperator( X_plus_dXi )
111 HX_moins_dXi = self.DirectOperator( X_moins_dXi )
113 HX_Diff = ( HX_plus_dXi - HX_moins_dXi ) / (2.*_dX[i])
115 Jacobienne.append( HX_Diff )
119 # Boucle de calcul des colonnes de la Jacobienne
120 # ----------------------------------------------
122 for i in range( len(_dX) ):
123 X_plus_dXi = numpy.array( _X.A1, dtype=float )
124 X_plus_dXi[i] = _X[i] + _dX[i]
126 HX_plus_dXi = self.DirectOperator( X_plus_dXi )
128 HX_plus_dX.append( HX_plus_dXi )
130 # Calcul de la valeur centrale
131 # ----------------------------
132 HX = self.DirectOperator( _X )
134 # Calcul effectif de la Jacobienne par différences finies
135 # -------------------------------------------------------
137 for i in range( len(_dX) ):
138 Jacobienne.append( numpy.ravel(( HX_plus_dX[i] - HX ) / _dX[i]) )
140 Jacobienne = numpy.matrix( numpy.vstack( Jacobienne ) ).T
141 logging.debug(" == Fin du calcul de la Jacobienne")
145 # ---------------------------------------------------------
146 def TangentOperator(self, (X, dX) ):
148 Calcul du tangent à l'aide de la Jacobienne.
150 Jacobienne = self.TangentMatrix( X )
151 if dX is None or len(dX) == 0:
153 # Calcul de la forme matricielle si le second argument est None
154 # -------------------------------------------------------------
158 # Calcul de la valeur linéarisée de H en X appliqué à dX
159 # ------------------------------------------------------
160 _dX = numpy.asmatrix(numpy.ravel( dX )).T
161 HtX = numpy.dot(Jacobienne, _dX)
164 # ---------------------------------------------------------
165 def AdjointOperator(self, (X, Y) ):
167 Calcul de l'adjoint à l'aide de la Jacobienne.
169 JacobienneT = self.TangentMatrix( X ).T
170 if Y is None or len(Y) == 0:
172 # Calcul de la forme matricielle si le second argument est None
173 # -------------------------------------------------------------
177 # Calcul de la valeur de l'adjoint en X appliqué à Y
178 # --------------------------------------------------
179 _Y = numpy.asmatrix(numpy.ravel( Y )).T
180 HaY = numpy.dot(JacobienneT, _Y)
183 # ==============================================================================
186 """ Direct non-linear simulation operator """
188 # NEED TO BE COMPLETED
189 # NEED TO BE COMPLETED
190 # NEED TO BE COMPLETED
192 # --------------------------------------> # EXAMPLE TO BE REMOVED
193 # Example of Identity operator # EXAMPLE TO BE REMOVED
194 if type(XX) is type(numpy.matrix([])): # EXAMPLE TO BE REMOVED
195 HX = XX.A1.tolist() # EXAMPLE TO BE REMOVED
196 elif type(XX) is type(numpy.array([])): # EXAMPLE TO BE REMOVED
197 HX = numpy.matrix(XX).A1.tolist() # EXAMPLE TO BE REMOVED
198 else: # EXAMPLE TO BE REMOVED
199 HX = XX # EXAMPLE TO BE REMOVED
200 # # EXAMPLE TO BE REMOVED
201 HHX = [] # EXAMPLE TO BE REMOVED
202 HHX.extend( HX ) # EXAMPLE TO BE REMOVED
203 HHX.extend( HX ) # EXAMPLE TO BE REMOVED
204 # --------------------------------------> # EXAMPLE TO BE REMOVED
206 return numpy.array( HHX )
208 # ==============================================================================
209 if __name__ == "__main__":
212 print "AUTODIAGNOSTIC"
213 print "=============="
217 FDA = FDApproximation( test1 )
218 print "H(X) =", FDA.DirectOperator( X0 )
219 print "Tg matrice =\n", FDA.TangentMatrix( X0 )
220 print "Tg(X) =", FDA.TangentOperator( (X0, X0) )
221 print "Ad((X,Y)) =", FDA.AdjointOperator( (X0,range(3,3+2*len(X0))) )
225 FDA = FDApproximation( test1, centeredDF=True )
226 print "H(X) =", FDA.DirectOperator( X0 )
227 print "Tg matrice =\n", FDA.TangentMatrix( X0 )
228 print "Tg(X) =", FDA.TangentOperator( (X0, X0) )
229 print "Ad((X,Y)) =", FDA.AdjointOperator( (X0,range(3,3+2*len(X0))) )
235 FDA = FDApproximation( test1 )
236 print "H(X) =", FDA.DirectOperator( X0 )
237 print "Tg matrice =\n", FDA.TangentMatrix( X0 )
238 print "Tg(X) =", FDA.TangentOperator( (X0, X0) )
239 print "Ad((X,Y)) =", FDA.AdjointOperator( (X0,range(7,7+2*len(X0))) )