]> SALOME platform Git repositories - modules/shaper.git/blob - src/XGUI/XGUI_ObjectsBrowser.cpp
Salome HOME
738b11aeebf15711a38975039858a499152ab753
[modules/shaper.git] / src / XGUI / XGUI_ObjectsBrowser.cpp
1 // Copyright (C) 2014-2019  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 email : webmaster.salome@opencascade.com
18 //
19
20 #include "XGUI_ObjectsBrowser.h"
21 #include "XGUI_Tools.h"
22 #include "XGUI_DataModel.h"
23
24 #include <ModelAPI_Data.h>
25 #include <ModelAPI_Session.h>
26 #include <ModelAPI_Document.h>
27 #include <ModelAPI_Tools.h>
28 #include <ModelAPI_ResultField.h>
29
30 #include <ModuleBase_Tools.h>
31 #include <ModuleBase_ITreeNode.h>
32
33 #include <XGUI_Workshop.h>
34 #include <XGUI_Displayer.h>
35
36 #include <QLayout>
37 #include <QLineEdit>
38 #include <QPixmap>
39 #include <QEvent>
40 #include <QMouseEvent>
41 #include <QAction>
42 #include <QStyledItemDelegate>
43 #include <QMessageBox>
44 #include <QApplication>
45
46 #ifdef DEBUG_INDXES
47 #include <QToolTip>
48 #endif
49
50 /// Width of second column (minimum acceptable = 27)
51 #define FIRST_COL_WIDTH 20
52 #define SECOND_COL_WIDTH 30
53
54 /**
55 * \ingroup GUI
56 * Tree item delegate for definition of data in column items editor
57 */
58 class XGUI_TreeViewItemDelegate: public QStyledItemDelegate
59 {
60 public:
61   /// Constructor
62   /// \param theParent a parent of the delegate
63   XGUI_TreeViewItemDelegate(XGUI_DataTree* theParent):QStyledItemDelegate(theParent),
64     myTreedView(theParent) {}
65
66   /// Set data for item editor (name of the item)
67   /// \param editor a widget of editor
68   /// \param index the tree item index
69   virtual void  setEditorData ( QWidget* editor, const QModelIndex& index ) const
70   {
71     QLineEdit* aEditor = dynamic_cast<QLineEdit*>(editor);
72     if (aEditor) {
73       XGUI_DataModel* aModel = myTreedView->dataModel();
74       ObjectPtr aObj = aModel->object(index);
75       if (aObj.get() != NULL) {
76         aEditor->setText(QString::fromStdWString(aObj->data()->name()));
77         return;
78       }
79     }
80     QStyledItemDelegate::setEditorData(editor, index);
81   }
82
83 private:
84   XGUI_DataTree* myTreedView;
85 };
86
87
88 XGUI_DataTree::XGUI_DataTree(QWidget* theParent)
89     : QTreeView(theParent)
90 {
91   setHeaderHidden(true);
92   setTreePosition(1);
93   setEditTriggers(QAbstractItemView::NoEditTriggers);
94   setSelectionBehavior(QAbstractItemView::SelectRows);
95   setSelectionMode(QAbstractItemView::ExtendedSelection);
96
97   setItemDelegateForColumn(1, new XGUI_TreeViewItemDelegate(this));
98
99   connect(this, SIGNAL(doubleClicked(const QModelIndex&)),
100     SLOT(onDoubleClick(const QModelIndex&)));
101 }
102
103 XGUI_DataTree::~XGUI_DataTree()
104 {
105 }
106
107 XGUI_DataModel* XGUI_DataTree::dataModel() const
108 {
109   return static_cast<XGUI_DataModel*>(model());
110 }
111
112 void XGUI_DataTree::contextMenuEvent(QContextMenuEvent* theEvent)
113 {
114   emit contextMenuRequested(theEvent);
115 }
116
117 void XGUI_DataTree::commitData(QWidget* theEditor)
118 {
119   static int aEntrance = 0;
120   if (aEntrance == 0) {
121     // We have to check number of enter and exit of this function because it can be called
122     // recursively by Qt in order to avoid double modifying of a data
123     aEntrance = 1;
124     QLineEdit* aEditor = dynamic_cast<QLineEdit*>(theEditor);
125     if (aEditor) {
126       QString aName = aEditor->text();
127       QModelIndexList aIndexList = selectionModel()->selectedIndexes();
128       XGUI_DataModel* aModel = dataModel();
129       ObjectPtr aObj = aModel->object(aIndexList.first());
130
131       if (XGUI_Tools::canRename(aObj, aName)) {
132         SessionPtr aMgr = ModelAPI_Session::get();
133         aMgr->startOperation("Rename");
134         std::wstring aaa = aName.toStdWString();
135         aObj->data()->setName(aName.toStdWString());
136         aMgr->finishOperation();
137       }
138     }
139   }
140   aEntrance = 0;
141 }
142
143 void XGUI_DataTree::clear()
144 {
145   dataModel()->clear();
146   reset();
147 }
148
149 void XGUI_DataTree::resizeEvent(QResizeEvent* theEvent)
150 {
151   QTreeView::resizeEvent(theEvent);
152   QSize aSize = theEvent->size();
153   if (aSize.isValid()) {
154     setColumnWidth(0, FIRST_COL_WIDTH);
155     setColumnWidth(1, aSize.width() - SECOND_COL_WIDTH - FIRST_COL_WIDTH - 10);
156     setColumnWidth(2, SECOND_COL_WIDTH);
157   }
158 }
159
160 #ifdef DEBUG_INDXES
161 void XGUI_DataTree::mousePressEvent(QMouseEvent* theEvent)
162 {
163   QTreeView::mousePressEvent(theEvent);
164   if (theEvent->button() != Qt::MidButton)
165     return;
166   QModelIndex aInd = indexAt(theEvent->pos());
167   QString aTxt =
168     QString("r=%1 c=%2 p=%3").arg(aInd.row()).arg(aInd.column()).arg((long)aInd.internalPointer());
169
170   QModelIndex aPar = aInd.parent();
171   QString aTxt1 =
172     QString("r=%1 c=%2 p=%3").arg(aPar.row()).arg(aPar.column()).arg((long)aPar.internalPointer());
173   QToolTip::showText(theEvent->globalPos(), aTxt + '\n' + aTxt1);
174 }
175 #endif
176
177 void XGUI_DataTree::mouseReleaseEvent(QMouseEvent* theEvent)
178 {
179   QTreeView::mouseReleaseEvent(theEvent);
180 #ifdef DEBUG_INDXES
181   if (theEvent->button() != Qt::MidButton)
182     return;
183   QToolTip::hideText();
184 #endif
185   if (theEvent->button() == Qt::LeftButton) {
186     QModelIndex aInd = indexAt(theEvent->pos());
187     if (aInd.column() == 0)
188       processEyeClick(aInd);
189   }
190 }
191
192 void XGUI_DataTree::processHistoryChange(const QModelIndex& theIndex)
193 {
194   SessionPtr aMgr = ModelAPI_Session::get();
195   // When operation is opened then we can not change history
196   if (aMgr->isOperation())
197     return;
198   XGUI_DataModel* aModel = dataModel();
199   if (aModel->flags(theIndex) == 0)
200     return;
201   ObjectPtr aObj = aModel->object(theIndex);
202
203   DocumentPtr aDoc = aMgr->activeDocument();
204
205   std::string aOpName = tr("History change").toStdString();
206   if (aObj.get()) {
207     if (aObj->document() != aDoc)
208       return;
209     aMgr->startOperation(aOpName);
210     aDoc->setCurrentFeature(std::dynamic_pointer_cast<ModelAPI_Feature>(aObj), true);
211     aMgr->finishOperation();
212   } else {
213     // Ignore clicks on folders outside current document
214     if ((theIndex.internalId() == 0) && (aDoc != aMgr->moduleDocument()))
215       // Clicked folder under root but active document is another
216       return;
217     if ((theIndex.internalId() != 0) && (aDoc != aModel->document(theIndex)))
218       // Cliced not on active document folder
219       return;
220
221     aMgr->startOperation(aOpName);
222     aDoc->setCurrentFeature(FeaturePtr(), true);
223     aMgr->finishOperation();
224   }
225   QModelIndex aParent = theIndex.parent();
226   int aSize = aModel->rowCount(aParent);
227   for (int i = 0; i < aSize; i++) {
228     update(aModel->index(i, 0, aParent));
229     update(aModel->index(i, 1, aParent));
230     update(aModel->index(i, 2, aParent));
231   }
232   XGUI_ObjectsBrowser* aObjBrowser = qobject_cast<XGUI_ObjectsBrowser*>(parent());
233   aObjBrowser->workshop()->displayer()->updateViewer();
234 }
235
236 void XGUI_DataTree::processEyeClick(const QModelIndex& theIndex)
237 {
238   XGUI_ObjectsBrowser* aObjBrowser = qobject_cast<XGUI_ObjectsBrowser*>(parent());
239   XGUI_DataModel* aModel = dataModel();
240   ObjectPtr aObj = aModel->object(theIndex);
241   if (aObj.get()) {
242     std::set<ObjectPtr> anObjects;
243     anObjects.insert(aObj);
244
245     bool hasHiddenState = aModel->hasHiddenState(theIndex);
246     if (aObjBrowser && hasHiddenState && !aObjBrowser->workshop()->prepareForDisplay(anObjects))
247       return;
248     if (hasHiddenState) { // #issue 2335(hide all faces then show solid problem)
249       if (aObj->isDisplayed())
250         aObj->setDisplayed(false);
251       aObj->setDisplayed(true);
252     }
253     else
254       aObj->setDisplayed(!aObj->isDisplayed());
255
256     // Update list of selected objects because this event happens after
257     // selection event in object browser
258     Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
259     update(theIndex);
260     if (aObjBrowser) {
261       aObjBrowser->onSelectionChanged();
262     }
263   }
264 }
265
266 void XGUI_DataTree::onDoubleClick(const QModelIndex& theIndex)
267 {
268   switch (theIndex.column()) {
269   case 2:
270     processHistoryChange(theIndex);
271     break;
272   }
273 }
274
275
276 //********************************************************************
277 //********************************************************************
278 //********************************************************************
279 XGUI_ActiveDocLbl::XGUI_ActiveDocLbl(const QString& theText, QWidget* theParent )
280   : QLabel(theText, theParent),
281   myPreSelectionStyle(""),
282   myNeutralStyle(""),
283   mySelectionStyle(""),
284   myIsSelected(false)
285 {
286 }
287
288 void XGUI_ActiveDocLbl::setTreeView(QTreeView* theView)
289 {
290   myTreeView = theView;
291   setFont(myTreeView->font());
292
293   QPalette aPalet = myTreeView->palette();
294   QColor aHighlight = aPalet.highlight().color();
295   QColor aHighlightText = aPalet.highlightedText().color();
296
297   myPreSelectionStyle = "QLabel {background-color: ";
298   myPreSelectionStyle += aHighlight.lighter(170).name() + "}";
299   //myPreSelectionStyle += "qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 white, stop:1 " +
300   //  aHighlight.lighter(170).name() + ");";
301   //myPreSelectionStyle += "border: 1px solid lightblue; border-radius: 2px }";
302
303   QString aName = aPalet.color(QPalette::Base).name();
304   myNeutralStyle = "QLabel { border: 1px solid " + aName + " }";
305
306
307 #if (!defined HAVE_SALOME) && (defined WIN32)
308   mySelectionStyle = "QLabel {background-color: ";
309   mySelectionStyle += "rgb(205, 232, 255); ";
310   //mySelectionStyle += "qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 rgb(236, 245, 255)";
311   //mySelectionStyle += ", stop:1 rgb(208, 229, 255));";
312   //mySelectionStyle += "border: 1px solid rgb(132, 172, 221); border-radius: 2px }";
313   mySelectionStyle += "border: 1px solid rgb(153, 209, 255) }";
314 #else
315   mySelectionStyle = "QLabel {background-color: " + aHighlight.name();
316   mySelectionStyle += "; color : " + aHighlightText.name() + "}";
317 #endif
318
319   myTreeView->viewport()->installEventFilter(this);
320 }
321
322
323 #if (!defined HAVE_SALOME) && (defined WIN32)
324 bool XGUI_ActiveDocLbl::event(QEvent* theEvent)
325 {
326   switch (theEvent->type()) {
327     case QEvent::Enter:
328       if (!myIsSelected)
329         setStyleSheet(myPreSelectionStyle);
330       break;
331     case QEvent::Leave:
332       if (!myIsSelected)
333         setStyleSheet(myNeutralStyle);
334       break;
335   }
336   return QLabel::event(theEvent);
337 }
338 #endif
339
340 bool XGUI_ActiveDocLbl::eventFilter(QObject* theObj, QEvent* theEvent)
341 {
342   if (theObj == myTreeView->viewport()) {
343     if (theEvent->type() == QEvent::MouseButtonRelease)
344       unselect();
345   }
346   return QLabel::eventFilter(theObj, theEvent);
347 }
348
349 static bool MYClearing = false;
350 void XGUI_ActiveDocLbl::mouseReleaseEvent( QMouseEvent* e)
351 {
352   MYClearing = true;
353   myIsSelected = true;
354   setStyleSheet(mySelectionStyle);
355   // We can not block signals because on
356   // clear selection the View state will not be updated
357   myTreeView->clearSelection();
358   QLabel::mouseReleaseEvent(e);
359   MYClearing = false;
360 }
361
362 void XGUI_ActiveDocLbl::unselect()
363 {
364   if (!MYClearing) {
365     myIsSelected = false;
366     setStyleSheet(myNeutralStyle);
367   }
368 }
369
370
371 //********************************************************************
372 //********************************************************************
373 //********************************************************************
374 XGUI_ObjectsBrowser::XGUI_ObjectsBrowser(QWidget* theParent, XGUI_Workshop* theWorkshop)
375     : QWidget(theParent), myDocModel(0), myWorkshop(theWorkshop)
376 {
377   QVBoxLayout* aLayout = new QVBoxLayout(this);
378   ModuleBase_Tools::zeroMargins(aLayout);
379   aLayout->setSpacing(0);
380
381   QWidget* aLabelWgt = new QWidget(this);
382   aLabelWgt->setAutoFillBackground(true);
383
384   aLayout->addWidget(aLabelWgt);
385   QHBoxLayout* aLabelLay = new QHBoxLayout(aLabelWgt);
386   ModuleBase_Tools::zeroMargins(aLabelLay);
387   aLabelLay->setSpacing(0);
388
389   QLabel* aLbl = new QLabel(aLabelWgt);
390   aLbl->setPixmap(QPixmap(":pictures/assembly.png"));
391   aLbl->setMargin(2);
392   // Do not paint background of the label (in order to show icon)
393   aLbl->setAutoFillBackground(false);
394
395   aLabelLay->addWidget(aLbl);
396
397   SessionPtr aMgr = ModelAPI_Session::get();
398   DocumentPtr aDoc = aMgr->moduleDocument();
399
400   myActiveDocLbl = new XGUI_ActiveDocLbl(tr("Part set"), aLabelWgt);
401 //  myActiveDocLbl->setReadOnly(true);
402 //  myActiveDocLbl->setFrame(false);
403   myActiveDocLbl->setContextMenuPolicy(Qt::CustomContextMenu);
404
405   aLabelLay->addWidget(myActiveDocLbl);
406   aLabelLay->setStretch(1, 1);
407
408   myTreeView = new XGUI_DataTree(this);
409   myTreeView->setFrameShape(QFrame::NoFrame);
410   aLayout->addWidget(myTreeView);
411
412   QPalette aTreePalet = myTreeView->palette();
413   QColor aTreeBack = aTreePalet.color(QPalette::Base);
414
415   QPalette aPalet;
416   aPalet.setColor(QPalette::Base, aTreeBack);
417   aPalet.setColor(QPalette::Window, aTreeBack);
418   aLabelWgt->setPalette(aPalet);
419
420   myDocModel = new XGUI_DataModel(this);
421   connect(myDocModel, SIGNAL(beforeTreeRebuild()), SLOT(onBeforeReset()));
422   connect(myDocModel, SIGNAL(treeRebuilt()), SLOT(onAfterModelReset()));
423
424   connect(myTreeView, SIGNAL(contextMenuRequested(QContextMenuEvent*)), this,
425           SLOT(onContextMenuRequested(QContextMenuEvent*)));
426 }
427
428 //***************************************************
429 XGUI_ObjectsBrowser::~XGUI_ObjectsBrowser()
430 {
431 }
432
433 void XGUI_ObjectsBrowser::initialize(ModuleBase_ITreeNode* theRoot)
434 {
435   //myDocModel->setXMLReader(theReader);
436   myDocModel->setRoot(theRoot);
437   myTreeView->setModel(myDocModel);
438
439   // It has to be done after setting of model
440   myActiveDocLbl->setTreeView(myTreeView);
441
442   QItemSelectionModel* aSelMod = myTreeView->selectionModel();
443   connect(aSelMod, SIGNAL(selectionChanged(const QItemSelection&, const QItemSelection&)),
444           this, SLOT(onSelectionChanged(const QItemSelection&, const QItemSelection&)));
445 }
446
447 //***************************************************
448 void XGUI_ObjectsBrowser::onContextMenuRequested(QContextMenuEvent* theEvent)
449 {
450   QModelIndexList aIndexes;
451   QObjectPtrList aSelectedData = selectedObjects(&aIndexes);
452   bool toEnable = false;
453
454   if (aSelectedData.size() == 1) {
455     QModelIndex aSelected = myTreeView->indexAt(theEvent->pos());
456     if (!aIndexes.contains(aSelected))
457       return; // menu is called on non selected item
458
459     Qt::ItemFlags aFlags = dataModel()->flags(aIndexes.first());
460     toEnable = ((aFlags & Qt::ItemIsEditable) != 0);
461   }
462   foreach(QAction* aCmd, actions()) {
463     aCmd->setEnabled(toEnable);
464   }
465   emit contextMenuRequested(theEvent);
466 }
467
468 //***************************************************
469 void XGUI_ObjectsBrowser::onLabelContextMenuRequested(const QPoint& thePnt)
470 {
471   myTreeView->selectionModel()->clearSelection();
472   //Empty feature pointer means that selected root document
473   foreach(QAction* aCmd, actions()) {
474     aCmd->setEnabled(true);
475   }
476   QContextMenuEvent aEvent(QContextMenuEvent::Mouse, thePnt, myActiveDocLbl->mapToGlobal(thePnt));
477   emit contextMenuRequested(&aEvent);
478 }
479
480 //***************************************************
481 void XGUI_ObjectsBrowser::onEditItem()
482 {
483   QObjectPtrList aSelectedData = selectedObjects();
484   if (aSelectedData.size() > 0) {
485     ObjectPtr anObject = aSelectedData.first();
486     if (anObject.get()) {  // Selection happens in TreeView
487       // check whether the object can be renamed. There should not be parts which are not loaded
488       std::set<FeaturePtr> aFeatures;
489       aFeatures.insert(ModelAPI_Feature::feature(anObject));
490       if (!XGUI_Tools::canRemoveOrRename((QWidget*)parent(), aFeatures))
491         return;
492
493       // Find index which corresponds the feature
494       QModelIndex aIndex;
495       foreach(QModelIndex aIdx, selectedIndexes()) {
496         if (aIdx.column() == 1) {
497           ObjectPtr aFea = dataModel()->object(aIdx);
498           if (dataModel()->object(aIdx)->isSame(anObject)) {
499             aIndex = aIdx;
500             break;
501           }
502         }
503       }
504       if (aIndex.isValid()) {
505         myTreeView->setCurrentIndex(aIndex);
506         myTreeView->edit(aIndex);
507       }
508       return;
509     }
510   }
511 }
512
513 //***************************************************
514 QList<ModuleBase_ITreeNode*> XGUI_ObjectsBrowser::expandedItems(const QModelIndex& theParent) const
515 {
516   QList<ModuleBase_ITreeNode*> aIndexes;
517   QModelIndex aIndex;
518   int aCount = myDocModel->rowCount(theParent);
519   for (int i = 0; i < aCount; i++) {
520     aIndex = myDocModel->index(i, 0, theParent);
521     if (myDocModel->hasChildren(aIndex)) {
522       if (myTreeView->isExpanded(aIndex)) {
523         aIndexes.append((ModuleBase_ITreeNode*)aIndex.internalPointer());
524         QList<ModuleBase_ITreeNode*> aSubIndexes = expandedItems(aIndex);
525         if (!aSubIndexes.isEmpty())
526           aIndexes.append(aSubIndexes);
527       }
528     }
529   }
530   return aIndexes;
531 }
532
533 //***************************************************
534 void XGUI_ObjectsBrowser::rebuildDataTree()
535 {
536   myDocModel->root()->update();
537   myDocModel->rebuildDataTree();
538   update();
539 }
540
541 //***************************************************
542 void XGUI_ObjectsBrowser::setObjectsSelected(const QObjectPtrList& theObjects)
543 {
544   QItemSelectionModel* aSelectModel = myTreeView->selectionModel();
545   QModelIndexList aIndexes = aSelectModel->selectedIndexes();
546   if (theObjects.size() == 0) {
547     bool aIsBlock = aSelectModel->blockSignals(true);
548     aSelectModel->clear();
549     aSelectModel->blockSignals(aIsBlock);
550     foreach(QModelIndex aIdx, aIndexes) {
551       myTreeView->update(aIdx);
552     }
553     return;
554   }
555
556   ObjectPtr aObject;
557   QModelIndexList aUnselect;
558   QObjectPtrList aToSelect = theObjects;
559   QHash<qint64, ObjectPtr> aNotChanged;
560   foreach(QModelIndex aIdx, aIndexes) {
561     aObject = myDocModel->object(aIdx);
562     if (aObject.get()) {
563       if (aToSelect.contains(aObject)) {
564         aNotChanged.insert((qint64)aObject.get(), aObject);
565       } else {
566         aUnselect.append(aIdx);
567       }
568     }
569     else {
570       aUnselect.append(aIdx);
571     }
572   }
573
574   foreach(ObjectPtr aObj, aNotChanged)
575     aToSelect.removeAll(aObj);
576
577   bool aIsBlock = aSelectModel->blockSignals(true);
578   foreach(QModelIndex aIdx, aUnselect) {
579     aSelectModel->select(aIdx, QItemSelectionModel::Deselect);
580     myTreeView->update(aIdx);
581   }
582
583   QModelIndex aIndex0, aIndex1, aIndex2, aCurrent;
584   foreach(ObjectPtr aFeature, aToSelect) {
585     aIndex1 = myDocModel->objectIndex(aFeature, 1);
586     if (aIndex1.isValid()) {
587       if (!aCurrent.isValid())
588         aCurrent = aIndex1;
589       aIndex0 = myDocModel->objectIndex(aFeature, 0);
590       aIndex2 = myDocModel->objectIndex(aFeature, 2);
591       aSelectModel->select(aIndex1, QItemSelectionModel::Select | QItemSelectionModel::Rows);
592       myTreeView->update(aIndex0);
593       myTreeView->update(aIndex1);
594       myTreeView->update(aIndex2);
595     }
596   }
597   aSelectModel->setCurrentIndex(aCurrent, QItemSelectionModel::NoUpdate);
598   aSelectModel->blockSignals(aIsBlock);
599 }
600
601 //***************************************************
602 void XGUI_ObjectsBrowser::ensureVisible(const ObjectPtr theObject)
603 {
604   QModelIndex aIndex = myDocModel->objectIndex(theObject);
605   if (aIndex.isValid())  {
606     QModelIndex aParent = aIndex.parent();
607     while (aParent.isValid()) {
608       myTreeView->expand(aParent);
609       aParent = aParent.parent();
610     }
611     myTreeView->scrollTo(aIndex);
612   }
613 }
614
615 //***************************************************
616 void XGUI_ObjectsBrowser::clearContent()
617 {
618   myTreeView->clear();
619 }
620
621 //***************************************************
622 void XGUI_ObjectsBrowser::onSelectionChanged(const QItemSelection& theSelected,
623                                        const QItemSelection& theDeselected)
624 {
625   onSelectionChanged();
626 }
627
628 //***************************************************
629 void XGUI_ObjectsBrowser::onSelectionChanged()
630 {
631   emit selectionChanged();
632 }
633
634 //***************************************************
635 QObjectPtrList XGUI_ObjectsBrowser::selectedObjects(QModelIndexList* theIndexes) const
636 {
637   QObjectPtrList aList;
638   QModelIndexList aIndexes = selectedIndexes();
639   XGUI_DataModel* aModel = dataModel();
640
641   foreach(QModelIndex aIdx, aIndexes) {
642     if (aIdx.column() == 1) {
643       ObjectPtr aObject = aModel->object(aIdx);
644       if (aObject) {
645         aList.append(aObject);
646         if (theIndexes)
647           theIndexes->append(aIdx);
648       }
649     }
650   }
651   return aList;
652 }
653
654 void XGUI_ObjectsBrowser::onBeforeReset()
655 {
656   myExpandedItems = expandedItems();
657 }
658
659 void XGUI_ObjectsBrowser::onAfterModelReset()
660 {
661   XGUI_DataModel* aModel = myTreeView->dataModel();
662   QModelIndex aIndex;
663   foreach(ModuleBase_ITreeNode* aNode, myExpandedItems) {
664     if (aModel->hasNode(aNode)) {
665       aIndex = aModel->getIndex(aNode, 0);
666       if (aIndex.isValid() && (myTreeView->dataModel()->hasIndex(aIndex)))
667         myTreeView->setExpanded(aIndex, true);
668     }
669   }
670   myExpandedItems.clear();
671 }
672
673 std::list<bool> XGUI_ObjectsBrowser::getStateForDoc(DocumentPtr theDoc) const
674 {
675   std::list<bool> aStates;
676   XGUI_DataModel* aModel = dataModel();
677   QModelIndex aRootIdx = aModel->documentRootIndex(theDoc);
678   int aNbChild = aModel->rowCount(aRootIdx);
679   for (int i = 0; i < aNbChild; i++) {
680     QModelIndex aIdx = aModel->index(i, 0, aRootIdx);
681     aStates.push_back(myTreeView->isExpanded(aIdx));
682   }
683   return aStates;
684 }
685
686 void XGUI_ObjectsBrowser::setStateForDoc(DocumentPtr theDoc, const std::list<bool>& theStates)
687 {
688   if (theStates.size() == 0)
689     return;
690   XGUI_DataModel* aModel = dataModel();
691   QModelIndex aRootIdx = aModel->documentRootIndex(theDoc);
692   int aNbChild = aModel->rowCount(aRootIdx);
693
694   std::list<bool>::const_iterator aIt;
695   int i = 0;
696   for (aIt = theStates.cbegin(); aIt != theStates.cend(); aIt++, i++) {
697     if (i >= aNbChild )
698       break;
699     QModelIndex aIdx = aModel->index(i, 0, aRootIdx);
700     myTreeView->setExpanded(aIdx, (*aIt));
701   }
702 }
703
704 void XGUI_ObjectsBrowser::updateAllIndexes(int theColumn, const QModelIndex& theParent)
705 {
706   int aNb = myDocModel->rowCount(theParent);
707   for (int i = 0; i < aNb; i++) {
708     QModelIndex aIdx = myDocModel->index(i, theColumn, theParent);
709     if (aIdx.isValid() && myDocModel->hasIndex(aIdx)) {
710       myTreeView->update(aIdx);
711       if (myTreeView->isExpanded(aIdx))
712         updateAllIndexes(theColumn, aIdx);
713     }
714   }
715 }
716
717 QMap<ObjectPtr, bool> XGUI_ObjectsBrowser::getFoldersState(DocumentPtr theDoc) const
718 {
719   QMap<ObjectPtr, bool> aMap;
720
721   int aNb = theDoc->size(ModelAPI_Folder::group());
722   ObjectPtr aObj;
723   for (int i = 0; i < aNb; i++) {
724     aObj = theDoc->object(ModelAPI_Folder::group(), i);
725     QModelIndex aIdx = myDocModel->objectIndex(aObj, 0);
726     aMap[aObj] = myTreeView->isExpanded(aIdx);
727   }
728   return aMap;
729 }
730
731 void XGUI_ObjectsBrowser::setFoldersState(const QMap<ObjectPtr, bool>& theStates)
732 {
733   QMap<ObjectPtr, bool>::const_iterator aIt;
734   for (aIt = theStates.constBegin(); aIt != theStates.constEnd(); aIt++) {
735     QModelIndex aIdx = myDocModel->objectIndex(aIt.key(), 0);
736     myTreeView->setExpanded(aIdx, aIt.value());
737   }
738 }
739
740
741 void XGUI_ObjectsBrowser::resizeEvent(QResizeEvent* theEvent)
742 {
743   QWidget::resizeEvent(theEvent);
744   emit sizeChanged();
745 }