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',config=None):
  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.valeur
  86       else :
  87          self.dictTempo[obj.nom]=obj.valeur
  88       return s
  89
  90
  91    def generETAPE(self,obj):
  92       print "IDM: generETAPE dans generator_openturns_study.py"
  93       print "IDM: obj.nom=", obj.nom
  94       if obj.nom in ( "DISTRIBUTION", ) :
  95          self.TraiteMCSIMP=0
  96          self.dictTempo={}
  97       s=PythonGenerator.generETAPE(self,obj)
  98       if obj.nom in ( "DISTRIBUTION", ) :
  99          self.dictMCLois[obj.sd]=self.dictTempo
 100          self.dictTempo={}
 101       self.TraiteMCSIMP=1
 102       return s
 103
 104    def generPROC_ETAPE(self,obj):
 105       print "IDM: generPROC_ETAPE dans generator_openturns_study.py"
 106       print "IDM: obj.nom=", obj.nom
 107       if obj.nom in ( "VARIABLE",  ) :
 108          self.TraiteMCSIMP=0
 109          self.dictTempo={}
 110       s=PythonGenerator.generPROC_ETAPE(self,obj)
 111       if obj.nom in ( "VARIABLE", ) :
 112          self.listeVariables.append(self.dictTempo)
 113          self.dictTempo={}
 114       self.TraiteMCSIMP=1
 115       return s
 116
 117    def genereSTD(self):
 118       print "IDM: genereSTD dans generator_openturns_study.py"
 119       print "IDM: self.listeVariables=", self.listeVariables
 120       MonGenerateur=self.getGenerateur()
 121       #try :
 122       if 1== 1 :
 123          self.texteSTD=MonGenerateur.CreeSTD()
 124       #except :
 125       else :
 126          self.texteSTD="Il y a un pb a la Creation du STD"
 127
 128    def writeDefault(self, fn):
 129       fileSTD = fn[:fn.rfind(".")] + '.py'
 130       f = open( str(fileSTD), 'wb')
 131       f.write( self.texteSTD )
 132       f.close()
 133
 134    def getGenerateur (self):
 135       print "IDM: getGenerateur dans generator_openturns_study.py"
 136       print "IDM: self.dictMCVal=", self.dictMCVal
 137       print "IDM: self.listeVariables=", self.listeVariables
 138       print "IDM: self.dictMCLois=", self.dictMCLois
 139       MonBaseGenerateur=Generateur(self.appli,self.dictMCVal, self.listeVariables, self.dictMCLois)
 140       MonGenerateur=MonBaseGenerateur.getSTDGenerateur()
 141       return MonGenerateur