Add "Torus" primitive and "Cone" primitive.

[modules/shaper.git] / src / PrimitivesPlugin / PrimitivesPlugin_Plugin.cpp

diff --git a/src/PrimitivesPlugin/PrimitivesPlugin_Plugin.cpp b/src/PrimitivesPlugin/PrimitivesPlugin_Plugin.cpp
index 36c77fa8a64cb7e172bbd753b71422baefb32b0c..8db8d5d167a3f41202f9535e4b1e02af089db6e2 100644 (file)
--- a/src/PrimitivesPlugin/PrimitivesPlugin_Plugin.cpp
+++ b/src/PrimitivesPlugin/PrimitivesPlugin_Plugin.cpp
@@ -7,8 +7,10 @@
 #include <PrimitivesPlugin_Plugin.h>
 
 #include <PrimitivesPlugin_Box.h>
+#include <PrimitivesPlugin_Cone.h>
 #include <PrimitivesPlugin_Cylinder.h>
 #include <PrimitivesPlugin_Sphere.h>
+#include <PrimitivesPlugin_Torus.h>
 
 #include <ModelAPI_Session.h>
 
@@ -29,10 +31,14 @@ FeaturePtr PrimitivesPlugin_Plugin::createFeature(std::string theFeatureID)
 {
   if (theFeatureID == PrimitivesPlugin_Box::ID()) {
     return FeaturePtr(new PrimitivesPlugin_Box);
+  } else if (theFeatureID == PrimitivesPlugin_Cone::ID()) {
+    return FeaturePtr(new PrimitivesPlugin_Cone);
   } else if (theFeatureID == PrimitivesPlugin_Cylinder::ID()) {
     return FeaturePtr(new PrimitivesPlugin_Cylinder);
   } else if (theFeatureID == PrimitivesPlugin_Sphere::ID()) {
     return FeaturePtr(new PrimitivesPlugin_Sphere);
+  } else if (theFeatureID == PrimitivesPlugin_Torus::ID()) {
+    return FeaturePtr(new PrimitivesPlugin_Torus);
   }
   // feature of such kind is not found
   return FeaturePtr();