X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FXGUI%2FXGUI_DocumentDataModel.cpp;h=b4cc62c0c9f4d5f8bc6a6fba6cd41dffd6301903;hb=47c18d4de2719126f6b045b5c37525cd8f170aa1;hp=2c4470cce95b8d88c0fb3c7dc4d5758499bbbd33;hpb=23378e19cee76b189c2bfdea631002b005f3237a;p=modules%2Fshaper.git diff --git a/src/XGUI/XGUI_DocumentDataModel.cpp b/src/XGUI/XGUI_DocumentDataModel.cpp index 2c4470cce..b4cc62c0c 100644 --- a/src/XGUI/XGUI_DocumentDataModel.cpp +++ b/src/XGUI/XGUI_DocumentDataModel.cpp @@ -1,12 +1,14 @@ #include "XGUI_DocumentDataModel.h" #include "XGUI_PartDataModel.h" #include "XGUI_Workshop.h" +#include "XGUI_Tools.h" #include #include #include #include #include +#include #include @@ -23,17 +25,13 @@ XGUI_DocumentDataModel::XGUI_DocumentDataModel(QObject* theParent) : QAbstractItemModel(theParent), myActivePart(0) { - // Find Document object - boost::shared_ptr aMgr = ModelAPI_PluginManager::get(); - myDocument = aMgr->currentDocument(); - // Register in event loop Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_FEATURE_CREATED)); Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_FEATURE_UPDATED)); Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_FEATURE_DELETED)); // Create a top part of data tree model - myModel = new XGUI_TopDataModel(myDocument, this); + myModel = new XGUI_TopDataModel(this); myModel->setItemsColor(ACTIVE_COLOR); } @@ -46,81 +44,95 @@ XGUI_DocumentDataModel::~XGUI_DocumentDataModel() void XGUI_DocumentDataModel::processEvent(const Events_Message* theMessage) { + DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument(); + // Created object event ******************* if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_CREATED)) { const Model_FeatureUpdatedMessage* aUpdMsg = dynamic_cast(theMessage); - boost::shared_ptr aFeature = aUpdMsg->feature(); - boost::shared_ptr aDoc = aFeature->document(); - - if (aDoc == myDocument) { // If root objects - if (aFeature->getGroup().compare(PARTS_GROUP) == 0) { // Update only Parts group - // Add a new part - int aStart = myPartModels.size() + 1; - XGUI_PartDataModel* aModel = new XGUI_PartDataModel(myDocument, this); - aModel->setPartId(myPartModels.count()); - myPartModels.append(aModel); - insertRow(aStart, partFolderNode()); - } else { // Update top groups (other except parts - QModelIndex aIndex = myModel->findParent(aFeature); - int aStart = myModel->rowCount(aIndex) - 1; - aIndex = createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)); - insertRow(aStart, aIndex); - } - } else { // if sub-objects of first level nodes - XGUI_PartModel* aPartModel = 0; - QList::const_iterator aIt; - for (aIt = myPartModels.constBegin(); aIt != myPartModels.constEnd(); ++aIt) { - if ((*aIt)->hasDocument(aDoc)) { - aPartModel = (*aIt); - break; + std::set aFeatures = aUpdMsg->features(); + + std::set::const_iterator aIt; + for (aIt = aFeatures.begin(); aIt != aFeatures.end(); ++aIt) { + FeaturePtr aFeature = (*aIt); + DocumentPtr aDoc = aFeature->document(); + if (aDoc == aRootDoc) { // If root objects + if (aFeature->getGroup().compare(PARTS_GROUP) == 0) { // Update only Parts group + // Add a new part + int aStart = myPartModels.size(); + XGUI_PartDataModel* aModel = new XGUI_PartDataModel(this); + aModel->setPartId(myPartModels.count()); + myPartModels.append(aModel); + insertRow(aStart, partFolderNode()); + } else { // Update top groups (other except parts + QModelIndex aIndex = myModel->findParent(aFeature); + int aStart = myModel->rowCount(aIndex) - 1; + aIndex = createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)); + insertRow(aStart, aIndex); + } + } else { // if sub-objects of first level nodes + XGUI_PartModel* aPartModel = 0; + QList::const_iterator aIt; + for (aIt = myPartModels.constBegin(); aIt != myPartModels.constEnd(); ++aIt) { + if ((*aIt)->hasDocument(aDoc)) { + aPartModel = (*aIt); + break; + } + } + if (aPartModel) { + QModelIndex aIndex = aPartModel->findParent(aFeature); + int aStart = aPartModel->rowCount(aIndex) - 1; + aIndex = createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)); + insertRow(aStart, aIndex); } - } - if (aPartModel) { - QModelIndex aIndex = aPartModel->findParent(aFeature); - int aStart = aPartModel->rowCount(aIndex) - 1; - aIndex = createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)); - insertRow(aStart, aIndex); } } - // Deleted object event *********************** } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_DELETED)) { const Model_FeatureDeletedMessage* aUpdMsg = dynamic_cast(theMessage); - boost::shared_ptr aDoc = aUpdMsg->document(); - - if (aDoc == myDocument) { // If root objects - if (aUpdMsg->group().compare(PARTS_GROUP) == 0) { // Updsate only Parts group - int aStart = myPartModels.size(); - removeSubModel(myPartModels.size() - 1); - removeRow(aStart, partFolderNode()); - } else { // Update top groups (other except parts - QModelIndex aIndex = myModel->findGroup(aUpdMsg->group()); - int aStart = myModel->rowCount(aIndex); - aIndex = createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)); - removeRow(aStart, aIndex); - } - } else { - XGUI_PartModel* aPartModel = 0; - QList::const_iterator aIt; - for (aIt = myPartModels.constBegin(); aIt != myPartModels.constEnd(); ++aIt) { - if ((*aIt)->hasDocument(aDoc)) { - aPartModel = (*aIt); - break; + DocumentPtr aDoc = aUpdMsg->document(); + std::set aGroups = aUpdMsg->groups(); + + std::set::const_iterator aIt; + for (aIt = aGroups.begin(); aIt != aGroups.end(); ++aIt) { + std::string aGroup = (*aIt); + if (aDoc == aRootDoc) { // If root objects + if (aGroup.compare(PARTS_GROUP) == 0) { // Updsate only Parts group + int aStart = myPartModels.size() - 1; + removeSubModel(aStart); + removeRow(aStart, partFolderNode()); + if (myActivePart && (!isPartSubModel(myActivePart))) { + myActivePart = 0; + myActivePartIndex = QModelIndex(); + myModel->setItemsColor(ACTIVE_COLOR); + } + } else { // Update top groups (other except parts + QModelIndex aIndex = myModel->findGroup(aGroup); + int aStart = myModel->rowCount(aIndex); + aIndex = createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)); + removeRow(aStart, aIndex); + } + } else { + XGUI_PartModel* aPartModel = 0; + QList::const_iterator aIt; + for (aIt = myPartModels.constBegin(); aIt != myPartModels.constEnd(); ++aIt) { + if ((*aIt)->hasDocument(aDoc)) { + aPartModel = (*aIt); + break; + } + } + if (aPartModel) { + QModelIndex aIndex = aPartModel->findGroup(aGroup); + int aStart = aPartModel->rowCount(aIndex); + aIndex = createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)); + removeRow(aStart, aIndex); } - } - if (aPartModel) { - QModelIndex aIndex = aPartModel->findGroup(aUpdMsg->group()); - int aStart = aPartModel->rowCount(aIndex); - aIndex = createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)); - removeRow(aStart, aIndex); } } - // Deleted object event *********************** } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_UPDATED)) { //const Model_FeatureUpdatedMessage* aUpdMsg = dynamic_cast(theMessage); - //boost::shared_ptr aFeature = aUpdMsg->feature(); - //boost::shared_ptr aDoc = aFeature->document(); + //FeaturePtr aFeature = aUpdMsg->feature(); + //DocumentPtr aDoc = aFeature->document(); // TODO: Identify the necessary index by the modified feature QModelIndex aIndex; @@ -128,22 +140,30 @@ void XGUI_DocumentDataModel::processEvent(const Events_Message* theMessage) // Reset whole tree ************************** } else { - beginResetModel(); - int aNbParts = myDocument->size(PARTS_GROUP); - if (myPartModels.size() != aNbParts) { // resize internal models - while (myPartModels.size() > aNbParts) { - delete myPartModels.last(); - myPartModels.removeLast(); - } - while (myPartModels.size() < aNbParts) { - myPartModels.append(new XGUI_PartDataModel(myDocument, this)); - } - for (int i = 0; i < myPartModels.size(); i++) - myPartModels.at(i)->setPartId(i); + rebuildDataTree(); + } +} + +void XGUI_DocumentDataModel::rebuildDataTree() +{ + DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument(); + + beginResetModel(); + clearModelIndexes(); + + int aNbParts = aRootDoc->size(PARTS_GROUP); + if (myPartModels.size() != aNbParts) { // resize internal models + while (myPartModels.size() > aNbParts) { + delete myPartModels.last(); + myPartModels.removeLast(); + } + while (myPartModels.size() < aNbParts) { + myPartModels.append(new XGUI_PartDataModel(this)); } - clearModelIndexes(); - endResetModel(); + for (int i = 0; i < myPartModels.size(); i++) + myPartModels.at(i)->setPartId(i); } + endResetModel(); } QVariant XGUI_DocumentDataModel::data(const QModelIndex& theIndex, int theRole) const @@ -171,7 +191,8 @@ QVariant XGUI_DocumentDataModel::data(const QModelIndex& theIndex, int theRole) case HistoryNode: { int aOffset = historyOffset(); - FeaturePtr aFeature = myDocument->feature(FEATURES_GROUP, theIndex.row() - aOffset); + DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument(); + FeaturePtr aFeature = aRootDoc->feature(FEATURES_GROUP, theIndex.row() - aOffset); if (!aFeature) return QVariant(); switch (theRole) { @@ -211,10 +232,11 @@ QVariant XGUI_DocumentDataModel::headerData(int theSection, Qt::Orientation theO int XGUI_DocumentDataModel::rowCount(const QModelIndex& theParent) const { if (!theParent.isValid()) { + DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument(); // Size of external models int aVal = historyOffset(); // Plus history size - aVal += myDocument->size(FEATURES_GROUP); + aVal += aRootDoc->size(FEATURES_GROUP); return aVal; } if (theParent.internalId() == PartsFolder) { @@ -340,8 +362,9 @@ FeaturePtr XGUI_DocumentDataModel::feature(const QModelIndex& theIndex) const if (theIndex.internalId() == PartsFolder) return FeaturePtr(); if (theIndex.internalId() == HistoryNode) { - int aOffset = historyOffset(); - return myDocument->feature(FEATURES_GROUP, theIndex.row() - aOffset); + DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument(); + int aOffset = historyOffset(); + return aRootDoc->feature(FEATURES_GROUP, theIndex.row() - aOffset); } QModelIndex* aIndex = toSourceModelIndex(theIndex); if (!isSubModel(aIndex->model())) @@ -463,5 +486,72 @@ void XGUI_DocumentDataModel::deactivatePart() if (myActivePart) myActivePart->setItemsColor(PASSIVE_COLOR); myActivePart = 0; + myActivePartIndex = QModelIndex(); myModel->setItemsColor(ACTIVE_COLOR); } + +Qt::ItemFlags XGUI_DocumentDataModel::flags(const QModelIndex& theIndex) const +{ + Qt::ItemFlags aFlags = QAbstractItemModel::flags(theIndex); + if (feature(theIndex)) { + aFlags |= Qt::ItemIsEditable; + } + return aFlags; +} + +QModelIndex XGUI_DocumentDataModel::partIndex(const FeaturePtr& theFeature) const +{ + FeaturePtr aFeature = XGUI_Tools::realFeature(theFeature); + + int aRow = -1; + XGUI_PartModel* aModel = 0; + foreach (XGUI_PartModel* aPartModel, myPartModels) { + aRow++; + if (aPartModel->part() == aFeature) { + aModel = aPartModel; + break; + } + } + if (aModel) { + return createIndex(aRow, 0, (void*)getModelIndex(aModel->index(0, 0, QModelIndex()))); + } + return QModelIndex(); +} + +QModelIndex XGUI_DocumentDataModel::featureIndex(const FeaturePtr theFeature) const +{ + // Check that this feature belongs to root document + DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument(); + DocumentPtr aDoc = theFeature->document(); + if (aDoc == aRootDoc) { + // This feature belongs to histrory or top model + if (theFeature->isInHistory()) { + int aId; + for (aId = 0; aId < aRootDoc->size(FEATURES_GROUP); aId++) { + if (theFeature == aRootDoc->feature(FEATURES_GROUP, aId)) + break; + } + return index(aId + historyOffset(), 0, QModelIndex()); + } else { + QModelIndex aIndex = myModel->featureIndex(theFeature); + return aIndex.isValid()? + createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)) : + QModelIndex(); + } + } else { + XGUI_PartModel* aPartModel = 0; + foreach(XGUI_PartModel* aModel, myPartModels) { + if (aModel->hasDocument(aDoc)) { + aPartModel = aModel; + break; + } + } + if (aPartModel) { + QModelIndex aIndex = aPartModel->featureIndex(theFeature); + return aIndex.isValid()? + createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)) : + QModelIndex(); + } + } + return QModelIndex(); +} \ No newline at end of file