generator/generator_dicoImbrique.py

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