1 # -*- coding: utf-8 -*-
2 #_____________________________________
4 import sys, os, re,types
6 from PyQt4.QtGui import QMessageBox
8 from salome.kernel.logger import Logger
9 logger = Logger( "EFICAS_SRC.EFICASGUI.eficasSalome.py" )
12 # eficasConfig definit le EFICAS_ROOT
13 # lignes de path ajoutees pour acceder aux packages python du
14 # logiciel Eficas. Le package Aster est ajoute explicitement pour
15 # acceder au module prefs.py. A
16 # ajout de InterfaceQT4 pour permettre l acces a la fenetre Option
17 sys.path[:0]=[eficasConfig.eficasPath,
18 os.path.join( eficasConfig.eficasPath,'Editeur'),
19 os.path.join( eficasConfig.eficasPath,'UiQT4'),
20 os.path.join( eficasConfig.eficasPath,'InterfaceQT4'),
21 eficasConfig.eficasPath,
26 from InterfaceQT4 import qtEficas
32 from salome.kernel.studyedit import getStudyEditor
35 # couleur pour visualisation des geometries
37 COLORS = colors.ListeColors
38 LEN_COLORS = len( COLORS )
41 class MyEficas( qtEficas.Appli ):
43 Classe de lancement du logiciel EFICAS dans SALOME
44 Cette classe specialise le logiciel Eficas par l'ajout de:
45 a)la creation de groupes de mailles dans le composant SMESH de SALOME
46 b)la visualisation d'elements geometrique dans le coposant GEOM de SALOME par selection dans EFICAS
48 def __init__( self, parent, code = "ASTER", fichier = None, module = "EFICAS",
49 version = None, componentName = "Eficas",multi=False):
53 @param parent: widget Qt parent
55 @param code: catalogue a lancer ( ASTER, HOMARD OPENTURNS ). optionnel ( defaut = ASTER ).
57 @param fichier: chemin absolu du fichier eficas a ouvrir a das le lancement. optionnel
60 dictPathCode={'ASTER':'Aster','OPENTURNS_STUDY':'Openturns_Study','CARMEL3D':'Carmel3D',
61 'OPENTURNS_WRAPPER':'Openturns_Wrapper','MAP':'MAP','SEP':'Sep'}
62 if code in dictPathCode.keys():
63 pathCode=dictPathCode[code]
64 sys.path[:0]=[os.path.join(eficasConfig.eficasPath,pathCode)]
66 if Editeur.__dict__.has_key( 'session' ):
67 from Editeur import session
71 eficasArg += [ fichier ]
73 eficasArg += [ "-c", version ]
76 session.parse( eficasArg )
78 self.editor = getStudyEditor() # Editeur de l'arbre d'etude
80 qtEficas.Appli.__init__( self,code=code,salome=1,parent=parent,multi=multi)
82 #--------------- specialisation EFICAS dans SALOME -------------------
84 self.salome = True #active les parties de code specifique dans Salome( pour le logiciel Eficas )
85 self.module = module #indique sous quel module dans l'arbre d'etude ajouter le JDC.
86 self.componentName = componentName
88 # donnee pour la creation de groupe de maille
89 self.mainShapeNames = {} #dictionnaire pour gerer les multiples fichiers possibles ouverts par
90 #eficas ( cle = identifiant du JDC ), une mainshape par fichier ouvert.
91 #dictionnaire des sous-geometrie de la geometrie principale ( cle = entry, valeur = name )
92 #----------------------------------------------------------------------
94 self.icolor = 0 # compteur pour memoriser la couleur courante
98 def closeEvent(self,event):
99 if hasattr(self,'readercata') :
106 # ___________________________ Methodes de l ex Pal __________________________________
108 #----------------------------------------------------------------
109 def getCORBAObjectInComponent( self, entry, composant ):
110 #----------------------------------------------------------------
112 mySO = self.editor.study.FindObjectID(entry)
114 object = mySO.GetObject()
116 myComponent = salome.lcc.FindOrLoadComponent("FactoryServer", composant)
117 SCom = self.editor.study.FindComponent( composant )
118 print myComponent , SCom
119 self.editor.builder.LoadWith( SCom , myComponent )
120 object = mySO.GetObject()
122 logger.debug("selectedEntry: An error occurs")
126 #----------------------------------------------
127 def giveMeshGroups( self,entry,label1,typeMesh):
128 #----------------------------------------------
133 monMaillage =self.getCORBAObjectInComponent(entry,"SMESH")
134 if monMaillage != None : # selection d'un groupe de SMESH
135 if monMaillage._narrow(SMESH.SMESH_Mesh):
136 mailSO = self.editor.study.FindObjectID(entry)
137 if mailSO == None : return names, msg
140 subIt = self.editor.study.NewChildIterator(mailSO)
142 subSO = subIt.Value()
145 if (subSO.GetName()[0:9]!=label1) : continue
146 subSSMeshit=self.editor.study.NewChildIterator(subSO)
147 while subSSMeshit.More():
148 subSSMeshSO = subSSMeshit.Value()
150 if subSSMeshSO.GetObject()._narrow(typeMesh):
151 names.append(subSSMeshSO.GetName())
153 msg=entry + " n est pas un maillage"
155 logger.debug(' giveMeshGroups pb avec ( entry = %s ) ' %entry )
156 msg=' giveMeshGroup pb avec ( entry = %s ) '+ entry
159 #-------------------------------------
160 def isMeshGroup( self,entry):
161 #-------------------------------------
165 monObjet =self.getCORBAObjectInComponent(entry,"SMESH")
166 if monObjet != None : # selection d'un groupe de SMESH
167 if monObjet._narrow(SMESH.SMESH_GroupBase):
170 logger.debug(' isMeshGroup pb avec ( entry = %s ) ' %entry )
173 #-------------------------------------
174 def isShape( self,entry):
175 #-------------------------------------
179 monObjet =self.getCORBAObjectInComponent(entry,"GEOM")
180 if monObjet != None : # selection d'un objet GEOM
181 if monObjet._narrow(GEOM.GEOM_Object ):
184 logger.debug(' isShape pb avec ( entry = %s ) ' %entry )
187 #-----------------------------------------------------------------
188 def getMainShapeEntry(self,entry):
189 #-----------------------------------------------------------------
192 mainShapeEntry = entry.split(':')[:4]
193 if len(mainShapeEntry) == 4:
194 strMainShapeEntry = '%s:%s:%s:%s'%tuple(mainShapeEntry)
195 if self.isMainShape(strMainShapeEntry):
196 result = strMainShapeEntry
198 logger.debug( 'Erreur pour SalomeStudy.getMainShapeEntry( entry = %s ) ' %entry )
202 #-----------------------------------------------------------------
203 def isMainShape(self,entry):
204 #-----------------------------------------------------------------
207 monObjet =self.getCORBAObjectInComponent(entry,"GEOM")
209 shape = monObjet._narrow( GEOM.GEOM_Object )
210 if shape.IsMainShape():
213 logger.debug( 'Errreur pour SalomeStudy.isMainShape( entry = %s ) ' %entry )
218 #-----------------------------------------------------------------
219 def ChercheType( self, shape ):
220 #-----------------------------------------------------------------
221 tgeo = shape.GetShapeType()
222 geomEngine = salome.lcc.FindOrLoadComponent( "FactoryServer", "GEOM" )
223 #print dir(self.editor.study)
224 groupIMeasureOp = geomEngine.GetIMeasureOperations(self.editor.study._get_StudyId())
225 if tgeo != "COMPOUND" : return tgeo
227 strInfo = groupIMeasureOp.WhatIs( shape )
229 l = strInfo.split('\n')
232 nom, valeur = couple.split(':')
233 dictInfo[ nom.strip() ] = valeur.strip()
235 ordre = [ "COMPSOLID", "SOLID", "SHELL", "FACE", "WIRE", "EDGE", "VERTEX" ]
237 if dictInfo[ t ] != '0':
243 #-----------------------------------------------------------------
244 def selectShape( self, editor, entry, kwType = None ):
245 #-----------------------------------------------------------------
247 selection sous-geometrie dans Salome:
248 -test1) si c'est un element sous-geometrique .
249 -test2) si appartient a la geometrie principale.
251 name, msgError = '',''
252 mySO = self.editor.study.FindObjectID(entry)
254 return name, msgError
255 object = mySO.GetObject()
257 return name, msgError
260 shape = object._narrow( GEOM.GEOM_Object )
262 return name, msgError
264 tGeo=self.ChercheType(shape)
266 return name, msgError
267 #if kwType == "GROUP_NO" and str(tGeo) != "VERTEX":
268 # name,msgError = '',"la selection n est pas un Vertex"
269 # return name, msgError
270 if kwType == "GROUP_MA" and str(tGeo) == "VERTEX":
271 name, msgError = '', "la selection n est pas un groupe de maille"
272 return name, msgError
274 mainShapeEntry = self.getMainShapeEntry( entry )
275 if self.mainShapeNames.has_key( editor ):
276 #print "------------- self.mainShapeNames[editor]" , self.mainShapeNames[editor]
277 if self.mainShapeNames[editor] == mainShapeEntry:
280 msgError="Le groupe reference la geometrie " + mainShapeEntry + " et non " + self.mainShapeNames[editor]
282 self.mainShapeNames[editor] = mainShapeEntry
285 return name, msgError
288 #-----------------------------------------------------------------
289 def selectMeshGroup( self, editor, selectedEntry, kwType = None ):
290 #-----------------------------------------------------------------
292 selection groupe de maille dans Salome:
293 -test 1) si c'est un groupe de maille
294 -test 2) si le maillage fait reference a la geometrie principale
296 name, msgError = '',''
298 mySO=self.editor.study.FindObjectID(selectedEntry )
299 from salome.smesh.smeshstudytools import SMeshStudyTools
300 monSMeshStudyTools=SMeshStudyTools(self.editor)
301 meshSO = monSMeshStudyTools.getMeshFromGroup(mySO)
302 if meshSO == None : return name, msgError
304 # on verifie que l entree selectionnee a le bon type (NODE ou EDGE...)
306 groupObject = self.getCORBAObjectInComponent(selectedEntry,"SMESH")
308 logger.debug("selectedMeshEntry: An error occurs")
311 aGroup = groupObject._narrow( SMESH.SMESH_GroupBase )
312 if aGroup: tGroup = aGroup.GetType()
314 if kwType == "GROUP_NO" and tGroup != SMESH.NODE:
315 msgError = "GROUP_NO attend un groupe de noeud"
316 return name, msgError
317 elif kwType == "GROUP_MA" and tGroup == SMESH.NODE:
318 msgError = "GROUP_MA attend un point goupe de maille"
319 return name, msgError
321 # on cherche la shape associee
322 #PN PN mesh_Object est un SOject
323 meshObject = meshSO.GetObject()
324 mesh = meshObject._narrow( SMESH.SMESH_Mesh )
325 if mesh: #c'est bien un objet maillage
326 shape = mesh.GetShapeToMesh()
328 ior = salome.orb.object_to_string( shape )
330 sObject = self.editor.study.FindObjectIOR( ior )
331 mainShapeID = sObject.GetID()
335 return name, "Type d objet non permis"
337 # on cherche si la shape associee est la bonne
338 #print "------------- mainShapeID" , mainShapeID
339 if self.mainShapeNames.has_key( editor ):
340 #print "------------- self.mainShapeNames[editor]" , self.mainShapeNames[editor]
341 if self.mainShapeNames[editor] == mainShapeID:
344 msgError="Le groupe reference la geometrie " + mainShapeID + " et non " + self.mainShapeNames[editor]
346 self.mainShapeNames[editor] = mainShapeID
349 #print "------------------------------ name :", name
350 #print "------------------------------ name :", name
351 #print "------------------------------ name :", name
355 def displayMeshGroups(self, meshGroupName):
357 visualisation group de maille de nom meshGroupName dans salome
359 ok, msgError = False, ''
362 sg = salome.ImportComponentGUI('SMESH')
363 meshGroupEntries = []
365 selMeshGroupEntry = None
367 # liste des groupes de maille de nom meshGroupName
368 listSO = self.editor.study.FindObjectByName(meshGroupName, "SMESH")
370 #print "liste des groupes de maille de nom %s: "%(meshGroupName), listSO
373 return 0,'Plusieurs objets portent ce nom'
375 return 0,'Aucun objet ne porte ce nom'
377 groupEntry = SObjet.GetID()
378 myComponent = salome.lcc.FindOrLoadComponent("FactoryServer", "SMESH")
379 SCom = self.editor.study.FindComponent("SMESH")
380 myBuilder = self.editor.study.NewBuilder()
381 myBuilder.LoadWith( SCom , myComponent )
382 sg.CreateAndDisplayActor(groupEntry)
383 #color = COLORS[ self.icolor % LEN_COLORS ]
384 #self.icolor = self.icolor + 1
385 #sg.SetColor(groupEntry, color[0], color[1], color[2])
386 salome.sg.Display(groupEntry)
392 msgError = "Impossible d afficher "+shapeName
396 # ___________________________ Methodes appelees par EFICAS __________________________________
397 #----------------------------------------------------------------
398 def selectGroupFromSalome( self, kwType = None, editor=None):
399 #----------------------------------------------------------------
401 Selection d'element(s) d'une geometrie ( sub-shape ) ou d'element(s) de maillage ( groupe de maille) partir de l'arbre salome
402 retourne ( la liste des noms des groupes, message d'erreur )
404 Note: Appele par EFICAS lorsqu'on clique sur le bouton ajouter la liste du panel GROUPMA
408 atLeastOneStudy = self.editor.study
409 if not atLeastOneStudy:
412 # recupere toutes les selections de l'utilsateur dans l'arbre Salome
413 entries = salome.sg.getAllSelected()
414 nbEntries = len( entries )
416 for entry in entries:
417 if self.isMeshGroup(entry): # selection d 'un sous maillage
418 name, msg = self.selectMeshGroup( editor, entry, kwType )
419 elif self.isShape(entry): # selection d'une sous-geometrie
420 name, msg = self.selectShape( editor, entry, kwType )
422 name, msg = None, "Selection SALOME non autorisee."
427 logger.debug("selectGroupFromSalome: An error occurs")
428 #print "=================== selectGroupFromSalome ", names, msg
429 #print "=================== selectGroupFromSalome ", names, msg
430 #print "=================== selectGroupFromSalome ", names, msg
433 #---------------------------------------------
434 def addJdcInSalome( self, jdcPath ):
435 #---------------------------------------------
437 Ajoute le Jeu De Commande dans l'arbre d'etude Salome dans la rubrique EFICAS
440 msgError = "Erreur dans l'export du fichier de commande dans l'arbre d'etude Salome"
443 atLeastOneStudy = self.editor.study
444 if not atLeastOneStudy:
447 fileType = { 'ASTER' : "FICHIER_EFICAS_ASTER",
448 'SEP' : "FICHIER_EFICAS_SEP",
449 'MAP' : "FICHIER_EFICAS_MAP",
450 'OPENTURNS': "FICHIER_EFICAS_OPENTURNS",
451 'OPENTURNS_STUDY': "FICHIER_EFICAS_OPENTURNS_STUDY",
452 'OPENTURNS_WRAPPER': "FICHIER_EFICAS_OPENTURNS_WRAPPER",
453 'CARMEL3D': "FICHIER_EFICAS_CARMEL3D",
456 folderName = { 'ASTER' : 'AsterFiles',
458 'CARMEL3D' : 'CARMEL3DFiles' ,
460 'OPENTURNS_STUDY': 'OpenturnsFiles',
461 'OPENTURNS_WRAPPER': 'OpenturnsFiles'}
463 folderType = { 'ASTER': "ASTER_FILE_FOLDER",
464 'SEP': "SEP_FILE_FOLDER",
465 'MAP': "MAP_FILE_FOLDER",
466 'CARMEL3D': "CARMEL3D_FILE_FOLDER",
467 'OPENTURNS_STUDY':"OPENTURNS_FILE_FOLDER",
468 'OPENTURNS_WRAPPER': "OPENTURNS_FILE_FOLDER"}
471 moduleEntry = self.editor.findOrCreateComponent(self.module, self.componentName)
472 itemName = re.split("/",jdcPath)[-1]
474 fatherEntry = self.editor.findOrCreateItem(
476 name = folderName[self.code],
477 #icon = "ICON_COMM_FOLDER",
478 fileType = folderType[self.code])
480 commEntry = self.editor.findOrCreateItem( fatherEntry ,
482 fileType = fileType[ self.code ],
484 #icon = "ICON_COMM_FILE",
485 comment = str( jdcPath ))
487 salome.sg.updateObjBrowser(1)
489 #print 'addJdcInSalome commEntry->', commEntry
491 ok, msgError = True, ''
492 except Exception, exc:
493 msgError = "Can't add Eficas file to Salome study tree"
494 logger.debug(msgError, exc_info = True)
495 QMessageBox.warning(self, self.tr("Warning"),
496 self.tr("%s. Reason:\n%s\n\nSee logs for "
497 "more details." % (msgError, exc)))
501 #---------------------------------------
502 def displayShape( self, shapeName ):
503 #---------------------------------------
505 visualisation de nom shapeName dans salome
507 ok, msgError = False, ''
511 sgPyQt = SalomePyQt.SalomePyQt()
512 myActiveView=sgPyQt.getActiveView()
513 if myActiveView < 0 :
514 return ok, 'pas de vue courante'
516 currentViewType=sgPyQt.getViewType(myActiveView)
517 if str(currentViewType) != "OCCViewer" : # maillage
518 ok, msgError = self.displayMeshGroups(shapeName)
520 current_color = COLORS[ self.icolor % LEN_COLORS ]
521 from salome.geom.geomtools import GeomStudyTools
522 myGeomTools=GeomStudyTools(self.editor)
523 ok = myGeomTools.displayShapeByName( shapeName, current_color )
525 self.icolor = self.icolor + 1
527 msgError = "Impossible d afficher "+shapeName
533 #---------------------------------------
534 def ChercheGrpMeshInSalome(self):
535 print "je passe par la"
539 entries = salome.sg.getAllSelected()
540 nbEntries = len( entries )
541 names, msg = None, "Selection SALOME non autorisee."
543 for entry in entries:
545 names,msg=self.giveMeshGroups(entry,"SubMeshes",SMESH.SMESH_subMesh)
551 #---------------------------------------
552 def ChercheGrpMailleInSalome(self):
557 entries = salome.sg.getAllSelected()
558 nbEntries = len( entries )
559 names, msg = None, "Selection SALOME non autorisee."
561 for entry in entries:
563 names,msg=self.giveMeshGroups(entry,"Groups of",SMESH.SMESH_GroupBase)
571 #---------------------------------------
573 # def buildCabriGeom( self, name, **param ):
575 visualisation dans GEOM d'une geometrie CABRI
578 # qt.QApplication.setOverrideCursor( qt.QCursor.waitCursor )
579 # cabri.tetra( name, **param )
580 # qt.QApplication.restoreOverrideCursor()
584 #-------------------------------------------------------------------------------------------------------
585 # Pilotage de la Visu des elements de structures
589 def envoievisu(self,liste_commandes):
591 from salome.geom.structelem import StructuralElementManager, InvalidParameterError
593 QMessageBox.critical(self, self.tr("Error"),
594 self.tr("Cannot display structural elements: "
595 "module GEOM is not installed."))
598 atLeastOneStudy = self.editor.study
599 if not atLeastOneStudy:
601 logger.debug(10*'#'+":envoievisu: creating a StructuralElementManager instance")
602 structElemManager = StructuralElementManager()
603 elem = structElemManager.createElement(liste_commandes)
605 salome.sg.updateObjBrowser(True)
606 except InvalidParameterError, err:
607 trStr = self.tr("Invalid parameter for group %(group)s: %(expr)s must be "
608 "greater than %(minval)g (actual value is %(value)g)")
609 msg = str(trStr) % {"group": err.groupName, "expr": err.expression,
610 "minval": err.minValue, "value": err.value}
611 QMessageBox.warning(self, self.tr("Error"), msg)
613 traceback.print_exc()
614 logger.debug(10*'#'+":pb dans envoievisu")
617 #-------------------------------------------------------------------------------------------------------
618 # Point d'entree lancement EFICAS
620 def runEficas( code=None, fichier=None, module = "EFICAS", version=None, componentName = "Eficas",multi=False):
621 logger.debug(10*'#'+":runEficas: START")
623 logger.debug(10*'#'+":runEficas: code="+str(code))
624 logger.debug(10*'#'+":runEficas: fichier="+str(fichier))
625 logger.debug(10*'#'+":runEficas: module="+str(module))
626 logger.debug(10*'#'+":runEficas: version="+str(version))
628 #if not appli: #une seul instance possible!
629 appli = MyEficas( SalomePyQt.SalomePyQt().getDesktop(), code = code, fichier = fichier,
630 module = module, version = version, componentName = componentName,multi=multi )
631 #if not appli: #une seul instance possible!
632 # appli = MyEficas( SalomePyQt.SalomePyQt().getDesktop(), code = code, fichier = fichier,
633 # module = module, componentName = componentName, version=version )
634 logger.debug(10*'#'+":runEficas: END")