mirror of
https://git.salome-platform.org/gitpub/modules/smesh.git
synced 2024-11-15 10:08:34 +05:00
Reverse the function of guihelper in smeshstudytool
This commit is contained in:
boulant
parent
fb93a51d7d
commit
b441d65e90
@ -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
|
||||
|
||||
@ -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()
|
||||
@ -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,6 +52,14 @@ 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):
|
||||
"""
|
||||
@ -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()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user