X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FXGUI%2FXGUI_ObjectsBrowser.cpp;h=a83a67251145cd80563734a6f633f691d16ae2d6;hb=ac53df4563f345f48d3a7274dbbb3f2618bacbda;hp=658a1d1e2e59003fc076cb6826c06886c9c94a17;hpb=3e3fc52de000b4d8e51fdcffe7f72f82237833ad;p=modules%2Fshaper.git diff --git a/src/XGUI/XGUI_ObjectsBrowser.cpp b/src/XGUI/XGUI_ObjectsBrowser.cpp index 658a1d1e2..a83a67251 100644 --- a/src/XGUI/XGUI_ObjectsBrowser.cpp +++ b/src/XGUI/XGUI_ObjectsBrowser.cpp @@ -1,4 +1,21 @@ -// Copyright (C) 2014-20xx CEA/DEN, EDF R&D --> +// Copyright (C) 2014-2019 CEA/DEN, EDF R&D +// +// This library is free software; you can redistribute it and/or +// modify it under the terms of the GNU Lesser General Public +// License as published by the Free Software Foundation; either +// version 2.1 of the License, or (at your option) any later version. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License along with this library; if not, write to the Free Software +// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +// +// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +// #include "XGUI_ObjectsBrowser.h" #include "XGUI_Tools.h" @@ -8,12 +25,14 @@ #include #include #include +#include #include -#include +#include + +#include #include -#include #include #include #include @@ -21,11 +40,16 @@ #include #include #include +#include + +#ifdef DEBUG_INDXES +#include +#endif /// Width of second column (minimum acceptable = 27) +#define FIRST_COL_WIDTH 20 #define SECOND_COL_WIDTH 30 - /** * \ingroup GUI * Tree item delegate for definition of data in column items editor @@ -35,7 +59,8 @@ class XGUI_TreeViewItemDelegate: public QStyledItemDelegate public: /// Constructor /// \param theParent a parent of the delegate - XGUI_TreeViewItemDelegate(XGUI_DataTree* theParent):QStyledItemDelegate(theParent), myTreedView(theParent) {} + XGUI_TreeViewItemDelegate(XGUI_DataTree* theParent):QStyledItemDelegate(theParent), + myTreedView(theParent) {} /// Set data for item editor (name of the item) /// \param editor a widget of editor @@ -44,7 +69,7 @@ public: { QLineEdit* aEditor = dynamic_cast(editor); if (aEditor) { - ModuleBase_IDocumentDataModel* aModel = myTreedView->dataModel(); + XGUI_DataModel* aModel = myTreedView->dataModel(); ObjectPtr aObj = aModel->object(index); if (aObj.get() != NULL) { aEditor->setText(aObj->data()->name().c_str()); @@ -63,20 +88,24 @@ XGUI_DataTree::XGUI_DataTree(QWidget* theParent) : QTreeView(theParent) { setHeaderHidden(true); + setTreePosition(1); setEditTriggers(QAbstractItemView::NoEditTriggers); setSelectionBehavior(QAbstractItemView::SelectRows); setSelectionMode(QAbstractItemView::ExtendedSelection); - setItemDelegateForColumn(0, new XGUI_TreeViewItemDelegate(this)); + setItemDelegateForColumn(1, new XGUI_TreeViewItemDelegate(this)); + + connect(this, SIGNAL(doubleClicked(const QModelIndex&)), + SLOT(onDoubleClick(const QModelIndex&))); } XGUI_DataTree::~XGUI_DataTree() { } -ModuleBase_IDocumentDataModel* XGUI_DataTree::dataModel() const +XGUI_DataModel* XGUI_DataTree::dataModel() const { - return static_cast(model()); + return static_cast(model()); } void XGUI_DataTree::contextMenuEvent(QContextMenuEvent* theEvent) @@ -86,53 +115,251 @@ void XGUI_DataTree::contextMenuEvent(QContextMenuEvent* theEvent) void XGUI_DataTree::commitData(QWidget* theEditor) { - QLineEdit* aEditor = dynamic_cast(theEditor); - if (aEditor) { - QString aName = aEditor->text(); - QModelIndexList aIndexList = selectionModel()->selectedIndexes(); - ModuleBase_IDocumentDataModel* aModel = dataModel(); - ObjectPtr aObj = aModel->object(aIndexList.first()); - SessionPtr aMgr = ModelAPI_Session::get(); - aMgr->startOperation("Rename"); - - if (!canRename(aObj, aName)) { - aMgr->abortOperation(); - return; + static int aEntrance = 0; + if (aEntrance == 0) { + // We have to check number of enter and exit of this function because it can be called + // recursively by Qt in order to avoid double modifying of a data + aEntrance = 1; + QLineEdit* aEditor = dynamic_cast(theEditor); + if (aEditor) { + QString aName = aEditor->text(); + QModelIndexList aIndexList = selectionModel()->selectedIndexes(); + XGUI_DataModel* aModel = dataModel(); + ObjectPtr aObj = aModel->object(aIndexList.first()); + + if (XGUI_Tools::canRename(aObj, aName)) { + SessionPtr aMgr = ModelAPI_Session::get(); + aMgr->startOperation("Rename"); + aObj->data()->setName(qPrintable(aName)); + aMgr->finishOperation(); + } } + } + aEntrance = 0; +} - aObj->data()->setName(qPrintable(aName)); +void XGUI_DataTree::clear() +{ + dataModel()->clear(); + reset(); +} + +void XGUI_DataTree::resizeEvent(QResizeEvent* theEvent) +{ + QTreeView::resizeEvent(theEvent); + QSize aSize = theEvent->size(); + if (aSize.isValid()) { + setColumnWidth(0, FIRST_COL_WIDTH); + setColumnWidth(1, aSize.width() - SECOND_COL_WIDTH - FIRST_COL_WIDTH - 10); + setColumnWidth(2, SECOND_COL_WIDTH); + } +} + +#ifdef DEBUG_INDXES +void XGUI_DataTree::mousePressEvent(QMouseEvent* theEvent) +{ + QTreeView::mousePressEvent(theEvent); + if (theEvent->button() != Qt::MidButton) + return; + QModelIndex aInd = indexAt(theEvent->pos()); + QString aTxt = + QString("r=%1 c=%2 p=%3").arg(aInd.row()).arg(aInd.column()).arg((long)aInd.internalPointer()); + + QModelIndex aPar = aInd.parent(); + QString aTxt1 = + QString("r=%1 c=%2 p=%3").arg(aPar.row()).arg(aPar.column()).arg((long)aPar.internalPointer()); + QToolTip::showText(theEvent->globalPos(), aTxt + '\n' + aTxt1); +} +#endif + +void XGUI_DataTree::mouseReleaseEvent(QMouseEvent* theEvent) +{ + QTreeView::mouseReleaseEvent(theEvent); +#ifdef DEBUG_INDXES + if (theEvent->button() != Qt::MidButton) + return; + QToolTip::hideText(); +#endif + if (theEvent->button() == Qt::LeftButton) { + QModelIndex aInd = indexAt(theEvent->pos()); + if (aInd.column() == 0) + processEyeClick(aInd); + } +} + +void XGUI_DataTree::processHistoryChange(const QModelIndex& theIndex) +{ + SessionPtr aMgr = ModelAPI_Session::get(); + // When operation is opened then we can not change history + if (aMgr->isOperation()) + return; + XGUI_DataModel* aModel = dataModel(); + if (aModel->flags(theIndex) == 0) + return; + ObjectPtr aObj = aModel->object(theIndex); + + DocumentPtr aDoc = aMgr->activeDocument(); + + std::string aOpName = tr("History change").toStdString(); + if (aObj.get()) { + if (aObj->document() != aDoc) + return; + aMgr->startOperation(aOpName); + aDoc->setCurrentFeature(std::dynamic_pointer_cast(aObj), true); aMgr->finishOperation(); + } else { + // Ignore clicks on folders outside current document + if ((theIndex.internalId() == 0) && (aDoc != aMgr->moduleDocument())) + // Clicked folder under root but active document is another + return; + if ((theIndex.internalId() != 0) && (aDoc != aModel->document(theIndex))) + // Cliced not on active document folder + return; + + aMgr->startOperation(aOpName); + aDoc->setCurrentFeature(FeaturePtr(), true); + aMgr->finishOperation(); + } + QModelIndex aParent = theIndex.parent(); + int aSize = aModel->rowCount(aParent); + for (int i = 0; i < aSize; i++) { + update(aModel->index(i, 0, aParent)); + update(aModel->index(i, 1, aParent)); + update(aModel->index(i, 2, aParent)); + } +} + +void XGUI_DataTree::processEyeClick(const QModelIndex& theIndex) +{ + XGUI_ObjectsBrowser* aObjBrowser = qobject_cast(parent()); + XGUI_DataModel* aModel = dataModel(); + ObjectPtr aObj = aModel->object(theIndex); + if (aObj.get()) { + std::set anObjects; + anObjects.insert(aObj); + + bool hasHiddenState = aModel->hasHiddenState(theIndex); + if (aObjBrowser && hasHiddenState && !aObjBrowser->workshop()->prepareForDisplay(anObjects)) + return; + if (hasHiddenState) { // #issue 2335(hide all faces then show solid problem) + if (aObj->isDisplayed()) + aObj->setDisplayed(false); + aObj->setDisplayed(true); + } + else + aObj->setDisplayed(!aObj->isDisplayed()); + + // Update list of selected objects because this event happens after + // selection event in object browser + if (aObjBrowser) { + aObjBrowser->onSelectionChanged(); + } + Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY)); + update(theIndex); } } -bool XGUI_DataTree::canRename(const ObjectPtr& theObject, const QString& theName) +void XGUI_DataTree::onDoubleClick(const QModelIndex& theIndex) { - double aValue; - ResultParameterPtr aParam; + switch (theIndex.column()) { + case 2: + processHistoryChange(theIndex); + break; + } +} - bool isVariableFound = ModelAPI_Tools::findVariable(theObject->document(), qPrintable(theName), aValue, aParam); - if (isVariableFound) - QMessageBox::information(this, tr("Rename parameter"), - QString(tr("Selected parameter can not be renamed to: %1. \ -There is a parameter with the same name. Its value is: %2.")).arg(qPrintable(theName)).arg(aValue)); +//******************************************************************** +//******************************************************************** +//******************************************************************** +XGUI_ActiveDocLbl::XGUI_ActiveDocLbl(const QString& theText, QWidget* theParent ) + : QLabel(theText, theParent), + myPreSelectionStyle(""), + myNeutralStyle(""), + mySelectionStyle(""), + myIsSelected(false) +{ +} + +void XGUI_ActiveDocLbl::setTreeView(QTreeView* theView) +{ + myTreeView = theView; + setFont(myTreeView->font()); + + QPalette aPalet = myTreeView->palette(); + QColor aHighlight = aPalet.highlight().color(); + QColor aHighlightText = aPalet.highlightedText().color(); + + myPreSelectionStyle = "QLabel {background-color: "; + myPreSelectionStyle += aHighlight.lighter(170).name() + "}"; + //myPreSelectionStyle += "qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 white, stop:1 " + + // aHighlight.lighter(170).name() + ");"; + //myPreSelectionStyle += "border: 1px solid lightblue; border-radius: 2px }"; + + QString aName = aPalet.color(QPalette::Base).name(); + myNeutralStyle = "QLabel { border: 1px solid " + aName + " }"; + + +#if (!defined HAVE_SALOME) && (defined WIN32) + mySelectionStyle = "QLabel {background-color: "; + mySelectionStyle += "rgb(205, 232, 255); "; + //mySelectionStyle += "qlineargradient(x1:0, y1:0, x2:0, y2:1, stop:0 rgb(236, 245, 255)"; + //mySelectionStyle += ", stop:1 rgb(208, 229, 255));"; + //mySelectionStyle += "border: 1px solid rgb(132, 172, 221); border-radius: 2px }"; + mySelectionStyle += "border: 1px solid rgb(153, 209, 255) }"; +#else + mySelectionStyle = "QLabel {background-color: " + aHighlight.name(); + mySelectionStyle += "; color : " + aHighlightText.name() + "}"; +#endif - return !isVariableFound; + myTreeView->viewport()->installEventFilter(this); } -void XGUI_DataTree::clear() + +#if (!defined HAVE_SALOME) && (defined WIN32) +bool XGUI_ActiveDocLbl::event(QEvent* theEvent) { - ModuleBase_IDocumentDataModel* aModel = dataModel(); - aModel->clear(); - reset(); + switch (theEvent->type()) { + case QEvent::Enter: + if (!myIsSelected) + setStyleSheet(myPreSelectionStyle); + break; + case QEvent::Leave: + if (!myIsSelected) + setStyleSheet(myNeutralStyle); + break; + } + return QLabel::event(theEvent); } +#endif -void XGUI_DataTree::resizeEvent(QResizeEvent* theEvent) +bool XGUI_ActiveDocLbl::eventFilter(QObject* theObj, QEvent* theEvent) { - QSize aSize = theEvent->size(); - if (aSize.isValid()) { - setColumnWidth(0, aSize.width() - SECOND_COL_WIDTH); - setColumnWidth(1, SECOND_COL_WIDTH); + if (theObj == myTreeView->viewport()) { + if (theEvent->type() == QEvent::MouseButtonRelease) + unselect(); + } + return QLabel::eventFilter(theObj, theEvent); +} + +static bool MYClearing = false; +void XGUI_ActiveDocLbl::mouseReleaseEvent( QMouseEvent* e) +{ + MYClearing = true; + myIsSelected = true; + setStyleSheet(mySelectionStyle); + // We can not block signals because on + // clear selection the View state will not be updated + myTreeView->clearSelection(); + QLabel::mouseReleaseEvent(e); + MYClearing = false; +} + +void XGUI_ActiveDocLbl::unselect() +{ + if (!MYClearing) { + myIsSelected = false; + setStyleSheet(myNeutralStyle); } } @@ -140,18 +367,15 @@ void XGUI_DataTree::resizeEvent(QResizeEvent* theEvent) //******************************************************************** //******************************************************************** //******************************************************************** -XGUI_ObjectsBrowser::XGUI_ObjectsBrowser(QWidget* theParent) - : QWidget(theParent), myDocModel(0) +XGUI_ObjectsBrowser::XGUI_ObjectsBrowser(QWidget* theParent, XGUI_Workshop* theWorkshop) + : QWidget(theParent), myDocModel(0), myWorkshop(theWorkshop) { QVBoxLayout* aLayout = new QVBoxLayout(this); ModuleBase_Tools::zeroMargins(aLayout); aLayout->setSpacing(0); - QFrame* aLabelWgt = new QFrame(this); + QWidget* aLabelWgt = new QWidget(this); aLabelWgt->setAutoFillBackground(true); - QPalette aPalet = aLabelWgt->palette(); - aPalet.setColor(QPalette::Window, Qt::white); - aLabelWgt->setPalette(aPalet); aLayout->addWidget(aLabelWgt); QHBoxLayout* aLabelLay = new QHBoxLayout(aLabelWgt); @@ -161,42 +385,40 @@ XGUI_ObjectsBrowser::XGUI_ObjectsBrowser(QWidget* theParent) QLabel* aLbl = new QLabel(aLabelWgt); aLbl->setPixmap(QPixmap(":pictures/assembly.png")); aLbl->setMargin(2); - - aLbl->setAutoFillBackground(true); + // Do not paint background of the label (in order to show icon) + aLbl->setAutoFillBackground(false); aLabelLay->addWidget(aLbl); SessionPtr aMgr = ModelAPI_Session::get(); DocumentPtr aDoc = aMgr->moduleDocument(); - // TODO: Find a name of the root document - myActiveDocLbl = new QLineEdit(tr("Part set"), aLabelWgt); - myActiveDocLbl->setReadOnly(true); - myActiveDocLbl->setFrame(false); - //myActiveDocLbl->setMargin(2); + myActiveDocLbl = new XGUI_ActiveDocLbl(tr("Part set"), aLabelWgt); +// myActiveDocLbl->setReadOnly(true); +// myActiveDocLbl->setFrame(false); myActiveDocLbl->setContextMenuPolicy(Qt::CustomContextMenu); - myActiveDocLbl->installEventFilter(this); - aLabelLay->addWidget(myActiveDocLbl); aLabelLay->setStretch(1, 1); myTreeView = new XGUI_DataTree(this); + myTreeView->setFrameShape(QFrame::NoFrame); aLayout->addWidget(myTreeView); - aLabelWgt->setFrameShape(myTreeView->frameShape()); - aLabelWgt->setFrameShadow(myTreeView->frameShadow()); + QPalette aTreePalet = myTreeView->palette(); + QColor aTreeBack = aTreePalet.color(QPalette::Base); + + QPalette aPalet; + aPalet.setColor(QPalette::Base, aTreeBack); + aPalet.setColor(QPalette::Window, aTreeBack); + aLabelWgt->setPalette(aPalet); + + myDocModel = new XGUI_DataModel(this); + connect(myDocModel, SIGNAL(beforeTreeRebuild()), SLOT(onBeforeReset())); + connect(myDocModel, SIGNAL(treeRebuilt()), SLOT(onAfterModelReset())); - connect(myActiveDocLbl, SIGNAL(customContextMenuRequested(const QPoint&)), this, - SLOT(onLabelContextMenuRequested(const QPoint&))); connect(myTreeView, SIGNAL(contextMenuRequested(QContextMenuEvent*)), this, SLOT(onContextMenuRequested(QContextMenuEvent*))); - - // Create internal actions - QAction* aAction = new QAction(QIcon(":pictures/rename_edit.png"), tr("Rename"), this); - aAction->setData("RENAME_CMD"); - connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onEditItem())); - addAction(aAction); } //*************************************************** @@ -204,54 +426,18 @@ XGUI_ObjectsBrowser::~XGUI_ObjectsBrowser() { } -//*************************************************** -bool XGUI_ObjectsBrowser::eventFilter(QObject* obj, QEvent* theEvent) -{ - if (obj == myActiveDocLbl) { - if (!myActiveDocLbl->isReadOnly()) { - // End of editing by mouse click - if (theEvent->type() == QEvent::MouseButtonRelease) { - QMouseEvent* aEvent = (QMouseEvent*) theEvent; - QPoint aPnt = mapFromGlobal(aEvent->globalPos()); - if (childAt(aPnt) != myActiveDocLbl) { - closeDocNameEditing(true); - } - } else if (theEvent->type() == QEvent::KeyRelease) { - QKeyEvent* aEvent = (QKeyEvent*) theEvent; - switch (aEvent->key()) { - case Qt::Key_Return: - case Qt::Key_Enter: // Accept current input - closeDocNameEditing(true); - break; - case Qt::Key_Escape: // Cancel the input - closeDocNameEditing(false); - break; - } - } - } else { - if (theEvent->type() == QEvent::MouseButtonDblClick) { - emit headerMouseDblClicked(QModelIndex()); - return true; - } - } - } - return QWidget::eventFilter(obj, theEvent); -} +void XGUI_ObjectsBrowser::initialize(ModuleBase_ITreeNode* theRoot) +{ + //myDocModel->setXMLReader(theReader); + myDocModel->setRoot(theRoot); + myTreeView->setModel(myDocModel); -//*************************************************** -void XGUI_ObjectsBrowser::closeDocNameEditing(bool toSave) -{ - myActiveDocLbl->deselect(); - myActiveDocLbl->clearFocus(); - myActiveDocLbl->releaseMouse(); - myActiveDocLbl->setReadOnly(true); - if (toSave) { - // TODO: Save the name of root document - SessionPtr aMgr = ModelAPI_Session::get(); - DocumentPtr aDoc = aMgr->moduleDocument(); - } else { - myActiveDocLbl->setText(myActiveDocLbl->property("OldText").toString()); - } + // It has to be done after setting of model + myActiveDocLbl->setTreeView(myTreeView); + + QItemSelectionModel* aSelMod = myTreeView->selectionModel(); + connect(aSelMod, SIGNAL(selectionChanged(const QItemSelection&, const QItemSelection&)), + this, SLOT(onSelectionChanged(const QItemSelection&, const QItemSelection&))); } //*************************************************** @@ -260,7 +446,12 @@ void XGUI_ObjectsBrowser::onContextMenuRequested(QContextMenuEvent* theEvent) QModelIndexList aIndexes; QObjectPtrList aSelectedData = selectedObjects(&aIndexes); bool toEnable = false; + if (aSelectedData.size() == 1) { + QModelIndex aSelected = myTreeView->indexAt(theEvent->pos()); + if (!aIndexes.contains(aSelected)) + return; // menu is called on non selected item + Qt::ItemFlags aFlags = dataModel()->flags(aIndexes.first()); toEnable = ((aFlags & Qt::ItemIsEditable) != 0); } @@ -287,21 +478,23 @@ void XGUI_ObjectsBrowser::onEditItem() { QObjectPtrList aSelectedData = selectedObjects(); if (aSelectedData.size() > 0) { - ObjectPtr aFeature = aSelectedData.first(); - if (aFeature) { // Selection happens in TreeView - QObjectPtrList aList; - aList.append(aFeature); - // check whether the object can be deleted. There should not be parts which are not loaded - if (!XGUI_Tools::canRemoveOrRename((QWidget*)parent(), aList)) + ObjectPtr anObject = aSelectedData.first(); + if (anObject.get()) { // Selection happens in TreeView + // check whether the object can be renamed. There should not be parts which are not loaded + std::set aFeatures; + aFeatures.insert(ModelAPI_Feature::feature(anObject)); + if (!XGUI_Tools::canRemoveOrRename((QWidget*)parent(), aFeatures)) return; // Find index which corresponds the feature QModelIndex aIndex; foreach(QModelIndex aIdx, selectedIndexes()) { - ObjectPtr aFea = dataModel()->object(aIdx); - if (dataModel()->object(aIdx)->isSame(aFeature)) { - aIndex = aIdx; - break; + if (aIdx.column() == 1) { + ObjectPtr aFea = dataModel()->object(aIdx); + if (dataModel()->object(aIdx)->isSame(anObject)) { + aIndex = aIdx; + break; + } } } if (aIndex.isValid()) { @@ -311,17 +504,32 @@ void XGUI_ObjectsBrowser::onEditItem() return; } } - //Selection happens in Upper label - myActiveDocLbl->setReadOnly(false); - myActiveDocLbl->setFocus(); - myActiveDocLbl->selectAll(); - myActiveDocLbl->grabMouse(); - myActiveDocLbl->setProperty("OldText", myActiveDocLbl->text()); +} + +//*************************************************** +QList XGUI_ObjectsBrowser::expandedItems(const QModelIndex& theParent) const +{ + QList aIndexes; + QModelIndex aIndex; + int aCount = myDocModel->rowCount(theParent); + for (int i = 0; i < aCount; i++) { + aIndex = myDocModel->index(i, 0, theParent); + if (myDocModel->hasChildren(aIndex)) { + if (myTreeView->isExpanded(aIndex)) { + aIndexes.append((ModuleBase_ITreeNode*)aIndex.internalPointer()); + QList aSubIndexes = expandedItems(aIndex); + if (!aSubIndexes.isEmpty()) + aIndexes.append(aSubIndexes); + } + } + } + return aIndexes; } //*************************************************** void XGUI_ObjectsBrowser::rebuildDataTree() { + myDocModel->root()->update(); myDocModel->rebuildDataTree(); update(); } @@ -329,63 +537,205 @@ void XGUI_ObjectsBrowser::rebuildDataTree() //*************************************************** void XGUI_ObjectsBrowser::setObjectsSelected(const QObjectPtrList& theObjects) { - QList theIndexes; QItemSelectionModel* aSelectModel = myTreeView->selectionModel(); - aSelectModel->clear(); + QModelIndexList aIndexes = aSelectModel->selectedIndexes(); + if (theObjects.size() == 0) { + bool aIsBlock = aSelectModel->blockSignals(true); + aSelectModel->clear(); + aSelectModel->blockSignals(aIsBlock); + foreach(QModelIndex aIdx, aIndexes) { + myTreeView->update(aIdx); + } + return; + } - foreach(ObjectPtr aFeature, theObjects) - { - QModelIndex aIndex = myDocModel->objectIndex(aFeature); - if (aIndex.isValid()) { - aSelectModel->select(aIndex, QItemSelectionModel::Select); + ObjectPtr aObject; + QModelIndexList aUnselect; + QObjectPtrList aToSelect = theObjects; + QHash aNotChanged; + foreach(QModelIndex aIdx, aIndexes) { + aObject = myDocModel->object(aIdx); + if (aObject.get()) { + if (aToSelect.contains(aObject)) { + aNotChanged.insert((qint64)aObject.get(), aObject); + } else { + aUnselect.append(aIdx); + } + } + else { + aUnselect.append(aIdx); + } + } + + foreach(ObjectPtr aObj, aNotChanged) + aToSelect.removeAll(aObj); + + bool aIsBlock = aSelectModel->blockSignals(true); + foreach(QModelIndex aIdx, aUnselect) { + aSelectModel->select(aIdx, QItemSelectionModel::Deselect); + myTreeView->update(aIdx); + } + + QModelIndex aIndex0, aIndex1, aIndex2, aCurrent; + foreach(ObjectPtr aFeature, aToSelect) { + aIndex1 = myDocModel->objectIndex(aFeature, 1); + if (aIndex1.isValid()) { + if (!aCurrent.isValid()) + aCurrent = aIndex1; + aIndex0 = myDocModel->objectIndex(aFeature, 0); + aIndex2 = myDocModel->objectIndex(aFeature, 2); + aSelectModel->select(aIndex1, QItemSelectionModel::Select | QItemSelectionModel::Rows); + myTreeView->update(aIndex0); + myTreeView->update(aIndex1); + myTreeView->update(aIndex2); } } + aSelectModel->setCurrentIndex(aCurrent, QItemSelectionModel::NoUpdate); + aSelectModel->blockSignals(aIsBlock); } //*************************************************** -void XGUI_ObjectsBrowser::clearContent() -{ - myTreeView->clear(); +void XGUI_ObjectsBrowser::ensureVisible(const ObjectPtr theObject) +{ + QModelIndex aIndex = myDocModel->objectIndex(theObject); + if (aIndex.isValid()) { + QModelIndex aParent = aIndex.parent(); + while (aParent.isValid()) { + myTreeView->expand(aParent); + aParent = aParent.parent(); + } + myTreeView->scrollTo(aIndex); + } } -void XGUI_ObjectsBrowser::setDataModel(ModuleBase_IDocumentDataModel* theModel) +//*************************************************** +void XGUI_ObjectsBrowser::clearContent() { -#ifdef ModuleDataModel - myDocModel = theModel; -#else - myDocModel = new XGUI_DataModel(this); -#endif - myTreeView->setModel(myDocModel); - QItemSelectionModel* aSelMod = myTreeView->selectionModel(); - connect(aSelMod, SIGNAL(selectionChanged(const QItemSelection&, const QItemSelection&)), - this, SLOT(onSelectionChanged(const QItemSelection&, const QItemSelection&))); + myTreeView->clear(); } +//*************************************************** void XGUI_ObjectsBrowser::onSelectionChanged(const QItemSelection& theSelected, const QItemSelection& theDeselected) +{ + onSelectionChanged(); +} + +//*************************************************** +void XGUI_ObjectsBrowser::onSelectionChanged() { emit selectionChanged(); } +//*************************************************** QObjectPtrList XGUI_ObjectsBrowser::selectedObjects(QModelIndexList* theIndexes) const { QObjectPtrList aList; QModelIndexList aIndexes = selectedIndexes(); -#ifdef ModuleDataModel - ModuleBase_IDocumentDataModel* aModel = dataModel(); -#else XGUI_DataModel* aModel = dataModel(); -#endif - QModelIndexList::const_iterator aIt; - for (aIt = aIndexes.constBegin(); aIt != aIndexes.constEnd(); ++aIt) { - if ((*aIt).column() == 0) { - ObjectPtr aObject = aModel->object(*aIt); + + foreach(QModelIndex aIdx, aIndexes) { + if (aIdx.column() == 1) { + ObjectPtr aObject = aModel->object(aIdx); if (aObject) { aList.append(aObject); if (theIndexes) - theIndexes->append(*aIt); + theIndexes->append(aIdx); } } } return aList; -} \ No newline at end of file +} + +void XGUI_ObjectsBrowser::onBeforeReset() +{ + myExpandedItems = expandedItems(); +} + +void XGUI_ObjectsBrowser::onAfterModelReset() +{ + XGUI_DataModel* aModel = myTreeView->dataModel(); + QModelIndex aIndex; + foreach(ModuleBase_ITreeNode* aNode, myExpandedItems) { + if (aModel->hasNode(aNode)) { + aIndex = aModel->getIndex(aNode, 0); + if (aIndex.isValid() && (myTreeView->dataModel()->hasIndex(aIndex))) + myTreeView->setExpanded(aIndex, true); + } + } + myExpandedItems.clear(); +} + +std::list XGUI_ObjectsBrowser::getStateForDoc(DocumentPtr theDoc) const +{ + std::list aStates; + XGUI_DataModel* aModel = dataModel(); + QModelIndex aRootIdx = aModel->documentRootIndex(theDoc); + int aNbChild = aModel->rowCount(aRootIdx); + for (int i = 0; i < aNbChild; i++) { + QModelIndex aIdx = aModel->index(i, 0, aRootIdx); + aStates.push_back(myTreeView->isExpanded(aIdx)); + } + return aStates; +} + +void XGUI_ObjectsBrowser::setStateForDoc(DocumentPtr theDoc, const std::list& theStates) +{ + if (theStates.size() == 0) + return; + XGUI_DataModel* aModel = dataModel(); + QModelIndex aRootIdx = aModel->documentRootIndex(theDoc); + int aNbChild = aModel->rowCount(aRootIdx); + + std::list::const_iterator aIt; + int i = 0; + for (aIt = theStates.cbegin(); aIt != theStates.cend(); aIt++, i++) { + if (i >= aNbChild ) + break; + QModelIndex aIdx = aModel->index(i, 0, aRootIdx); + myTreeView->setExpanded(aIdx, (*aIt)); + } +} + +void XGUI_ObjectsBrowser::updateAllIndexes(int theColumn, const QModelIndex& theParent) +{ + int aNb = myDocModel->rowCount(theParent); + for (int i = 0; i < aNb; i++) { + QModelIndex aIdx = myDocModel->index(i, theColumn, theParent); + if (aIdx.isValid() && myDocModel->hasIndex(aIdx)) { + myTreeView->update(aIdx); + if (myTreeView->isExpanded(aIdx)) + updateAllIndexes(theColumn, aIdx); + } + } +} + +QMap XGUI_ObjectsBrowser::getFoldersState(DocumentPtr theDoc) const +{ + QMap aMap; + + int aNb = theDoc->size(ModelAPI_Folder::group()); + ObjectPtr aObj; + for (int i = 0; i < aNb; i++) { + aObj = theDoc->object(ModelAPI_Folder::group(), i); + QModelIndex aIdx = myDocModel->objectIndex(aObj, 0); + aMap[aObj] = myTreeView->isExpanded(aIdx); + } + return aMap; +} + +void XGUI_ObjectsBrowser::setFoldersState(const QMap& theStates) +{ + QMap::const_iterator aIt; + for (aIt = theStates.constBegin(); aIt != theStates.constEnd(); aIt++) { + QModelIndex aIdx = myDocModel->objectIndex(aIt.key(), 0); + myTreeView->setExpanded(aIdx, aIt.value()); + } +} + + +void XGUI_ObjectsBrowser::resizeEvent(QResizeEvent* theEvent) +{ + QWidget::resizeEvent(theEvent); + emit sizeChanged(); +}