Some slight improvements and bugs fixing.

[modules/shaper.git] / src / PartSet / PartSet_Module.h

diff --git a/src/PartSet/PartSet_Module.h b/src/PartSet/PartSet_Module.h
index 561c97fe0224741d042221903ba80f27d388a000..261427642304dff1cf6afc7cae2e37ab20790962 100644 (file)
--- a/src/PartSet/PartSet_Module.h
+++ b/src/PartSet/PartSet_Module.h
@@ -3,8 +3,10 @@
 
 #include "PartSet.h"
 
-#include <XGUI_Module.h>
+#include <ModuleBase_IModule.h>
+#include <ModuleBase_Definitions.h>
 #include <XGUI_Command.h>
+#include <ModelAPI_Feature.h>
 
 #include <QMap>
 #include <QObject>
@@ -18,8 +20,9 @@ class QKeyEvent;
 class PartSet_Listener;
 class ModelAPI_Feature;
 class XGUI_ViewerPrs;
+class ModuleBase_Operation;
 
-class PARTSET_EXPORT PartSet_Module: public QObject, public XGUI_Module
+class PARTSET_EXPORT PartSet_Module: public ModuleBase_IModule
 {
 Q_OBJECT
 
@@ -31,8 +34,13 @@ public:
   /// \returns a workshop instance
   XGUI_Workshop* workshop() const;
 
+  /// Reads description of features from XML file 
   virtual void createFeatures();
+
+  /// Called on creation of menu item in desktop
   virtual void featureCreated(QAction* theFeature);
+
+  /// Returnc list of nested commands for the given feature
   virtual QStringList nestedFeatures(QString theFeature);
   std::string featureFile(const std::string&);
 
@@ -40,6 +48,9 @@ public:
   /// \param theCmdId the operation name
   virtual void launchOperation(const QString& theCmdId);
 
+  /// Called when it is necessary to update a command state (enable or disable it)
+  virtual bool isFeatureEnabled(const QString& theCmdId) const;
+
   /// Displays or erase the current operation preview, if it has it.
   /// \param theFeature the feature instance to be displayed
   /// \param isDisplay the state whether the presentation should be displayed or erased