X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FXGUI%2FXGUI_DocumentDataModel.cpp;h=6082e445bb56df332252a752caa11c14477e13dd;hb=4224f4dbe7ceaefe74b5d6b79a5840a9f5df2d7a;hp=066ebdcae6985bf5c7d038401f4f726f173f98fe;hpb=057e426145a26c02a1c5d6e1c220f92a17d8e53d;p=modules%2Fshaper.git diff --git a/src/XGUI/XGUI_DocumentDataModel.cpp b/src/XGUI/XGUI_DocumentDataModel.cpp index 066ebdcae..6082e445b 100644 --- a/src/XGUI/XGUI_DocumentDataModel.cpp +++ b/src/XGUI/XGUI_DocumentDataModel.cpp @@ -7,7 +7,8 @@ #include #include #include -#include +#include +#include #include #include @@ -18,6 +19,7 @@ #include #include +#include #define ACTIVE_COLOR QColor(0,72,140) #define PASSIVE_COLOR Qt::black @@ -25,17 +27,13 @@ XGUI_DocumentDataModel::XGUI_DocumentDataModel(QObject* theParent) : QAbstractItemModel(theParent), myActivePart(0) { - // Find Document object - PluginManagerPtr 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)); + Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_CREATED)); + Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_UPDATED)); + Events_Loop::loop()->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_DELETED)); // Create a top part of data tree model - myModel = new XGUI_TopDataModel(myDocument, this); + myModel = new XGUI_TopDataModel(this); myModel->setItemsColor(ACTIVE_COLOR); } @@ -48,26 +46,33 @@ 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); - 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 == myDocument) { // If root objects - if (aFeature->getGroup().compare(PARTS_GROUP) == 0) { // Update only Parts group + if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_CREATED)) { + const ModelAPI_ObjectUpdatedMessage* aUpdMsg = dynamic_cast(theMessage); + std::set aObjects = aUpdMsg->objects(); + + std::set::const_iterator aIt; + for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) { + ObjectPtr aObject = (*aIt); + FeaturePtr aFeature = boost::dynamic_pointer_cast(aObject); + if (aFeature && (!aFeature->isInHistory())) + continue; + + DocumentPtr aDoc = aObject->document(); + if (aDoc == aRootDoc) { // If root objects + if (aObject->groupName() == ModelAPI_ResultPart::group()) { // Update only Parts group // Add a new part int aStart = myPartModels.size(); - XGUI_PartDataModel* aModel = new XGUI_PartDataModel(myDocument, this); + 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); + QModelIndex aIndex = myModel->findParent(aObject); int aStart = myModel->rowCount(aIndex) - 1; + if (aStart < 0) aStart = 0; aIndex = createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)); insertRow(aStart, aIndex); } @@ -81,24 +86,24 @@ void XGUI_DocumentDataModel::processEvent(const Events_Message* theMessage) } } if (aPartModel) { - QModelIndex aIndex = aPartModel->findParent(aFeature); - int aStart = aPartModel->rowCount(aIndex) - 1; + QModelIndex aIndex = aPartModel->findParent(aObject); + int aStart = aPartModel->rowCount(aIndex); // check this index 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); + } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED)) { + const ModelAPI_ObjectDeletedMessage* aUpdMsg = dynamic_cast(theMessage); 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 == myDocument) { // If root objects - if (aGroup.compare(PARTS_GROUP) == 0) { // Updsate only Parts group + if (aDoc == aRootDoc) { // If root objects + if (aGroup == ModelAPI_ResultPart::group()) { // Update only Parts group int aStart = myPartModels.size() - 1; removeSubModel(aStart); removeRow(aStart, partFolderNode()); @@ -131,9 +136,9 @@ void XGUI_DocumentDataModel::processEvent(const Events_Message* theMessage) } } // Deleted object event *********************** - } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_UPDATED)) { - //const Model_FeatureUpdatedMessage* aUpdMsg = dynamic_cast(theMessage); - //FeaturePtr aFeature = aUpdMsg->feature(); + } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED)) { + //const ModelAPI_ObjectUpdatedMessage* aUpdMsg = dynamic_cast(theMessage); + //ObjectPtr aFeature = aUpdMsg->feature(); //DocumentPtr aDoc = aFeature->document(); // TODO: Identify the necessary index by the modified feature @@ -142,22 +147,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(ModelAPI_ResultPart::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 @@ -185,13 +198,15 @@ 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(); + ObjectPtr aObj = aRootDoc->object(ModelAPI_Feature::group(), theIndex.row() - aOffset); + FeaturePtr aFeature = boost::dynamic_pointer_cast(aObj); if (!aFeature) return QVariant(); switch (theRole) { case Qt::DisplayRole: if (aFeature) - return aFeature->data()->getName().c_str(); + return aFeature->data()->name().c_str(); else return QVariant(); case Qt::DecorationRole: @@ -225,13 +240,15 @@ 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(ModelAPI_Feature::group()); return aVal; } if (theParent.internalId() == PartsFolder) { + int aSize = myPartModels.size(); return myPartModels.size(); } if (theParent.internalId() == HistoryNode) { @@ -349,20 +366,21 @@ void XGUI_DocumentDataModel::clearModelIndexes() myIndexes.clear(); } -FeaturePtr XGUI_DocumentDataModel::feature(const QModelIndex& theIndex) const +ObjectPtr XGUI_DocumentDataModel::object(const QModelIndex& theIndex) const { if (theIndex.internalId() == PartsFolder) - return FeaturePtr(); + return ObjectPtr(); 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->object(ModelAPI_Feature::group(), theIndex.row() - aOffset); } QModelIndex* aIndex = toSourceModelIndex(theIndex); if (!isSubModel(aIndex->model())) - return FeaturePtr(); + return ObjectPtr(); const XGUI_FeaturesModel* aModel = dynamic_cast(aIndex->model()); - return aModel->feature(*aIndex); + return aModel->object(*aIndex); } bool XGUI_DocumentDataModel::insertRows(int theRow, int theCount, const QModelIndex& theParent) @@ -465,11 +483,11 @@ bool XGUI_DocumentDataModel::activatedIndex(const QModelIndex& theIndex) return false; } -FeaturePtr XGUI_DocumentDataModel::activePart() const +ResultPartPtr XGUI_DocumentDataModel::activePart() const { if (myActivePart) return myActivePart->part(); - return FeaturePtr(); + return ResultPartPtr(); } void XGUI_DocumentDataModel::deactivatePart() @@ -484,24 +502,19 @@ void XGUI_DocumentDataModel::deactivatePart() Qt::ItemFlags XGUI_DocumentDataModel::flags(const QModelIndex& theIndex) const { Qt::ItemFlags aFlags = QAbstractItemModel::flags(theIndex); - if (feature(theIndex)) { + if (object(theIndex)) { aFlags |= Qt::ItemIsEditable; } return aFlags; } -QModelIndex XGUI_DocumentDataModel::partIndex(const FeaturePtr& theFeature) const +QModelIndex XGUI_DocumentDataModel::partIndex(const ResultPartPtr& theObject) const { - FeaturePtr aFeature = theFeature; - if (XGUI_Tools::isModelObject(aFeature)) { - ObjectPtr aObject = boost::dynamic_pointer_cast(aFeature); - aFeature = aObject->featureRef(); - } int aRow = -1; XGUI_PartModel* aModel = 0; foreach (XGUI_PartModel* aPartModel, myPartModels) { aRow++; - if (aPartModel->part() == aFeature) { + if (aPartModel->part() == theObject) { aModel = aPartModel; break; } @@ -511,3 +524,44 @@ QModelIndex XGUI_DocumentDataModel::partIndex(const FeaturePtr& theFeature) cons } return QModelIndex(); } + +QModelIndex XGUI_DocumentDataModel::objectIndex(const ObjectPtr theObject) const +{ + // Check that this feature belongs to root document + DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument(); + DocumentPtr aDoc = theObject->document(); + if (aDoc == aRootDoc) { + // This feature belongs to histrory or top model + FeaturePtr aFeature = boost::dynamic_pointer_cast(theObject); + if (aFeature) { + int aId; + int aNb = aRootDoc->size(ModelAPI_Feature::group()); + for (aId = 0; aId < aNb; aId++) { + if (theObject == aRootDoc->object(ModelAPI_Feature::group(), aId)) + break; + } + if (aId < aNb) + return index(aId + historyOffset(), 0, QModelIndex()); + } else { + QModelIndex aIndex = myModel->objectIndex(theObject); + 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->objectIndex(theObject); + return aIndex.isValid()? + createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)) : + QModelIndex(); + } + } + return QModelIndex(); +}