Ability for plugins to manage state (on/off) of their features added

[modules/shaper.git] / src / PartSetPlugin / PartSetPlugin_Plugin.h

diff --git a/src/PartSetPlugin/PartSetPlugin_Plugin.h b/src/PartSetPlugin/PartSetPlugin_Plugin.h
index 9969cb403b4dc407299a26cff2e1999e575ac7ea..364ebb48dfbf4f12e2fbd571930136890e3c3b0b 100644 (file)
--- a/src/PartSetPlugin/PartSetPlugin_Plugin.h
+++ b/src/PartSetPlugin/PartSetPlugin_Plugin.h
@@ -7,19 +7,28 @@
 #ifndef PartSetPlugin_Plugin_H_
 #define PartSetPlugin_Plugin_H_
 
-#include "PartSetPlugin.h"
-#include "ModelAPI_Plugin.h"
-#include "ModelAPI_Feature.h"
+#include <PartSetPlugin.h>
+#include <ModelAPI_Plugin.h>
+#include <ModelAPI_Feature.h>
+#include <ModelAPI_Events.h>
 
-class PARTSETPLUGIN_EXPORT PartSetPlugin_Plugin : public ModelAPI_Plugin
+#include <Events_Listener.h>
+
+#include <memory>
+
+class PARTSETPLUGIN_EXPORT PartSetPlugin_Plugin : public ModelAPI_Plugin,
+                                                  public Events_Listener
 {
  public:
   /// Creates the feature object of this plugin by the feature string ID
   virtual FeaturePtr createFeature(std::string theFeatureID);
 
- public:
   /// Is needed for python wrapping by swig
   PartSetPlugin_Plugin();
+
+  //! Redefinition of Events_Listener method
+  virtual void processEvent(const std::shared_ptr<Events_Message>& theMessage);
+  std::shared_ptr<ModelAPI_FeatureStateMessage> getFeaturesState();
 };
 
 #endif