Task "Make the size of the selection area even bigger, especially for points"

[modules/shaper.git] / src / ConstructionPlugin / ConstructionPlugin_Point.h

diff --git a/src/ConstructionPlugin/ConstructionPlugin_Point.h b/src/ConstructionPlugin/ConstructionPlugin_Point.h
index 84cb35cb35085d1b7461b84cf3da54ca5ea7f07f..d1f15287f64d62d6e50bc67a555eac16a8e23ee9 100644 (file)
--- a/src/ConstructionPlugin/ConstructionPlugin_Point.h
+++ b/src/ConstructionPlugin/ConstructionPlugin_Point.h
@@ -10,17 +10,19 @@
 #include "ConstructionPlugin.h"
 #include <ModelAPI_Feature.h>
 #include <ModelAPI_Result.h>
+#include <GeomAPI_ICustomPrs.h>
 
 /**\class ConstructionPlugin_Point
  * \ingroup Plugins
  * \brief Feature for creation of the new part in PartSet.
  */
-class ConstructionPlugin_Point : public ModelAPI_Feature
+class ConstructionPlugin_Point : public ModelAPI_Feature, public GeomAPI_ICustomPrs
 {
  public:
   /// Returns the kind of a feature
   CONSTRUCTIONPLUGIN_EXPORT virtual const std::string& getKind();
 
+  /// Point kind
   inline static const std::string& ID()
   {
     static const std::string CONSTRUCTION_POINT_KIND("Point");
@@ -57,6 +59,10 @@ class ConstructionPlugin_Point : public ModelAPI_Feature
 
   /// Use plugin manager for features creation
   ConstructionPlugin_Point();
+
+  /// Customize presentation of the feature
+  virtual bool customisePresentation(ResultPtr theResult, AISObjectPtr thePrs,
+                                     std::shared_ptr<GeomAPI_ICustomPrs> theDefaultPrs);
 };
 
 #endif