generator/generator_vers3DSalome.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     python pour EFICAS.
  24     PN
  25
  26 """
  27 import traceback
  28 import types,string,re
  29 import math
  30
  31 from Noyau import N_CR
  32 from Noyau.N_utils import repr_float
  33 import Accas
  34 import Extensions
  35 from Extensions.parametre import ITEM_PARAMETRE
  36 from Formatage import Formatage
  37 from generator_python import PythonGenerator
  38 #from Editeur.widgets import showerror
  39
  40 def entryPoint():
  41    """
  42        Retourne les informations nécessaires pour le chargeur de plugins
  43
  44        Ces informations sont retournées dans un dictionnaire
  45    """
  46    return {
  47         # Le nom du plugin
  48           'name' : 'vers3DSalome',
  49         # La factory pour créer une instance du plugin
  50           'factory' : vers3DSalomeGenerator,
  51           }
  52
  53
  54 class vers3DSalomeGenerator(PythonGenerator):
  55    """
  56        Ce generateur parcourt un objet AFFE-CARA_ELEM
  57        et produit un fichier au format texte contenant
  58        les instructions idl pour PAL
  59    """
  60
  61    def __init__(self,cr=None):
  62       self.list_commandes=[];
  63       self.jdc=None
  64       self.node=None
  65       self.clefs=None
  66       self.liste_motetat = ("AFFE_CARA_ELEM", "ORIG_AXE", "AXE" ,
  67                             "BARRE", "CABLE", "CARA", "COQUE", "EPAIS",
  68                             "EXCENTREMENT", "GROUP_MA", "ORIENTATION",
  69                             "POUTRE", "SECTION", "VALE", "VARI_SECT",
  70                             "GRILLE", "ANGL_REP", "VECTEUR",
  71                              "b_constant", "b_homothetique",
  72                             "b_rectangle", "b_affine", "b_cercle" )
  73       self.dict_deb_com={"POUTRE":"VisuPoutre", "CABLE" : "VisuCable",
  74                          "COQUE" : "VisuCoque", "GRILLE" : "VisuGrille",
  75                          "ORIENTATION" : "Orientation", "BARRE" : "VisuBarre"}
  76
  77       self.dict_suite_com={"RECTANGLE":"Rectangle","GENERALE":"Generale",
  78                            "CERCLE":"Cercle"}
  79
  80       self.dict_traduit={"VARI_SECT":"extrusion","EXCENTREMENT":"Excentre","EPAIS":"Epais","VECTEUR":"Vecteur"}
  81
  82       self.init_ligne()
  83
  84    def init_jdc(self,jdc) :
  85       self.jdc=jdc
  86
  87    def init_ligne (self) :
  88       self.boolGpMa = 0
  89       self.commande = ""
  90       self.dict_attributs = {}
  91
  92    def gener(self,node,config=None):
  93       """
  94       """
  95       self.node=node
  96       self.list_commandes=[];
  97       self.generator(self.node.object)
  98       #print self.list_commandes
  99       return self.list_commandes
 100
 101    def generator(self,obj):
 102       if (obj.nom in self.liste_motetat) and (self.calcule_ouinon(obj)):
 103          PythonGenerator.generator(self,obj)
 104       """
 105         f1=PythonGenerator.generator(self,obj)
 106       else :
 107         return ""
 108       """
 109
 110    def calcule_ouinon(self,obj):
 111       ouinon=1
 112       for l in obj.get_genealogie() :
 113           if not l in self.liste_motetat :
 114              ouinon=0
 115              break
 116       return ouinon
 117
 118
 119    def generETAPE(self,obj):
 120       """
 121       """
 122       if obj.isvalid() == 0 :
 123          #showerror("Element non valide","Salome ne sait pas traiter les élements non valides")
 124          return
 125       for v in obj.mc_liste:
 126          liste=self.generator(v)
 127
 128
 129    def generMCSIMP(self,obj) :
 130       """
 131       """
 132       if obj.nom in dir(self) :
 133          suite = self.__class__.__dict__[obj.nom](self,obj)
 134       else :
 135          clef=self.dict_traduit[obj.nom]
 136          # Traitement des parametres
 137          try :
 138              self.dict_attributs[clef]=obj.val.eval()
 139          except :
 140              self.dict_attributs[clef]=obj.val
 141
 142
 143    def generMCFACT(self,obj):
 144       """
 145           Convertit un objet MCFACT en une liste de chaines de caractères à la
 146           syntaxe python
 147       """
 148       self.init_ligne()
 149       self.commande=self.dict_deb_com[obj.nom]
 150       for v in obj.mc_liste:
 151          self.generator(v)
 152       if self.boolGpMa == 1:
 153          self.list_commandes.append((self.commande,self.dict_attributs))
 154       else :
 155          #showerror("Elements ne portant pas sur un Groupe de Maille","Salome ne sait pas montrer ce type d' element")
 156          print ("Elements ne portant pas sur un Groupe de Maille","Salome ne sait pas montrer ce type d' element")
 157          pass
 158
 159    def generMCList(self,obj):
 160       """
 161       """
 162       for mcfact in obj.data:
 163           self.generator(mcfact)
 164
 165    def generMCBLOC(self,obj):
 166       """
 167       """
 168       for v in obj.mc_liste:
 169          self.generator(v)
 170
 171    def GROUP_MA(self,obj):
 172       self.boolGpMa = 1
 173       self.dict_attributs["Group_Maille"]=obj.val
 174
 175    def SECTION(self,obj):
 176       assert (self.commande != "" )
 177       if self.commande == "VisuCable" :
 178          self.dict_attributs["R"]= math.sqrt(obj.val/math.pi).eval()
 179       elif (self.commande !="VisuGrille")  :
 180          self.commande=self.commande+self.dict_suite_com[obj.valeur]
 181
 182    def CARA(self,obj) :
 183        self.clefs=obj.val
 184        if type(self.clefs) == types.StringType :
 185           self.clefs=(obj.val,)
 186
 187    def VALE(self,obj) :
 188        atraiter=obj.val
 189        if len(self.clefs) > 1 :
 190            assert (len(atraiter) == len(self.clefs))
 191        else :
 192            atraiter=(atraiter,)
 193        for k in range(len(atraiter)) :
 194            clef=self.clefs[k]
 195            val =atraiter[k]
 196            if isinstance (val, Extensions.parametre.PARAMETRE):
 197               val=val.valeur
 198               print val.__class__
 199               context={}
 200               if type(val) == type("aaa") :
 201                  for p in self.jdc.params:
 202                      context[p.nom]=eval(p.val,self.jdc.const_context, context)
 203                      print context[p.nom]
 204                  res=eval(val,self.jdc.const_context, context)
 205                  val=res
 206            self.dict_attributs[clef]=val
 207
 208    def ANGL_REP(self,obj) :
 209       assert (len(obj.val) == 2)
 210       alpha,beta=obj.val
 211       self.dict_attributs["angleAlpha"]=alpha
 212       self.dict_attributs["angleBeta"]=beta
 213
 214    def ORIG_AXE(self,obj) :
 215       assert (len(obj.val) == 3)
 216       alpha,beta,gamma=obj.val
 217       self.dict_attributs["origAxeX"]=alpha
 218       self.dict_attributs["origAxeY"]=beta
 219       self.dict_attributs["origAxeZ"]=gamma
 220
 221    def AXE(self,obj) :
 222       assert (len(obj.val) == 3)
 223       alpha,beta,gamma=obj.val
 224       self.dict_attributs["axeX"]=alpha
 225       self.dict_attributs["axeY"]=beta
 226       self.dict_attributs["axeZ"]=gamma