refs #80 - Sketch base GUI: create/draw point, circle and arc

[modules/shaper.git] / src / PartSet / PartSet_OperationCreateConstraint.cpp

diff --git a/src/PartSet/PartSet_OperationCreateConstraint.cpp b/src/PartSet/PartSet_OperationCreateConstraint.cpp
index bc4463b94ff000e982dc9d13fffc3e16d87bf678..d4e9e51758b96e6851e449b7816c223319a30495 100644 (file)
--- a/src/PartSet/PartSet_OperationCreateConstraint.cpp
+++ b/src/PartSet/PartSet_OperationCreateConstraint.cpp
@@ -13,12 +13,17 @@
 #include <PartSet_EditLine.h>
 
 #include <SketchPlugin_Feature.h>
-#include <SketchPlugin_Point.h>
+/*#include <SketchPlugin_Point.h>
 #include <SketchPlugin_Line.h>
 #include <SketchPlugin_Circle.h>
 #include <SketchPlugin_Arc.h>
+*/
 
-#include <SketchPlugin_ConstraintLength.h>
+#include <PartSet_FeatureLengthPrs.h>
+#include <PartSet_FeatureDistancePrs.h>
+#include <PartSet_FeatureRadiusPrs.h>
+
+#include <SketchPlugin_Constraint.h>
 
 #include <ModuleBase_OperationDescription.h>
 
@@ -59,7 +64,9 @@ PartSet_OperationCreateConstraint::~PartSet_OperationCreateConstraint()
 bool PartSet_OperationCreateConstraint::canProcessKind(const std::string& theId)
 {
   // changed
-  return theId == SKETCH_CONSTRAINT_LENGTH_KIND;
+  return theId == PartSet_FeatureLengthPrs::getKind() ||
+         theId == PartSet_FeatureDistancePrs::getKind() ||
+         theId == PartSet_FeatureRadiusPrs::getKind();
 }
 
 bool PartSet_OperationCreateConstraint::canBeCommitted() const