#include <ModelAPI_Feature.h>
#include <ModelAPI_CompositeFeature.h>
#include <ModelAPI_ResultCompSolid.h>
+#include <ModelAPI_ResultField.h>
#include <ModelAPI_Tools.h>
#include <Config_FeatureMessage.h>
#include <Config_DataModelReader.h>
#include <Events_Loop.h>
-#include <Events_Error.h>
#include <QIcon>
#include <QBrush>
-#define ACTIVE_COLOR Qt::black
+#define ACTIVE_COLOR QColor(Qt::black)
//#define ACTIVE_COLOR QColor(0,72,140)
//#define PASSIVE_COLOR Qt::black
#define SELECTABLE_COLOR QColor(80, 80, 80)
#define DISABLED_COLOR QColor(200, 200, 200)
+
ResultPartPtr getPartResult(ModelAPI_Object* theObj)
{
ModelAPI_Feature* aFeature = dynamic_cast<ModelAPI_Feature*>(theObj);
// Constructor *************************************************
-XGUI_DataModel::XGUI_DataModel(QObject* theParent) : QAbstractItemModel(theParent)
+XGUI_DataModel::XGUI_DataModel(QObject* theParent) : QAbstractItemModel(theParent)//,
+ //myIsEventsProcessingBlocked(false)
{
Events_Loop* aLoop = Events_Loop::loop();
aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_CREATED));
XGUI_DataModel::~XGUI_DataModel()
{
+ clear();
}
//******************************************************
void XGUI_DataModel::processEvent(const std::shared_ptr<Events_Message>& theMessage)
{
+ //if (myIsEventsProcessingBlocked)
+ // return;
DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
std::string aRootType = myXMLReader->rootType();
std::string aSubType = myXMLReader->subType();
} else {
int aFolderId = myXMLReader->rootFolderId(aObjType);
if (aFolderId != -1) {
- insertRow(aRow, createIndex(aFolderId, 0, -1));
+ insertRow(aRow, createIndex(aFolderId, 0, (void*)Q_NULLPTR));
}
- }
+ }
}
} else {
// Object created in sub-document
}
}
}
+ } else {
+ rebuildDataTree();
+ break;
}
- } else
+ } else {
rebuildDataTree();
+ break;
+ }
}
}
// Deleted object event ***********************
std::shared_ptr<ModelAPI_ObjectDeletedMessage> aUpdMsg =
std::dynamic_pointer_cast<ModelAPI_ObjectDeletedMessage>(theMessage);
DocumentPtr aDoc = aUpdMsg->document();
- std::set<std::string> aGroups = aUpdMsg->groups();
- std::set<std::string>::const_iterator aIt;
+ std::set<std::string> aMsgGroups = aUpdMsg->groups();
+
+ /// Sort groups because RootType deletion has to be done after others
+ std::string aType = (aDoc == aRootDoc)? aRootType : aSubType;
+ std::list<std::string> aGroups;
+ std::set<std::string>::const_iterator aSetIt;
+ for (aSetIt = aMsgGroups.begin(); aSetIt != aMsgGroups.end(); ++aSetIt) {
+ std::string aGroup = (*aSetIt);
+ if (aGroup == aType)
+ aGroups.push_back(aGroup);
+ else
+ aGroups.push_front(aGroup);
+ }
+
+ std::list<std::string>::const_iterator aIt;
for (aIt = aGroups.begin(); aIt != aGroups.end(); ++aIt) {
std::string aGroup = (*aIt);
if (aDoc == aRootDoc) { // If root objects
// Process root sub-folder
int aFolderId = myXMLReader->rootFolderId(aGroup);
if (aFolderId != -1) {
- QModelIndex aFolderIndex = createIndex(aFolderId, 0, -1);
+ QModelIndex aFolderIndex = createIndex(aFolderId, 0, (void*)Q_NULLPTR);
removeRow(aRow, aFolderIndex);
//rebuildBranch(0, aRow);
}
break;
}
}
+ } else {
+ rebuildDataTree();
+ break;
}
}
}
for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) {
ObjectPtr aObject = (*aIt);
if (aObject->data()->isValid()) {
- QModelIndex aIndex = objectIndex(aObject);
- if (aIndex.isValid()) {
- emit dataChanged(aIndex, aIndex);
+ FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObject);
+ if (aFeature.get() && aFeature->firstResult().get()
+ && (aFeature->firstResult()->groupName() == ModelAPI_ResultField::group())) {
+ ResultFieldPtr aResult =
+ std::dynamic_pointer_cast<ModelAPI_ResultField>(aFeature->firstResult());
+ QModelIndex aIndex = objectIndex(aResult);
+ removeRows(0, aResult->stepsSize(), aIndex);
+ } else {
+ QModelIndex aIndex = objectIndex(aObject);
+ if (aIndex.isValid()) {
+ emit dataChanged(aIndex, aIndex);
+ }
}
} else {
rebuildDataTree();
} else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_ORDER_UPDATED)) {
std::shared_ptr<ModelAPI_OrderUpdatedMessage> aUpdMsg =
std::dynamic_pointer_cast<ModelAPI_OrderUpdatedMessage>(theMessage);
- DocumentPtr aDoc = aUpdMsg->document();
- std::string aGroup = aUpdMsg->group();
-
- QModelIndex aParent;
- int aStartId = 0;
- if (aDoc == aRootDoc) {
- // Update a group under root
- if (aGroup == myXMLReader->rootType()) // Update objects under root
- aStartId = foldersCount();
- else // Update objects in folder under root
- aParent = createIndex(folderId(aGroup), 0, -1);
+ if (aUpdMsg->reordered().get()) {
+ DocumentPtr aDoc = aUpdMsg->reordered()->document();
+ std::string aGroup = aUpdMsg->reordered()->group();
+
+ QModelIndex aParent;
+ int aStartId = 0;
+ if (aDoc == aRootDoc) {
+ // Update a group under root
+ if (aGroup == myXMLReader->rootType()) // Update objects under root
+ aStartId = foldersCount();
+ else // Update objects in folder under root
+ aParent = createIndex(folderId(aGroup), 0, (void*)Q_NULLPTR);
+ } else {
+ // Update a sub-document
+ if (aGroup == myXMLReader->subType()) {
+ // Update sub-document root
+ aParent = findDocumentRootIndex(aDoc.get());
+ aStartId = foldersCount(aDoc.get());
+ } else
+ // update folder in sub-document
+ aParent = createIndex(folderId(aGroup, aDoc.get()), 0, aDoc.get());
+ }
+ int aChildNb = rowCount(aParent);
+ rebuildBranch(aStartId, aChildNb - aStartId, aParent);
} else {
- // Update a sub-document
- if (aGroup == myXMLReader->subType()) {
- // Update sub-document root
- aParent = findDocumentRootIndex(aDoc.get());
- aStartId = foldersCount(aDoc.get());
- } else
- // update folder in sub-document
- aParent = createIndex(folderId(aGroup, aDoc.get()), 0, aDoc.get());
+ rebuildDataTree();
}
- int aChildNb = rowCount(aParent);
- rebuildBranch(aStartId, aChildNb - aStartId, aParent);
} else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_DOCUMENT_CHANGED)) {
DocumentPtr aDoc = ModelAPI_Session::get()->activeDocument();
if (aDoc != aRootDoc) {
QModelIndex aDocRoot = findDocumentRootIndex(aDoc.get());
if (aDocRoot.isValid())
emit dataChanged(aDocRoot, aDocRoot);
- else
+ else
// We have got a new document
rebuildDataTree();
}
- }
+ }
}
//******************************************************
void XGUI_DataModel::clear()
{
-
+ beginResetModel();
+ endResetModel();
}
//******************************************************
//******************************************************
ObjectPtr XGUI_DataModel::object(const QModelIndex& theIndex) const
{
- if (theIndex.internalId() < 0) // this is a folder
+ if (theIndex.internalId() == 0) // this is a folder
+ return ObjectPtr();
+ ModelAPI_Object* aObj =
+ dynamic_cast<ModelAPI_Object*>((ModelAPI_Entity*)theIndex.internalPointer());
+ if (!aObj)
return ObjectPtr();
- ModelAPI_Object* aObj = (ModelAPI_Object*)theIndex.internalPointer();
if (getSubDocument(aObj)) // the selected index is a folder of sub-document
return ObjectPtr();
}
if (aRow == -1)
return QModelIndex();
- else
+ else
return createIndex(aRow, 0, theObject.get());
}
SessionPtr aSession = ModelAPI_Session::get();
DocumentPtr aRootDoc = aSession->moduleDocument();
- if (aDoc == aRootDoc && myXMLReader->rootType() == aType) {
+ if (aDoc == aRootDoc && myXMLReader->rootType() == aType) {
// The object from root document
aRow += foldersCount();
- } else if (myXMLReader->subType() == aType) {
+ } else if (myXMLReader->subType() == aType) {
// The object from sub document
aRow += foldersCount(aDoc.get());
}
if (theIndex.column() == 1)
return QVariant();
- int aParentId = theIndex.internalId();
- if (aParentId == -1) { // root folders
+ quintptr aParentId = theIndex.internalId();
+ if (aParentId == 0) { // root folders
switch (theRole) {
case Qt::DisplayRole:
- return QString(myXMLReader->rootFolderName(theIndexRow).c_str()) +
+ return QString(myXMLReader->rootFolderName(theIndexRow).c_str()) +
QString(" (%1)").arg(rowCount(theIndex));
case Qt::DecorationRole:
return QIcon(myXMLReader->rootFolderIcon(theIndexRow).c_str());
if (aSubDoc) { // this is a folder of sub document
QIntList aMissedIdx = missedFolderIndexes(aSubDoc);
int aRow = theIndexRow;
- while (aMissedIdx.contains(aRow))
+ while (aMissedIdx.contains(aRow))
aRow++;
-
- switch (theRole) {
- case Qt::DisplayRole:
- return QString(myXMLReader->subFolderName(aRow).c_str()) +
- QString(" (%1)").arg(rowCount(theIndex));
- case Qt::DecorationRole:
- return QIcon(myXMLReader->subFolderIcon(aRow).c_str());
+ if (aRow < myXMLReader->subFoldersNumber()) {
+ switch (theRole) {
+ case Qt::DisplayRole:
+ return QString(myXMLReader->subFolderName(aRow).c_str()) +
+ QString(" (%1)").arg(rowCount(theIndex));
+ case Qt::DecorationRole:
+ return QIcon(myXMLReader->subFolderIcon(aRow).c_str());
+ }
}
} else {
- ModelAPI_Object* aObj = (ModelAPI_Object*)theIndex.internalPointer();
- switch (theRole) {
- case Qt::DisplayRole:
- {
- if (aObj->groupName() == ModelAPI_ResultParameter::group()) {
- ModelAPI_ResultParameter* aParam = dynamic_cast<ModelAPI_ResultParameter*>(aObj);
- AttributeDoublePtr aValueAttribute = aParam->data()->real(ModelAPI_ResultParameter::VALUE());
- QString aVal = QString::number(aValueAttribute->value());
- QString aTitle = QString(aObj->data()->name().c_str());
- return aTitle + " = " + aVal;
+ ModelAPI_Object* aObj =
+ dynamic_cast<ModelAPI_Object*>((ModelAPI_Entity*)theIndex.internalPointer());
+ if (aObj) {
+ switch (theRole) {
+ case Qt::DisplayRole:
+ {
+ if (aObj->groupName() == ModelAPI_ResultParameter::group()) {
+ ModelAPI_ResultParameter* aParam = dynamic_cast<ModelAPI_ResultParameter*>(aObj);
+ AttributeDoublePtr aValueAttribute =
+ aParam->data()->real(ModelAPI_ResultParameter::VALUE());
+ QString aVal = QString::number(aValueAttribute->value());
+ QString aTitle = QString(aObj->data()->name().c_str());
+ return aTitle + " = " + aVal;
+ }
+ QString aSuffix;
+ if (aObj->groupName() == myXMLReader->subType()) {
+ ResultPartPtr aPartRes = getPartResult(aObj);
+ if (aPartRes.get()) {
+ if (aPartRes->partDoc().get() == NULL)
+ aSuffix = " (Not loaded)";
+ }
+ }
+ return aObj->data()->name().c_str() + aSuffix;
}
- QString aSuffix;
- if (aObj->groupName() == myXMLReader->subType()) {
- ResultPartPtr aPartRes = getPartResult(aObj);
- if (aPartRes.get()) {
- if (aPartRes->partDoc().get() == NULL)
- aSuffix = " (Not loaded)";
+ case Qt::DecorationRole:
+ return ModuleBase_IconFactory::get()->getIcon(object(theIndex));
+ }
+ } else {
+ switch (theRole) {
+ case Qt::DisplayRole:
+ {
+ ModelAPI_ResultField::ModelAPI_FieldStep* aStep =
+ dynamic_cast<ModelAPI_ResultField::ModelAPI_FieldStep*>
+ ((ModelAPI_Entity*)theIndex.internalPointer());
+ if (aStep) {
+ return "Step " + QString::number(aStep->id() + 1) + " " +
+ aStep->field()->textLine(aStep->id()).c_str();
}
}
- return aObj->data()->name().c_str() + aSuffix;
+ break;
}
- case Qt::DecorationRole:
- return ModuleBase_IconFactory::get()->getIcon(object(theIndex));
}
}
}
return aNbFolders + aNbItems;
}
- int aId = theParent.internalId();
- if (aId == -1) {
+ quintptr aId = theParent.internalId();
+ if (aId == 0) {
// this is a folder under root
int aParentPos = theParent.row();
std::string aType = myXMLReader->rootFolderType(aParentPos);
} else {
// It is an object which could have children
ModelAPI_Document* aDoc = getSubDocument(theParent.internalPointer());
- if (aDoc) {
+ if (aDoc) {
// a folder of sub-document
QIntList aMissedIdx = missedFolderIndexes(aDoc);
int aRow = theParent.row();
- while (aMissedIdx.contains(aRow))
+ while (aMissedIdx.contains(aRow))
aRow++;
- std::string aType = myXMLReader->subFolderType(aRow);
- return aDoc->size(aType);
+ if (aRow < myXMLReader->subFoldersNumber()) {
+ std::string aType = myXMLReader->subFolderType(aRow);
+ return aDoc->size(aType);
+ }
} else {
- ModelAPI_Object* aObj = (ModelAPI_Object*)theParent.internalPointer();
+ ModelAPI_Object* aObj =
+ dynamic_cast<ModelAPI_Object*>((ModelAPI_Entity*)theParent.internalPointer());
// Check for Part feature
ResultPartPtr aPartRes = getPartResult(aObj);
if (aPartRes.get()) {
} else {
// Check for composite object
ModelAPI_CompositeFeature* aCompFeature = dynamic_cast<ModelAPI_CompositeFeature*>(aObj);
- if (aCompFeature)
+ if (aCompFeature)
return aCompFeature->numberOfSubs(true);
ModelAPI_ResultCompSolid* aCompRes = dynamic_cast<ModelAPI_ResultCompSolid*>(aObj);
- if (aCompRes)
+ if (aCompRes)
return aCompRes->numberOfSubs(true);
+ ModelAPI_ResultField* aFieldRes = dynamic_cast<ModelAPI_ResultField*>(aObj);
+ if (aFieldRes)
+ return aFieldRes->stepsSize();
}
}
}
if (!theParent.isValid()) {
if (theRow < aNbFolders) // Return first level folder index
- return createIndex(theRow, theColumn, -1);
+ return createIndex(theRow, theColumn, (void*)Q_NULLPTR);
else { // return object under root index
std::string aType = myXMLReader->rootType();
int aObjId = theRow - aNbFolders;
}
}
} else {
- int aId = theParent.internalId();
+ quintptr aId = theParent.internalId();
int aParentPos = theParent.row();
- if (aId == -1) { // return object index inside of first level of folders
+ if (aId == 0) { // return object index inside of first level of folders
std::string aType = myXMLReader->rootFolderType(aParentPos);
if (theRow < aRootDoc->size(aType)) {
ObjectPtr aObj = aRootDoc->object(aType, theRow);
} else {
// It is an object which could have children
ModelAPI_Document* aDoc = getSubDocument(theParent.internalPointer());
- if (aDoc) {
+ if (aDoc) {
// It is a folder of sub-document
int aParentRow = aParentPos;
QIntList aMissedIdx = missedFolderIndexes(aDoc);
while (aMissedIdx.contains(aParentRow))
aParentRow++;
- std::string aType = myXMLReader->subFolderType(aParentRow);
- if (theRow < aDoc->size(aType)) {
- ObjectPtr aObj = aDoc->object(aType, theRow);
- aIndex = objectIndex(aObj);
+ if (aParentRow < myXMLReader->subFoldersNumber()) {
+ std::string aType = myXMLReader->subFolderType(aParentRow);
+ if (theRow < aDoc->size(aType)) {
+ ObjectPtr aObj = aDoc->object(aType, theRow);
+ aIndex = objectIndex(aObj);
+ }
}
} else {
- ModelAPI_Object* aParentObj = (ModelAPI_Object*)theParent.internalPointer();
+ ModelAPI_Object* aParentObj =
+ dynamic_cast<ModelAPI_Object*>((ModelAPI_Entity*)theParent.internalPointer());
// Check for Part feature
ResultPartPtr aPartRes = getPartResult(aParentObj);
}
} else {
// Check for composite object
- ModelAPI_CompositeFeature* aCompFeature = dynamic_cast<ModelAPI_CompositeFeature*>(aParentObj);
+ ModelAPI_CompositeFeature* aCompFeature =
+ dynamic_cast<ModelAPI_CompositeFeature*>(aParentObj);
if (aCompFeature) {
aIndex = objectIndex(aCompFeature->subFeature(theRow));
} else {
- ModelAPI_ResultCompSolid* aCompRes = dynamic_cast<ModelAPI_ResultCompSolid*>(aParentObj);
- if (aCompRes)
+ ModelAPI_ResultCompSolid* aCompRes =
+ dynamic_cast<ModelAPI_ResultCompSolid*>(aParentObj);
+ if (aCompRes)
aIndex = objectIndex(aCompRes->subResult(theRow));
+ else {
+ ModelAPI_ResultField* aFieldRes =
+ dynamic_cast<ModelAPI_ResultField*>(aParentObj);
+ if (aFieldRes) {
+ aIndex = createIndex(theRow, 0, aFieldRes->step(theRow));
+ }
+ }
}
}
}
if (theIndex == MYLastDeleted)
return QModelIndex();
- int aId = theIndex.internalId();
- if (aId != -1) { // The object is not a root folder
+ SessionPtr aSession = ModelAPI_Session::get();
+ quintptr aId = theIndex.internalId();
+ if (aId != 0) { // The object is not a root folder
ModelAPI_Document* aDoc = getSubDocument(theIndex.internalPointer());
- if (aDoc) {
+ if (aDoc) {
// It is a folder of sub-document
return findDocumentRootIndex(aDoc);
}
ObjectPtr aObj = object(theIndex);
if (!aObj.get()) {
+ // It can b e a step of a field
+ ModelAPI_ResultField::ModelAPI_FieldStep* aStep =
+ dynamic_cast<ModelAPI_ResultField::ModelAPI_FieldStep*>
+ ((ModelAPI_Entity*)theIndex.internalPointer());
+ if (aStep) {
+ ModelAPI_ResultField* aField = aStep->field();
+ DocumentPtr aDoc = aSession->activeDocument();
+ ObjectPtr aFld;
+ for(int i = 0; i < aDoc->size(ModelAPI_ResultField::group()); i++) {
+ aFld = aDoc->object(ModelAPI_ResultField::group(), i);
+ if (aFld.get() == aField)
+ return objectIndex(aFld);
+ }
+ }
// To avoid additional request about index which was already deleted
// If deleted it causes a crash on delete object from Part
MYLastDeleted = theIndex;
}
// Use as ordinary object
std::string aType = aObj->groupName();
- SessionPtr aSession = ModelAPI_Session::get();
DocumentPtr aRootDoc = aSession->moduleDocument();
DocumentPtr aSubDoc = aObj->document();
if (aSubDoc == aRootDoc) {
// return first level of folder index
int aFolderId = myXMLReader->rootFolderId(aType);
// Items in a one row must have the same parent
- return createIndex(aFolderId, 0, -1);
+ return createIndex(aFolderId, 0, (void*)Q_NULLPTR);
}
} else {
if (aType == myXMLReader->subType())
return findDocumentRootIndex(aSubDoc.get());
else {
// return first level of folder index
- int aFolderId = myXMLReader->subFolderId(aType);
+ int aFolderId = folderId(aType, aSubDoc.get());
// Items in a one row must have the same parent
return createIndex(aFolderId, 0, aSubDoc.get());
}
}
- }
+ }
return QModelIndex();
}
//******************************************************
Qt::ItemFlags XGUI_DataModel::flags(const QModelIndex& theIndex) const
{
- qint64 aIt = theIndex.internalId();
+ quintptr aIt = theIndex.internalId();
ModelAPI_Object* aObj = 0;
ModelAPI_Document* aDoc = 0;
SessionPtr aSession = ModelAPI_Session::get();
Qt::ItemFlags aEditingFlag = Qt::ItemIsSelectable | Qt::ItemIsEnabled | Qt::ItemIsEditable;
- if (aIt == -1) {
+ if (aIt == 0) {
// Folders under root
DocumentPtr aRootDoc = aSession->moduleDocument();
if (aRootDoc != aActiveDoc)
} else {
aDoc = getSubDocument(theIndex.internalPointer());
if (!aDoc)
- aObj = (ModelAPI_Object*) theIndex.internalPointer();
+ aObj = dynamic_cast<ModelAPI_Object*>((ModelAPI_Entity*)theIndex.internalPointer());
}
if (aObj) {
// An object
- if (aObj->isDisabled())
+ if (aObj->isDisabled())
return theIndex.column() == 1? Qt::ItemIsSelectable : aNullFlag;
-
+
+ if (aSession->moduleDocument() != aObj->document())
+ if (aActiveDoc != aObj->document())
+ return theIndex.column() == 1? Qt::ItemIsSelectable : aNullFlag;
+
bool isCompositeSub = false;
// An object which is sub-object of a composite object can not be accessible in column 1
if (theIndex.column() == 1) {
FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObjPtr);
if (aFeature.get()) {
CompositeFeaturePtr aCompFea = ModelAPI_Tools::compositeOwner(aFeature);
- if (aCompFea.get())
+ if (aCompFea.get())
isCompositeSub = true;
} else {
ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(aObjPtr);
if (aResult.get()) {
ResultCompSolidPtr aCompRes = ModelAPI_Tools::compSolidOwner(aResult);
- if (aCompRes.get())
+ if (aCompRes.get())
isCompositeSub = true;
}
}
DocumentPtr aRootDoc = aSession->moduleDocument();
if (theDoc == aRootDoc)
return QModelIndex();
- else
+ else
return findDocumentRootIndex(theDoc.get());
}
} else {
if (aCurDoc == aSession->moduleDocument())
return createIndex(foldersCount() - 1, 1, -1);
- else
+ else
return createIndex(foldersCount(aCurDoc.get()) - 1, 1, aCurDoc.get());
}
}
//******************************************************
-int XGUI_DataModel::folderId(std::string theType, ModelAPI_Document* theDoc)
+int XGUI_DataModel::folderId(std::string theType, ModelAPI_Document* theDoc) const
{
SessionPtr aSession = ModelAPI_Session::get();
ModelAPI_Document* aDoc = theDoc;
insertRows(theRow, theCount, theParent);
}
}
+
+//******************************************************
+//bool XGUI_DataModel::blockEventsProcessing(const bool theState)
+//{
+// bool aPreviousState = myIsEventsProcessingBlocked;
+// myIsEventsProcessingBlocked = theState;
+// return aPreviousState;
+//}
