Reverse the function of guihelper in smeshstudytool

src/SMESH_PY/Makefile.am

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

src/SMESH_PY/guihelper.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()

src/SMESH_PY/smeshstudytools.py

@@ -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()