-// Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
+// Copyright (C) 2014-2017 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<mailto:webmaster.salome@opencascade.com>
+//
#include "XGUI_ObjectsBrowser.h"
#include "XGUI_Tools.h"
#include <ModelAPI_Tools.h>
#include <ModuleBase_Tools.h>
-#include <ModuleBase_IDocumentDataModel.h>
#include <QLayout>
-#include <QLabel>
#include <QLineEdit>
#include <QPixmap>
#include <QEvent>
#include <QStyledItemDelegate>
#include <QMessageBox>
+#ifdef DEBUG_INDXES
+#include <QToolTip>
+#endif
+
/// Width of second column (minimum acceptable = 27)
+#define FIRST_COL_WIDTH 20
#define SECOND_COL_WIDTH 30
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
{
QLineEdit* aEditor = dynamic_cast<QLineEdit*>(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());
: 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<ModuleBase_IDocumentDataModel*>(model());
+ return static_cast<XGUI_DataModel*>(model());
}
void XGUI_DataTree::contextMenuEvent(QContextMenuEvent* theEvent)
void XGUI_DataTree::commitData(QWidget* theEditor)
{
- QLineEdit* aEditor = dynamic_cast<QLineEdit*>(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<QLineEdit*>(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;
+}
+
+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);
- aObj->data()->setName(qPrintable(aName));
+ 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<ModelAPI_Feature>(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.get() != theIndex.internalPointer()))
+ // Cliced not on active document folder
+ return;
+
+ aMgr->startOperation(aOpName);
+ aDoc->setCurrentFeature(FeaturePtr(), true);
aMgr->finishOperation();
}
+ QModelIndex aNewIndex = aModel->lastHistoryIndex();
+ 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));
+ }
}
-bool XGUI_DataTree::canRename(const ObjectPtr& theObject, const QString& theName)
+void XGUI_DataTree::processEyeClick(const QModelIndex& theIndex)
{
- double aValue;
- ResultParameterPtr aParam;
+ XGUI_DataModel* aModel = dataModel();
+ ObjectPtr aObj = aModel->object(theIndex);
+ if (aObj.get()) {
+ ResultPtr aResObj = std::dynamic_pointer_cast<ModelAPI_Result>(aObj);
+ if (aResObj.get()) {
+ aResObj->setDisplayed(!aResObj->isDisplayed());
+ Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
+ update(theIndex);
+ }
+ // Update list of selected objects because this event happens after
+ // selection event in object browser
+ XGUI_ObjectsBrowser* aObjBrowser = qobject_cast<XGUI_ObjectsBrowser*>(parent());
+ if (aObjBrowser) {
+ aObjBrowser->onSelectionChanged();
+ }
+ }
+}
- bool isVariableFound = ModelAPI_Tools::findVariable(theObject->document(), qPrintable(theName), aValue, aParam);
+void XGUI_DataTree::onDoubleClick(const QModelIndex& theIndex)
+{
+ switch (theIndex.column()) {
+ case 2:
+ processHistoryChange(theIndex);
+ break;
+ }
+}
- 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));
- return !isVariableFound;
+//********************************************************************
+//********************************************************************
+//********************************************************************
+XGUI_ActiveDocLbl::XGUI_ActiveDocLbl(const QString& theText, QWidget* theParent )
+ : QLabel(theText, theParent),
+ myPreSelectionStyle(""),
+ myNeutralStyle(""),
+ mySelectionStyle(""),
+ myIsSelected(false)
+{
}
-void XGUI_DataTree::clear()
+void XGUI_ActiveDocLbl::setTreeView(QTreeView* theView)
{
- ModuleBase_IDocumentDataModel* aModel = dataModel();
- aModel->clear();
- reset();
+ 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
+
+ myTreeView->viewport()->installEventFilter(this);
}
-void XGUI_DataTree::resizeEvent(QResizeEvent* theEvent)
+
+#if (!defined HAVE_SALOME) && (defined WIN32)
+bool XGUI_ActiveDocLbl::event(QEvent* theEvent)
{
- QSize aSize = theEvent->size();
- if (aSize.isValid()) {
- setColumnWidth(0, aSize.width() - SECOND_COL_WIDTH);
- setColumnWidth(1, SECOND_COL_WIDTH);
+ switch (theEvent->type()) {
+ case QEvent::Enter:
+ if (!myIsSelected)
+ setStyleSheet(myPreSelectionStyle);
+ break;
+ case QEvent::Leave:
+ if (!myIsSelected)
+ setStyleSheet(myNeutralStyle);
+ break;
+ }
+ return QLabel::event(theEvent);
+}
+#endif
+
+bool XGUI_ActiveDocLbl::eventFilter(QObject* theObj, QEvent* theEvent)
+{
+ 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);
}
}
//********************************************************************
//********************************************************************
//********************************************************************
-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);
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(modelAboutToBeReset()), 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);
}
//***************************************************
{
}
-//***************************************************
-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::setXMLReader(Config_DataModelReader* theReader)
+{
+ myDocModel->setXMLReader(theReader);
+ 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&)));
}
//***************************************************
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);
}
{
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<FeaturePtr> 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)) {
+ if (dataModel()->object(aIdx)->isSame(anObject)) {
aIndex = aIdx;
break;
}
return;
}
}
- //Selection happens in Upper label
- myActiveDocLbl->setReadOnly(false);
- myActiveDocLbl->setFocus();
- myActiveDocLbl->selectAll();
- myActiveDocLbl->grabMouse();
- myActiveDocLbl->setProperty("OldText", myActiveDocLbl->text());
+}
+
+//***************************************************
+QModelIndexList XGUI_ObjectsBrowser::expandedItems(const QModelIndex& theParent) const
+{
+ QModelIndexList aIndexes;
+ QModelIndex aIndex;
+ for (int i = 0; i < myDocModel->rowCount(theParent); i++) {
+ aIndex = myDocModel->index(i, 0, theParent);
+ if (myDocModel->hasChildren(aIndex)) {
+ if (myTreeView->isExpanded(aIndex)) {
+ aIndexes.append(aIndex);
+ QModelIndexList aSubIndexes = expandedItems(aIndex);
+ if (!aSubIndexes.isEmpty())
+ aIndexes.append(aSubIndexes);
+ }
+ }
+ }
+ return aIndexes;
}
//***************************************************
}
//***************************************************
-void XGUI_ObjectsBrowser::clearContent()
-{
- myTreeView->clear();
-}
-
-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) {
+ if ((*aIt).column() == 1) {
ObjectPtr aObject = aModel->object(*aIt);
if (aObject) {
aList.append(aObject);
}
}
return aList;
-}
\ No newline at end of file
+}
+
+void XGUI_ObjectsBrowser::onBeforeReset()
+{
+ myExpandedItems = expandedItems();
+}
+
+void XGUI_ObjectsBrowser::onAfterModelReset()
+{
+ foreach(QModelIndex aIndex, myExpandedItems) {
+ myTreeView->setExpanded(aIndex, true);
+ }
+}
+
+std::list<bool> XGUI_ObjectsBrowser::getStateForDoc(DocumentPtr theDoc) const
+{
+ std::list<bool> 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, 1, aRootIdx);
+ aStates.push_back(myTreeView->isExpanded(aIdx));
+ }
+ return aStates;
+}
+
+void XGUI_ObjectsBrowser::setStateForDoc(DocumentPtr theDoc, const std::list<bool>& theStates)
+{
+ if (theStates.size() == 0)
+ return;
+ XGUI_DataModel* aModel = dataModel();
+ QModelIndex aRootIdx = aModel->documentRootIndex(theDoc);
+ int aNbChild = aModel->rowCount(aRootIdx);
+
+ std::list<bool>::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));
+ }
+}