FIX+IMP (display/erase a shape identified by its entry)

diff --git a/src/GEOM_PY/geomtools.py b/src/GEOM_PY/geomtools.py
index 4b5c1630243ed0ef84a297bbd79457d5a42976c4..890959cfb9e1482420ecddc3a4e40953eae0727c 100644 (file)
--- a/src/GEOM_PY/geomtools.py
+++ b/src/GEOM_PY/geomtools.py
@@ -91,7 +91,7 @@ class GeomStudyTools:
         if GEOM is None:
             GEOM = __import__("GEOM")
         if studyEditor is None:
-            studyEditor = helper.getStudyEditor()
+            studyEditor = getStudyEditor()
         self.editor = studyEditor
 
     def displayShapeByName(self, shapeName, color = None):
@@ -114,15 +114,33 @@ class GeomStudyTools:
             geomObj = self.editor.getOrLoadObject(sObj)
             if geomObj:
                 shape = geomObj._narrow(GEOM.GEOM_Object)
-                if shape:                
-                    geomgui = salome.ImportComponentGUI("GEOM")            
-                    geomgui.createAndDisplayGO(entry)
-                    geomgui.setDisplayMode(entry, DisplayMode)
-                    if color is not None:
-                        geomgui.setColor(entry, color[0], color[1], color[2])
-                    return True
+                if shape:
+                    return self.displayShapeByEntry(entry,color)
         return False
 
+    def displayShapeByEntry(self, entry, color = None):
+        """
+        Display the geometrical shape whose entry is given by `entry`.
+        """
+        geomgui = salome.ImportComponentGUI("GEOM")            
+        geomgui.createAndDisplayGO(entry)
+        geomgui.setDisplayMode(entry, DisplayMode)
+        if color is not None:
+            geomgui.setColor(entry, color[0], color[1], color[2])
+        return True
+
+    def eraseShapeByEntry(self, entry):
+        """
+        Erase the geometrical shape whose entry is given by
+        `entry`. Please note that the shape is just erased from the
+        viewer. The associated study object still exists in the study,
+        and the geom object still exists in the GEOM engine.
+        """
+        geomgui = salome.ImportComponentGUI("GEOM")
+        eraseFromAllWindows=True
+        geomgui.eraseGO(entry,eraseFromAllWindows)
+        return True
+    
     def getGeomObjectSelected(self):
         '''
         Returns the GEOM object currently selected in the objects browser.