X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FXGUI%2FXGUI_DocumentDataModel.cpp;h=1ee46e331e9d07ca4d2c47980bf40cdea6af65a0;hb=d86c77d1c6210bbe04fbc3e5b00f9e212e1ec930;hp=ba23d472d96ad5b660d5cdd01126aeb3805a0dfe;hpb=275e39f8c9408e23882702f27f0d561193caeeb3;p=modules%2Fshaper.git diff --git a/src/XGUI/XGUI_DocumentDataModel.cpp b/src/XGUI/XGUI_DocumentDataModel.cpp index ba23d472d..1ee46e331 100644 --- a/src/XGUI/XGUI_DocumentDataModel.cpp +++ b/src/XGUI/XGUI_DocumentDataModel.cpp @@ -5,28 +5,30 @@ #include #include #include -#include -#include +#include +#include -#include +#include #include +#include XGUI_DocumentDataModel::XGUI_DocumentDataModel(QObject* theParent) : QAbstractItemModel(theParent) { // Find Document object - std::shared_ptr aMgr = ModelAPI_PluginManager::get(); + boost::shared_ptr aMgr = ModelAPI_PluginManager::get(); myDocument = aMgr->currentDocument(); // Register in event loop - Event_Loop::loop()->registerListener(this, Event_Loop::eventByName(EVENT_FEATURE_UPDATED)); + 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(this); - myModel->setDocument(myDocument); + myModel = new XGUI_TopDataModel(myDocument, this); } @@ -36,30 +38,106 @@ XGUI_DocumentDataModel::~XGUI_DocumentDataModel() } -void XGUI_DocumentDataModel::processEvent(const Event_Message* theMessage) +void XGUI_DocumentDataModel::processEvent(const Events_Message* theMessage) { - beginResetModel(); - int aNbParts = myDocument->featuresIterator(PARTS_GROUP)->numIterationsLeft(); - if (myPartModels.size() != aNbParts) { // resize internal models - while (myPartModels.size() > aNbParts) { - delete myPartModels.last(); - myPartModels.removeLast(); + // Created object event ******************* + if (QString(theMessage->eventID().eventText()) == 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) { // Updsate only Parts group + // Add a new part + int aStart = myModel->rowCount(QModelIndex()) + myPartModels.size() + 1; + XGUI_PartDataModel* aModel = new XGUI_PartDataModel(myDocument, this); + aModel->setPartId(myPartModels.count()); + myPartModels.append(aModel); + insertRows(QModelIndex(), aStart, aStart); + } 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)); + insertRows(aIndex, aStart, aStart); + } + } 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)); + insertRows(aIndex, aStart, aStart); + } } - while (myPartModels.size() < aNbParts) { - myPartModels.append(new XGUI_PartDataModel(this)); + + // Deteted object event *********************** + } else if (QString(theMessage->eventID().eventText()) == 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 = myModel->rowCount(QModelIndex()) + myPartModels.size() - 1; + beginRemoveRows(QModelIndex(), aStart, aStart); + removeSubModel(myPartModels.size() - 1); + endRemoveRows(); + } 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)); + beginRemoveRows(aIndex, aStart, aStart); + endRemoveRows(); + } + } 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(aUpdMsg->group()); + int aStart = aPartModel->rowCount(aIndex); + aIndex = createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)); + beginRemoveRows(aIndex, aStart, aStart); + endRemoveRows(); + } } - for (int i = 0; i < myPartModels.size(); i++) - myPartModels.at(i)->setDocument(myDocument, i); + + // Reset whole tree ************************** + } else { + beginResetModel(); + int aNbParts = myDocument->featuresIterator(PARTS_GROUP)->numIterationsLeft(); + 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); + } + clearModelIndexes(); + endResetModel(); } - clearModelIndexes(); - endResetModel(); } QVariant XGUI_DocumentDataModel::data(const QModelIndex& theIndex, int theRole) const { if (!theIndex.isValid()) return QVariant(); - return toSourceModel(theIndex).data(theRole); + return toSourceModelIndex(theIndex).data(theRole); } @@ -70,10 +148,14 @@ QVariant XGUI_DocumentDataModel::headerData(int theSection, Qt::Orientation theO int XGUI_DocumentDataModel::rowCount(const QModelIndex& theParent) const { - if (!theParent.isValid()) + if (!theParent.isValid()) { + int aVal = myModel->rowCount(theParent) + myPartModels.size(); return myModel->rowCount(theParent) + myPartModels.size(); + } + QModelIndex aParent = toSourceModelIndex(theParent); + if (!hasSubModel(aParent.model())) + return 0; - QModelIndex aParent = toSourceModel(theParent); return aParent.model()->rowCount(aParent); } @@ -89,9 +171,15 @@ QModelIndex XGUI_DocumentDataModel::index(int theRow, int theColumn, const QMode int aOffs = myModel->rowCount(); if (theRow < aOffs) aIndex = myModel->index(theRow, theColumn, theParent); - else - aIndex = myPartModels.at(theRow - aOffs)->index(theRow - aOffs, theColumn, theParent); - + else { + if (myPartModels.size() > 0) { + int aPos = theRow - aOffs; + if (aPos >= myPartModels.size()) + aPos = 0; + aIndex = myPartModels.at(aPos)->index(aPos, theColumn, theParent); + } else + return aIndex; + } aIndex = createIndex(theRow, theColumn, (void*)getModelIndex(aIndex)); } else { QModelIndex* aParent = (QModelIndex*)theParent.internalPointer(); @@ -105,7 +193,10 @@ QModelIndex XGUI_DocumentDataModel::index(int theRow, int theColumn, const QMode QModelIndex XGUI_DocumentDataModel::parent(const QModelIndex& theIndex) const { - QModelIndex aParent = toSourceModel(theIndex); + QModelIndex aParent = toSourceModelIndex(theIndex); + if (!hasSubModel(aParent.model())) + return QModelIndex(); + aParent = aParent.model()->parent(aParent); if (aParent.isValid()) return createIndex(aParent.row(), aParent.column(), (void*)getModelIndex(aParent)); @@ -121,7 +212,7 @@ bool XGUI_DocumentDataModel::hasChildren(const QModelIndex& theParent) const } -QModelIndex XGUI_DocumentDataModel::toSourceModel(const QModelIndex& theProxy) const +QModelIndex XGUI_DocumentDataModel::toSourceModelIndex(const QModelIndex& theProxy) const { QModelIndex* aIndexPtr = static_cast(theProxy.internalPointer()); return (*aIndexPtr); @@ -156,4 +247,52 @@ void XGUI_DocumentDataModel::clearModelIndexes() for (aIt = myIndexes.constBegin(); aIt != myIndexes.constEnd(); ++aIt) delete (*aIt); myIndexes.clear(); -} \ No newline at end of file +} + +FeaturePtr XGUI_DocumentDataModel::feature(const QModelIndex& theIndex) const +{ + QModelIndex aIndex = toSourceModelIndex(theIndex); + if (!hasSubModel(aIndex.model())) + return FeaturePtr(); + + const XGUI_FeaturesModel* aModel = dynamic_cast(aIndex.model()); + return aModel->feature(aIndex); +} + +void XGUI_DocumentDataModel::insertRows(const QModelIndex& theParent, int theStart, int theEnd) +{ + beginInsertRows(theParent, theStart, theEnd); + endInsertRows(); + if (theStart == 0) // Update parent if this is a first child in order to update node decoration + emit dataChanged(theParent, theParent); +} + +void XGUI_DocumentDataModel::removeSubModel(int theModelId) +{ + XGUI_PartModel* aModel = myPartModels.at(theModelId); + QIntList aToRemove; + for (int i = 0; i < myIndexes.size(); i++) { + if (myIndexes.at(i)->model() == aModel) + aToRemove.append(i); + } + int aId; + while(aToRemove.size() > 0) { + aId = aToRemove.last(); + delete myIndexes.at(aId); + myIndexes.removeAt(aId); + aToRemove.removeLast(); + } + delete aModel; + myPartModels.removeAt(theModelId); +} + +bool XGUI_DocumentDataModel::hasSubModel(const QAbstractItemModel* theModel) const +{ + if (theModel == myModel) + return true; + QList::const_iterator aIt; + for (aIt = myPartModels.constBegin(); aIt != myPartModels.constEnd(); ++aIt) + if ((*aIt) == theModel) + return true; + return false; +}