X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROGUI%2FHYDROGUI_PriorityTableModel.h;h=9133843769deec66b81c69263f3be4624c7aaed7;hb=e37037b0cca2ae883a605ddb242e36045e280e32;hp=28b058db7993addbe4f636131d6d3a3fd170f2fd;hpb=439579ec24edd8b147cab07f688d446d59029a1e;p=modules%2Fhydro.git

diff --git a/src/HYDROGUI/HYDROGUI_PriorityTableModel.h b/src/HYDROGUI/HYDROGUI_PriorityTableModel.h
index 28b058db..91338437 100644
--- a/src/HYDROGUI/HYDROGUI_PriorityTableModel.h
+++ b/src/HYDROGUI/HYDROGUI_PriorityTableModel.h
@@ -58,19 +58,21 @@ public:
   bool removeRows( const QList<int> theRows );
   bool removeAll();
 
-  void setObjects( const QList<Handle(HYDROData_Object)>& theObjects );
+  void setObjects( const QList<Handle(HYDROData_Entity)>& theObjects );
 
   void setRules( const HYDROData_ListOfRules& theObjects );
   HYDROData_ListOfRules getRules() const;
 
   bool createNewRule();
   bool canCreateNewRule() const;
+
+  void setColumnCount( int theColumnCount );
   
 protected:
-  bool isUsed( const Handle(HYDROData_Object)& theObj1, 
-               const Handle(HYDROData_Object)& theObj2 ) const;
+  bool isUsed( const Handle(HYDROData_Entity)& theObj1, 
+               const Handle(HYDROData_Entity)& theObj2 ) const;
 
-  QStringList getAvailablePairs( const Handle(HYDROData_Object)& theObject ) const;
+  QStringList getAvailablePairs( const Handle(HYDROData_Entity)& theObject ) const;
   QStringList getAvailableObjectNames() const;
 
   QString priorityToString( const int thePriority ) const;
@@ -80,12 +82,16 @@ signals:
   void showError( const QString& theMsg );
 
 private:
+  const ObjectKind getObjectsKind() const;
+
   friend class test_HYDROGUI_PriorityTableModel;
 
   HYDROGUI_Module* myModule;
 
   HYDROData_ListOfRules myRules;
-  QList<Handle(HYDROData_Object)> myObjects;
+  QList<Handle(HYDROData_Entity)> myObjects;
+
+  int myColumnCount;
 };
 
 #endif
\ No newline at end of file