From 7a9f064633a57e79683a8fe75c228a0c76d383a0 Mon Sep 17 00:00:00 2001
From: vsv <vitaly.smetannikov@opencascade.com>
Date: Wed, 10 Jun 2015 19:08:09 +0300
Subject: [PATCH] Cancel ExtrusionCut if there are no bodies

---
 .../FeaturesPlugin_ExtrusionCut.cpp           |  2 +-
 src/PartSet/PartSet_WidgetSketchCreator.cpp   | 38 ++++++++++++++++---
 2 files changed, 33 insertions(+), 7 deletions(-)

diff --git a/src/FeaturesPlugin/FeaturesPlugin_ExtrusionCut.cpp b/src/FeaturesPlugin/FeaturesPlugin_ExtrusionCut.cpp
index 6cd449d82..2f8704e02 100755
--- a/src/FeaturesPlugin/FeaturesPlugin_ExtrusionCut.cpp
+++ b/src/FeaturesPlugin/FeaturesPlugin_ExtrusionCut.cpp
@@ -55,7 +55,7 @@ std::shared_ptr<ModelAPI_Feature> FeaturesPlugin_ExtrusionCut::addFeature(std::s
     data()->reference(SKETCH_OBJECT_ID())->setValue(aNew);
   }
    // set as current also after it becomes sub to set correctly enabled for other sketch subs
-  //document()->setCurrentFeature(aNew, false);
+  document()->setCurrentFeature(aNew, false);
   return aNew;
 }
 
diff --git a/src/PartSet/PartSet_WidgetSketchCreator.cpp b/src/PartSet/PartSet_WidgetSketchCreator.cpp
index ac1d449d8..a2d51b7cc 100644
--- a/src/PartSet/PartSet_WidgetSketchCreator.cpp
+++ b/src/PartSet/PartSet_WidgetSketchCreator.cpp
@@ -15,6 +15,7 @@
 #include <GeomAPI_Face.h>
 
 #include <ModelAPI_Session.h>
+#include <ModelAPI_ResultBody.h>
 
 #include <ModuleBase_Tools.h>
 #include <ModuleBase_Operation.h>
@@ -24,6 +25,7 @@
 #include <QLabel>
 #include <QLineEdit>
 #include <QFormLayout>
+#include <QMessageBox>
 
 PartSet_WidgetSketchCreator::PartSet_WidgetSketchCreator(QWidget* theParent, 
                                                          PartSet_Module* theModule,
@@ -89,13 +91,37 @@ void PartSet_WidgetSketchCreator::onStarted()
 {
   disconnect(myModule, SIGNAL(operationLaunched()), this, SLOT(onStarted()));
 
-  CompositeFeaturePtr aCompFeature = 
-    std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(myFeature);
-  FeaturePtr aSketch = aCompFeature->addFeature("Sketch");
+  // Check that model already has bodies
+  XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(myModule->workshop());
+  XGUI_Workshop* aWorkshop = aConnector->workshop();
+  XGUI_Displayer* aDisp = aWorkshop->displayer();
+  QObjectPtrList aObjList = aDisp->displayedObjects();
+  bool aHasBody = false;
+  ResultBodyPtr aBody;
+  foreach(ObjectPtr aObj, aObjList) {
+    aBody = std::dynamic_pointer_cast<ModelAPI_ResultBody>(aObj);
+    if (aBody.get() != NULL) {
+      aHasBody = true;
+      break;
+    }
+  }
 
-  ModuleBase_Operation* anOperation = myModule->createOperation("Sketch");
-  anOperation->setFeature(aSketch);
-  myModule->sendOperation(anOperation);
+  if (aHasBody) {
+    // Launch Sketch operation
+    CompositeFeaturePtr aCompFeature = 
+      std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(myFeature);
+    FeaturePtr aSketch = aCompFeature->addFeature("Sketch");
+
+    ModuleBase_Operation* anOperation = myModule->createOperation("Sketch");
+    anOperation->setFeature(aSketch);
+    myModule->sendOperation(anOperation);
+  } else {
+    // Break current operation
+    QMessageBox::warning(this, tr("Extrusion Cut"),
+        tr("There are no bodies found. Operation aborted."), QMessageBox::Ok);
+    ModuleBase_Operation* aOp = myModule->workshop()->currentOperation();
+    aOp->abort();
+  }
 }
 
 bool PartSet_WidgetSketchCreator::focusTo()
-- 
2.39.2