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

[modules/shaper.git] / src / SketchPlugin / SketchPlugin_Plugin.cpp

diff --git a/src/SketchPlugin/SketchPlugin_Plugin.cpp b/src/SketchPlugin/SketchPlugin_Plugin.cpp
index 1149409813a5645b6b24018d974e55aa79b3f29b..14e61b2362c7806c27657e8017b92cb77c2f95c8 100644 (file)
--- a/src/SketchPlugin/SketchPlugin_Plugin.cpp
+++ b/src/SketchPlugin/SketchPlugin_Plugin.cpp
@@ -1,5 +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>
 
@@ -14,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 == 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);
   }
-  /*else if (theFeatureID == "Point") {
-    return shared_ptr<ModelAPI_Feature>(new SketchPlugin_Point);
-  }*/
   // feature of such kind is not found
-  return boost::shared_ptr<ModelAPI_Feature>();
+  return FeaturePtr();
 }