1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
3 #include "PartSet_PartDataModel.h"
4 #include "PartSet_Module.h"
5 #include "PartSet_DocumentDataModel.h"
7 #include <ModelAPI_Session.h>
8 #include <ModelAPI_Document.h>
9 #include <ModelAPI_Feature.h>
10 #include <ModelAPI_Result.h>
11 #include <ModelAPI_Data.h>
12 #include <ModelAPI_AttributeDocRef.h>
13 #include <ModelAPI_Object.h>
14 #include <ModelAPI_ResultPart.h>
15 #include <ModelAPI_ResultConstruction.h>
16 #include <ModelAPI_ResultParameter.h>
17 #include <ModelAPI_ResultBody.h>
18 #include <ModelAPI_ResultGroup.h>
19 #include <ModelAPI_AttributeDouble.h>
20 #include <ModelAPI_Events.h>
21 #include <ModelAPI_Tools.h>
23 #include <Events_Loop.h>
29 PartSet_PartDataModel::PartSet_PartDataModel(QObject* theParent)
30 : PartSet_PartModel(theParent)
34 PartSet_PartDataModel::~PartSet_PartDataModel()
38 QVariant PartSet_PartDataModel::data(const QModelIndex& theIndex, int theRole) const
40 DocumentPtr aPartDoc = partDocument();
41 if (theIndex.column() == 1) {
42 DocumentPtr aActiveDoc = ModelAPI_Session::get()->activeDocument();
43 QModelIndex aParent = theIndex.parent();
44 if (aActiveDoc == aPartDoc) {
45 if (!aParent.isValid()) {
47 case Qt::DecorationRole:
48 if (theIndex.row() == lastHistoryRow())
49 return QIcon(":pictures/arrow.png");
56 if (theIndex.internalId() >= 0) {
57 ObjectPtr aObj = object(theIndex);
60 return aObj->data()->name().c_str();
61 case Qt::DecorationRole:
63 FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
65 return PartSet_DocumentDataModel::featureIcon(aFeature);
68 case Qt::ForegroundRole:
69 if (theIndex.internalId() > lastHistoryRow())
70 return QBrush(Qt::lightGray);
71 return QBrush(myItemsColor);
78 switch (theIndex.internalId()) {
80 // DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
81 // ObjectPtr aObject = aRootDoc->object(ModelAPI_ResultPart::group(), myId);
83 // return std::dynamic_pointer_cast<ModelAPI_Object>(aObject)->data()->name().c_str();
86 return tr("Parameters") + QString(" (%1)").arg(rowCount(theIndex));
88 return tr("Constructions") + QString(" (%1)").arg(rowCount(theIndex));
90 return tr("Bodies") + QString(" (%1)").arg(rowCount(theIndex));
92 return tr("Groups") + QString(" (%1)").arg(rowCount(theIndex));
94 ObjectPtr aObject = aPartDoc->object(ModelAPI_ResultParameter::group(), theIndex.row());
96 ResultParameterPtr aParam = std::dynamic_pointer_cast<ModelAPI_ResultParameter>(aObject);
97 AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
98 QString aVal = QString::number(aValueAttribute->value());
99 QString aTitle = QString(aObject->data()->name().c_str());
100 return aTitle + " = " + aVal;
104 case ConstructObject: {
105 ObjectPtr aObject = aPartDoc->object(ModelAPI_ResultConstruction::group(), theIndex.row());
107 return std::dynamic_pointer_cast<ModelAPI_Object>(aObject)->data()->name().c_str();
111 ObjectPtr aObject = aPartDoc->object(ModelAPI_ResultBody::group(), theIndex.row());
113 return aObject->data()->name().c_str();
117 ObjectPtr aObject = aPartDoc->object(ModelAPI_ResultGroup::group(), theIndex.row());
119 return aObject->data()->name().c_str();
121 case HistoryObject: {
122 ObjectPtr aObject = aPartDoc->object(ModelAPI_Feature::group(), theIndex.row() - getRowsNumber());
124 return aObject->data()->name().c_str();
128 case Qt::DecorationRole:
130 switch (theIndex.internalId()) {
132 // return QIcon(":pictures/part_ico.png");
134 return QIcon(":pictures/params_folder.png");
135 case ConstructFolder:
137 return QIcon(":pictures/constr_folder.png");
139 return QIcon(":pictures/constr_folder.png");
140 case ConstructObject:
143 std::string aGroup = theIndex.internalId() == ConstructObject ?
144 ModelAPI_ResultConstruction::group() : ModelAPI_ResultBody::group();
145 ObjectPtr anObject = aPartDoc->object(aGroup, theIndex.row());
146 if (anObject && anObject->data() &&
147 anObject->data()->execState() == ModelAPI_StateMustBeUpdated) {
148 return QIcon(":pictures/constr_object_modified.png");
150 return QIcon(":pictures/constr_object.png");
152 case HistoryObject: {
153 ObjectPtr aObject = aPartDoc->object(ModelAPI_Feature::group(), theIndex.row() - getRowsNumber());
154 FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObject);
156 return PartSet_DocumentDataModel::featureIcon(aFeature);
160 case Qt::ToolTipRole:
163 case Qt::ForegroundRole:
164 if (theIndex.internalId() == HistoryObject) {
165 if (theIndex.row() > lastHistoryRow())
166 return QBrush(Qt::lightGray);
168 return QBrush(myItemsColor);
173 QVariant PartSet_PartDataModel::headerData(int section, Qt::Orientation orientation, int role) const
178 int PartSet_PartDataModel::rowCount(const QModelIndex& parent) const
180 if (!parent.isValid()) {
181 DocumentPtr aDoc = partDocument();
183 return getRowsNumber() + aDoc->size(ModelAPI_Feature::group());
187 switch (parent.internalId()) {
189 return partDocument()->size(ModelAPI_ResultParameter::group());
190 case ConstructFolder:
191 return partDocument()->size(ModelAPI_ResultConstruction::group());
193 return partDocument()->size(ModelAPI_ResultBody::group());
195 return partDocument()->size(ModelAPI_ResultGroup::group());
198 ObjectPtr aObj = object(parent);
199 CompositeFeaturePtr aCompFeature =
200 std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(aObj);
201 if (aCompFeature.get()) {
202 return aCompFeature->numberOfSubs(true);
209 int PartSet_PartDataModel::columnCount(const QModelIndex &parent) const
214 QModelIndex PartSet_PartDataModel::index(int theRow, int theColumn, const QModelIndex &theParent) const
216 if (!theParent.isValid()) {
219 return createIndex(theRow, theColumn, (qint32) ParamsFolder);
221 return createIndex(theRow, theColumn, (qint32) ConstructFolder);
223 return createIndex(theRow, theColumn, (qint32) BodiesFolder);
226 int aSize = partDocument()->size(ModelAPI_ResultGroup::group());
228 return createIndex(theRow, theColumn, (qint32) GroupsFolder);
230 return createIndex(theRow, theColumn, (qint32) HistoryObject);
233 return createIndex(theRow, theColumn, (qint32) HistoryObject);
236 int aId = (int) theParent.internalId();
239 return createIndex(theRow, theColumn, (qint32) ParamObject);
240 case ConstructFolder:
241 return createIndex(theRow, theColumn, (qint32) ConstructObject);
243 return createIndex(theRow, theColumn, (qint32) BodiesObject);
245 return createIndex(theRow, theColumn, (qint32) GroupObject);
248 return createIndex(theRow, theColumn, (qint32) theParent.row());
252 return QModelIndex();
255 QModelIndex PartSet_PartDataModel::parent(const QModelIndex& theIndex) const
257 if (theIndex.internalId() >= 0) {
258 int aPRow = theIndex.internalId();
259 return createIndex(aPRow, 0, (qint32) HistoryObject);
261 switch (theIndex.internalId()) {
263 case ConstructFolder:
267 return QModelIndex();
270 return createIndex(0, 0, (qint32) ParamsFolder);
271 case ConstructObject:
272 return createIndex(1, 0, (qint32) ConstructFolder);
274 return createIndex(2, 0, (qint32) BodiesFolder);
276 return createIndex(3, 0, (qint32) GroupsFolder);
278 return QModelIndex();
281 bool PartSet_PartDataModel::hasChildren(const QModelIndex& theParent) const
283 return rowCount(theParent) > 0;
286 DocumentPtr PartSet_PartDataModel::partDocument() const
288 ResultPartPtr aPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(myPart->firstResult());
289 if (aPart.get()) // this may be null is Part feature is disabled
290 return aPart->partDoc();
291 return DocumentPtr();
294 ObjectPtr PartSet_PartDataModel::object(const QModelIndex& theIndex) const
296 if (theIndex.internalId() >= 0) {
297 int aPRow = theIndex.internalId();
299 partDocument()->object(ModelAPI_Feature::group(), aPRow - getRowsNumber());
300 CompositeFeaturePtr aCompFeature =
301 std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(aObj);
302 if (aCompFeature.get()) {
303 return aCompFeature->subFeature(theIndex.row(), true);
307 switch (theIndex.internalId()) {
309 case ConstructFolder:
315 return partDocument()->object(ModelAPI_ResultParameter::group(), theIndex.row());
316 case ConstructObject:
317 return partDocument()->object(ModelAPI_ResultConstruction::group(), theIndex.row());
319 return partDocument()->object(ModelAPI_ResultBody::group(), theIndex.row());
321 return partDocument()->object(ModelAPI_ResultGroup::group(), theIndex.row());
323 return partDocument()->object(ModelAPI_Feature::group(), theIndex.row() - getRowsNumber());
328 bool PartSet_PartDataModel::hasDocument(const DocumentPtr& theDoc) const
330 return (partDocument() == theDoc);
333 QModelIndex PartSet_PartDataModel::findParent(const ObjectPtr& theObject) const
335 return findGroup(theObject->groupName().c_str());
338 QModelIndex PartSet_PartDataModel::findGroup(const std::string& theGroup) const
340 if (theGroup == ModelAPI_ResultParameter::group())
341 return createIndex(0, 0, (qint32) ParamsFolder);
342 if (theGroup == ModelAPI_ResultConstruction::group())
343 return createIndex(1, 0, (qint32) ConstructFolder);
344 if (theGroup == ModelAPI_ResultBody::group())
345 return createIndex(2, 0, (qint32) BodiesFolder);
346 if (theGroup == ModelAPI_ResultGroup::group())
347 return createIndex(3, 0, (qint32) GroupsFolder);
348 return QModelIndex();
351 QModelIndex PartSet_PartDataModel::objectIndex(const ObjectPtr& theObject) const
355 if (part() == theObject)
358 std::string aGroup = theObject->groupName();
359 DocumentPtr aDoc = theObject->document();
360 int aNb = aDoc->size(aGroup);
362 for (int i = 0; i < aNb; i++) {
363 if (aDoc->object(aGroup, i) == theObject) {
370 if (aGroup == ModelAPI_ResultParameter::group())
371 return createIndex(aRow, 0, (qint32) ParamObject);
372 else if (aGroup == ModelAPI_ResultConstruction::group())
373 return createIndex(aRow, 0, (qint32) ConstructObject);
374 else if (aGroup == ModelAPI_ResultBody::group())
375 return createIndex(aRow, 0, (qint32) BodiesObject);
376 else if (aGroup == ModelAPI_ResultGroup::group())
377 return createIndex(aRow, 0, (qint32) GroupObject);
379 return createIndex(aRow + getRowsNumber(), 0, (qint32) HistoryObject);
385 int PartSet_PartDataModel::getRowsNumber() const
387 int aSize = partDocument()->size(ModelAPI_ResultGroup::group());
388 if (aSize == 0) // If there are no groups then do not show group folder
393 int PartSet_PartDataModel::lastHistoryRow() const
395 DocumentPtr aDoc = partDocument();
396 FeaturePtr aFeature = aDoc->currentFeature(true);
398 return getRowsNumber() + aDoc->index(aFeature);
400 return getRowsNumber() - 1;
403 void PartSet_PartDataModel::setLastHistoryItem(const QModelIndex& theIndex)
405 SessionPtr aMgr = ModelAPI_Session::get();
406 DocumentPtr aDoc = partDocument();
407 std::string aOpName = tr("History change").toStdString();
408 if (theIndex.internalId() == HistoryObject) {
409 ObjectPtr aObject = object(theIndex);
410 aMgr->startOperation(aOpName);
411 aDoc->setCurrentFeature(std::dynamic_pointer_cast<ModelAPI_Feature>(aObject), true);
412 aMgr->finishOperation();
414 aMgr->startOperation(aOpName);
415 aDoc->setCurrentFeature(FeaturePtr(), true);
416 aMgr->finishOperation();
420 QModelIndex PartSet_PartDataModel::lastHistoryItem() const
422 return index(lastHistoryRow(), 1);
425 Qt::ItemFlags PartSet_PartDataModel::flags(const QModelIndex& theIndex) const
427 // Disable sub-features at column 1
428 if ((theIndex.column() == 1) && (theIndex.internalId() >= 0))
431 Qt::ItemFlags aFlags = Qt::ItemIsSelectable;
432 if (object(theIndex)) {
433 aFlags |= Qt::ItemIsEditable;
436 if (theIndex.internalId() == HistoryObject) {
437 if (theIndex.row() <= lastHistoryRow() || (theIndex.column() == 1))
438 aFlags |= Qt::ItemIsEnabled;
440 aFlags |= Qt::ItemIsEnabled;