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',config=None):
  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 not( type(obj.valeur) in (list, tuple)) and (obj.get_min_max()[1] != 1):
  87           valeur=(obj.valeur,)
  88        else :
  89           valeur=obj.valeur
  90        if self.traiteMCSIMP == 1 :
  91           self.dictMCVal[obj.nom]=valeur
  92        else :
  93           self.dictTempo[obj.nom]=valeur
  94        return s
  95
  96    def generETAPE(self,obj):
  97        #print "generETAPE" , obj.nom
  98        if obj.nom == "VARIABLE" :
  99           self.traiteMCSIMP=0
 100           self.dictTempo={}
 101        s=PythonGenerator.generETAPE(self,obj)
 102        if obj.nom == "VARIABLE" :
 103           self.dictTempo["numOrdre"]=self.numOrdre
 104           self.numOrdre = self.numOrdre +1
 105           if obj.sd == None :
 106              self.dictVariables["SansNom"]=self.dictTempo
 107           else :
 108              self.dictVariables[obj.sd.nom]=self.dictTempo
 109           self.dictTempo={}
 110        self.traiteMCSIMP=1
 111        return s
 112
 113    def generMCFACT(self,obj):
 114        # Il n est pas possible d utiliser obj.valeur qui n est pas
 115        # a jour pour les nouvelles variables ou les modifications
 116        if obj.nom in ( "Files", ) :
 117           self.traiteMCSIMP=0
 118           self.dictTempo={}
 119        s=PythonGenerator.generMCFACT(self,obj)
 120        self.listeFichiers.append(self.dictTempo)
 121        self.traiteMCSIMP=1
 122        return s
 123
 124    def genereXML(self):
 125        print "IDM: genereXML dans generator_openturns_wrapper.py"
 126        #print "appli.CONFIGURATION=",self.appli.CONFIGURATION.__dict__
 127        if self.listeFichiers != [] :
 128           self.dictMCVal["Files"]=self.listeFichiers
 129        print "dictMCVal", self.dictMCVal, "dictVariables", self.dictVariables
 130        MonBaseGenerateur=Generateur(self.appli,self.dictMCVal, [], {} ,self.dictVariables)
 131        MonGenerateur=MonBaseGenerateur.getXMLGenerateur()
 132        try :
 133        #if 1== 1 :
 134           self.wrapperXML=MonGenerateur.CreeXML()
 135        except :
 136        #else :
 137           self.wrapperXML=None
 138
 139    def writeDefault(self, filename):
 140       fileXML = filename[:filename.rfind(".")] + '.xml'
 141       self.wrapperXML.writeFile( str(fileXML) )