Salome HOME
refs #80 - Sketch base GUI: create/draw point, circle and arc
[modules/shaper.git] / src / SketchPlugin / SketchPlugin_Plugin.cpp
index ab95647eedc7b0abec58d235f20d2536a1b23dab..14e61b2362c7806c27657e8017b92cb77c2f95c8 100644 (file)
@@ -1,6 +1,15 @@
 #include "SketchPlugin_Plugin.h"
 #include "SketchPlugin_Sketch.h"
 #include "SketchPlugin_Line.h"
+#include "SketchPlugin_Point.h"
+#include "SketchPlugin_Circle.h"
+#include "SketchPlugin_Arc.h"
+#include "SketchPlugin_ConstraintCoincidence.h"
+#include "SketchPlugin_ConstraintDistance.h"
+#include "SketchPlugin_ConstraintLength.h"
+#include "SketchPlugin_ConstraintParallel.h"
+#include "SketchPlugin_ConstraintPerpendicular.h"
+#include "SketchPlugin_ConstraintRadius.h"
 #include <ModelAPI_PluginManager.h>
 #include <ModelAPI_Document.h>
 
@@ -15,14 +24,38 @@ SketchPlugin_Plugin::SketchPlugin_Plugin()
   ModelAPI_PluginManager::get()->registerPlugin(this);
 }
 
-boost::shared_ptr<ModelAPI_Feature> SketchPlugin_Plugin::createFeature(string theFeatureID)
+FeaturePtr SketchPlugin_Plugin::createFeature(string theFeatureID)
 {
-  if (theFeatureID == "Sketch") {
-    return boost::shared_ptr<ModelAPI_Feature>(new SketchPlugin_Sketch);
+  if (theFeatureID == SKETCH_KIND) {
+    return FeaturePtr(new SketchPlugin_Sketch);
   }
-  else if (theFeatureID == "SketchLine") {
-    return boost::shared_ptr<ModelAPI_Feature>(new SketchPlugin_Line);
+  else if (theFeatureID == SKETCH_POINT_KIND) {
+    return FeaturePtr(new SketchPlugin_Point);
+  }
+  else if (theFeatureID == SKETCH_LINE_KIND) {
+    return FeaturePtr(new SketchPlugin_Line);
+  }
+  else if (theFeatureID == SKETCH_CIRCLE_KIND) {
+    return FeaturePtr(new SketchPlugin_Circle);
+  }
+  else if (theFeatureID == SKETCH_CONSTRAINT_COINCIDENCE_KIND) {
+    return FeaturePtr(new SketchPlugin_ConstraintCoincidence);
+  }
+  else if (theFeatureID == SKETCH_CONSTRAINT_DISTANCE_KIND) {
+    return FeaturePtr(new SketchPlugin_ConstraintDistance);
+  }
+  else if (theFeatureID == SKETCH_CONSTRAINT_LENGTH_KIND) {
+    return FeaturePtr(new SketchPlugin_ConstraintLength);
+  }
+  else if (theFeatureID == SKETCH_CONSTRAINT_PARALLEL_KIND) {
+    return FeaturePtr(new SketchPlugin_ConstraintParallel);
+  }
+  else if (theFeatureID == SKETCH_CONSTRAINT_PERPENDICULAR_KIND) {
+    return FeaturePtr(new SketchPlugin_ConstraintPerpendicular);
+  }
+  else if (theFeatureID == SKETCH_CONSTRAINT_RADIUS_KIND) {
+    return FeaturePtr(new SketchPlugin_ConstraintRadius);
   }
   // feature of such kind is not found
-  return boost::shared_ptr<ModelAPI_Feature>();
+  return FeaturePtr();
 }