generator/OpenturnsBase.py

   1 #@ AJOUT OpenturnsSolver Macro
   2 # -*- coding: iso-8859-1 -*-
   3 # RESPONSABLE
   4
   5 """
   6 Ce module contient la partie commune
   7 aux generateurs XML et Etude d Openturns
   8 """
   9
  10 __revision__ = "V1.0"
  11
  12 import os
  13 import sys
  14
  15 path=os.getcwd()
  16 pathDef=path+"DefautOpenturns"
  17
  18 sys.path.append(pathDef)
  19
  20
  21 #=============================================
  22 # La classe generale
  23 #=============================================
  24
  25 class Generateur :
  26
  27   '''
  28   Classe generale du generateur
  29   DictMCVal : dictionnaire des mots-cles
  30   ListeVariables : chaque variable est decrite par un dictionnaire ; cette liste les regroupe
  31   DictLois : dictionnaires des lois
  32   '''
  33   def __init__ (self, appli, DictMCVal = {}, ListeVariables = [], DictLois = {}, DictVariables = {} ) :
  34   #---------------------------------------------------------#
  35     self.ListeVariables = ListeVariables
  36     self.ListeVariablesIn = []
  37     self.DictLois = DictLois
  38     self.DictVariables = DictVariables
  39     self.DictMCVal = DictMCVal
  40     self.DictTypeVar = {}
  41     self.nbVarIn = 0
  42     self.creeInfoVar()
  43     self.appli = appli
  44     #
  45     # On charge eventuellement le Solver par defaut
  46     # et les valeurs par defaut du Solver (dans l init)
  47     #
  48     try :
  49     #if 1 :
  50         Solver = self.DictMCVal["PhysicalSolver"]
  51         import_name = "Defaut"+Solver
  52         self.module = __import__( import_name, globals(), locals() )
  53         monDefaut = self.module.Defaut( self )
  54     #else :
  55     except:
  56         self.module = None
  57
  58
  59   def getSTDGenerateur(self) :
  60   #--------------------------#
  61     try :
  62         gener = self.module.__dict__["MonSTDGenerateur"]
  63         monSTDGenerateur=gener( self.DictMCVal, self.ListeVariablesIn, self.DictLois )
  64     except :
  65         from OpenturnsSTD import STDGenerateur
  66         monSTDGenerateur = STDGenerateur( self.appli, self.DictMCVal, self.ListeVariablesIn, self.DictLois )
  67     return monSTDGenerateur
  68
  69   def getXMLGenerateur(self) :
  70   #--------------------------#
  71     try :
  72         gener = self.module.__dict__["MonXMLGenerateur"]
  73         monXMLGenerateur=gener( self.DictMCVal, self.ListeVariables, self.DictLois )
  74     except :
  75         from OpenturnsXML import XMLGenerateur
  76         monXMLGenerateur = XMLGenerateur( self.appli, self.DictMCVal, self.DictVariables )
  77     return monXMLGenerateur
  78
  79   def creeInfoVar (self) :
  80   #----------------------#
  81     """
  82     On repere les variables in/out et on les numerote.
  83     """
  84     num = 0
  85     liste = []
  86     for DictVariable in self.ListeVariables :
  87       if not DictVariable.has_key("Type") : DictVariable["Type"] = "in"
  88       self.DictTypeVar[num] = DictVariable["Type"]
  89       if DictVariable["Type"] == "in" :
  90          self.nbVarIn = self.nbVarIn + 1
  91          self.ListeVariablesIn.append( DictVariable )
  92       liste.append( DictVariable )
  93       num = num + 1
  94     self.ListeVariables = liste
  95
  96
  97   def ajouteDictMCVal(self, dicoPlus) :
  98   #-----------------------------------#
  99   # Appele par le classe Defaut du python specifique au code (exple DefautASTER.py)
 100   # enrichit self.DictMCVal avec les valeurs donnees dans dicoPlus
 101   # si elles ne sont pas deja dans le dictionnaire
 102
 103     for clef in dicoPlus.keys():
 104         if not self.DictMCVal.has_key(clef) :
 105            self.DictMCVal[clef] = dicoPlus[clef]
 106
 107   def ajouteInfoVariables (self, dicoVariablesIn, dicoVariablesOut) :
 108   #-----------------------------------------------------------------#
 109   # Appele par le classe Defaut du python specifique au code (exple DefautASTER.py)
 110   # met a jour les dictionnaires qui decrivent les variables (regexp par exemple)
 111     liste=[]
 112     num = 0
 113     for dictVariable in self.ListeVariables:
 114          if self.DictTypeVar[num] == "in" :
 115             dico = dicoVariablesIn
 116          else :
 117             dico = dicoVariablesOut
 118          for nouvelleVariable in dico.keys() :
 119             if not dictVariable.has_key(nouvelleVariable):
 120                dictVariable[nouvelleVariable] = dico[nouvelleVariable]
 121          liste.append( dictVariable )
 122          num = num + 1