ForkMaster/smesh
1
0
Fork 0
mirror of https://git.salome-platform.org/gitpub/modules/smesh.git synced 2024-11-15 10:08:34 +05:00
Code Issues Packages Projects Releases Wiki Activity

Reverse the function of guihelper in smeshstudytool

Browse Source
This commit is contained in:
boulant 2011-10-28 17:00:01 +00:00
parent fb93a51d7d
commit b441d65e90
3 changed files with 141 additions and 170 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/SMESH_PY/Makefile.am
View File

@ -22,5 +22,4 @@ include $(top_srcdir)/adm_local/unix/make_common_starter.am
salomepypkgdir = $(salomepythondir)/salome/smesh
salomepypkg_PYTHON = \
__init__.py \
smeshstudytools.py \
guihelper.py
smeshstudytools.py

168
src/SMESH_PY/guihelper.py
View File

@ -1,168 +0,0 @@
# -*- coding: iso-8859-1 -*-
# Copyright (C) 2007-2011 CEA/DEN, EDF R&D, OPEN CASCADE
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
#
# Author: Guillaume Boulant (EDF/R&D)
# ==================================================================
# This file provides helper functions to drive some SMESH gui features
# of a SALOME Application (the selection of mesh objects in the object
# browser for now, further development coming soon).
# ==================================================================
from salome.gui import helper
#
# ==================================================================
# Special functions to deal with SMESH objects
# ==================================================================
#
import salome
salome.salome_init()
import SMESH
def getSmeshObjectFromSObject(sobject):
'''
Returns the SMESH object associated to the specified SObject,
(the SObject is an item in the objects browser).
'''
if sobject is None:
return None
#theObject = sobject.GetObject()
#if theObject is not None:
# return theObject
# The engine must be loaded
theStudy = helper.getActiveStudy()
theBuilder = theStudy.NewBuilder()
strContainer, strComponentName = "FactoryServer", "SMESH"
engine = salome.lcc.FindOrLoadComponent( strContainer, strComponentName )
scomponent = theStudy.FindComponent( strComponentName )
theBuilder.LoadWith( scomponent , engine )
# The servant can be retrieved from the sobject
theObject = sobject.GetObject()
# Then cast to the smesh desired class (supposed to be a mesh in
# this example).
smeshObject = theObject._narrow(SMESH.SMESH_Mesh)
return smeshObject
import smesh
def getSmeshObjectFromEntry(entry):
'''
Returns the SMESH object associated to the specified entry,
(the entry is the identifier of an item in the objects browser).
'''
if entry is None:
return None
theStudy = helper.getActiveStudy()
smesh.SetCurrentStudy(theStudy)
smeshObject=smesh.IDToObject(entry)
return smeshObject
def getSmeshObjectSelected():
'''
Returns the SMESH object currently selected in the objects browser.
'''
sobject, entry = helper.getSObjectSelected()
# You can retrieve the smesh object either from the sobject or
# from the entry. From the entry is quicker.
#smeshObject = getSmeshObjectFromSObject(sobject)
smeshObject = getSmeshObjectFromEntry(entry)
return smeshObject
SmeshGUI = salome.ImportComponentGUI("SMESH")
def displaySmeshObject(entry):
'''
Display the SMESH object associated to the specified entry
(the entry is the identifier of an item in the objects browser).
'''
if not helper.SalomeGUI.hasDesktop():
print "displayGeomObjects: no desktop available"
return
SmeshGUI.CreateAndDisplayActor(entry)
#
# ==================================================================
# Use cases and demo functions
# ==================================================================
#
# CAUTION: Before running this test functions, you first have to
# create (or import) an smesh object and select this object in the
# objects browser. You can run the box mesh creation procedure below
# instead.
def TEST_createBoxMesh():
theStudy = helper.getActiveStudy()
import geompy
geompy.init_geom(theStudy)
box = geompy.MakeBoxDXDYDZ(200, 200, 200)
import smesh, SMESH, SALOMEDS
smesh.SetCurrentStudy(theStudy)
import StdMeshers
boxmesh = smesh.Mesh(box)
Regular_1D = boxmesh.Segment()
Nb_Segments_1 = Regular_1D.NumberOfSegments(15)
Nb_Segments_1.SetDistrType( 0 )
Quadrangle_2D = boxmesh.Quadrangle()
Hexa_3D = smesh.CreateHypothesis('Hexa_3D')
status = boxmesh.AddHypothesis(Hexa_3D)
isDone = boxmesh.Compute()
smesh.SetName(boxmesh.GetMesh(), 'boxmesh')
if salome.sg.hasDesktop():
salome.sg.updateObjBrowser(1)
#
# Definitions:
# - the SObject is an item in the study (Study Object).
# - the entry is the identifier of an item.
# - the object (geom object or smesh object) is a CORBA servant
# embedded in the SALOME component container and with a reference in
# the SALOME study, so that it can be retrieved.
#
def TEST_selectAndExport_01():
myMesh = getSmeshObjectSelected()
myMesh.ExportUNV("/tmp/myMesh.unv")
def TEST_selectAndExport_02():
# In this case, we want to retrieve the name of the mesh in the
# object browser. Note that in SALOME, a mesh object has no
# name. Only the SObject in the object browser has a name
# attribute.
mySObject, myEntry = helper.getSObjectSelected()
myName = mySObject.GetName()
myMesh = getSmeshObjectFromEntry(myEntry)
exportFileName = "/tmp/"+myName+".unv"
myMesh.ExportUNV(exportFileName)
def TEST_display():
mySObject, myEntry = helper.getSObjectSelected()
displaySmeshObject(myEntry)
if __name__ == "__main__":
TEST_selectAndExport_01()
TEST_selectAndExport_02()
TEST_display()

140
src/SMESH_PY/smeshstudytools.py
View File

@ -27,6 +27,7 @@ import salome
SMESH = None # SMESH module is loaded only when needed
from salome.kernel.studyedit import getStudyEditor
from salome.gui import helper
class SMeshStudyTools:
"""
@ -51,7 +52,15 @@ class SMeshStudyTools:
if studyEditor is None:
studyEditor = getStudyEditor()
self.editor = studyEditor
self.smeshGui = None
def updateStudy(self, studyId=None):
"""
This function updates the tools so that it works on the
specified study.
"""
self.editor = getStudyEditor(studyId)
def getMeshFromGroup(self, meshGroupItem):
"""
Get the mesh item owning the mesh group `meshGroupItem`.
@ -69,3 +78,134 @@ class SMeshStudyTools:
meshObj = group.GetMesh()
meshItem = salome.ObjectToSObject(meshObj)
return meshItem
def getMeshObjectSelected(self):
'''
Returns the MESH object currently selected in the active study.
'''
sobject, entry = helper.getSObjectSelected()
meshObject = self.getMeshObjectFromEntry(entry)
return meshObject
def getMeshObjectFromEntry(self, entry):
'''
Returns the MESH object associated to the specified entry,
(the entry is the identifier of an item in the objects browser).
'''
if entry is None:
return None
import smesh
smesh.SetCurrentStudy(self.editor.study)
meshObject=smesh.IDToObject(entry)
return meshObject
def getMeshObjectFromSObject(self, sobject):
'''
Returns the SMESH object associated to the specified SObject,
(the SObject is an item in the objects browser).
'''
if sobject is None:
return None
obj = self.editor.getOrLoadObject(sobject)
meshObject = obj._narrow(SMESH.SMESH_Mesh)
return meshObject
def displayMeshObjectFromEntry(self,entry):
'''
Display the SMESH object associated to the specified entry
(the entry is the identifier of an item in the objects browser).
'''
if self.smeshGui is None:
self.smeshGui = salome.ImportComponentGUI("SMESH")
if not helper.SalomeGUI.hasDesktop():
print "displayMeshObject: no desktop available"
return
self.smeshGui.CreateAndDisplayActor(entry)
#
# ==================================================================
# Use cases and demo functions
# ==================================================================
#
# CAUTION: Before running this test functions, you first have to
# create (or import) an smesh object and select this object in the
# objects browser. You can run the box mesh creation procedure below
# instead.
# How to test?
# 1. Run a SALOME application including GEOM and SMESH, and create a new study
# 2. In the console, enter:
# >>> from salome.smesh import smeshstudytools
# >>> smeshstudytools.TEST_createBoxMesh()
# 3. Select the object named "boxmesh" in the browser
# 4. In the console, enter:
# >>> smeshstudytools.TEST_selectAndExport_01()
# >>> smeshstudytools.TEST_selectAndExport_02()
# >>> smeshstudytools.TEST_display()
def TEST_createBoxMesh():
theStudy = helper.getActiveStudy()
import geompy
geompy.init_geom(theStudy)
box = geompy.MakeBoxDXDYDZ(200, 200, 200)
import smesh, SMESH, SALOMEDS
smesh.SetCurrentStudy(theStudy)
import StdMeshers
boxmesh = smesh.Mesh(box)
Regular_1D = boxmesh.Segment()
Nb_Segments_1 = Regular_1D.NumberOfSegments(15)
Nb_Segments_1.SetDistrType( 0 )
Quadrangle_2D = boxmesh.Quadrangle()
Hexa_3D = smesh.CreateHypothesis('Hexa_3D')
status = boxmesh.AddHypothesis(Hexa_3D)
isDone = boxmesh.Compute()
smesh.SetName(boxmesh.GetMesh(), 'boxmesh')
if salome.sg.hasDesktop():
salome.sg.updateObjBrowser(1)
#
# Definitions:
# - the SObject is an item in the study (Study Object).
# - the entry is the identifier of an item.
# - the object (geom object or smesh object) is a CORBA servant
# embedded in the SALOME component container and with a reference in
# the SALOME study, so that it can be retrieved.
#
def TEST_selectAndExport_01():
tool = SMeshStudyTools()
myMesh = tool.getMeshObjectSelected()
myMesh.ExportUNV("/tmp/myMesh.unv")
def TEST_selectAndExport_02():
# In this case, we want to retrieve the name of the mesh in the
# object browser. Note that in SALOME, a mesh object has no
# name. Only the SObject in the object browser has a name
# attribute.
tool = SMeshStudyTools()
mySObject, myEntry = helper.getSObjectSelected()
myName = mySObject.GetName()
myMesh = tool.getMeshObjectFromEntry(myEntry)
exportFileName = "/tmp/"+myName+".unv"
myMesh.ExportUNV(exportFileName)
def TEST_display():
mySObject, myEntry = helper.getSObjectSelected()
tool = SMeshStudyTools()
tool.displayMeshObjectFromEntry(myEntry)
if __name__ == "__main__":
TEST_selectAndExport_01()
TEST_selectAndExport_02()
TEST_display()