Validator for partition

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

diff --git a/src/ConstructionPlugin/ConstructionPlugin_Point.h b/src/ConstructionPlugin/ConstructionPlugin_Point.h
index 84cb35cb35085d1b7461b84cf3da54ca5ea7f07f..8e0f0fcda6f74a4b959fdbbe664db39ab332ed0a 100644 (file)
--- a/src/ConstructionPlugin/ConstructionPlugin_Point.h
+++ b/src/ConstructionPlugin/ConstructionPlugin_Point.h
@@ -10,12 +10,13 @@
 #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
@@ -57,6 +58,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