Apply warning icon to folder only in case if it is not expanded

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

diff --git a/src/XGUI/XGUI_ObjectsBrowser.h b/src/XGUI/XGUI_ObjectsBrowser.h
index fd522c61a8dd253aaafc940fec2aadac2b942018..e2a4579404b2eaa8217652c051bf1d578a22e85a 100644 (file)
--- a/src/XGUI/XGUI_ObjectsBrowser.h
+++ b/src/XGUI/XGUI_ObjectsBrowser.h
@@ -26,15 +26,18 @@
 #include <ModelAPI_Object.h>
 #include <ModelAPI_ResultPart.h>
 #include <ModelAPI_Events.h>
+#include <ModelAPI_Folder.h>
 
 #include <QWidget>
 #include <QTreeView>
 #include <QLabel>
+#include <QMap>
 
 class ModuleBase_IDocumentDataModel;
 class XGUI_DataModel;
 class Config_DataModelReader;
 class XGUI_Workshop;
+class ModuleBase_ITreeNode;
 
 //#define DEBUG_INDXES
 
@@ -87,7 +90,7 @@ private:
 * \ingroup GUI
 * Implementation of Data Tree object for Object Browser
 */
-class XGUI_DataTree : public QTreeView
+class XGUI_EXPORT XGUI_DataTree : public QTreeView
 {
 Q_OBJECT
  public:
@@ -166,10 +169,17 @@ Q_OBJECT
   /// \param theObjects list of objects to select
   void setObjectsSelected(const QObjectPtrList& theObjects);
 
+  //! Scroll TreeView to make given object visible
+  //! \param theObject object to make it visible
+  void ensureVisible(const ObjectPtr theObject);
+
   //! Returns currently selected indexes
   QModelIndexList selectedIndexes() const
   {
-    return myTreeView->selectionModel()->selectedIndexes();
+    if (myTreeView->selectionModel())
+      return myTreeView->selectionModel()->selectedIndexes();
+    else
+      return QModelIndexList();
   }
 
   //! Returns TreeView widget
@@ -187,9 +197,8 @@ Q_OBJECT
   /// Resets the object browser into initial state
   void clearContent();
 
-  /// Set XML reader object for data model
-  /// \param theReader the reader object
-  void setXMLReader(Config_DataModelReader* theReader);
+  /// Initialize the Object browser
+  void initialize(ModuleBase_ITreeNode* theRoot);
 
   /// Returns list of folders opened state for the given document
   /// \param theDoc the document
@@ -206,6 +215,14 @@ Q_OBJECT
 
   void onSelectionChanged();
 
+  /// Updates all items of object browser
+  /// \param theColumn - column of items
+  /// \param theParent - a parent item (by default from root)
+  void updateAllIndexes(int theColumn = 0, const QModelIndex& theParent = QModelIndex());
+
+  QMap<ObjectPtr, bool> getFoldersState(DocumentPtr theDoc) const;
+
+  void setFoldersState(const QMap<ObjectPtr, bool>& theStates);
 
 public slots:
   //! Called on Edit command request