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       #print (self.text)
  64       return self.text
  65
  66
  67
  68 #----------------------------------------------------------------------------------------
  69 # initialisations
  70 #----------------------------------------------------------------------------------------
  71
  72    def initDico(self) :
  73
  74       self.Dico={}
  75       self.DicoDejaLa={}
  76       self.Entete = ''
  77
  78
  79 #----------------------------------------------------------------------------------------
  80 # ecriture
  81 #----------------------------------------------------------------------------------------
  82
  83    def writeDefault(self,fn) :
  84        fileDico = fn[:fn.rfind(".")] + '.py'
  85        f = open( str(fileDico), 'wb')
  86
  87        f.write( self.Entete + "Dico =" + str(self.Dico) )
  88        f.close()
  89
  90 #----------------------------------------------------------------------------------------
  91 #  analyse de chaque noeud de l'arbre
  92 #----------------------------------------------------------------------------------------
  93
  94    def generMCSIMP(self,obj) :
  95         """recuperation de l objet MCSIMP"""
  96
  97         s=PythonGenerator.generMCSIMP(self,obj)
  98         if obj.isInformation() : return s
  99         if not obj.isvalid() :  return s
 100
 101         liste=obj.get_genealogie_precise()
 102
 103         if obj.etape.nom=='MODIFICATION_CATALOGUE' : return s
 104         nom = obj.etape.nom
 105
 106         if hasattr(obj.etape,'sdnom') and obj.etape.sdnom != None and obj.etape.sdnom != "" :
 107            nom = nom+ obj.etape.sdnom
 108
 109         if not(nom in self.Dico) : dicoCourant={}
 110         else : dicoCourant=self.Dico [nom]
 111
 112         nomFeuille=liste[-1]
 113         if nomFeuille in dicoCourant or nomFeuille in self.DicoDejaLa:
 114            if nomFeuille in self.DicoDejaLa:
 115               nomTravail= nomFeuille +'_'+str(self.DicoDejaLa[nomFeuille])
 116               self.DicoDejaLa[nomFeuille]=self.DicoDejaLa[nomFeuille]+1
 117               nomFeuille=nomTravail
 118            else :
 119               self.DicoDejaLa[nomFeuille]=3
 120               nom1=nomFeuille +'_1'
 121               dicoCourant[nom1]= dicoCourant[nomFeuille]
 122               del dicoCourant[nomFeuille]
 123               nomFeuille=nomFeuille +'_2'
 124
 125
 126         if hasattr(obj.valeur,'nom'): dicoCourant[nomFeuille]=obj.valeur.nom
 127         else :
 128            if type(obj.valeur)  in (list,tuple):
 129               try :
 130 #PNPNPN a remplacer par plus propre
 131                  if obj.definition.validators.typeDesTuples[0] !='R' :
 132                     val=[]
 133                     elt=[]
 134                     for tupleElt in obj.valeur :
 135                         elt=(str(tupleElt[0]),tupleElt[1])
 136                         val.append(elt)
 137                     dicoCourant[nomFeuille]=val
 138                  else :
 139                     dicoCourant[nomFeuille]=obj.valeur
 140               except :
 141                  dicoCourant[nomFeuille]=obj.valeurFormatee
 142            #else :dicoCourant[nomFeuille]=obj.valeurFormatee
 143            else :
 144               dicoCourant[nomFeuille]=obj.valeurFormatee
 145               #print nomFeuille, obj.valeurFormatee
 146         self.Dico[nom]=dicoCourant
 147
 148         return s
 149
 150