Salome HOME
Merge remote-tracking branch 'remotes/origin/master'
[modules/shaper.git] / src / XGUI / XGUI_DocumentDataModel.h
index 592308ae8adcdbcb30e069f62c5a3f05fedb64ae..7214c82f4e443a68244b4e8428c49ad6618fb6bf 100644 (file)
@@ -6,7 +6,7 @@
 #include "XGUI_Constants.h"
 
 #include <QAbstractItemModel>
-#include <Event_Listener.h>
+#include <Events_Listener.h>
 
 #include <QList>
 
@@ -19,7 +19,7 @@ class XGUI_TopDataModel;
  * \brief This is a proxy data model for Object Browser (QTreeView).
  * It contains several sub-models for generation of each sub-part of data tree.
  */
-class XGUI_EXPORT XGUI_DocumentDataModel : public QAbstractItemModel, public Event_Listener
+class XGUI_EXPORT XGUI_DocumentDataModel : public QAbstractItemModel, public Events_Listener
 {
   Q_OBJECT
 public:
@@ -29,7 +29,7 @@ public:
   virtual ~XGUI_DocumentDataModel();
 
   // Event Listener method
-  virtual void processEvent(const Event_Message* theMessage);
+  virtual void processEvent(const Events_Message* theMessage);
 
 
   virtual QVariant data(const QModelIndex& theIndex, int theRole) const;
@@ -46,12 +46,19 @@ public:
 
   virtual bool hasChildren(const QModelIndex& theParent = QModelIndex()) const;
 
+  bool insertRows(int theRow, int theCount, const QModelIndex& theParent = QModelIndex());
+
+  bool removeRows(int theRow, int theCount, const QModelIndex& theParent = QModelIndex());
+
   //! Returns Feature object by the given Model index.
   //! Returns 0 if the given index is not index of a feature
   FeaturePtr feature(const QModelIndex& theIndex) const;
 
 private:
 
+  enum {PartsFolder, HistoryNode};
+
+
   //! Converts QModelIndex of this model to QModelIndex of a one of sub-models.
   QModelIndex toSourceModelIndex(const QModelIndex& theProxy) const;
 
@@ -64,14 +71,20 @@ private:
   //! Deletes all saved pointers on QModelIndex objects.
   void clearModelIndexes();
 
-  //! Causes inserting of new nodes for given parent and indexes
-  void insertRows(const QModelIndex& theParent, int theStart, int theEnd);
-
   //! Removes sub-model on removing a part object. Also it removes QModelIndex-es which refer to this model
   void removeSubModel(int theModelId);
 
-  //! 
-  bool hasSubModel(const QAbstractItemModel* theModel) const;
+  //! Returns true if the given model is a one of sub-models (of both types)
+  bool isSubModel(const QAbstractItemModel* theModel) const;
+
+  //! Returns true if the given model is a one of sub-models of Part type
+  bool isPartSubModel(const QAbstractItemModel* theModel) const;
+
+  //! Returns Parts Folder node
+  QModelIndex partFolderNode() const;
+
+  int historyOffset() const;
+
 
   //! Document
   boost::shared_ptr<ModelAPI_Document> myDocument;
@@ -84,6 +97,7 @@ private:
 
   //! List of saved QModelIndexes created by sub-models
   QList<QModelIndex*> myIndexes;
+
 };
 
 #endif
\ No newline at end of file