From b08ee19f03ef6ec4ef049384d4712f8deb8eb1eb Mon Sep 17 00:00:00 2001
From: nds <nds@opencascade.com>
Date: Thu, 24 Mar 2016 11:49:22 +0300
Subject: [PATCH] Issue #1343 Improvement of Extrusion and Revolution
 operations: removeFeature is corrected to remove Sketch from reference;
 extrusion/revolution sketches are hidden; display is flushed after sketch
 hide in the sketch operation widget.

---
 .../FeaturesPlugin_CompositeSketch.cpp        |  2 +
 src/FeaturesPlugin/plugin-Features.xml        |  8 +-
 src/FeaturesPlugin/revolution_widget.xml      | 14 ---
 .../GeomValidators_FeatureKind.cpp            |  2 +-
 src/PartSet/PartSet_WidgetSketchCreator.cpp   | 89 ++++++++++---------
 5 files changed, 53 insertions(+), 62 deletions(-)

diff --git a/src/FeaturesPlugin/FeaturesPlugin_CompositeSketch.cpp b/src/FeaturesPlugin/FeaturesPlugin_CompositeSketch.cpp
index 3d8ce83ef..57a711e0c 100644
--- a/src/FeaturesPlugin/FeaturesPlugin_CompositeSketch.cpp
+++ b/src/FeaturesPlugin/FeaturesPlugin_CompositeSketch.cpp
@@ -88,6 +88,8 @@ void FeaturesPlugin_CompositeSketch::removeFeature(std::shared_ptr<ModelAPI_Feat
   AttributeSelectionListPtr aFacesSelectionList = selectionList(LIST_ID());
   if (aFacesSelectionList.get() && aFacesSelectionList->size() > 0)
     aFacesSelectionList->clear();
+
+  data()->reference(SKETCH_OBJECT_ID())->setValue(ObjectPtr());
 }
 
 //=================================================================================================
diff --git a/src/FeaturesPlugin/plugin-Features.xml b/src/FeaturesPlugin/plugin-Features.xml
index fe897b2a5..9f5cbd74e 100644
--- a/src/FeaturesPlugin/plugin-Features.xml
+++ b/src/FeaturesPlugin/plugin-Features.xml
@@ -6,9 +6,9 @@
       <feature id="Extrusion" title="Extrusion" tooltip="Create a solid by extrusion of a face" icon=":icons/extrusion.png">
           <source path="extrusion_widget.xml"/>
       </feature>
-      <feature id="ExtrusionSketch" title="ExtrusionSketch" tooltip="Create a solids by extrusion of a sketch" icon=":icons/extrusionsketch.png">
+      <!--<feature id="ExtrusionSketch" title="ExtrusionSketch" tooltip="Create a solids by extrusion of a sketch" icon=":icons/extrusionsketch.png">
           <source path="extrusionsketch_widget.xml"/>
-      </feature>
+      </feature>-->
       <feature id="ExtrusionCut" title="ExtrusionCut" tooltip="" icon=":icons/extrusion_cut.png">
         <source path="extrusioncut_widget.xml"/>
       </feature>
@@ -20,9 +20,9 @@
       <feature id="Revolution" title="Revolution" tooltip="Create a solid by revolution of a face" icon=":icons/revol.png">
           <source path="revolution_widget.xml"/>
       </feature>
-      <feature id="RevolutionSketch" title="RevolutionSketch" tooltip="Create a solids by revolution of a sketch" icon=":icons/revolsketch.png">
+      <!--<feature id="RevolutionSketch" title="RevolutionSketch" tooltip="Create a solids by revolution of a sketch" icon=":icons/revolsketch.png">
           <source path="revolutionsketch_widget.xml"/>
-      </feature>
+      </feature>-->
       <feature id="RevolutionCut" title="RevolutionCut" tooltip="" icon=":icons/revol_cut.png">
         <source path="revolutioncut_widget.xml"/>
       </feature>
diff --git a/src/FeaturesPlugin/revolution_widget.xml b/src/FeaturesPlugin/revolution_widget.xml
index 73ed217c7..a7a5b82d2 100644
--- a/src/FeaturesPlugin/revolution_widget.xml
+++ b/src/FeaturesPlugin/revolution_widget.xml
@@ -18,13 +18,6 @@
   </composite_multi_selector>
   <toolbox id="CreationMethod">
     <box id="ByAngles" title="By angles" icon=":icons/angle_up_down_32x32.png">
-      <multi_selector id="base"
-        label="Select a sketch face"
-        icon=":icons/sketch.png"
-        tooltip="Select a sketch face"
-        type_choice="Faces">
-        <validator id="GeomValidators_FeatureKind" parameters="Sketch"/>
-      </multi_selector>
       <shape_selector id="axis_object"
                       icon=":icons/axis.png"
                       label="Axis"
@@ -55,13 +48,6 @@
       </groupbox>
     </box>
     <box id="ByPlanesAndOffsets" title="By bounding planes and angles" icon=":icons/plane_inverted_32x32.png">
-      <multi_selector id="base"
-        label="Select a sketch face"
-        icon=":icons/sketch.png"
-        tooltip="Select a sketch face"
-        type_choice="Faces">
-        <validator id="GeomValidators_FeatureKind" parameters="Sketch"/>
-      </multi_selector>
       <shape_selector id="axis_object"
                       icon=":icons/axis.png"
                       label="Axis"
diff --git a/src/GeomValidators/GeomValidators_FeatureKind.cpp b/src/GeomValidators/GeomValidators_FeatureKind.cpp
index edf205f05..3ae820082 100755
--- a/src/GeomValidators/GeomValidators_FeatureKind.cpp
+++ b/src/GeomValidators/GeomValidators_FeatureKind.cpp
@@ -11,7 +11,7 @@
 #include <ModelAPI_AttributeRefAttr.h>
 #include <ModelAPI_Object.h>
 
-//#define DEBUG_EXTRUSION_INVALID_SKETCH
+#define DEBUG_EXTRUSION_INVALID_SKETCH
 #ifdef DEBUG_EXTRUSION_INVALID_SKETCH
   #include <ModelAPI_CompositeFeature.h>
 #endif
diff --git a/src/PartSet/PartSet_WidgetSketchCreator.cpp b/src/PartSet/PartSet_WidgetSketchCreator.cpp
index 8243a253a..2c44b55fd 100644
--- a/src/PartSet/PartSet_WidgetSketchCreator.cpp
+++ b/src/PartSet/PartSet_WidgetSketchCreator.cpp
@@ -202,8 +202,6 @@ bool PartSet_WidgetSketchCreator::canCommitCurrentSketch(ModuleBase_IWorkshop* t
       if (!aIsOp)
         aMgr->startOperation();
       aPCompositeFeature->execute(); // to fill attribute selection list
-      if (!aIsOp)
-        aMgr->finishOperation();
 
       std::list<AttributePtr> aSelListAttributes = aParentFeature->data()->attributes(
                                                         ModelAPI_AttributeSelectionList::typeId());
@@ -226,6 +224,12 @@ bool PartSet_WidgetSketchCreator::canCommitCurrentSketch(ModuleBase_IWorkshop* t
             aCanCommit = false;
         }
       }
+      if (!aIsOp) {
+        if (aCanCommit)
+          aMgr->finishOperation();
+        else
+          aMgr->abortOperation();
+      }
     }
   }
   return aCanCommit;
@@ -383,49 +387,48 @@ void PartSet_WidgetSketchCreator::onResumed(ModuleBase_Operation* theOp)
 
       // do nothing, selection control should be shown
       activateSelectionControl();
-      return;
-    }
-    // do nothing, selection control should be hidden
-    setVisibleSelectionControl(false);
-    // Update value in attribute selection list
-    XGUI_PropertyPanel* aPanel = aWorkshop->propertyPanel();
-    const QList<ModuleBase_ModelWidget*>& aWidgets = aPanel->modelWidgets();
-    foreach(ModuleBase_ModelWidget* aWidget, aWidgets) {
-      if (aWidget->attributeID() == myAttributeListID)
-        aWidget->restoreValue();
-    }
-
-    // Hide sketcher result
-    CompositeFeaturePtr aSketchFeature = 
-      std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(aCompFeature->subFeature(0));
-    std::list<ResultPtr> aResults = aSketchFeature->results();
-    std::list<ResultPtr>::const_iterator aIt;
-    for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
-      (*aIt)->setDisplayed(false);
     }
-    aSketchFeature->setDisplayed(false);
-
-    // restore value in the selection control
-
+    else {
+      setVisibleSelectionControl(false);
+      // Update value in attribute selection list
+      XGUI_PropertyPanel* aPanel = aWorkshop->propertyPanel();
+      const QList<ModuleBase_ModelWidget*>& aWidgets = aPanel->modelWidgets();
+      foreach(ModuleBase_ModelWidget* aWidget, aWidgets) {
+        if (aWidget->attributeID() == myAttributeListID)
+          aWidget->restoreValue();
+      }
 
-    // Add Selected body were created the sketcher to list of selected objects
-    std::string anObjectsAttribute = FeaturesPlugin_CompositeBoolean::BOOLEAN_OBJECTS_ID();
-    AttributeSelectionListPtr aSelList = aCompFeature->data()->selectionList(anObjectsAttribute);
-    if (aSelList.get()) {
-      DataPtr aData = aSketchFeature->data();
-      AttributeSelectionPtr aSelAttr = std::dynamic_pointer_cast<ModelAPI_AttributeSelection>
-                                    (aData->attribute(SketchPlugin_SketchEntity::EXTERNAL_ID()));
-      ResultPtr aRes = aSelAttr.get() ? aSelAttr->context() : ResultPtr();
-      if (aRes.get()) {
-        SessionPtr aMgr = ModelAPI_Session::get();
-        ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
-        AttributePtr anAttribute = myFeature->attribute(anObjectsAttribute);
-        std::string aValidatorID, anError;
-        aSelList->append(aRes, GeomShapePtr());
-        if (aFactory->validate(anAttribute, aValidatorID, anError))
-          updateObject(aCompFeature);
-        else
-          aSelList->clear();
+      // Hide sketcher result
+      CompositeFeaturePtr aSketchFeature = 
+        std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(aCompFeature->subFeature(0));
+      std::list<ResultPtr> aResults = aSketchFeature->results();
+      std::list<ResultPtr>::const_iterator aIt;
+      for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
+        (*aIt)->setDisplayed(false);
+      }
+      aSketchFeature->setDisplayed(false);
+      static Events_Loop* aLoop = Events_Loop::loop();
+      aLoop->flush(aLoop->eventByName(EVENT_OBJECT_TO_REDISPLAY));
+
+      // Add Selected body were created the sketcher to list of selected objects
+      std::string anObjectsAttribute = FeaturesPlugin_CompositeBoolean::BOOLEAN_OBJECTS_ID();
+      AttributeSelectionListPtr aSelList = aCompFeature->data()->selectionList(anObjectsAttribute);
+      if (aSelList.get()) {
+        DataPtr aData = aSketchFeature->data();
+        AttributeSelectionPtr aSelAttr = std::dynamic_pointer_cast<ModelAPI_AttributeSelection>
+                                      (aData->attribute(SketchPlugin_SketchEntity::EXTERNAL_ID()));
+        ResultPtr aRes = aSelAttr.get() ? aSelAttr->context() : ResultPtr();
+        if (aRes.get()) {
+          SessionPtr aMgr = ModelAPI_Session::get();
+          ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
+          AttributePtr anAttribute = myFeature->attribute(anObjectsAttribute);
+          std::string aValidatorID, anError;
+          aSelList->append(aRes, GeomShapePtr());
+          if (aFactory->validate(anAttribute, aValidatorID, anError))
+            updateObject(aCompFeature);
+          else
+            aSelList->clear();
+        }
       }
     }
   }
-- 
2.39.2