Salome HOME
Fix for the issue #2456 : Fatal error when remove part
[modules/shaper.git] / src / XGUI / XGUI_ObjectsBrowser.h
1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
19 //
20
21 #ifndef XGUI_ObjectsBrowser_H
22 #define XGUI_ObjectsBrowser_H
23
24 #include "XGUI.h"
25 #include <ModuleBase_Definitions.h>
26 #include <ModelAPI_Object.h>
27 #include <ModelAPI_ResultPart.h>
28 #include <ModelAPI_Events.h>
29 #include <ModelAPI_Folder.h>
30
31 #include <QWidget>
32 #include <QTreeView>
33 #include <QLabel>
34 #include <QMap>
35
36 class ModuleBase_IDocumentDataModel;
37 class XGUI_DataModel;
38 class Config_DataModelReader;
39 class XGUI_Workshop;
40
41 //#define DEBUG_INDXES
42
43 /**
44 * \ingroup GUI
45 * Implementation of root label in Object Browser
46 */
47 class XGUI_ActiveDocLbl: public QLabel
48 {
49 Q_OBJECT
50  public:
51    /// Constructor
52    /// \param theText a text
53    /// \param theParent a parent widget
54    XGUI_ActiveDocLbl(const QString& theText, QWidget* theParent );
55
56    /// Sets tree view
57    /// \param theView a view
58    void setTreeView(QTreeView* theView);
59
60    /// Returns tree view
61    QTreeView* treePalette() const { return myTreeView;}
62
63 #if (!defined HAVE_SALOME) && (defined WIN32)
64    virtual bool event(QEvent* theEvent);
65 #endif
66
67 public slots:
68   /// On unselect
69   void unselect();
70
71 protected:
72   /// On mouse release
73   virtual void mouseReleaseEvent( QMouseEvent* e);
74
75   /// Filter event
76   bool eventFilter(QObject* theObj, QEvent* theEvent);
77
78 private:
79   QString myPreSelectionStyle;
80   QString myNeutralStyle;
81   QString mySelectionStyle;
82
83   QTreeView* myTreeView;
84   bool myIsSelected;
85 };
86
87
88 /**
89 * \ingroup GUI
90 * Implementation of Data Tree object for Object Browser
91 */
92 class XGUI_DataTree : public QTreeView
93 {
94 Q_OBJECT
95  public:
96    /// Constructor
97    /// \param theParent a parent widget
98   XGUI_DataTree(QWidget* theParent);
99
100   virtual ~XGUI_DataTree();
101
102   /// Returns current data model
103   XGUI_DataModel* dataModel() const;
104
105 signals:
106   //! Emited on context menu request
107   void contextMenuRequested(QContextMenuEvent* theEvent);
108
109 public slots:
110   /// Clear content of data tree
111   virtual void clear();
112
113  protected slots:
114   /// Commit modified data (used for renaming of objects)
115   virtual void commitData(QWidget* theEditor);
116
117   /// A slot which is called on mouse double click
118   void onDoubleClick(const QModelIndex&);
119
120  protected:
121    /// Redefinition of virtual method
122   virtual void contextMenuEvent(QContextMenuEvent* theEvent);
123
124    /// Redefinition of virtual method
125   virtual void resizeEvent(QResizeEvent* theEvent);
126
127    /// Redefinition of virtual method
128   virtual void mouseReleaseEvent(QMouseEvent* theEvent);
129
130 #ifdef DEBUG_INDXES
131   virtual void mousePressEvent(QMouseEvent* theEvent);
132 #endif
133
134 private:
135   /// Process a history change request
136   /// \param theIndex a clicked data index
137   void processHistoryChange(const QModelIndex& theIndex);
138
139   /// Process a visibility change request
140   /// \param theIndex a clicked data index
141   void processEyeClick(const QModelIndex& theIndex);
142 };
143
144 /**\class XGUI_ObjectsBrowser
145  * \ingroup GUI
146  * \brief Object browser window object. Represents data tree of current data structure
147  */
148 class XGUI_EXPORT XGUI_ObjectsBrowser : public QWidget
149 {
150 Q_OBJECT
151  public:
152    /// Constructor
153    /// \param theParent a parent widget
154   XGUI_ObjectsBrowser(QWidget* theParent, XGUI_Workshop* theWorkshop);
155   virtual ~XGUI_ObjectsBrowser();
156
157   //! Returns Model which provides access to data objects
158   XGUI_DataModel* dataModel() const
159   {
160     return myDocModel;
161   }
162
163   //! Returns list of currently selected objects
164   //! \param theIndexes - output list of corresponded indexes (can be NULL)
165   QObjectPtrList selectedObjects(QModelIndexList* theIndexes = 0) const;
166
167   /// Set selected list of objects
168   /// \param theObjects list of objects to select
169   void setObjectsSelected(const QObjectPtrList& theObjects);
170
171   //! Returns currently selected indexes
172   QModelIndexList selectedIndexes() const
173   {
174     return myTreeView->selectionModel()->selectedIndexes();
175   }
176
177   //! Returns TreeView widget
178   XGUI_DataTree* treeView() const
179   {
180     return myTreeView;
181   }
182
183   /// Returns active doc label object
184   QLabel* activeDocLabel() const { return myActiveDocLbl; }
185
186   /// Rebuild data tree
187   void rebuildDataTree();
188
189   /// Resets the object browser into initial state
190   void clearContent();
191
192   /// Set XML reader object for data model
193   /// \param theReader the reader object
194   void setXMLReader(Config_DataModelReader* theReader);
195
196   /// Returns list of folders opened state for the given document
197   /// \param theDoc the document
198   /// \return list of booleans with state expanded or not
199   std::list<bool> getStateForDoc(DocumentPtr theDoc) const;
200
201   /// Set folders opened state for the given document
202   /// \param theDoc the document
203   /// \param theStates list of booleans with state expanded or not
204   void setStateForDoc(DocumentPtr theDoc, const std::list<bool>& theStates);
205
206   /// Returns current workshop
207   XGUI_Workshop* workshop() const { return myWorkshop; }
208
209   void onSelectionChanged();
210
211   /// Updates all items of object browser
212   /// \param theColumn - column of items
213   /// \param theParent - a parent item (by default from root)
214   void updateAllIndexes(int theColumn = 0, const QModelIndex& theParent = QModelIndex());
215
216   QMap<ObjectPtr, bool> getFoldersState(DocumentPtr theDoc) const;
217
218   void setFoldersState(const QMap<ObjectPtr, bool>& theStates);
219
220 public slots:
221   //! Called on Edit command request
222   void onEditItem();
223
224 signals:
225   //! Emited when selection is changed
226   void selectionChanged();
227
228   //! Emited on context menu request
229   void contextMenuRequested(QContextMenuEvent* theEvent);
230
231   //! Segnal is emitted when user cliks by mouse in header label of object browser
232   void headerMouseDblClicked(const QModelIndex&);
233
234  private slots:
235   /// Show context menu
236   /// \param theEvent a context menu event
237   void onContextMenuRequested(QContextMenuEvent* theEvent);
238
239   /// Show context menu on upper label
240   /// \param thePnt a position of context menu
241   void onLabelContextMenuRequested(const QPoint& thePnt);
242
243   //! Called when selection in Data Tree is changed
244   void onSelectionChanged(const QItemSelection& theSelected, const QItemSelection& theDeselected);
245
246   void onBeforeReset();
247
248   void onAfterModelReset();
249
250  private:
251   QModelIndexList expandedItems(const QModelIndex& theParent = QModelIndex()) const;
252
253   //! Internal model
254   XGUI_DataModel* myDocModel;
255   XGUI_ActiveDocLbl* myActiveDocLbl;
256   XGUI_DataTree* myTreeView;
257   XGUI_Workshop* myWorkshop;
258
259   /// A field to store expanded items before model reset
260   QModelIndexList myExpandedItems;
261 };
262
263 #endif