1 # -*- coding: iso-8859-1 -*-
3 def handleAjoutGroupFiltre(editor,listeGroup):
4 """CARMEL3D : obtention des groupes de maille du maillage selectionne dans Salome
5 Les groupes de mailles sont filtres en utilisant une liste des prefixes autorises pour code Code_Carmel3D,
6 i.e. un nom de groupe de mailles est DIEL_toto_foo par exemple, qui deviendra toto_foo.
7 La creation du MESH_GROUPE est type (materiau ou source), d'après le prefixe.
8 ATTENTION! Le nom devenant un concept, i.e. une variable Python, certains signes sont interdits dans le nom du groupe,
9 e.g. les signes moins (-), plus (+), etc. Une erreur est retournee en ce cas.
11 from string import join
13 listePrefixesMateriaux = ('DIEL', 'NOCOND','COND', 'ZS', 'ZJ', 'NILMAT') # liste des prefixes pour les materiaux
14 listePrefixesSources = ('CURRENT', 'EPORT', 'HPORT') # liste des prefixes pour les sources
15 listePrefixes = listePrefixesMateriaux + listePrefixesSources # liste de tous les prefixes autorises
16 listePrefixesGroupesMultiples = ('CURRENT', ) # listes des prefixes autorises pour groupes multiples, i.e. plusieurs groupes de mailles associes en une seule caracteistique materiau ou source
17 sep = '_' # separateur entre le prefixe et le nom reel du groupe (qui peut lui aussi contenir ce separateur)
18 dictGroupesMultiplesNomsPossibles = {} # dictionnaire contenant les noms reels possibles de groupes multiples et leur occurence dans la liste, i.e. 1 par defaut et > 1 pour une groupe multiple, e.g. pour un inducteur bobine en plusieurs morceaux CURRENT_toto_1, CURRENT_toto_2, ce dictionnaire contiendra 'toto':2
19 listeGroupesMultiples = [] # liste contenant les noms possibles de groupes multiples, e.g. pour un inducteur bobine en plusieurs morceaux CURRENT_toto_1, CURRENT_toto_2, cette liste contiendra 'toto'
20 for groupe in listeGroup:
21 partiesGroupe = groupe.split(sep) # parties du nom, separees initialement par le separateur du prefixe, e.g. 'CURRENT_toto_foo' devient ['CURRENT','toto','foo'] et 'toto' devient ['toto']
22 prefix = partiesGroupe[0] # prefixe possible de ce nom, ou nom lui-meme
23 if len(partiesGroupe) >= 2 and prefix in listePrefixesGroupesMultiples: # prefixe existant et autorise
24 nomGroupeMultiple = partiesGroupe[1] # nom possible d'un groupe multiple
25 if dictGroupesMultiplesNomsPossibles.has_key(nomGroupeMultiple): # comptage du nombre d'occurrences de ce nom de groupe multiple possible
26 dictGroupesMultiplesNomsPossibles[nomGroupeMultiple] += 1
28 dictGroupesMultiplesNomsPossibles[nomGroupeMultiple] = 1
29 for nom in dictGroupesMultiplesNomsPossibles: # suppression des noms avec une seule occurence, i.e. ils ne sont pas des groupes multiples
30 if dictGroupesMultiplesNomsPossibles[nom] > 1: listeGroupesMultiples.append(nom)
32 print "listeGroup=", listeGroup
33 print "dictGroupesMultiplesNomPossibles=", dictGroupesMultiplesNomsPossibles
34 print "listeGroupesMultiples=", listeGroupesMultiples
35 print "listePrefixes=", listePrefixes
36 # retourne le dernier element du JdC, ou None si le JdC est vide, afin de savoir a quelle place ajouter les MESH_GROUPE (en dernier)
38 dernier=editor.tree.racine.children[-1]
41 for groupe in listeGroup: # parcours de la liste de tous les groupes de maille trouves (volumiques et les autres)
42 if debug: print 'groupe=', groupe
43 partiesGroupe = groupe.split(sep) # parties du nom, separees initialement par le separateur du prefixe, e.g. 'CURRENT_toto_foo' devient ['CURRENT','toto','foo'] et 'toto' devient ['toto']
44 prefix = partiesGroupe[0] # prefixe possible de ce nom, ou nom lui-meme
45 if len(partiesGroupe) == 1: # pas de prefixe
46 print u"ERREUR: ce nom de groupe ("+groupe+") ne peut pas etre utilise car il n'a pas de prefixe"
47 elif len(partiesGroupe) >= 2 and prefix in listePrefixes: # prefixe existant et autorise
48 nomReel = None # initialisation du nom reel, qui provoquera une erreur par la suite (evaluation de None=None) s'il reste ainsi
49 if prefix in listePrefixesGroupesMultiples: # ce groupe pourrait faire partie d'un groupe multiple
50 nomGroupeMultiple = partiesGroupe[1] # nom possible d'un groupe multiple
51 if nomGroupeMultiple in listeGroupesMultiples: # ce groupe est multiple et n'a pas encore ete cree
52 nomReel = nomGroupeMultiple # ce groupe pourrait etre utilise...
53 listeGroupesMultiples.remove(nomGroupeMultiple) #... une seule fois
54 if debug: print u"ce nom de groupe ("+nomReel+") est multiple et sera utilise une fois seulement"
55 elif dictGroupesMultiplesNomsPossibles[nomGroupeMultiple] == 1: # ce groupe existe dans le dictionnaire et n'est pas multiple (occurence =1)
56 nomReel = join(partiesGroupe[1:], sep) # reconstruction du nom reel, i.e. sans le prefixe
57 if debug: print u"ce nom de groupe ("+nomReel+") n'est pas multiple et sera utilise"
58 else: # ce groupe est multiple et a deja ete utilise
59 if debug: print u"ce nom de groupe ("+groupe+") est multiple et a deja ete utilise"
60 else: # ce groupe n'est pas multiple, il pourrait etre utilise tel quel
61 nomReel = join(partiesGroupe[1:], sep) # reconstruction du nom reel, i.e. sans le prefixe
62 if nomReel is not None: # on a un nom de groupe possible, il faut realiser des tests plus pousses
63 try: # test de conformite du nom pour un concept, i.e. une variable Python
64 exec(nomReel+'=None') # le test consiste a tenter de creer une variable, initialisee a None, a partir du nom, e.g. toto=None est bon mais toto-foo=None ne fonctionne pas.
65 # creation du groupe MESH_GROUPE
67 new_node = dernier.append_brother("MESHGROUP",'after')
69 new_node=editor.tree.racine.append_child("MESHGROUP",pos='first')
70 test,mess = new_node.item.nomme_sd(nomReel) # precision du nom (de concept) du groupe
71 if debug: print u"ce nom de groupe ("+nomReel+") est utilise..."
72 if prefix in listePrefixesMateriaux: # ce groupe est associe a un materiau
73 new_node.append_child('MATERIAL') # on rajoute la propriete de materiau, qu'il suffit d'associer ensuite a la liste des materiaux presents
74 if debug: print u" et c'est un materiau."
75 elif prefix in listePrefixesSources: # ce groupe est associe a une source
76 new_node.append_child('SOURCE') # on rajoute la propriete de la source, qu'il suffit d'associer ensuite a la liste des sources presentes
77 if debug: print u" et c'est une source."
78 else: # ce cas ne devrait pas se produire
80 dernier=new_node # mise a jour du dernier noeud du JdC, afin de rajouter les autres MESH_GROUPE eventuels a sa suite
82 print u"ERREUR: ce nom de groupe ("+nomReel+") ne peut pas etre utilise car il ne peut pas servir de concept a cause de caractères interdits, e.g. signes moins (-), plus (+), etc."
83 else: # ce nom de groupe est ecarte car le groupe multiple deja ete cree
84 print u"Ce nom de groupe ("+groupe+") ne peut pas etre utilise car il appartient a un groupe multiple qui a deja ete cree."
85 else: # prefixe existant mais non autorise
86 print u"ERREUR: ce nom de groupe ("+groupe+") ne peut pas etre utilise car son prefixe ("+partiesGroupe[0]+") n'est pas dans la liste autorisee "+str(listePrefixes)