2 #ifndef XGUI_DocumentDataModel_H
3 #define XGUI_DocumentDataModel_H
6 #include "XGUI_Constants.h"
8 #include <QAbstractItemModel>
9 #include <Events_Listener.h>
13 class ModelAPI_Document;
15 class XGUI_TopDataModel;
17 /**\class XGUI_DocumentDataModel
19 * \brief This is a proxy data model for Object Browser (QTreeView).
20 * It contains several sub-models for generation of each sub-part of data tree.
22 class XGUI_EXPORT XGUI_DocumentDataModel : public QAbstractItemModel, public Events_Listener
28 XGUI_DocumentDataModel(QObject* theParent);
29 virtual ~XGUI_DocumentDataModel();
31 // Event Listener method
32 virtual void processEvent(const Events_Message* theMessage);
35 virtual QVariant data(const QModelIndex& theIndex, int theRole) const;
36 virtual QVariant headerData(int theSection, Qt::Orientation theOrient,
37 int theRole = Qt::DisplayRole) const;
39 virtual int rowCount(const QModelIndex& theParent = QModelIndex()) const;
40 virtual int columnCount(const QModelIndex& theParent = QModelIndex()) const;
42 virtual QModelIndex index(int theRow, int theColumn,
43 const QModelIndex &parent = QModelIndex()) const;
45 virtual QModelIndex parent(const QModelIndex& theIndex) const;
47 virtual bool hasChildren(const QModelIndex& theParent = QModelIndex()) const;
49 bool insertRows(int theRow, int theCount, const QModelIndex& theParent = QModelIndex());
51 bool removeRows(int theRow, int theCount, const QModelIndex& theParent = QModelIndex());
53 Qt::ItemFlags flags(const QModelIndex& theIndex) const;
55 //! Returns Feature object by the given Model index.
56 //! Returns 0 if the given index is not index of a feature
57 FeaturePtr feature(const QModelIndex& theIndex) const;
59 //! Returns QModelIndex which corresponds to the given feature if this is a part
60 //! If the feature is not found then index is not valid
61 QModelIndex partIndex(const FeaturePtr& theFeature) const;
63 //! Activates a part data model if the index is a Part node index.
64 //! Returns true if active part changed.
65 bool activatedIndex(const QModelIndex& theIndex);
67 FeaturePtr activePart() const;
69 QModelIndex activePartIndex() const { return myActivePartIndex; }
71 //! Deactivates a Part
72 void deactivatePart();
74 void rebuildDataTree();
78 enum {PartsFolder, HistoryNode};
81 //! Converts QModelIndex of this model to QModelIndex of a one of sub-models.
82 QModelIndex* toSourceModelIndex(const QModelIndex& theProxy) const;
84 //! Finds a pointer on QModelIndex which is equal to the given one
85 QModelIndex* findModelIndex(const QModelIndex& theIndex) const;
87 //! Returns pointer on QModelIndex which is equal to the given one.
88 QModelIndex* getModelIndex(const QModelIndex& theIndex) const;
90 //! Deletes all saved pointers on QModelIndex objects.
91 void clearModelIndexes();
93 //! Removes sub-model on removing a part object. Also it removes QModelIndex-es which refer to this model
94 void removeSubModel(int theModelId);
96 //! Returns true if the given model is a one of sub-models (of both types)
97 bool isSubModel(const QAbstractItemModel* theModel) const;
99 //! Returns true if the given model is a one of sub-models of Part type
100 bool isPartSubModel(const QAbstractItemModel* theModel) const;
102 //! Returns Parts Folder node
103 QModelIndex partFolderNode() const;
105 int historyOffset() const;
108 //! Data model of top part of data tree (not parts object)
109 XGUI_TopDataModel* myModel;
111 //! Data models for Parts data tree representation (one data model per a one part)
112 QList<XGUI_PartModel*> myPartModels;
114 //! Active part in part editing mode
115 XGUI_PartModel* myActivePart;
117 QModelIndex myActivePartIndex;
119 //! List of saved QModelIndexes created by sub-models
120 QList<QModelIndex*> myIndexes;