From 7144e51cb8c2806269b1aff754d73e4e44e104b9 Mon Sep 17 00:00:00 2001
From: vsv <vsv@opencascade.com>
Date: Wed, 24 Jun 2020 20:46:05 +0300
Subject: [PATCH] Issue #3236: Add buttons for shape type choice

---
 src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp | 3 +++
 src/FeaturesPlugin/FeaturesPlugin_Extrusion.h   | 7 +++++++
 src/FeaturesPlugin/extrusion_widget.xml         | 9 +++++++++
 src/PartSet/PartSet_WidgetSketchCreator.cpp     | 3 +++
 4 files changed, 22 insertions(+)

diff --git a/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp b/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp
index 57b1ba874..6ae36c31a 100644
--- a/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp
+++ b/src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp
@@ -23,6 +23,7 @@
 #include <ModelAPI_AttributeDouble.h>
 #include <ModelAPI_AttributeSelection.h>
 #include <ModelAPI_AttributeString.h>
+#include <ModelAPI_AttributeInteger.h>
 #include <ModelAPI_Session.h>
 #include <ModelAPI_Validator.h>
 
@@ -57,6 +58,8 @@ void FeaturesPlugin_Extrusion::initAttributes()
 
   data()->addAttribute(DIRECTION_OBJECT_ID(), ModelAPI_AttributeSelection::typeId());
 
+  data()->addAttribute(USE_SHAPES_ID(), ModelAPI_AttributeInteger::typeId());
+
   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), TO_OBJECT_ID());
   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), FROM_OBJECT_ID());
   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), DIRECTION_OBJECT_ID());
diff --git a/src/FeaturesPlugin/FeaturesPlugin_Extrusion.h b/src/FeaturesPlugin/FeaturesPlugin_Extrusion.h
index 1f657b30a..91a30f16d 100644
--- a/src/FeaturesPlugin/FeaturesPlugin_Extrusion.h
+++ b/src/FeaturesPlugin/FeaturesPlugin_Extrusion.h
@@ -123,6 +123,13 @@ public:
     return MY_FROM_OFFSET_ID;
   }
 
+  /// Attribute name of using shape types.
+  inline static const std::string& USE_SHAPES_ID()
+  {
+    static const std::string MY_USE_SHAPEST_ID("use_sapes");
+    return MY_USE_SHAPEST_ID;
+  }
+
   /// \return the kind of a feature.
   FEATURESPLUGIN_EXPORT virtual const std::string& getKind()
   {
diff --git a/src/FeaturesPlugin/extrusion_widget.xml b/src/FeaturesPlugin/extrusion_widget.xml
index 85ca64d71..7d833c095 100644
--- a/src/FeaturesPlugin/extrusion_widget.xml
+++ b/src/FeaturesPlugin/extrusion_widget.xml
@@ -8,6 +8,15 @@
                    shape_types="vertices edges wires faces shells solids compsolids compounds">
     <validator id="FeaturesPlugin_ValidatorCompositeLauncher" parameters="base,Sketch"/>
   </sketch_launcher>
+  <choice id="use_sapes"
+    widget_type="radiobuttons"
+    buttons_dir="horizontal"
+    label="Type"
+    tooltip="Using shapes type"
+    string_list="Vertices Edges Faces"
+    icons_list=":pictures/vertex32.png :pictures/edge32.png :pictures/face32.png"
+    default="2">
+  </choice>
   <multi_selector id="base"
                   label="Base objects:"
                   tooltip="Select a base objects"
diff --git a/src/PartSet/PartSet_WidgetSketchCreator.cpp b/src/PartSet/PartSet_WidgetSketchCreator.cpp
index de888dc75..9d6d90cf2 100644
--- a/src/PartSet/PartSet_WidgetSketchCreator.cpp
+++ b/src/PartSet/PartSet_WidgetSketchCreator.cpp
@@ -212,6 +212,9 @@ void PartSet_WidgetSketchCreator::setVisibleSelectionControl(const bool theSelec
   XGUI_PropertyPanel* aPanel = aWorkshop->propertyPanel();
   const QList<ModuleBase_ModelWidget*>& aWidgets = aPanel->modelWidgets();
   foreach(ModuleBase_ModelWidget* aWidget, aWidgets) {
+    QString aType(aWidget->metaObject()->className());
+    if (aType == "ModuleBase_WidgetChoice")
+      continue;
     if (theSelectionControl) { // hide other controls
       if (aWidget != this)
         aWidget->setVisible(false);
-- 
2.39.2