Salome HOME
Issue #2130: arc is done not as desired
[modules/shaper.git] / src / SketchPlugin / SketchPlugin_ConstraintBase.h
index f6f6b07822cee2254a24221a45c956f4a87b401a..508980aa819e92230763f57ee5ebe347db08ae12 100644 (file)
@@ -14,6 +14,7 @@
 #include <ModelAPI_AttributeRefAttr.h>
 
 #include <GeomAPI_IPresentable.h>
+#include <GeomAPI_ICustomPrs.h>
 
 #include <list>
 
@@ -38,7 +39,8 @@
  *         Some feature's methods implemented here as dummy to
  *         Base class for all constraints.
  */
-class SketchPlugin_ConstraintBase : public SketchPlugin_Constraint, public GeomAPI_IPresentable
+class SketchPlugin_ConstraintBase : public SketchPlugin_Constraint, public GeomAPI_IPresentable,
+                                    public GeomAPI_ICustomPrs
 {
  public:
   /// Returns the AIS preview
@@ -53,9 +55,12 @@ class SketchPlugin_ConstraintBase : public SketchPlugin_Constraint, public GeomA
   /// \param theDeltaY the delta for Y coordinate is moved
   SKETCHPLUGIN_EXPORT virtual void move(const double theDeltaX, const double theDeltaY);
 
-  /// Return the distance between the feature and the point
-  /// \param thePoint the point
-  virtual double distanceToPoint(const std::shared_ptr<GeomAPI_Pnt2d>& thePoint);
+      /// Customize presentation of the feature
+  virtual bool customisePresentation(ResultPtr theResult, AISObjectPtr thePrs,
+                                     std::shared_ptr<GeomAPI_ICustomPrs> theDefaultPrs)
+  {
+    return theDefaultPrs->customisePresentation(theResult, thePrs, theDefaultPrs);
+  }
 
  protected:
   /// \brief Use plugin manager for features creation