generator/generator_openturns_study.py

   1 # -*- coding: utf-8 -*-
   2 #            CONFIGURATION MANAGEMENT OF EDF VERSION
   3 # ======================================================================
   4 # COPYRIGHT (C) 1991 - 2002  EDF R&D                  WWW.CODE-ASTER.ORG
   5 # THIS PROGRAM IS FREE SOFTWARE; YOU CAN REDISTRIBUTE IT AND/OR MODIFY
   6 # IT UNDER THE TERMS OF THE GNU GENERAL PUBLIC LICENSE AS PUBLISHED BY
   7 # THE FREE SOFTWARE FOUNDATION; EITHER VERSION 2 OF THE LICENSE, OR
   8 # (AT YOUR OPTION) ANY LATER VERSION.
   9 #
  10 # THIS PROGRAM IS DISTRIBUTED IN THE HOPE THAT IT WILL BE USEFUL, BUT
  11 # WITHOUT ANY WARRANTY; WITHOUT EVEN THE IMPLIED WARRANTY OF
  12 # MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE. SEE THE GNU
  13 # GENERAL PUBLIC LICENSE FOR MORE DETAILS.
  14 #
  15 # YOU SHOULD HAVE RECEIVED A COPY OF THE GNU GENERAL PUBLIC LICENSE
  16 # ALONG WITH THIS PROGRAM; IF NOT, WRITE TO EDF R&D CODE_ASTER,
  17 #    1 AVENUE DU GENERAL DE GAULLE, 92141 CLAMART CEDEX, FRANCE.
  18 #
  19 #
  20 # ======================================================================
  21 """
  22    Ce module contient le plugin generateur de fichier au format
  23    openturns pour EFICAS.
  24
  25 """
  26 import traceback
  27 import types,string,re
  28
  29 from generator_python import PythonGenerator
  30 from OpenturnsBase import Generateur
  31 #from OpenturnsXML import XMLGenerateur
  32 #from OpenturnsSTD import STDGenerateur
  33
  34 def entryPoint():
  35    """
  36       Retourne les informations necessaires pour le chargeur de plugins
  37
  38       Ces informations sont retournees dans un dictionnaire
  39    """
  40    return {
  41         # Le nom du plugin
  42           'name' : 'openturns_study',
  43         # La factory pour creer une instance du plugin
  44           'factory' : OpenturnsGenerator,
  45           }
  46
  47
  48 class OpenturnsGenerator(PythonGenerator):
  49    """
  50       Ce generateur parcourt un objet de type JDC et produit
  51       un texte au format eficas et
  52       un texte au format xml
  53
  54    """
  55    # Les extensions de fichier permis?
  56    extensions=('.comm',)
  57
  58    def initDico(self):
  59       self.dictMCVal={}
  60       self.listeVariables=[]
  61       self.listeFichiers=[]
  62       self.dictMCLois={}
  63       self.dictTempo={}
  64       self.TraiteMCSIMP=1
  65       self.texteSTD="""#!/usr/bin/env python
  66       import sys
  67       print "Invalid file. Check build process."
  68       sys.exit(1)
  69       """
  70
  71    def gener(self,obj,format='brut'):
  72       print "IDM: gener dans generator_openturns_study.py"
  73       self.initDico()
  74       self.text=PythonGenerator.gener(self,obj,format)
  75       self.genereSTD()
  76       return self.text
  77
  78    def generMCSIMP(self,obj) :
  79       """
  80       Convertit un objet MCSIMP en texte python
  81       Remplit le dictionnaire des MCSIMP si nous ne sommes ni dans une loi, ni dans une variable
  82       """
  83       s=PythonGenerator.generMCSIMP(self,obj)
  84       if self.TraiteMCSIMP == 1 :
  85          self.dictMCVal[obj.nom]=obj.val
  86       else :
  87          self.dictTempo[obj.nom]=obj.valeur
  88       return s
  89
  90 #   def generMCFACT(self,obj):
  91 #      # Il n est pas possible d utiliser obj.valeur qui n est pas
  92 #      # a jour pour les nouvelles variables ou les modifications
  93 #      if obj.nom in ( "Variables", "Files", ) :
  94 #         self.TraiteMCSIMP=0
  95 #         self.dictTempo={}
  96 #      s=PythonGenerator.generMCFACT(self,obj)
  97 #      if obj.nom in ( "Variables", ) :
  98 #         self.listeVariables.append(self.dictTempo)
  99 #         self.dictTempo={}
 100 #      else :
 101 #         self.listeFichiers.append(self.dictTempo)
 102 #      self.TraiteMCSIMP=1
 103 #      return s
 104
 105    def generETAPE(self,obj):
 106       print "IDM: generETAPE dans generator_openturns_study.py"
 107       print "IDM: obj.nom=", obj.nom
 108       if obj.nom in ( "DISTRIBUTION", ) :
 109          self.TraiteMCSIMP=0
 110          self.dictTempo={}
 111       s=PythonGenerator.generETAPE(self,obj)
 112       if obj.nom in ( "DISTRIBUTION", ) :
 113          self.dictMCLois[obj.sd]=self.dictTempo
 114          self.dictTempo={}
 115       self.TraiteMCSIMP=1
 116       return s
 117
 118    def generPROC_ETAPE(self,obj):
 119       print "IDM: generPROC_ETAPE dans generator_openturns_study.py"
 120       print "IDM: obj.nom=", obj.nom
 121       if obj.nom in ( "VARIABLE",  ) :
 122          self.TraiteMCSIMP=0
 123          self.dictTempo={}
 124       s=PythonGenerator.generPROC_ETAPE(self,obj)
 125       if obj.nom in ( "VARIABLE", ) :
 126          self.listeVariables.append(self.dictTempo)
 127          self.dictTempo={}
 128       self.TraiteMCSIMP=1
 129       return s
 130
 131    def genereSTD(self):
 132       print "IDM: genereSTD dans generator_openturns_study.py"
 133       print "IDM: self.listeVariables=", self.listeVariables
 134       MonBaseGenerateur=Generateur(self.appli,self.dictMCVal, self.listeVariables, self.dictMCLois)
 135       MonGenerateur=MonBaseGenerateur.getSTDGenerateur()
 136       #try :
 137       if 1== 1 :
 138          self.texteSTD=MonGenerateur.CreeSTD()
 139       #except :
 140       else :
 141          self.texteSTD="Il y a un pb a la Creation du STD"
 142
 143    def writeOpenturnsSTD(self, filename):
 144       f = open( str(filename), 'wb')
 145       f.write( self.texteSTD )
 146       f.close()
 147
 148