Make concealment of results working on compsolids: if at least one sub-body is concea...

[modules/shaper.git] / src / SketchPlugin / SketchPlugin_ConstraintDistance.h

diff --git a/src/SketchPlugin/SketchPlugin_ConstraintDistance.h b/src/SketchPlugin/SketchPlugin_ConstraintDistance.h
index c8c7cb78129f01627acced76b0ebef54a77cb702..a1e315d2bc0246a0a75b8b7fec4d8aff02847b8f 100644 (file)
--- a/src/SketchPlugin/SketchPlugin_ConstraintDistance.h
+++ b/src/SketchPlugin/SketchPlugin_ConstraintDistance.h
@@ -12,6 +12,8 @@
 #include "SketchPlugin_Sketch.h"
 #include "ModelAPI_Data.h"
 
+#include <GeomAPI_ICustomPrs.h>
+
 #include <list>
 
 class SketchPlugin_Line;
@@ -25,7 +27,7 @@ class GeomDataAPI_Point2D;
  *  This constraint has three attributes:
  *  SketchPlugin_Constraint::VALUE(), SketchPlugin_Constraint::ENTITY_A() and SketchPlugin_Constraint::ENTITY_B()
  */
-class SketchPlugin_ConstraintDistance : public SketchPlugin_ConstraintBase
+class SketchPlugin_ConstraintDistance : public SketchPlugin_ConstraintBase, public GeomAPI_ICustomPrs
 {
  public:
   /// Distance constraint kind
@@ -66,7 +68,11 @@ class SketchPlugin_ConstraintDistance : public SketchPlugin_ConstraintBase
   SKETCHPLUGIN_EXPORT virtual void attributeChanged(const std::string& theID);
 
   /// Returns the current distance between the feature attributes
-  double calculateCurrentDistance() const;
+  double calculateCurrentDistance();
+
+    /// Customize presentation of the feature
+  virtual bool customisePresentation(ResultPtr theResult, AISObjectPtr thePrs,
+                                     std::shared_ptr<GeomAPI_ICustomPrs> theDefaultPrs);
 
   /// \brief Use plugin manager for features creation
   SketchPlugin_ConstraintDistance();