]> SALOME platform Git repositories - modules/shaper.git/blob - src/XGUI/XGUI_DocumentDataModel.cpp
Salome HOME
Make property panel as a GUI of an operation
[modules/shaper.git] / src / XGUI / XGUI_DocumentDataModel.cpp
1 #include "XGUI_DocumentDataModel.h"
2 #include "XGUI_PartDataModel.h"
3 #include "XGUI_Workshop.h"
4 #include "XGUI_Tools.h"
5
6 #include <ModelAPI_Session.h>
7 #include <ModelAPI_Document.h>
8 #include <ModelAPI_Feature.h>
9 #include <ModelAPI_Data.h>
10 #include <ModelAPI_ResultPart.h>
11 #include <ModelAPI_Events.h>
12 #include <ModelAPI_Object.h>
13
14 #include <Events_Loop.h>
15
16 #include <Config_FeatureMessage.h>
17
18 #include <QIcon>
19 #include <QString>
20 #include <QBrush>
21
22 #include <set>
23
24 #define ACTIVE_COLOR QColor(0,72,140)
25 #define PASSIVE_COLOR Qt::black
26
27 XGUI_DocumentDataModel::XGUI_DocumentDataModel(QObject* theParent)
28     : QAbstractItemModel(theParent),
29       myActivePart(0)
30 {
31   // Register in event loop
32   //Events_Loop* aLoop = Events_Loop::loop();
33   //aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_CREATED));
34   //aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
35   //aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_DELETED));
36
37   // Create a top part of data tree model
38   myModel = new XGUI_TopDataModel(this);
39   myModel->setItemsColor(ACTIVE_COLOR);
40 }
41
42 XGUI_DocumentDataModel::~XGUI_DocumentDataModel()
43 {
44   clearModelIndexes();
45 }
46
47 void XGUI_DocumentDataModel::processEvent(const boost::shared_ptr<Events_Message>& theMessage)
48 {
49   DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
50
51   // Created object event *******************
52   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_CREATED)) {
53     boost::shared_ptr<ModelAPI_ObjectUpdatedMessage> aUpdMsg =
54         boost::dynamic_pointer_cast<ModelAPI_ObjectUpdatedMessage>(theMessage);
55     std::set<ObjectPtr> aObjects = aUpdMsg->objects();
56
57     std::set<ObjectPtr>::const_iterator aIt;
58     for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) {
59       ObjectPtr aObject = (*aIt);
60       FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(aObject);
61       if (aFeature && (!aFeature->isInHistory()))
62         continue;
63
64       DocumentPtr aDoc = aObject->document();
65       if (aDoc == aRootDoc) {  // If root objects
66         if (aObject->groupName() == ModelAPI_ResultPart::group()) {  // Update only Parts group
67             // Add a new part
68           int aStart = myPartModels.size();
69           XGUI_PartDataModel* aModel = new XGUI_PartDataModel(this);
70           aModel->setPartId(myPartModels.count());
71           myPartModels.append(aModel);
72           insertRow(aStart, partFolderNode());
73         } else {  // Update top groups (other except parts
74           QModelIndex aIndex = myModel->findParent(aObject);
75           int aStart = myModel->rowCount(aIndex) - 1;
76           if (aStart < 0)
77             aStart = 0;
78           aIndex = createIndex(aIndex.row(), aIndex.column(), (void*) getModelIndex(aIndex));
79           insertRow(aStart, aIndex);
80         }
81       } else {  // if sub-objects of first level nodes
82         XGUI_PartModel* aPartModel = 0;
83         QList<XGUI_PartModel*>::const_iterator aIt;
84         for (aIt = myPartModels.constBegin(); aIt != myPartModels.constEnd(); ++aIt) {
85           if ((*aIt)->hasDocument(aDoc)) {
86             aPartModel = (*aIt);
87             break;
88           }
89         }
90         if (aPartModel) {
91           QModelIndex aIndex = aPartModel->findParent(aObject);
92           int aStart = aPartModel->rowCount(aIndex);  // check this index
93           aIndex = createIndex(aIndex.row(), aIndex.column(), (void*) getModelIndex(aIndex));
94           insertRow(aStart, aIndex);
95         }
96       }
97     }
98     // Deleted object event ***********************
99   } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED)) {
100     boost::shared_ptr<ModelAPI_ObjectDeletedMessage> aUpdMsg =
101         boost::dynamic_pointer_cast<ModelAPI_ObjectDeletedMessage>(theMessage);
102     DocumentPtr aDoc = aUpdMsg->document();
103     std::set<std::string> aGroups = aUpdMsg->groups();
104
105     std::set<std::string>::const_iterator aIt;
106     for (aIt = aGroups.begin(); aIt != aGroups.end(); ++aIt) {
107       std::string aGroup = (*aIt);
108       if (aDoc == aRootDoc) {  // If root objects
109         if (aGroup == ModelAPI_ResultPart::group()) {  // Update only Parts group
110           int aStart = myPartModels.size() - 1;
111           removeSubModel(aStart);
112           removeRow(aStart, partFolderNode());
113           if (myActivePart && (!isPartSubModel(myActivePart))) {
114             myActivePart = 0;
115             myActivePartIndex = QModelIndex();
116             myModel->setItemsColor(ACTIVE_COLOR);
117           }
118         } else {  // Update top groups (other except parts
119           QModelIndex aIndex = myModel->findGroup(aGroup);
120           int aStart = myModel->rowCount(aIndex);
121           aIndex = createIndex(aIndex.row(), aIndex.column(), (void*) getModelIndex(aIndex));
122           removeRow(aStart, aIndex);
123         }
124       } else {
125         XGUI_PartModel* aPartModel = 0;
126         QList<XGUI_PartModel*>::const_iterator aIt;
127         for (aIt = myPartModels.constBegin(); aIt != myPartModels.constEnd(); ++aIt) {
128           if ((*aIt)->hasDocument(aDoc)) {
129             aPartModel = (*aIt);
130             break;
131           }
132         }
133         if (aPartModel) {
134           QModelIndex aIndex = aPartModel->findGroup(aGroup);
135           int aStart = aPartModel->rowCount(aIndex);
136           aIndex = createIndex(aIndex.row(), aIndex.column(), (void*) getModelIndex(aIndex));
137           removeRow(aStart, aIndex);
138         }
139       }
140     }
141     // Deleted object event ***********************
142   } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED)) {
143     //boost::shared_ptr<ModelAPI_ObjectUpdatedMessage> aUpdMsg = boost::dynamic_pointer_cast<ModelAPI_ObjectUpdatedMessage>(theMessage);
144     //ObjectPtr aFeature = aUpdMsg->feature();
145     //DocumentPtr aDoc = aFeature->document();
146
147     // TODO: Identify the necessary index by the modified feature
148     QModelIndex aIndex;
149     emit dataChanged(aIndex, aIndex);
150
151     // Reset whole tree **************************
152   } else {
153     rebuildDataTree();
154   }
155 }
156
157 void XGUI_DocumentDataModel::rebuildDataTree()
158 {
159   DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
160
161   beginResetModel();
162   clearModelIndexes();
163
164   int aNbParts = aRootDoc->size(ModelAPI_ResultPart::group());
165   if (myPartModels.size() != aNbParts) {  // resize internal models
166     while (myPartModels.size() > aNbParts) {
167       delete myPartModels.last();
168       myPartModels.removeLast();
169     }
170     while (myPartModels.size() < aNbParts) {
171       myPartModels.append(new XGUI_PartDataModel(this));
172     }
173     for (int i = 0; i < myPartModels.size(); i++)
174       myPartModels.at(i)->setPartId(i);
175   }
176   endResetModel();
177 }
178
179 QVariant XGUI_DocumentDataModel::data(const QModelIndex& theIndex, int theRole) const
180 {
181   if (!theIndex.isValid())
182     return QVariant();
183   switch (theIndex.internalId()) {
184     case PartsFolder:
185       switch (theRole) {
186         case Qt::DisplayRole:
187           return tr("Parts") + QString(" (%1)").arg(rowCount(theIndex));
188         case Qt::DecorationRole:
189           return QIcon(":pictures/constr_folder.png");
190         case Qt::ToolTipRole:
191           return tr("Parts folder");
192         case Qt::ForegroundRole:
193           if (myActivePart)
194             return QBrush(PASSIVE_COLOR);
195             else
196             return QBrush(ACTIVE_COLOR);
197             default:
198             return QVariant();
199           }
200           break;
201           case HistoryNode:
202           {
203             int aOffset = historyOffset();
204             DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
205             ObjectPtr aObj = aRootDoc->object(ModelAPI_Feature::group(), theIndex.row() - aOffset);
206             FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
207             if (!aFeature)
208             return QVariant();
209             switch (theRole) {
210               case Qt::DisplayRole:
211               if (aFeature)
212               return aFeature->data()->name().c_str();
213               else
214               return QVariant();
215               case Qt::DecorationRole:
216               return QIcon(XGUI_Workshop::featureIcon(aFeature->getKind()));
217               case Qt::ToolTipRole:
218               return tr("Feature object");
219               case Qt::ForegroundRole:
220               if (myActivePart)
221               return QBrush(PASSIVE_COLOR);
222               else
223               return QBrush(ACTIVE_COLOR);
224               default:
225               return QVariant();
226             }
227           }
228           break;
229         }
230   QModelIndex aParent = theIndex.parent();
231   if (aParent.isValid() && (aParent.internalId() == PartsFolder)) {
232     return myPartModels.at(theIndex.row())->data(QModelIndex(), theRole);
233   }
234   return toSourceModelIndex(theIndex)->data(theRole);
235 }
236
237 QVariant XGUI_DocumentDataModel::headerData(int theSection, Qt::Orientation theOrient,
238                                             int theRole) const
239 {
240   return QVariant();
241 }
242
243 int XGUI_DocumentDataModel::rowCount(const QModelIndex& theParent) const
244 {
245   if (!theParent.isValid()) {
246     DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
247     // Size of external models
248     int aVal = historyOffset();
249     // Plus history size
250     aVal += aRootDoc->size(ModelAPI_Feature::group());
251     return aVal;
252   }
253   if (theParent.internalId() == PartsFolder) {
254     int aSize = myPartModels.size();
255     return myPartModels.size();
256   }
257   if (theParent.internalId() == HistoryNode) {
258     return 0;
259   }
260   QModelIndex* aParent = toSourceModelIndex(theParent);
261   const QAbstractItemModel* aModel = aParent->model();
262   if (!isSubModel(aModel))
263     return 0;
264
265   /*if (isPartSubModel(aModel)) {
266    if (aModel != myActivePart)
267    return 0;
268    }*/
269   return aModel->rowCount(*aParent);
270 }
271
272 int XGUI_DocumentDataModel::columnCount(const QModelIndex& theParent) const
273 {
274   return 1;
275 }
276
277 QModelIndex XGUI_DocumentDataModel::index(int theRow, int theColumn,
278                                           const QModelIndex& theParent) const
279 {
280   QModelIndex aIndex;
281   if (!theParent.isValid()) {
282     int aOffs = myModel->rowCount();
283     if (theRow < aOffs) {
284       aIndex = myModel->index(theRow, theColumn, theParent);
285       aIndex = createIndex(theRow, theColumn, (void*) getModelIndex(aIndex));
286     } else {
287       if (theRow == aOffs)  // Create Parts node
288         aIndex = partFolderNode();
289       else
290         // create history node
291         aIndex = createIndex(theRow, theColumn, HistoryNode);
292     }
293   } else {
294     if (theParent.internalId() == PartsFolder) {
295       aIndex = myPartModels.at(theRow)->index(0, theColumn, QModelIndex());
296     } else {
297       QModelIndex* aParent = (QModelIndex*) theParent.internalPointer();
298       aIndex = aParent->model()->index(theRow, theColumn, (*aParent));
299     }
300     aIndex = createIndex(theRow, theColumn, (void*) getModelIndex(aIndex));
301   }
302   return aIndex;
303 }
304
305 QModelIndex XGUI_DocumentDataModel::parent(const QModelIndex& theIndex) const
306 {
307   if ((theIndex.internalId() == PartsFolder) || (theIndex.internalId() == HistoryNode))
308     return QModelIndex();
309
310   QModelIndex* aIndex = toSourceModelIndex(theIndex);
311   const QAbstractItemModel* aModel = aIndex->model();
312   if (!isSubModel(aModel))
313     return QModelIndex();
314
315   if (isPartSubModel(aModel)) {
316     if (!aModel->parent(*aIndex).isValid()) {
317       return partFolderNode();
318     }
319   }
320
321   QModelIndex aIndex1 = aModel->parent(*aIndex);
322   if (aIndex1.isValid())
323     return createIndex(aIndex1.row(), aIndex1.column(), (void*) getModelIndex(aIndex1));
324   return aIndex1;
325 }
326
327 bool XGUI_DocumentDataModel::hasChildren(const QModelIndex& theParent) const
328 {
329   if (!theParent.isValid())
330     return true;
331   return rowCount(theParent) > 0;
332 }
333
334 QModelIndex* XGUI_DocumentDataModel::toSourceModelIndex(const QModelIndex& theProxy) const
335 {
336   QModelIndex* aIndexPtr = static_cast<QModelIndex*>(theProxy.internalPointer());
337   return aIndexPtr;
338 }
339
340 QModelIndex* XGUI_DocumentDataModel::findModelIndex(const QModelIndex& theIndex) const
341 {
342   QList<QModelIndex*>::const_iterator aIt;
343   for (aIt = myIndexes.constBegin(); aIt != myIndexes.constEnd(); ++aIt) {
344     QModelIndex* aIndex = (*aIt);
345     if ((*aIndex) == theIndex)
346       return aIndex;
347   }
348   return 0;
349 }
350
351 QModelIndex* XGUI_DocumentDataModel::getModelIndex(const QModelIndex& theIndex) const
352 {
353   QModelIndex* aIndexPtr = findModelIndex(theIndex);
354   if (!aIndexPtr) {
355     aIndexPtr = new QModelIndex(theIndex);
356     XGUI_DocumentDataModel* that = (XGUI_DocumentDataModel*) this;
357     that->myIndexes.append(aIndexPtr);
358   }
359   return aIndexPtr;
360 }
361
362 void XGUI_DocumentDataModel::clearModelIndexes()
363 {
364   QList<QModelIndex*>::const_iterator aIt;
365   for (aIt = myIndexes.constBegin(); aIt != myIndexes.constEnd(); ++aIt)
366     delete (*aIt);
367   myIndexes.clear();
368 }
369
370 ObjectPtr XGUI_DocumentDataModel::object(const QModelIndex& theIndex) const
371 {
372   if (theIndex.internalId() == PartsFolder)
373     return ObjectPtr();
374   if (theIndex.internalId() == HistoryNode) {
375     DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
376     int aOffset = historyOffset();
377     return aRootDoc->object(ModelAPI_Feature::group(), theIndex.row() - aOffset);
378   }
379   QModelIndex* aIndex = toSourceModelIndex(theIndex);
380   if (!isSubModel(aIndex->model()))
381     return ObjectPtr();
382
383   const XGUI_FeaturesModel* aModel = dynamic_cast<const XGUI_FeaturesModel*>(aIndex->model());
384   return aModel->object(*aIndex);
385 }
386
387 bool XGUI_DocumentDataModel::insertRows(int theRow, int theCount, const QModelIndex& theParent)
388 {
389   beginInsertRows(theParent, theRow, theRow + theCount - 1);
390   //endInsertRows();
391
392   // Update history
393   QModelIndex aRoot;
394   int aRow = rowCount(aRoot);
395   beginInsertRows(aRoot, aRow, aRow);
396   endInsertRows();
397
398   return true;
399 }
400
401 bool XGUI_DocumentDataModel::removeRows(int theRow, int theCount, const QModelIndex& theParent)
402 {
403   beginRemoveRows(theParent, theRow, theRow + theCount - 1);
404   endRemoveRows();
405   return true;
406 }
407
408 void XGUI_DocumentDataModel::removeSubModel(int theModelId)
409 {
410   XGUI_PartModel* aModel = myPartModels.at(theModelId);
411   QIntList aToRemove;
412   for (int i = 0; i < myIndexes.size(); i++) {
413     if (myIndexes.at(i)->model() == aModel)
414       aToRemove.append(i);
415   }
416   int aId;
417   while (aToRemove.size() > 0) {
418     aId = aToRemove.last();
419     delete myIndexes.at(aId);
420     myIndexes.removeAt(aId);
421     aToRemove.removeLast();
422   }
423   delete aModel;
424   myPartModels.removeAt(theModelId);
425 }
426
427 bool XGUI_DocumentDataModel::isSubModel(const QAbstractItemModel* theModel) const
428 {
429   if (theModel == myModel)
430     return true;
431   return isPartSubModel(theModel);
432 }
433
434 bool XGUI_DocumentDataModel::isPartSubModel(const QAbstractItemModel* theModel) const
435 {
436   return myPartModels.contains((XGUI_PartModel*) theModel);
437 }
438
439 QModelIndex XGUI_DocumentDataModel::partFolderNode() const
440 {
441   int aPos = myModel->rowCount(QModelIndex());
442   return createIndex(aPos, columnCount() - 1, PartsFolder);
443 }
444
445 int XGUI_DocumentDataModel::historyOffset() const
446 {
447   // Nb of rows of top model + Parts folder
448   return myModel->rowCount(QModelIndex()) + 1;
449 }
450
451 bool XGUI_DocumentDataModel::activatedIndex(const QModelIndex& theIndex)
452 {
453   if ((theIndex.internalId() == PartsFolder) || (theIndex.internalId() == HistoryNode))
454     return false;
455
456   QModelIndex* aIndex = toSourceModelIndex(theIndex);
457   if (!aIndex)
458     return false;
459
460   const QAbstractItemModel* aModel = aIndex->model();
461
462   if (isPartSubModel(aModel)) {
463     // if this is root node (Part item index)
464     if (!aIndex->parent().isValid()) {
465       if (myActivePart)
466         myActivePart->setItemsColor(PASSIVE_COLOR);
467
468         if (myActivePart == aModel) {
469           myActivePart = 0;
470           myActivePartIndex = QModelIndex();
471         } else {
472           myActivePart = (XGUI_PartModel*)aModel;
473           myActivePartIndex = theIndex;
474         }
475
476         if (myActivePart) {
477           myActivePart->setItemsColor(ACTIVE_COLOR);
478           myModel->setItemsColor(PASSIVE_COLOR);
479         } else
480           myModel->setItemsColor(ACTIVE_COLOR);
481         return true;
482       }
483     }
484   return false;
485 }
486
487 ResultPartPtr XGUI_DocumentDataModel::activePart() const
488 {
489   if (myActivePart)
490     return myActivePart->part();
491   return ResultPartPtr();
492 }
493
494 void XGUI_DocumentDataModel::deactivatePart()
495 {
496   if (myActivePart)
497     myActivePart->setItemsColor(PASSIVE_COLOR);
498     myActivePart = 0;
499     myActivePartIndex = QModelIndex();
500     myModel->setItemsColor(ACTIVE_COLOR);
501   }
502
503 Qt::ItemFlags XGUI_DocumentDataModel::flags(const QModelIndex& theIndex) const
504 {
505   Qt::ItemFlags aFlags = QAbstractItemModel::flags(theIndex);
506   if (object(theIndex)) {
507     aFlags |= Qt::ItemIsEditable;
508   }
509   return aFlags;
510 }
511
512 QModelIndex XGUI_DocumentDataModel::partIndex(const ResultPartPtr& theObject) const
513 {
514   int aRow = -1;
515   XGUI_PartModel* aModel = 0;
516   foreach (XGUI_PartModel* aPartModel, myPartModels)
517   {
518     aRow++;
519     if (aPartModel->part() == theObject) {
520       aModel = aPartModel;
521       break;
522     }
523   }
524   if (aModel) {
525     return createIndex(aRow, 0, (void*) getModelIndex(aModel->index(0, 0, QModelIndex())));
526   }
527   return QModelIndex();
528 }
529
530 QModelIndex XGUI_DocumentDataModel::objectIndex(const ObjectPtr theObject) const
531 {
532   // Check that this feature belongs to root document
533   DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
534   DocumentPtr aDoc = theObject->document();
535   if (aDoc == aRootDoc) {
536     // This feature belongs to histrory or top model
537     FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(theObject);
538     if (aFeature) {
539       int aId;
540       int aNb = aRootDoc->size(ModelAPI_Feature::group());
541       for (aId = 0; aId < aNb; aId++) {
542         if (theObject == aRootDoc->object(ModelAPI_Feature::group(), aId))
543           break;
544       }
545       if (aId < aNb)
546         return index(aId + historyOffset(), 0, QModelIndex());
547     } else {
548       QModelIndex aIndex = myModel->objectIndex(theObject);
549       return
550           aIndex.isValid() ?
551               createIndex(aIndex.row(), aIndex.column(), (void*) getModelIndex(aIndex)) :
552               QModelIndex();
553     }
554   } else {
555     XGUI_PartModel* aPartModel = 0;
556     foreach(XGUI_PartModel* aModel, myPartModels)
557     {
558       if (aModel->hasDocument(aDoc)) {
559         aPartModel = aModel;
560         break;
561       }
562     }
563     if (aPartModel) {
564       QModelIndex aIndex = aPartModel->objectIndex(theObject);
565       return
566           aIndex.isValid() ?
567               createIndex(aIndex.row(), aIndex.column(), (void*) getModelIndex(aIndex)) :
568               QModelIndex();
569     }
570   }
571   return QModelIndex();
572 }