generator/generator_openturns.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',
  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
  66    def gener(self,obj,format='brut',config=None):
  67        #print "IDM: gener dans generator_openturns.py"
  68        self.initDico()
  69        self.text=PythonGenerator.gener(self,obj,format)
  70        self.genereXML()
  71        self.genereSTD()
  72        return self.text
  73
  74    def generMCSIMP(self,obj) :
  75        """
  76        Convertit un objet MCSIMP en texte python
  77        Remplit le dictionnaire des MCSIMP si nous ne sommes pas ni dans une loi, ni dans une variable
  78        """
  79        s=PythonGenerator.generMCSIMP(self,obj)
  80        if self.TraiteMCSIMP == 1 :
  81           self.dictMCVal[obj.nom]=obj.valeur
  82        else :
  83           self.dictTempo[obj.nom]=obj.valeur
  84        return s
  85
  86    def generMCFACT(self,obj):
  87        # Il n est pas possible d utiliser obj.valeur qui n est pas
  88        # a jour pour les nouvelles variables ou les modifications
  89        if obj.nom == "Variables" or "Files":
  90           self.TraiteMCSIMP=0
  91           self.dictTempo={}
  92        s=PythonGenerator.generMCFACT(self,obj)
  93        if obj.nom == "Variables" :
  94           self.listeVariables.append(self.dictTempo)
  95           self.dictTempo={}
  96        else :
  97           self.listeFichiers.append(self.dictTempo)
  98        self.TraiteMCSIMP=1
  99        return s
 100
 101    def generETAPE(self,obj):
 102        if obj.nom == "DISTRIBUTION" :
 103           self.TraiteMCSIMP=0
 104           self.dictTempo={}
 105        s=PythonGenerator.generETAPE(self,obj)
 106        if obj.nom == "DISTRIBUTION" :
 107           self.dictMCLois[obj.sd]=self.dictTempo
 108           self.dictTempo={}
 109        self.TraiteMCSIMP=1
 110        return s
 111
 112    def genereXML(self):
 113        #print "IDM: genereXML dans generator_openturns.py"
 114        if self.listeFichiers != [] :
 115           self.dictMCVal["exchange_file"]=self.listeFichiers
 116        MonBaseGenerateur=Generateur(self.dictMCVal, self.listeVariables, self.dictMCLois)
 117        MonGenerateur=MonBaseGenerateur.getXMLGenerateur()
 118        #try :
 119        if 1== 1 :
 120           self.texteXML=MonGenerateur.CreeXML()
 121        #except :
 122        else :
 123           self.texteXML="Il y a un pb a la Creation du XML"
 124
 125    def genereSTD(self):
 126        MonBaseGenerateur=Generateur(self.dictMCVal, self.listeVariables, self.dictMCLois)
 127        MonGenerateur=MonBaseGenerateur.getSTDGenerateur()
 128        #try :
 129        if 1== 1 :
 130           self.texteSTD=MonGenerateur.CreeSTD()
 131        #except :
 132        else :
 133           self.texteSTD="Il y a un pb a la Creation du STD"
 134
 135    def getOpenturnsXML(self):
 136        return self.texteXML
 137
 138    def getOpenturnsSTD(self):
 139        return self.texteSTD