From 05cff8624330db6a190eff65adf1a66860a234cb Mon Sep 17 00:00:00 2001
From: nds <nds@opencascade.com>
Date: Fri, 18 Mar 2016 12:13:34 +0300
Subject: [PATCH] Issue #1343 Improvement of Extrusion and Revolution
 operations: sketch is filled by selected presentation.

---
 src/PartSet/PartSet_WidgetSketchCreator.cpp |  8 ++++++--
 src/PartSet/PartSet_WidgetSketchCreator.h   |  3 ++-
 src/PartSet/PartSet_WidgetSketchLabel.cpp   | 17 ++++++++++++-----
 src/PartSet/PartSet_WidgetSketchLabel.h     |  6 +++++-
 4 files changed, 25 insertions(+), 9 deletions(-)

diff --git a/src/PartSet/PartSet_WidgetSketchCreator.cpp b/src/PartSet/PartSet_WidgetSketchCreator.cpp
index 71069fe36..874d917b1 100644
--- a/src/PartSet/PartSet_WidgetSketchCreator.cpp
+++ b/src/PartSet/PartSet_WidgetSketchCreator.cpp
@@ -6,6 +6,7 @@
 
 #include "PartSet_WidgetSketchCreator.h"
 #include "PartSet_Module.h"
+#include "PartSet_WidgetSketchLabel.h"
 
 #include <Config_Keywords.h>
 
@@ -180,7 +181,7 @@ void PartSet_WidgetSketchCreator::onSelectionChanged()
   QList<ModuleBase_ViewerPrs> aSelected = getFilteredSelected();
 
   if (aSelected.size() == 1) { // plane or planar face of not sketch object
-    startSketchOperation();
+    startSketchOperation(aSelected.front());
   }
   else if (aSelected.size() > 1) {
     QList<ModuleBase_ViewerPrs>::const_iterator anIt = aSelected.begin(), aLast = aSelected.end();
@@ -223,7 +224,7 @@ void PartSet_WidgetSketchCreator::onStarted()
   setVisibleSelectionControl(true);
 }
 
-void PartSet_WidgetSketchCreator::startSketchOperation()
+void PartSet_WidgetSketchCreator::startSketchOperation(const ModuleBase_ViewerPrs& theValue)
 {
   // Check that model already has bodies
   /*XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(myModule->workshop());
@@ -254,6 +255,9 @@ void PartSet_WidgetSketchCreator::startSketchOperation()
     DocumentPtr aDoc = aMgr->activeDocument();
     FeaturePtr aPreviousCurrentFeature = aDoc->currentFeature(false);
     FeaturePtr aSketch = aCompFeature->addFeature("Sketch");
+
+    PartSet_WidgetSketchLabel::fillSketchPlaneBySelection(aSketch, theValue);
+
     aDoc->setCurrentFeature(aPreviousCurrentFeature, false);
 
     // start edit operation for the sketch
diff --git a/src/PartSet/PartSet_WidgetSketchCreator.h b/src/PartSet/PartSet_WidgetSketchCreator.h
index 7349d1615..e14352a0c 100644
--- a/src/PartSet/PartSet_WidgetSketchCreator.h
+++ b/src/PartSet/PartSet_WidgetSketchCreator.h
@@ -93,7 +93,8 @@ private slots:
 
 private:
   /// Append new Sketch, set the selected plane for the sketch and start Edit operation.
-  void startSketchOperation();
+  /// \param theValue a selected object to obtain a plane for sketch
+  void startSketchOperation(const ModuleBase_ViewerPrs& theValue);
 
 private:
   std::string myAttributeListID;
diff --git a/src/PartSet/PartSet_WidgetSketchLabel.cpp b/src/PartSet/PartSet_WidgetSketchLabel.cpp
index d424aa7e7..ea65b9315 100644
--- a/src/PartSet/PartSet_WidgetSketchLabel.cpp
+++ b/src/PartSet/PartSet_WidgetSketchLabel.cpp
@@ -316,14 +316,20 @@ void PartSet_WidgetSketchLabel::restoreAttributeValue(const bool theValid)
 }
 
 bool PartSet_WidgetSketchLabel::setSelectionCustom(const ModuleBase_ViewerPrs& thePrs)
+{
+  return fillSketchPlaneBySelection(feature(), thePrs);
+}
+
+bool PartSet_WidgetSketchLabel::fillSketchPlaneBySelection(const FeaturePtr& theFeature,
+                                                           const ModuleBase_ViewerPrs& thePrs)
 {
   bool isOwnerSet = false;
 
   const TopoDS_Shape& aShape = thePrs.shape();
   std::shared_ptr<GeomAPI_Dir> aDir;
 
-  if (thePrs.object() && (feature() != thePrs.object())) {
-    DataPtr aData = feature()->data();
+  if (thePrs.object() && (theFeature != thePrs.object())) {
+    DataPtr aData = theFeature->data();
     AttributeSelectionPtr aSelAttr = 
       std::dynamic_pointer_cast<ModelAPI_AttributeSelection>
       (aData->attribute(SketchPlugin_SketchEntity::EXTERNAL_ID()));
@@ -345,7 +351,7 @@ bool PartSet_WidgetSketchLabel::setSelectionCustom(const ModuleBase_ViewerPrs& t
     }
   }
   else if (!aShape.IsNull()) {
-    aDir = setSketchPlane(aShape);
+    aDir = setSketchPlane(theFeature, aShape);
     isOwnerSet = aDir.get();
   }
   return isOwnerSet;
@@ -476,7 +482,8 @@ AISObjectPtr PartSet_WidgetSketchLabel::createPreviewPlane(std::shared_ptr<GeomA
 }
 
 
-std::shared_ptr<GeomAPI_Dir> PartSet_WidgetSketchLabel::setSketchPlane(const TopoDS_Shape& theShape)
+std::shared_ptr<GeomAPI_Dir> PartSet_WidgetSketchLabel::setSketchPlane(const FeaturePtr& theFeature,
+                                                                       const TopoDS_Shape& theShape)
 {
   if (theShape.IsNull())
     return std::shared_ptr<GeomAPI_Dir>();
@@ -493,7 +500,7 @@ std::shared_ptr<GeomAPI_Dir> PartSet_WidgetSketchLabel::setSketchPlane(const Top
     return std::shared_ptr<GeomAPI_Dir>();
 
   // set plane parameters to feature
-  std::shared_ptr<ModelAPI_Data> aData = feature()->data();
+  std::shared_ptr<ModelAPI_Data> aData = theFeature->data();
   double anA, aB, aC, aD;
   aPlane->coefficients(anA, aB, aC, aD);
 
diff --git a/src/PartSet/PartSet_WidgetSketchLabel.h b/src/PartSet/PartSet_WidgetSketchLabel.h
index 90c291bcd..ff6689d1c 100644
--- a/src/PartSet/PartSet_WidgetSketchLabel.h
+++ b/src/PartSet/PartSet_WidgetSketchLabel.h
@@ -79,6 +79,9 @@ public:
   virtual void setHighlighted(bool) { /*do nothing*/ };
   virtual void enableFocusProcessing();
 
+  static bool fillSketchPlaneBySelection(const FeaturePtr& theFeature,
+                                         const ModuleBase_ViewerPrs& thePrs);
+
 signals:
   /// Signal on plane selection
   void planeSelected(const std::shared_ptr<GeomAPI_Pln>& thePln);
@@ -163,7 +166,8 @@ protected:
 
   /// Set sketch plane by shape
   /// \param theShape a planar face
-  std::shared_ptr<GeomAPI_Dir> setSketchPlane(const TopoDS_Shape& theShape);
+  static std::shared_ptr<GeomAPI_Dir> setSketchPlane(const FeaturePtr& theFeature,
+                                                     const TopoDS_Shape& theShape);
 
   /// Erase preview planes
   void erasePreviewPlanes();
-- 
2.39.2