X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FXGUI%2FXGUI_PartDataModel.h;h=4ef9cd58fe5e77611c73fb0f32d3fd04994183b1;hb=b742df475446566e44a011eecd82d4507c04ad44;hp=15c863519bb69f76eed5b8bf05ad4c7277ca3df0;hpb=2c7c361d0ab4687935496e4bb6fe2b2e12702fc9;p=modules%2Fshaper.git

diff --git a/src/XGUI/XGUI_PartDataModel.h b/src/XGUI/XGUI_PartDataModel.h
index 15c863519..4ef9cd58f 100644
--- a/src/XGUI/XGUI_PartDataModel.h
+++ b/src/XGUI/XGUI_PartDataModel.h
@@ -1,3 +1,4 @@
+// Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
 
 #ifndef XGUI_PartDataModel_H
 #define XGUI_PartDataModel_H
@@ -12,12 +13,12 @@
  */
 class XGUI_EXPORT XGUI_TopDataModel : public XGUI_FeaturesModel
 {
-  Q_OBJECT
-public:
-  XGUI_TopDataModel(const std::shared_ptr<ModelAPI_Document>& theDocument, QObject* theParent);
+Q_OBJECT
+ public:
+  XGUI_TopDataModel(QObject* theParent);
   virtual ~XGUI_TopDataModel();
 
-  // Reimplementation from QAbstractItemModel
+  // Reimpl from QAbstractItemModel
   virtual QVariant data(const QModelIndex& theIndex, int theRole) const;
   virtual QVariant headerData(int section, Qt::Orientation orientation,
                               int role = Qt::DisplayRole) const;
@@ -25,35 +26,41 @@ public:
   virtual int rowCount(const QModelIndex &parent = QModelIndex()) const;
   virtual int columnCount(const QModelIndex &parent = QModelIndex()) const;
 
-  virtual QModelIndex index(int theRow, int theColumn, 
-                            const QModelIndex& theParent = QModelIndex()) const;
+  virtual QModelIndex index(int theRow, int theColumn, const QModelIndex& theParent =
+                                QModelIndex()) const;
 
   virtual QModelIndex parent(const QModelIndex& theIndex) const;
 
   virtual bool hasChildren(const QModelIndex& theParent = QModelIndex()) const;
 
-  //! Returns Feature object by the given Model index.
-  //! Returns 0 if the given index is not index of a feature
-  virtual FeaturePtr feature(const QModelIndex& theIndex) const;
+  //! Returns object by the given Model index.
+  //! Returns 0 if the given index is not index of a object
+  virtual ObjectPtr object(const QModelIndex& theIndex) const;
 
-  //! Returns parent index of the given feature
-  virtual QModelIndex findParent(const std::shared_ptr<ModelAPI_Feature>& theFeature) const;
+  //! Returns QModelIndex which corresponds to the given object
+  //! If the object is not found then index is not valid
+  virtual QModelIndex objectIndex(const ObjectPtr& theObject) const;
+
+  //! Returns parent index of the given object
+  virtual QModelIndex findParent(const ObjectPtr& theObject) const;
 
   //! Returns index corresponded to the group
   virtual QModelIndex findGroup(const std::string& theGroup) const;
 
-private:
+ private:
   //! Types of QModelIndexes
-  enum DataIds {
+  enum DataIds
+  {
     ParamsFolder,
     ParamObject,
     ConstructFolder,
     ConstructObject
+    //GroupsFolder,
+    //GroupObject
   };
 
 };
 
-
 /**\class XGUI_PartDataModel
  * \ingroup GUI
  * \brief This is a data model for Object Browser (QTreeView).
@@ -61,12 +68,12 @@ private:
  */
 class XGUI_PartDataModel : public XGUI_PartModel
 {
-  Q_OBJECT
-public:
-  XGUI_PartDataModel(const std::shared_ptr<ModelAPI_Document>& theDocument, QObject* theParent);
+Q_OBJECT
+ public:
+  XGUI_PartDataModel(QObject* theParent);
   virtual ~XGUI_PartDataModel();
 
-  // Reimplementation from QAbstractItemModel
+  // Reimpl from QAbstractItemModel
   virtual QVariant data(const QModelIndex& theIndex, int theRole) const;
   virtual QVariant headerData(int section, Qt::Orientation orientation,
                               int role = Qt::DisplayRole) const;
@@ -74,38 +81,55 @@ public:
   virtual int rowCount(const QModelIndex &parent = QModelIndex()) const;
   virtual int columnCount(const QModelIndex &parent = QModelIndex()) const;
 
-  virtual QModelIndex index(int theRow, int theColumn, 
-                            const QModelIndex& theParent = QModelIndex()) const;
+  virtual QModelIndex index(int theRow, int theColumn, const QModelIndex& theParent =
+                                QModelIndex()) const;
 
   virtual QModelIndex parent(const QModelIndex& theIndex) const;
 
   virtual bool hasChildren(const QModelIndex& theParent = QModelIndex()) const;
 
-  //! Returns Feature object by the given Model index.
-  //! Returns 0 if the given index is not index of a feature
-  virtual FeaturePtr feature(const QModelIndex& theIndex) const;
+  //! Returns object by the given Model index.
+  //! Returns 0 if the given index is not index of a object
+  virtual ObjectPtr object(const QModelIndex& theIndex) const;
+
+  //! Returns QModelIndex which corresponds to the given object
+  //! If the object is not found then index is not valid
+  virtual QModelIndex objectIndex(const ObjectPtr& theObject) const;
 
   //! Returns true if the given document is a sub-document of this tree
-  virtual bool hasDocument(const std::shared_ptr<ModelAPI_Document>& theDoc) const;
+  virtual bool hasDocument(const DocumentPtr& theDoc) const;
 
-  //! Returns parent index of the given feature
-  virtual QModelIndex findParent(const std::shared_ptr<ModelAPI_Feature>& theFeature) const;
+  //! Returns parent index of the given object
+  virtual QModelIndex findParent(const ObjectPtr& theObject) const;
 
   //! Returns index corresponded to the group
   virtual QModelIndex findGroup(const std::string& theGroup) const;
 
-private: 
-  std::shared_ptr<ModelAPI_Document> featureDocument() const;
+  //! Return a Part object
+  virtual ResultPartPtr part() const;
+
+ private:
+
+  //! Returns document of the current part
+  DocumentPtr partDocument() const;
+
+  int getRowsNumber() const;
 
   //! Types of QModelIndexes
-  enum DataIds {
+  enum DataIds
+  {
     MyRoot,
     ParamsFolder,
     ParamObject,
     ConstructFolder,
-    ConstructObject
+    ConstructObject,
+    BodiesFolder,
+    BodiesObject,
+    GroupsFolder,
+    GroupObject,
+    HistoryObject
   };
 
 };
 
-#endif
\ No newline at end of file
+#endif