#include "XGUI_DocumentDataModel.h"
#include "XGUI_PartDataModel.h"
#include "XGUI_Workshop.h"
+#include "XGUI_Tools.h"
#include <ModelAPI_PluginManager.h>
#include <ModelAPI_Document.h>
#include <ModelAPI_Feature.h>
#include <ModelAPI_Data.h>
#include <Model_Events.h>
+#include <ModelAPI_Object.h>
#include <Events_Loop.h>
XGUI_DocumentDataModel::XGUI_DocumentDataModel(QObject* theParent)
: QAbstractItemModel(theParent), myActivePart(0)
{
- // Find Document object
- PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
- myDocument = aMgr->currentDocument();
-
// Register in event loop
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(myDocument, this);
+ myModel = new XGUI_TopDataModel(this);
myModel->setItemsColor(ACTIVE_COLOR);
}
void XGUI_DocumentDataModel::processEvent(const Events_Message* theMessage)
{
+ DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument();
+
// Created object event *******************
if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_FEATURE_CREATED)) {
const Model_FeatureUpdatedMessage* aUpdMsg = dynamic_cast<const Model_FeatureUpdatedMessage*>(theMessage);
for (aIt = aFeatures.begin(); aIt != aFeatures.end(); ++aIt) {
FeaturePtr aFeature = (*aIt);
DocumentPtr aDoc = aFeature->document();
- if (aDoc == myDocument) { // If root objects
+ if (aDoc == aRootDoc) { // If root objects
if (aFeature->getGroup().compare(PARTS_GROUP) == 0) { // Update only Parts group
// Add a new part
- int aStart = myPartModels.size() + 1;
- XGUI_PartDataModel* aModel = new XGUI_PartDataModel(myDocument, this);
+ int aStart = myPartModels.size();
+ XGUI_PartDataModel* aModel = new XGUI_PartDataModel(this);
aModel->setPartId(myPartModels.count());
myPartModels.append(aModel);
insertRow(aStart, partFolderNode());
std::set<std::string>::const_iterator aIt;
for (aIt = aGroups.begin(); aIt != aGroups.end(); ++aIt) {
std::string aGroup = (*aIt);
- if (aDoc == myDocument) { // If root objects
+ if (aDoc == aRootDoc) { // If root objects
if (aGroup.compare(PARTS_GROUP) == 0) { // Updsate only Parts group
int aStart = myPartModels.size() - 1;
removeSubModel(aStart);
removeRow(aStart, partFolderNode());
+ if (myActivePart && (!isPartSubModel(myActivePart))) {
+ myActivePart = 0;
+ myActivePartIndex = QModelIndex();
+ myModel->setItemsColor(ACTIVE_COLOR);
+ }
} else { // Update top groups (other except parts
QModelIndex aIndex = myModel->findGroup(aGroup);
int aStart = myModel->rowCount(aIndex);
// Reset whole tree **************************
} else {
- beginResetModel();
- int aNbParts = myDocument->size(PARTS_GROUP);
- 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);
+ rebuildDataTree();
+ }
+}
+
+void XGUI_DocumentDataModel::rebuildDataTree()
+{
+ DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument();
+
+ beginResetModel();
+ clearModelIndexes();
+
+ int aNbParts = aRootDoc->size(PARTS_GROUP);
+ 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(this));
}
- clearModelIndexes();
- endResetModel();
+ for (int i = 0; i < myPartModels.size(); i++)
+ myPartModels.at(i)->setPartId(i);
}
+ endResetModel();
}
QVariant XGUI_DocumentDataModel::data(const QModelIndex& theIndex, int theRole) const
case HistoryNode:
{
int aOffset = historyOffset();
- FeaturePtr aFeature = myDocument->feature(FEATURES_GROUP, theIndex.row() - aOffset);
+ DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument();
+ FeaturePtr aFeature = aRootDoc->feature(FEATURES_GROUP, theIndex.row() - aOffset);
if (!aFeature)
return QVariant();
switch (theRole) {
int XGUI_DocumentDataModel::rowCount(const QModelIndex& theParent) const
{
if (!theParent.isValid()) {
+ DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument();
// Size of external models
int aVal = historyOffset();
// Plus history size
- aVal += myDocument->size(FEATURES_GROUP);
+ aVal += aRootDoc->size(FEATURES_GROUP);
return aVal;
}
if (theParent.internalId() == PartsFolder) {
if (theIndex.internalId() == PartsFolder)
return FeaturePtr();
if (theIndex.internalId() == HistoryNode) {
- int aOffset = historyOffset();
- return myDocument->feature(FEATURES_GROUP, theIndex.row() - aOffset);
+ DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument();
+ int aOffset = historyOffset();
+ return aRootDoc->feature(FEATURES_GROUP, theIndex.row() - aOffset);
}
QModelIndex* aIndex = toSourceModelIndex(theIndex);
if (!isSubModel(aIndex->model()))
if (myActivePart)
myActivePart->setItemsColor(PASSIVE_COLOR);
myActivePart = 0;
+ myActivePartIndex = QModelIndex();
myModel->setItemsColor(ACTIVE_COLOR);
}
QModelIndex XGUI_DocumentDataModel::partIndex(const FeaturePtr& theFeature) const
{
+ FeaturePtr aFeature = XGUI_Tools::realFeature(theFeature);
+
int aRow = -1;
XGUI_PartModel* aModel = 0;
foreach (XGUI_PartModel* aPartModel, myPartModels) {
aRow++;
- if (aPartModel->part() == theFeature) {
+ if (aPartModel->part() == aFeature) {
aModel = aPartModel;
break;
}
}
return QModelIndex();
}
+
+QModelIndex XGUI_DocumentDataModel::featureIndex(const FeaturePtr theFeature) const
+{
+ // Check that this feature belongs to root document
+ DocumentPtr aRootDoc = ModelAPI_PluginManager::get()->rootDocument();
+ DocumentPtr aDoc = theFeature->document();
+ if (aDoc == aRootDoc) {
+ // This feature belongs to histrory or top model
+ if (theFeature->isInHistory()) {
+ int aId;
+ for (aId = 0; aId < aRootDoc->size(FEATURES_GROUP); aId++) {
+ if (theFeature == aRootDoc->feature(FEATURES_GROUP, aId))
+ break;
+ }
+ return index(aId + historyOffset(), 0, QModelIndex());
+ } else {
+ QModelIndex aIndex = myModel->featureIndex(theFeature);
+ return aIndex.isValid()?
+ createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)) :
+ QModelIndex();
+ }
+ } else {
+ XGUI_PartModel* aPartModel = 0;
+ foreach(XGUI_PartModel* aModel, myPartModels) {
+ if (aModel->hasDocument(aDoc)) {
+ aPartModel = aModel;
+ break;
+ }
+ }
+ if (aPartModel) {
+ QModelIndex aIndex = aPartModel->featureIndex(theFeature);
+ return aIndex.isValid()?
+ createIndex(aIndex.row(), aIndex.column(), (void*)getModelIndex(aIndex)) :
+ QModelIndex();
+ }
+ }
+ return QModelIndex();
+}
\ No newline at end of file
