generator/generator_openturns_wrapper.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_wrapper',
  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.dictVariables={}
  61        self.listeFichiers=[]
  62        self.dictTempo={}
  63        self.traiteMCSIMP=1
  64        self.numOrdre=0
  65        self.texteSTD="""#!/usr/bin/env python
  66        import sys
  67        print "Invalid file. Check build process."
  68        sys.exit(1)
  69        """
  70        self.wrapperXML=None
  71
  72    def gener(self,obj,format='brut'):
  73        #print "IDM: gener dans generator_openturns_wrapper.py"
  74        self.initDico()
  75        self.text=PythonGenerator.gener(self,obj,format)
  76        self.genereXML()
  77        #self.genereSTD()
  78        return self.text
  79
  80    def generMCSIMP(self,obj) :
  81        """
  82        Convertit un objet MCSIMP en texte python
  83        Remplit le dictionnaire des MCSIMP si nous ne sommes ni dans une loi, ni dans une variable
  84        """
  85        s=PythonGenerator.generMCSIMP(self,obj)
  86        if self.traiteMCSIMP == 1 :
  87           self.dictMCVal[obj.nom]=obj.val
  88        else :
  89           self.dictTempo[obj.nom]=obj.valeur
  90        return s
  91
  92    def generETAPE(self,obj):
  93        #print "generETAPE" , obj.nom
  94        if obj.nom == "VARIABLE" :
  95           self.traiteMCSIMP=0
  96           self.dictTempo={}
  97        s=PythonGenerator.generETAPE(self,obj)
  98        if obj.nom == "VARIABLE" :
  99           self.dictTempo["numOrdre"]=self.numOrdre
 100           self.numOrdre = self.numOrdre +1
 101           if obj.sd == None :
 102              self.dictVariables["SansNom"]=self.dictTempo
 103           else :
 104              self.dictVariables[obj.sd.nom]=self.dictTempo
 105           self.dictTempo={}
 106        self.traiteMCSIMP=1
 107        return s
 108
 109    def generMCFACT(self,obj):
 110        # Il n est pas possible d utiliser obj.valeur qui n est pas
 111        # a jour pour les nouvelles variables ou les modifications
 112        #print "generMCFACT" , obj.nom
 113        if obj.nom in ( "Files", ) :
 114           self.traiteMCSIMP=0
 115           self.dictTempo={}
 116        s=PythonGenerator.generMCFACT(self,obj)
 117        self.listeFichiers.append(self.dictTempo)
 118        self.traiteMCSIMP=1
 119        return s
 120
 121    def genereXML(self):
 122        #print "IDM: genereXML dans generator_openturns_wrapper.py"
 123        #print "appli.CONFIGURATION=",self.appli.CONFIGURATION.__dict__
 124        if self.listeFichiers != [] :
 125           self.dictMCVal["Files"]=self.listeFichiers
 126        print "dictMCVal", self.dictMCVal, "dictVariables", self.dictVariables
 127        MonBaseGenerateur=Generateur(self.appli,self.dictMCVal, [], {} ,self.dictVariables)
 128        MonGenerateur=MonBaseGenerateur.getXMLGenerateur()
 129        #try :
 130        if 1== 1 :
 131           self.wrapperXML=MonGenerateur.CreeXML()
 132        #except :
 133        else :
 134           self.wrapperXML=None
 135
 136    def writeOpenturnsXML(self, filename):
 137       self.wrapperXML.writeFile( str(filename) )
 138