Issue #387 #388 Fix for the crash in the swig generated code on export

[modules/shaper.git] / src / XGUI / XGUI_PartDataModel.h

diff --git a/src/XGUI/XGUI_PartDataModel.h b/src/XGUI/XGUI_PartDataModel.h
index b1f808995a545a200dd94796eb67be3853dca6e1..4ef9cd58fe5e77611c73fb0f32d3fd04994183b1 100644 (file)
--- 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,8 +13,8 @@
  */
 class XGUI_EXPORT XGUI_TopDataModel : public XGUI_FeaturesModel
 {
-  Q_OBJECT
-public:
+Q_OBJECT
+ public:
   XGUI_TopDataModel(QObject* theParent);
   virtual ~XGUI_TopDataModel();
 
@@ -25,8 +26,8 @@ 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;
 
@@ -46,18 +47,20 @@ public:
   //! 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).
@@ -65,8 +68,8 @@ private:
  */
 class XGUI_PartDataModel : public XGUI_PartModel
 {
-  Q_OBJECT
-public:
+Q_OBJECT
+ public:
   XGUI_PartDataModel(QObject* theParent);
   virtual ~XGUI_PartDataModel();
 
@@ -78,8 +81,8 @@ 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;
 
@@ -103,15 +106,18 @@ public:
   virtual QModelIndex findGroup(const std::string& theGroup) const;
 
   //! Return a Part object
-  virtual ObjectPtr part() const;
+  virtual ResultPartPtr part() const;
 
-private: 
+ private:
 
   //! Returns document of the current part
   DocumentPtr partDocument() const;
 
+  int getRowsNumber() const;
+
   //! Types of QModelIndexes
-  enum DataIds {
+  enum DataIds
+  {
     MyRoot,
     ParamsFolder,
     ParamObject,
@@ -119,9 +125,11 @@ private:
     ConstructObject,
     BodiesFolder,
     BodiesObject,
+    GroupsFolder,
+    GroupObject,
     HistoryObject
   };
 
 };
 
-#endif
\ No newline at end of file
+#endif