Issue #1865: Model Widget for fields

[modules/shaper.git] / src / CollectionPlugin / CollectionPlugin_Plugin.cpp

diff --git a/src/CollectionPlugin/CollectionPlugin_Plugin.cpp b/src/CollectionPlugin/CollectionPlugin_Plugin.cpp
index 87c0dda9304a107cf9ad216503ca8b2b084d2886..6f0ca30e7b8d7c9b237637c39ec589845918b227 100644 (file)
--- a/src/CollectionPlugin/CollectionPlugin_Plugin.cpp
+++ b/src/CollectionPlugin/CollectionPlugin_Plugin.cpp
@@ -3,8 +3,13 @@
 #include <CollectionPlugin_Plugin.h>
 
 #include <CollectionPlugin_Group.h>
+#include <CollectionPlugin_Field.h>
 #include <ModelAPI_Session.h>
 
+#include <ModuleBase_WidgetCreatorFactory.h>
+
+#include "CollectionPlugin_WidgetCreator.h"
+
 #include <string>
 #include <memory>
 
@@ -13,6 +18,10 @@ static CollectionPlugin_Plugin* MY_COLLECTION_INSTANCE = new CollectionPlugin_Pl
 
 CollectionPlugin_Plugin::CollectionPlugin_Plugin()
 {
+  WidgetCreatorFactoryPtr aWidgetCreatorFactory = ModuleBase_WidgetCreatorFactory::get();
+  aWidgetCreatorFactory->registerCreator(
+   std::shared_ptr<CollectionPlugin_WidgetCreator>(new CollectionPlugin_WidgetCreator()));
+
   SessionPtr aMgr = ModelAPI_Session::get();
   // register this plugin
   ModelAPI_Session::get()->registerPlugin(this);
@@ -22,8 +31,11 @@ FeaturePtr CollectionPlugin_Plugin::createFeature(std::string theFeatureID)
 {
   if (theFeatureID == CollectionPlugin_Group::ID()) {
     return FeaturePtr(new CollectionPlugin_Group);
+  }else if (theFeatureID == CollectionPlugin_Field::ID()) {
+    return FeaturePtr(new CollectionPlugin_Field);
   }
 
+
   // feature of such kind is not found
   return FeaturePtr();
 }