1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
3 #include "XGUI_PartDataModel.h"
4 #include "XGUI_Workshop.h"
6 #include <ModelAPI_Session.h>
7 #include <ModelAPI_Document.h>
8 #include <ModelAPI_Feature.h>
9 #include <ModelAPI_Result.h>
10 #include <ModelAPI_Data.h>
11 #include <ModelAPI_AttributeDocRef.h>
12 #include <ModelAPI_Object.h>
13 #include <ModelAPI_ResultPart.h>
14 #include <ModelAPI_ResultConstruction.h>
15 #include <ModelAPI_ResultParameters.h>
16 #include <ModelAPI_ResultBody.h>
17 #include <ModelAPI_ResultGroup.h>
22 //ObjectPtr featureObj(const ObjectPtr& theFeature)
24 // ObjectPtr aObject = std::dynamic_pointer_cast<ModelAPI_Object>(theFeature);
26 // return aObject->featureRef();
30 XGUI_TopDataModel::XGUI_TopDataModel(QObject* theParent)
31 : XGUI_FeaturesModel(theParent)
35 XGUI_TopDataModel::~XGUI_TopDataModel()
39 QVariant XGUI_TopDataModel::data(const QModelIndex& theIndex, int theRole) const
44 switch (theIndex.internalId()) {
46 return tr("Parameters") + QString(" (%1)").arg(rowCount(theIndex));
48 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
49 ObjectPtr aObject = aRootDoc->object(ModelAPI_ResultParameters::group(), theIndex.row());
51 return aObject->data()->name().c_str();
55 return tr("Constructions") + QString(" (%1)").arg(rowCount(theIndex));
56 case ConstructObject: {
57 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
58 ObjectPtr aObject = aRootDoc->object(ModelAPI_ResultConstruction::group(),
61 return aObject->data()->name().c_str();
65 // return tr("Groups") + QString(" (%1)").arg(rowCount(theIndex));
67 // DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
68 // ObjectPtr aObject = aRootDoc->object(ModelAPI_ResultGroup::group(),
71 // return aObject->data()->name().c_str();
77 case Qt::DecorationRole:
80 switch (theIndex.internalId()) {
82 return QIcon(":pictures/params_folder.png");
84 return QIcon(":pictures/constr_folder.png");
86 return QIcon(":pictures/constr_object.png");
88 // return QIcon(":pictures/constr_folder.png");
96 case Qt::ForegroundRole:
97 return QBrush(myItemsColor);
103 QVariant XGUI_TopDataModel::headerData(int section, Qt::Orientation orientation, int role) const
108 int XGUI_TopDataModel::rowCount(const QModelIndex& theParent) const
110 if (!theParent.isValid())
111 return 2; // In case of groups using it has to be +1
113 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
114 if (theParent.internalId() == ParamsFolder)
115 return aRootDoc->size(ModelAPI_ResultParameters::group());
117 if (theParent.internalId() == ConstructFolder)
118 return aRootDoc->size(ModelAPI_ResultConstruction::group());
120 //if (theParent.internalId() == GroupsFolder)
121 // return aRootDoc->size(ModelAPI_ResultGroup::group());
126 int XGUI_TopDataModel::columnCount(const QModelIndex &parent) const
131 QModelIndex XGUI_TopDataModel::index(int theRow, int theColumn, const QModelIndex& theParent) const
133 if (!theParent.isValid()) {
136 return createIndex(theRow, theColumn, (qint32) ParamsFolder);
138 return createIndex(theRow, theColumn, (qint32) ConstructFolder);
140 // return createIndex(theRow, theColumn, (qint32) GroupsFolder);
143 if (theParent.internalId() == ParamsFolder)
144 return createIndex(theRow, theColumn, (qint32) ParamObject);
146 if (theParent.internalId() == ConstructFolder)
147 return createIndex(theRow, theColumn, (qint32) ConstructObject);
149 //if (theParent.internalId() == GroupsFolder)
150 // return createIndex(theRow, theColumn, (qint32) GroupObject);
152 return QModelIndex();
155 QModelIndex XGUI_TopDataModel::parent(const QModelIndex& theIndex) const
157 int aId = (int) theIndex.internalId();
160 case ConstructFolder:
162 return QModelIndex();
164 return createIndex(0, 0, (qint32) ParamsFolder);
165 case ConstructObject:
166 return createIndex(1, 0, (qint32) ConstructFolder);
168 // return createIndex(2, 0, (qint32) GroupsFolder);
170 return QModelIndex();
173 bool XGUI_TopDataModel::hasChildren(const QModelIndex& theParent) const
175 return rowCount(theParent) > 0;
178 ObjectPtr XGUI_TopDataModel::object(const QModelIndex& theIndex) const
180 switch (theIndex.internalId()) {
182 case ConstructFolder:
185 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
186 return aRootDoc->object(ModelAPI_ResultParameters::group(), theIndex.row());
188 case ConstructObject: {
189 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
190 return aRootDoc->object(ModelAPI_ResultConstruction::group(), theIndex.row());
192 //case GroupObject: {
193 // DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
194 // return aRootDoc->object(ModelAPI_ResultGroup::group(), theIndex.row());
200 QModelIndex XGUI_TopDataModel::findParent(const ObjectPtr& theObject) const
202 return findGroup(theObject->groupName().c_str());
205 QModelIndex XGUI_TopDataModel::findGroup(const std::string& theGroup) const
207 if (theGroup == ModelAPI_ResultParameters::group())
208 return createIndex(0, 0, (qint32) ParamsFolder);
209 if (theGroup == ModelAPI_ResultConstruction::group())
210 return createIndex(1, 0, (qint32) ConstructFolder);
211 //if (theGroup == ModelAPI_ResultGroup::group())
212 // return createIndex(2, 0, (qint32) ConstructFolder);
213 return QModelIndex();
216 QModelIndex XGUI_TopDataModel::objectIndex(const ObjectPtr& theObject) const
220 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
221 std::string aGroup = theObject->groupName();
222 int aNb = aRootDoc->size(aGroup);
224 for (int i = 0; i < aNb; i++) {
225 if (aRootDoc->object(aGroup, i) == theObject) {
231 if (aGroup == ModelAPI_ResultParameters::group())
232 return createIndex(aRow, 0, (qint32) ParamObject);
233 if (aGroup == ModelAPI_ResultConstruction::group())
234 return createIndex(aRow, 0, (qint32) ConstructObject);
235 //if (aGroup == ModelAPI_ResultGroup::group())
236 // return createIndex(aRow, 0, (qint32) GroupObject);
242 //******************************************************************
243 //******************************************************************
244 //******************************************************************
245 XGUI_PartDataModel::XGUI_PartDataModel(QObject* theParent)
246 : XGUI_PartModel(theParent)
250 XGUI_PartDataModel::~XGUI_PartDataModel()
254 QVariant XGUI_PartDataModel::data(const QModelIndex& theIndex, int theRole) const
257 case Qt::DisplayRole:
259 switch (theIndex.internalId()) {
261 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
262 ObjectPtr aObject = aRootDoc->object(ModelAPI_ResultPart::group(), myId);
264 return std::dynamic_pointer_cast<ModelAPI_Object>(aObject)->data()->name().c_str();
267 return tr("Parameters") + QString(" (%1)").arg(rowCount(theIndex));
268 case ConstructFolder:
269 return tr("Constructions") + QString(" (%1)").arg(rowCount(theIndex));
271 return tr("Bodies") + QString(" (%1)").arg(rowCount(theIndex));
273 return tr("Groups") + QString(" (%1)").arg(rowCount(theIndex));
275 ObjectPtr aObject = partDocument()->object(ModelAPI_ResultParameters::group(),
278 return std::dynamic_pointer_cast<ModelAPI_Object>(aObject)->data()->name().c_str();
281 case ConstructObject: {
282 ObjectPtr aObject = partDocument()->object(ModelAPI_ResultConstruction::group(),
285 return std::dynamic_pointer_cast<ModelAPI_Object>(aObject)->data()->name().c_str();
289 ObjectPtr aObject = partDocument()->object(ModelAPI_ResultBody::group(), theIndex.row());
291 return aObject->data()->name().c_str();
295 ObjectPtr aObject = partDocument()->object(ModelAPI_ResultGroup::group(), theIndex.row());
297 return aObject->data()->name().c_str();
299 case HistoryObject: {
300 ObjectPtr aObject = partDocument()->object(ModelAPI_Feature::group(), theIndex.row() - getRowsNumber());
302 return aObject->data()->name().c_str();
306 case Qt::DecorationRole:
308 switch (theIndex.internalId()) {
310 return QIcon(":pictures/part_ico.png");
312 return QIcon(":pictures/params_folder.png");
313 case ConstructFolder:
315 return QIcon(":pictures/constr_folder.png");
317 return QIcon(":pictures/constr_folder.png");
318 case ConstructObject:
321 std::string aGroup = theIndex.internalId() == ConstructObject ?
322 ModelAPI_ResultConstruction::group() : ModelAPI_ResultBody::group();
323 ObjectPtr anObject = partDocument()->object(aGroup, theIndex.row());
324 if (anObject && anObject->data() &&
325 anObject->data()->execState() == ModelAPI_StateMustBeUpdated) {
326 return QIcon(":pictures/constr_object_modified.png");
328 return QIcon(":pictures/constr_object.png");
330 case HistoryObject: {
331 ObjectPtr aObject = partDocument()->object(ModelAPI_Feature::group(), theIndex.row() - getRowsNumber());
332 FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObject);
334 return XGUI_Workshop::featureIcon(aFeature);
338 case Qt::ToolTipRole:
341 case Qt::ForegroundRole:
342 return QBrush(myItemsColor);
348 QVariant XGUI_PartDataModel::headerData(int section, Qt::Orientation orientation, int role) const
353 int XGUI_PartDataModel::rowCount(const QModelIndex& parent) const
355 if (!parent.isValid()) {
356 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
357 if (aRootDoc->object(ModelAPI_ResultPart::group(), myId))
362 switch (parent.internalId()) {
365 DocumentPtr aDoc = partDocument();
367 return getRowsNumber() + aDoc->size(ModelAPI_Feature::group());
372 return partDocument()->size(ModelAPI_ResultParameters::group());
373 case ConstructFolder:
374 return partDocument()->size(ModelAPI_ResultConstruction::group());
376 return partDocument()->size(ModelAPI_ResultBody::group());
378 return partDocument()->size(ModelAPI_ResultGroup::group());
383 int XGUI_PartDataModel::columnCount(const QModelIndex &parent) const
388 QModelIndex XGUI_PartDataModel::index(int theRow, int theColumn, const QModelIndex &theParent) const
390 if (!theParent.isValid())
391 return createIndex(theRow, 0, (qint32) MyRoot);
393 int aId = (int) theParent.internalId();
398 return createIndex(theRow, 0, (qint32) ParamsFolder);
400 return createIndex(theRow, 0, (qint32) ConstructFolder);
402 return createIndex(theRow, 0, (qint32) BodiesFolder);
405 int aSize = partDocument()->size(ModelAPI_ResultGroup::group());
407 return createIndex(theRow, 0, (qint32) GroupsFolder);
409 return createIndex(theRow, theColumn, (qint32) HistoryObject);
412 return createIndex(theRow, theColumn, (qint32) HistoryObject);
415 return createIndex(theRow, 0, (qint32) ParamObject);
416 case ConstructFolder:
417 return createIndex(theRow, 0, (qint32) ConstructObject);
419 return createIndex(theRow, 0, (qint32) BodiesObject);
421 return createIndex(theRow, 0, (qint32) GroupObject);
423 return QModelIndex();
426 QModelIndex XGUI_PartDataModel::parent(const QModelIndex& theIndex) const
428 switch (theIndex.internalId()) {
430 return QModelIndex();
432 case ConstructFolder:
436 return createIndex(0, 0, (qint32) MyRoot);
439 return createIndex(0, 0, (qint32) ParamsFolder);
440 case ConstructObject:
441 return createIndex(1, 0, (qint32) ConstructFolder);
443 return createIndex(2, 0, (qint32) BodiesFolder);
445 return createIndex(3, 0, (qint32) GroupsFolder);
447 return QModelIndex();
450 bool XGUI_PartDataModel::hasChildren(const QModelIndex& theParent) const
452 return rowCount(theParent) > 0;
455 DocumentPtr XGUI_PartDataModel::partDocument() const
457 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
458 ObjectPtr aObject = aRootDoc->object(ModelAPI_ResultPart::group(), myId);
459 ResultPartPtr aPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aObject);
461 return aPart->partDoc();
462 return DocumentPtr(); // null if not found
465 ObjectPtr XGUI_PartDataModel::object(const QModelIndex& theIndex) const
467 switch (theIndex.internalId()) {
469 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
470 return aRootDoc->object(ModelAPI_ResultPart::group(), myId);
473 case ConstructFolder:
479 return partDocument()->object(ModelAPI_ResultParameters::group(), theIndex.row());
480 case ConstructObject:
481 return partDocument()->object(ModelAPI_ResultConstruction::group(), theIndex.row());
483 return partDocument()->object(ModelAPI_ResultBody::group(), theIndex.row());
485 return partDocument()->object(ModelAPI_ResultGroup::group(), theIndex.row());
487 return partDocument()->object(ModelAPI_Feature::group(), theIndex.row() - getRowsNumber());
492 bool XGUI_PartDataModel::hasDocument(const DocumentPtr& theDoc) const
494 return (partDocument() == theDoc);
497 QModelIndex XGUI_PartDataModel::findParent(const ObjectPtr& theObject) const
499 return findGroup(theObject->groupName().c_str());
502 QModelIndex XGUI_PartDataModel::findGroup(const std::string& theGroup) const
504 if (theGroup == ModelAPI_ResultParameters::group())
505 return createIndex(0, 0, (qint32) ParamsFolder);
506 if (theGroup == ModelAPI_ResultConstruction::group())
507 return createIndex(1, 0, (qint32) ConstructFolder);
508 if (theGroup == ModelAPI_ResultBody::group())
509 return createIndex(2, 0, (qint32) BodiesFolder);
510 if (theGroup == ModelAPI_ResultGroup::group())
511 return createIndex(3, 0, (qint32) GroupsFolder);
512 return QModelIndex();
515 ResultPartPtr XGUI_PartDataModel::part() const
517 DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
518 ObjectPtr aObj = aRootDoc->object(ModelAPI_ResultPart::group(), myId);
519 return std::dynamic_pointer_cast<ModelAPI_ResultPart>(aObj);
522 QModelIndex XGUI_PartDataModel::objectIndex(const ObjectPtr& theObject) const
526 if (part() == theObject)
529 std::string aGroup = theObject->groupName();
530 DocumentPtr aDoc = theObject->document();
531 int aNb = aDoc->size(aGroup);
533 for (int i = 0; i < aNb; i++) {
534 if (aDoc->object(aGroup, i) == theObject) {
541 if (aGroup == ModelAPI_ResultParameters::group())
542 return createIndex(aRow, 0, (qint32) ParamObject);
543 else if (aGroup == ModelAPI_ResultConstruction::group())
544 return createIndex(aRow, 0, (qint32) ConstructObject);
545 else if (aGroup == ModelAPI_ResultBody::group())
546 return createIndex(aRow, 0, (qint32) BodiesObject);
547 else if (aGroup == ModelAPI_ResultGroup::group())
548 return createIndex(aRow, 0, (qint32) GroupObject);
550 return createIndex(aRow + getRowsNumber(), 0, (qint32) HistoryObject);
556 int XGUI_PartDataModel::getRowsNumber() const
558 int aSize = partDocument()->size(ModelAPI_ResultGroup::group());
559 if (aSize == 0) // If there are no groups then do not show group folder