generator/generator_dicoImbrique.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2007-2013   EDF R&D
   3 #
   4 # This library is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU Lesser General Public
   6 # License as published by the Free Software Foundation; either
   7 # version 2.1 of the License.
   8 #
   9 # This library is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # Lesser General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public
  15 # License along with this library; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20 """Ce module contient le plugin generateur de fichier au format  Code_Carmel3D pour EFICAS.
  21 """
  22
  23 from __future__ import absolute_import
  24 try :
  25    from builtins import str
  26 except : pass
  27
  28 import traceback
  29 import types,re,os
  30 from Extensions.i18n import tr
  31 from .generator_python import PythonGenerator
  32
  33 def entryPoint():
  34    """
  35       Retourne les informations necessaires pour le chargeur de plugins
  36       Ces informations sont retournees dans un dictionnaire
  37    """
  38    return {
  39         # Le nom du plugin
  40           'name' : 'dicoImbrique',
  41         # La factory pour creer une instance du plugin
  42           'factory' : DicoImbriqueGenerator,
  43           }
  44
  45
  46 class DicoImbriqueGenerator(PythonGenerator):
  47    """
  48       Ce generateur parcourt un objet de type JDC et produit
  49       un texte au format eficas et
  50       un texte au format dictionnaire
  51
  52    """
  53    # Les extensions de fichier permis?
  54    extensions=('.comm',)
  55
  56 #----------------------------------------------------------------------------------------
  57    def gener(self,obj,format='brut',config=None, appli=None):
  58
  59       self.initDico()
  60
  61       # Cette instruction genere le contenu du fichier de commandes (persistance)
  62       self.text=PythonGenerator.gener(self,obj,format)
  63       return self.text
  64
  65
  66 #----------------------------------------------------------------------------------------
  67 # initialisations
  68 #----------------------------------------------------------------------------------------
  69
  70    def initDico(self) :
  71
  72       self.Dico={}
  73       self.DicoDejaLa={}
  74       self.Entete = ''
  75
  76
  77 #----------------------------------------------------------------------------------------
  78 # ecriture
  79 #----------------------------------------------------------------------------------------
  80
  81    def writeDefault(self,fn) :
  82        fileDico = fn[:fn.rfind(".")] + '.py'
  83        f = open( str(fileDico), 'wb')
  84
  85        f.write( self.Entete + "Dico =" + str(self.Dico) )
  86        f.close()
  87
  88 #----------------------------------------------------------------------------------------
  89 #  analyse de chaque noeud de l'arbre
  90 #----------------------------------------------------------------------------------------
  91
  92    def generMCSIMP(self,obj) :
  93         """recuperation de l objet MCSIMP"""
  94
  95         s=PythonGenerator.generMCSIMP(self,obj)
  96         if obj.isInformation() : return s
  97         if not obj.isvalid() :  return s
  98
  99         liste=obj.get_genealogie_precise()
 100
 101         if obj.etape.nom=='MODIFICATION_CATALOGUE' : return s
 102         nom = obj.etape.nom
 103
 104         if hasattr(obj.etape,'sdnom') and obj.etape.sdnom != None and obj.etape.sdnom != "" :
 105            nom = nom+ obj.etape.sdnom
 106
 107         if not(nom in self.Dico) : dicoCourant={}
 108         else : dicoCourant=self.Dico [nom]
 109
 110         nomFeuille=liste[-1]
 111         if nomFeuille in dicoCourant or nomFeuille in self.DicoDejaLa:
 112            if nomFeuille in self.DicoDejaLa:
 113               nomTravail= nomFeuille +'_'+str(self.DicoDejaLa[nomFeuille])
 114               self.DicoDejaLa[nomFeuille]=self.DicoDejaLa[nomFeuille]+1
 115               nomFeuille=nomTravail
 116            else :
 117               self.DicoDejaLa[nomFeuille]=3
 118               nom1=nomFeuille +'_1'
 119               dicoCourant[nom1]= dicoCourant[nomFeuille]
 120               del dicoCourant[nomFeuille]
 121               nomFeuille=nomFeuille +'_2'
 122
 123
 124         if hasattr(obj.valeur,'nom'): dicoCourant[nomFeuille]=obj.valeur.nom
 125         else :
 126            if type(obj.valeur)  in (list,tuple):
 127               try :
 128 #PNPNPN a remplacer par plus propre
 129                  if obj.definition.validators.typeDesTuples[0] !='R' :
 130                     val=[]
 131                     elt=[]
 132                     for tupleElt in obj.valeur :
 133                         elt=(str(tupleElt[0]),tupleElt[1])
 134                         val.append(elt)
 135                     dicoCourant[nomFeuille]=val
 136                  else :
 137                     dicoCourant[nomFeuille]=obj.valeur
 138               except :
 139                  dicoCourant[nomFeuille]=obj.valeurFormatee
 140            #else :dicoCourant[nomFeuille]=obj.valeurFormatee
 141            else :
 142               dicoCourant[nomFeuille]=obj.valeurFormatee
 143               #print nomFeuille, obj.valeurFormatee
 144         self.Dico[nom]=dicoCourant
 145
 146         return s
 147
 148