1 #-*-coding:iso-8859-1-*-
3 # Copyright (C) 2008-2015 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 Interface de scripting pour une étude ADAO
26 __author__ = "Jean-Philippe ARGAUD"
30 from daCore import AssimilationStudy
34 Creation TUI d'un cas ADAO
36 def __init__(self, name = ""):
37 self.__adaoStudy = AssimilationStudy.AssimilationStudy(name)
38 self.__dumper = _DumpLogger(name)
40 # -----------------------------------------------------------
46 DiagonalSparseMatrix = None,
51 ScalarSparseMatrix = None,
56 ThreeFunctions = None,
61 "Interface unique de définition de variables d'entrées par argument"
62 self.__dumper.register("set",dir(),locals(),None,True)
64 if Concept == "Background":
65 self.setBackground(Vector,VectorSerie,Script,Stored)
66 elif Concept == "BackgroundError":
67 self.setBackgroundError(Matrix,ScalarSparseMatrix,
68 DiagonalSparseMatrix,Script,Stored)
69 elif Concept == "CheckingPoint":
70 self.setCheckingPoint(Vector,VectorSerie,Script,Stored)
71 elif Concept == "ControlModel":
72 self.setControlModel(Matrix,OneFunction,ThreeFunctions,
73 Parameters,Script,Stored)
74 elif Concept == "ControlInput":
75 self.setControlInput(Vector,VectorSerie,Script,Stored)
76 elif Concept == "EvolutionError":
77 self.setEvolutionError(Matrix,ScalarSparseMatrix,
78 DiagonalSparseMatrix,Script,Stored)
79 elif Concept == "EvolutionModel":
80 self.setEvolutionModel(Matrix,OneFunction,ThreeFunctions,
81 Parameters,Script,Stored)
82 elif Concept == "Observation":
83 self.setObservation(Vector,VectorSerie,Script,Stored)
84 elif Concept == "ObservationError":
85 self.setObservationError(Matrix,ScalarSparseMatrix,
86 DiagonalSparseMatrix,Script,Stored)
87 elif Concept == "ObservationOperator":
88 self.setObservationOperator(Matrix,OneFunction,ThreeFunctions,
89 Parameters,Script,Stored)
90 elif Concept == "AlgorithmParameters":
91 self.setAlgorithmParameters(Algorithm,Parameters,Script)
92 elif Concept == "Debug":
94 elif Concept == "NoDebug":
96 elif Concept == "Observer":
97 self.setObserver(Variable,Template,String,Script,Info)
99 raise ValueError("the variable named '%s' is not allowed."%str(Concept))
100 except Exception as e:
101 if type(e) == type(SyntaxError()): msg = "at %s: %s"%(e.offset, e.text)
103 raise ValueError("during settings, the following error occurs:\n\n%s %s\n\nSee also the potential messages, which can show the origin of the above error, in the launching terminal."%(str(e),msg))
105 # -----------------------------------------------------------
113 "Définition d'une entrée de calcul"
114 self.__dumper.register("setBackground", dir(), locals())
115 if Script is not None:
116 __Vector, __PersistentVector = None, None
118 __PersistentVector = _ImportFromScript(Script).getvalue( "Background" )
120 __Vector = _ImportFromScript(Script).getvalue( "Background" )
122 __Vector, __PersistentVector = Vector, VectorSerie
124 self.__adaoStudy.setBackground(
126 asPersistentVector = __PersistentVector,
130 def setBackgroundError(
133 ScalarSparseMatrix = None,
134 DiagonalSparseMatrix = None,
137 "Définition d'une entrée de calcul"
138 self.__dumper.register("setBackgroundError", dir(), locals())
139 if Script is not None:
140 __Covariance, __Scalar, __Vector = None, None, None
141 if ScalarSparseMatrix:
142 __Scalar = _ImportFromScript(Script).getvalue( "BackgroundError" )
143 elif DiagonalSparseMatrix:
144 __Vector = _ImportFromScript(Script).getvalue( "BackgroundError" )
146 __Covariance = _ImportFromScript(Script).getvalue( "BackgroundError" )
148 __Covariance, __Scalar, __Vector = Matrix, ScalarSparseMatrix, DiagonalSparseMatrix
150 self.__adaoStudy.setBackgroundError(
151 asCovariance = __Covariance,
152 asEyeByScalar = __Scalar,
153 asEyeByVector = __Vector,
157 def setCheckingPoint(
163 "Définition d'une entrée de vérification"
164 self.__dumper.register("setCheckingPoint", dir(), locals())
165 if Script is not None:
166 __Vector, __PersistentVector = None, None
168 __PersistentVector = _ImportFromScript(Script).getvalue( "CheckingPoint" )
170 __Vector = _ImportFromScript(Script).getvalue( "CheckingPoint" )
172 __Vector, __PersistentVector = Vector, VectorSerie
174 self.__adaoStudy.setBackground(
176 asPersistentVector = __PersistentVector,
184 ThreeFunctions = None,
188 "Définition d'une entrée de calcul"
189 self.__dumper.register("setControlModel", dir(), locals())
191 if Parameters is not None and type(Parameters) == type({}):
192 if Parameters.has_key("DifferentialIncrement"):
193 __Parameters["withIncrement"] = Parameters["DifferentialIncrement"]
194 if Parameters.has_key("CenteredFiniteDifference"):
195 __Parameters["withCenteredDF"] = Parameters["CenteredFiniteDifference"]
196 if Script is not None:
197 __Matrix, __Function = None, None
199 __Matrix = _ImportFromScript(Script).getvalue( "ObservationOperator" )
201 __Function = { "Direct":_ImportFromScript(Script).getvalue( "DirectOperator" ) }
202 __Function.update({"useApproximatedDerivatives":True})
203 __Function.update(__Parameters)
206 "Direct" :_ImportFromScript(Script).getvalue( "DirectOperator" ),
207 "Tangent":_ImportFromScript(Script).getvalue( "TangentOperator" ),
208 "Adjoint":_ImportFromScript(Script).getvalue( "AdjointOperator" ),
210 __Function.update(__Parameters)
213 if OneFunction is not None:
214 __Function = { "Direct":OneFunction }
215 __Function.update({"useApproximatedDerivatives":True})
216 __Function.update(__Parameters)
217 elif ThreeFunctions is not None:
218 if (type(ThreeFunctions) is not type({})) or \
219 not ThreeFunctions.has_key("Direct") or \
220 not ThreeFunctions.has_key("Tangent") or \
221 not ThreeFunctions.has_key("Adjoint"):
222 raise ValueError("ThreeFunctions has to be a dictionnary and to have the 3 keys Direct, Tangent, Adjoint")
223 __Function = ThreeFunctions
224 __Function.update(__Parameters)
228 self.__adaoStudy.setControlModel(
229 asFunction = __Function,
240 "Définition d'une entrée de calcul"
241 self.__dumper.register("setControlInput", dir(), locals())
242 if Script is not None:
243 __Vector, __PersistentVector = None, None
245 __PersistentVector = _ImportFromScript(Script).getvalue( "ControlInput" )
247 __Vector = _ImportFromScript(Script).getvalue( "ControlInput" )
249 __Vector, __PersistentVector = Vector, VectorSerie
251 self.__adaoStudy.setControlInput(
253 asPersistentVector = __PersistentVector,
257 def setEvolutionError(
260 ScalarSparseMatrix = None,
261 DiagonalSparseMatrix = None,
264 "Définition d'une entrée de calcul"
265 self.__dumper.register("setEvolutionError", dir(), locals())
266 if Script is not None:
267 __Covariance, __Scalar, __Vector = None, None, None
268 if ScalarSparseMatrix:
269 __Scalar = _ImportFromScript(Script).getvalue( "EvolutionError" )
270 elif DiagonalSparseMatrix:
271 __Vector = _ImportFromScript(Script).getvalue( "EvolutionError" )
273 __Covariance = _ImportFromScript(Script).getvalue( "EvolutionError" )
275 __Covariance, __Scalar, __Vector = Matrix, ScalarSparseMatrix, DiagonalSparseMatrix
277 self.__adaoStudy.setEvolutionError(
278 asCovariance = __Covariance,
279 asEyeByScalar = __Scalar,
280 asEyeByVector = __Vector,
284 def setEvolutionModel(
288 ThreeFunctions = None,
292 "Définition d'une entrée de calcul"
293 self.__dumper.register("setEvolutionModel", dir(), locals())
295 if Parameters is not None and type(Parameters) == type({}):
296 if Parameters.has_key("DifferentialIncrement"):
297 __Parameters["withIncrement"] = Parameters["DifferentialIncrement"]
298 if Parameters.has_key("CenteredFiniteDifference"):
299 __Parameters["withCenteredDF"] = Parameters["CenteredFiniteDifference"]
300 if Parameters.has_key("EnableMultiProcessing"):
301 __Parameters["withmpEnabled"] = Parameters["EnableMultiProcessing"]
302 if Parameters.has_key("NumberOfProcesses"):
303 __Parameters["withmpWorkers"] = Parameters["NumberOfProcesses"]
304 if Script is not None:
305 __Matrix, __Function = None, None
307 __Matrix = _ImportFromScript(Script).getvalue( "ObservationOperator" )
309 __Function = { "Direct":_ImportFromScript(Script).getvalue( "DirectOperator" ) }
310 __Function.update({"useApproximatedDerivatives":True})
311 __Function.update(__Parameters)
314 "Direct" :_ImportFromScript(Script).getvalue( "DirectOperator" ),
315 "Tangent":_ImportFromScript(Script).getvalue( "TangentOperator" ),
316 "Adjoint":_ImportFromScript(Script).getvalue( "AdjointOperator" ),
318 __Function.update(__Parameters)
321 if OneFunction is not None:
322 __Function = { "Direct":OneFunction }
323 __Function.update({"useApproximatedDerivatives":True})
324 __Function.update(__Parameters)
325 elif ThreeFunctions is not None:
326 if (type(ThreeFunctions) is not type({})) or \
327 not ThreeFunctions.has_key("Direct") or \
328 not ThreeFunctions.has_key("Tangent") or \
329 not ThreeFunctions.has_key("Adjoint"):
330 raise ValueError("ThreeFunctions has to be a dictionnary and to have the 3 keys Direct, Tangent, Adjoint")
331 __Function = ThreeFunctions
332 __Function.update(__Parameters)
336 self.__adaoStudy.setEvolutionModel(
337 asFunction = __Function,
348 "Définition d'une entrée de calcul"
349 self.__dumper.register("setObservation", dir(), locals())
350 if Script is not None:
351 __Vector, __PersistentVector = None, None
353 __PersistentVector = _ImportFromScript(Script).getvalue( "Observation" )
355 __Vector = _ImportFromScript(Script).getvalue( "Observation" )
357 __Vector, __PersistentVector = Vector, VectorSerie
359 self.__adaoStudy.setObservation(
361 asPersistentVector = __PersistentVector,
365 def setObservationError(
368 ScalarSparseMatrix = None,
369 DiagonalSparseMatrix = None,
372 "Définition d'une entrée de calcul"
373 self.__dumper.register("setObservationError", dir(), locals())
374 if Script is not None:
375 __Covariance, __Scalar, __Vector = None, None, None
376 if ScalarSparseMatrix:
377 __Scalar = _ImportFromScript(Script).getvalue( "ObservationError" )
378 elif DiagonalSparseMatrix:
379 __Vector = _ImportFromScript(Script).getvalue( "ObservationError" )
381 __Covariance = _ImportFromScript(Script).getvalue( "ObservationError" )
383 __Covariance, __Scalar, __Vector = Matrix, ScalarSparseMatrix, DiagonalSparseMatrix
385 self.__adaoStudy.setObservationError(
386 asCovariance = __Covariance,
387 asEyeByScalar = __Scalar,
388 asEyeByVector = __Vector,
392 def setObservationOperator(
396 ThreeFunctions = None,
400 "Définition d'une entrée de calcul"
401 self.__dumper.register("setObservationOperator", dir(), locals())
403 if Parameters is not None and type(Parameters) == type({}):
404 if Parameters.has_key("DifferentialIncrement"):
405 __Parameters["withIncrement"] = Parameters["DifferentialIncrement"]
406 if Parameters.has_key("CenteredFiniteDifference"):
407 __Parameters["withCenteredDF"] = Parameters["CenteredFiniteDifference"]
408 if Parameters.has_key("EnableMultiProcessing"):
409 __Parameters["withmpEnabled"] = Parameters["EnableMultiProcessing"]
410 if Parameters.has_key("NumberOfProcesses"):
411 __Parameters["withmpWorkers"] = Parameters["NumberOfProcesses"]
412 if Script is not None:
413 __Matrix, __Function = None, None
415 __Matrix = _ImportFromScript(Script).getvalue( "ObservationOperator" )
417 __Function = { "Direct":_ImportFromScript(Script).getvalue( "DirectOperator" ) }
418 __Function.update({"useApproximatedDerivatives":True})
419 __Function.update(__Parameters)
422 "Direct" :_ImportFromScript(Script).getvalue( "DirectOperator" ),
423 "Tangent":_ImportFromScript(Script).getvalue( "TangentOperator" ),
424 "Adjoint":_ImportFromScript(Script).getvalue( "AdjointOperator" ),
426 __Function.update(__Parameters)
429 if OneFunction is not None:
430 __Function = { "Direct":OneFunction }
431 __Function.update({"useApproximatedDerivatives":True})
432 __Function.update(__Parameters)
433 elif ThreeFunctions is not None:
434 if (type(ThreeFunctions) is not type({})) or \
435 not ThreeFunctions.has_key("Direct") or \
436 not ThreeFunctions.has_key("Tangent") or \
437 not ThreeFunctions.has_key("Adjoint"):
438 raise ValueError("ThreeFunctions has to be a dictionnary and to have the 3 keys Direct, Tangent, Adjoint")
439 __Function = ThreeFunctions
440 __Function.update(__Parameters)
444 self.__adaoStudy.setObservationOperator(
445 asFunction = __Function,
450 # -----------------------------------------------------------
452 def setAlgorithmParameters(
457 "Définition d'un paramétrage du calcul"
458 self.__dumper.register("setAlgorithmParameters", dir(), locals())
459 if Script is not None:
460 __Algorithm = _ImportFromScript(Script).getvalue( "Algorithm" )
461 __Parameters = _ImportFromScript(Script).getvalue( "AlgorithmParameters", "Parameters" )
463 __Algorithm = Algorithm
464 __Parameters = Parameters
465 self.__adaoStudy.setAlgorithm( choice = __Algorithm )
466 self.__adaoStudy.setAlgorithmParameters( asDico = __Parameters )
469 "Définition d'un paramétrage du calcul"
470 self.__dumper.register("setDebug",dir(),locals())
471 return self.__adaoStudy.setDebug()
473 def setNoDebug(self):
474 "Définition d'un paramétrage du calcul"
475 self.__dumper.register("setNoDebug",dir(),locals())
476 return self.__adaoStudy.unsetDebug()
485 "Définition d'un paramétrage du calcul"
486 self.__dumper.register("setObserver", dir(), locals())
488 raise ValueError("setting an observer has to be done over a variable name, not over None.")
490 __Variable = str(Variable)
492 __Info = str(Variable)
496 if String is not None:
497 __FunctionText = String
498 elif Template is not None:
499 if Template == "ValuePrinter":
500 __FunctionText = "print info,var[-1]"
501 if Template == "ValueSeriePrinter":
502 __FunctionText = "print info,var[:]"
503 if Template == "ValueSaver":
504 __FunctionText = r"import numpy,re\nv=numpy.array((var[-1]))\nglobal istep\ntry:\n istep += 1\nexcept:\n istep = 0\nf='/tmp/value_%s_%05i.txt'%(info,istep)\nf=re.sub('\s','_',f)\nprint 'Value saved in \"%s\"'%f\nnumpy.savetxt(f,v)"
505 if Template == "ValueSerieSaver":
506 __FunctionText = r"import numpy,re\nv=numpy.array((var[:])) \nglobal istep\ntry:\n istep += 1\nexcept:\n istep = 0\nf='/tmp/value_%s_%05i.txt'%(info,istep)\nf=re.sub('\s','_',f)\nprint 'Value saved in \"%s\"'%f\nnumpy.savetxt(f,v)"
507 if Template == "ValuePrinterAndSaver":
508 __FunctionText = r"import numpy,re\nv=numpy.array((var[-1]))\nprint info,v\nglobal istep\ntry:\n istep += 1\nexcept:\n istep = 0\nf='/tmp/value_%s_%05i.txt'%(info,istep)\nf=re.sub('\s','_',f)\nprint 'Value saved in \"%s\"'%f\nnumpy.savetxt(f,v)"
509 if Template == "ValueSeriePrinterAndSaver":
510 __FunctionText = r"import numpy,re\nv=numpy.array((var[:])) \nprint info,v\nglobal istep\ntry:\n istep += 1\nexcept:\n istep = 0\nf='/tmp/value_%s_%05i.txt'%(info,istep)\nf=re.sub('\s','_',f)\nprint 'Value saved in \"%s\"'%f\nnumpy.savetxt(f,v)"
511 if Template == "ValueGnuPlotter":
512 __FunctionText = r"import Gnuplot\nglobal ifig,gp\ntry:\n ifig += 1\n gp('set style data lines')\nexcept:\n ifig = 0\n gp = Gnuplot.Gnuplot(persist=1)\n gp('set style data lines')\ngp('set title \"%s (Figure %i)\"'%(info,ifig))\ngp.plot( Gnuplot.Data( var[-1], with_='lines lw 2' ) )"
513 if Template == "ValueSerieGnuPlotter":
514 __FunctionText = r"import Gnuplot\nglobal ifig,gp\ntry:\n ifig += 1\n gp('set style data lines')\nexcept:\n ifig = 0\n gp = Gnuplot.Gnuplot(persist=1)\n gp('set style data lines')\ngp('set title \"%s (Figure %i)\"'%(info,ifig))\ngp.plot( Gnuplot.Data( var[:], with_='lines lw 2' ) )"
515 elif Script is not None:
516 __FunctionText = _ImportFromScript(Script).getstring()
519 __Function = _ObserverF(__FunctionText)
521 self.__adaoStudy.setDataObserver(
522 VariableName = __Variable,
523 HookFunction = __Function.getfunc(),
524 HookParameters = __Info,
527 # -----------------------------------------------------------
529 def executePythonScheme(self):
530 "Lancement du calcul"
531 self.__dumper.register("executePythonScheme", dir(), locals())
533 self.__adaoStudy.analyze()
534 except Exception as e:
535 if type(e) == type(SyntaxError()): msg = "at %s: %s"%(e.offset, e.text)
537 raise ValueError("during execution, the following error occurs:\n\n%s %s\n\nSee also the potential messages, which can show the origin of the above error, in the launching terminal."%(str(e),msg))
539 execute = executePythonScheme
541 def executeYACSScheme(self, File=None):
542 "Lancement du calcul"
543 self.__dumper.register("executeYACSScheme", dir(), locals())
544 raise NotImplementedError()
546 # -----------------------------------------------------------
548 def get(self, Concept=None):
549 "Récupération d'une sortie du calcul"
550 self.__dumper.register("get",dir(),locals(),Concept)
551 return self.__adaoStudy.get(Concept)
553 def dumpNormalizedCommands(self, filename=None):
554 "Récupération de la liste des commandes du cas"
555 return self.__dumper.dump(filename)
558 return ['set', 'get', 'execute', '__doc__', '__init__', '__module__']
560 class _DumpLogger(object):
562 Conservation des commandes de création d'un cas
564 def __init__(self, __name="", __objname="case"):
565 self.__name = str(__name)
566 self.__objname = str(__objname)
568 self.__switchoff = False
569 self.__logSerie.append("#\n# Python script for ADAO TUI\n#")
570 self.__logSerie.append("from numpy import array, matrix")
571 self.__logSerie.append("import adaoBuilder")
572 self.__logSerie.append("%s = adaoBuilder.New('%s')"%(self.__objname, self.__name))
573 def register(self, __command=None, __keys=None, __local=None, __pre=None, __switchoff=False):
574 "Enregistrement d'une commande individuelle"
575 if __command is not None and __keys is not None and __local is not None and not self.__switchoff:
577 if __pre is not None:
578 __text += "%s = "%__pre
579 __text += "%s.%s( "%(self.__objname,str(__command))
580 __keys.remove("self")
583 if __v is None: continue
584 __text += "%s=%s, "%(k,repr(__v))
586 self.__logSerie.append(__text)
588 self.__switchoff = True
590 self.__switchoff = False
591 def dump(self, __filename=None):
592 "Restitution de la liste des commandes de création d'un cas"
593 __text = "\n".join(self.__logSerie)
594 if __filename is not None:
595 fid = open(__filename,"w")
600 class _ObserverF(object):
602 Création d'une fonction d'observateur à partir de son texte
604 def __init__(self, corps=""):
606 def func(self,var,info):
607 "Fonction d'observation"
610 "Restitution du pointeur de fonction dans l'objet"
613 class _ImportFromScript(object):
615 Obtention d'une variable nommée depuis un fichier script importé
617 def __init__(self, __filename=None):
618 "Verifie l'existence et importe le script"
619 __filename = __filename.rstrip(".py")
620 if __filename is None:
621 raise ValueError("The name of the file containing the variable to be imported has to be specified.")
622 if not os.path.isfile(str(__filename)+".py"):
623 raise ValueError("The file containing the variable to be imported doesn't seem to exist. The given file name is:\n \"%s\""%__filename)
624 self.__scriptfile = __import__(__filename, globals(), locals(), [])
625 self.__scriptstring = open(__filename+".py",'r').read()
626 def getvalue(self, __varname=None, __synonym=None ):
627 "Renvoie la variable demandee"
628 if __varname is None:
629 raise ValueError("The name of the variable to be imported has to be specified.")
630 if not hasattr(self.__scriptfile, __varname):
631 if __synonym is None:
632 raise ValueError("The imported script file doesn't contain the specified variable \"%s\"."%__varname)
633 elif not hasattr(self.__scriptfile, __synonym):
634 raise ValueError("The imported script file doesn't contain the specified variable \"%s\"."%__synonym)
636 return getattr(self.__scriptfile, __synonym)
638 return getattr(self.__scriptfile, __varname)
640 return self.__scriptstring
642 # ==============================================================================
643 if __name__ == "__main__":
644 print '\n AUTODIAGNOSTIC \n'