Operation presentation stayed in the viewer when argument is deselected in extrusion...

diff --git a/src/PartSet/PartSet_Module.cpp b/src/PartSet/PartSet_Module.cpp
index 164253ada2f9b80fcc2847e34355aa8127bd7002..ab86f1bf9d792d337597aa8fe4476aed5e29b840 100644 (file)
--- a/src/PartSet/PartSet_Module.cpp
+++ b/src/PartSet/PartSet_Module.cpp
@@ -709,6 +709,14 @@ void PartSet_Module::onObjectDisplayed(ObjectPtr theObject, AISObjectPtr theAIS)
   }
 }
 
+void PartSet_Module::onBeforeObjectErase(ObjectPtr theObject, AISObjectPtr theAIS)
+{
+  // it should be recomputed in order to disappear in the viewer if the corresponded object
+  // is erased
+  if (myCustomPrs->isActive())
+    myCustomPrs->customize(theObject);
+}
+
 void PartSet_Module::onViewTransformed(int theTrsfType)
 {
   // Set length of arrows constant in pixel size

diff --git a/src/PartSet/PartSet_Module.h b/src/PartSet/PartSet_Module.h
index 532e874f53465df94c5f9b273a3c26b1c2c1ee8a..d2f9c163b48239808792d3debfd1f1057a8dc407 100644 (file)
--- a/src/PartSet/PartSet_Module.h
+++ b/src/PartSet/PartSet_Module.h
@@ -189,6 +189,11 @@ public slots:
   /// \param theAIS a presentation object
   virtual void onObjectDisplayed(ObjectPtr theObject, AISObjectPtr theAIS);
 
+  /// Slot called on before object erase
+  /// \param theObject a data object
+  /// \param theAIS a presentation object
+  virtual void onBeforeObjectErase(ObjectPtr theObject, AISObjectPtr theAIS);
+
   /// Called on transformation in current viewer
   /// \param theTrsfType type of tranformation
   void onViewTransformed(int theTrsfType = 2);