src/PartSet/PartSet_DocumentDataModel.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
   2
   3 #include "PartSet_DocumentDataModel.h"
   4 #include "PartSet_PartDataModel.h"
   5 #include "PartSet_Module.h"
   6 //#include "XGUI_Tools.h"
   7
   8 #include <ModelAPI_Session.h>
   9 #include <ModelAPI_Document.h>
  10 #include <ModelAPI_Feature.h>
  11 #include <ModelAPI_Data.h>
  12 #include <ModelAPI_ResultPart.h>
  13 #include <ModelAPI_Events.h>
  14 #include <ModelAPI_Object.h>
  15
  16 #include <Events_Loop.h>
  17
  18 #include <Config_FeatureMessage.h>
  19 #include <ModuleBase_Tools.h>
  20 #include <ModuleBase_ActionInfo.h>
  21
  22 #include <PartSetPlugin_Part.h>
  23
  24 #include <QIcon>
  25 #include <QString>
  26 #include <QBrush>
  27 #include <QTreeView>
  28
  29 #include <set>
  30
  31 #define ACTIVE_COLOR QColor(0,72,140)
  32 #define PASSIVE_COLOR Qt::black
  33
  34 QMap<QString, QString> PartSet_DocumentDataModel::myIcons;
  35
  36
  37 PartSet_DocumentDataModel::PartSet_DocumentDataModel(QObject* theParent)
  38     : ModuleBase_IDocumentDataModel(theParent),
  39       myActivePartModel(0)
  40 {
  41   // Create a top part of data tree model
  42   myModel = new PartSet_TopDataModel(this);
  43   myModel->setItemsColor(ACTIVE_COLOR);
  44
  45   Events_Loop* aLoop = Events_Loop::loop();
  46   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
  47   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_CREATED));
  48   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_DELETED));
  49   aLoop->registerListener(this, Events_Loop::eventByName(Config_FeatureMessage::GUI_EVENT()));
  50 }
  51
  52 PartSet_DocumentDataModel::~PartSet_DocumentDataModel()
  53 {
  54   clearModelIndexes();
  55   clearSubModels();
  56 }
  57
  58 void PartSet_DocumentDataModel::processEvent(const std::shared_ptr<Events_Message>& theMessage)
  59 {
  60   DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
  61
  62
  63   // Created object event *******************
  64   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_CREATED)) {
  65     std::shared_ptr<ModelAPI_ObjectUpdatedMessage> aUpdMsg =
  66         std::dynamic_pointer_cast<ModelAPI_ObjectUpdatedMessage>(theMessage);
  67     std::set<ObjectPtr> aObjects = aUpdMsg->objects();
  68
  69     std::set<ObjectPtr>::const_iterator aIt;
  70     for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) {
  71       ObjectPtr aObject = (*aIt);
  72       FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObject);
  73       if (aFeature && (!aFeature->isInHistory()))
  74         continue;
  75
  76       DocumentPtr aDoc = aObject->document();
  77       if (aDoc == aRootDoc) {  // If root objects
  78         if (aObject->groupName() == ModelAPI_ResultPart::group()) {  // Update only Parts group
  79             // Add a new part
  80           int aStart = aRootDoc->size(ModelAPI_ResultPart::group());
  81           if (aStart > 0) {
  82             FeaturePtr aPartFeature = ModelAPI_Feature::feature(aObject);
  83             PartSet_PartDataModel* aModel = new PartSet_PartDataModel(this);
  84             int anId = aRootDoc->index(aPartFeature);
  85             aModel->setPart(aPartFeature);
  86             myPartModels.append(aModel);
  87             insertRow(aStart, partFolderNode(0));
  88           }
  89         } else {  // Update top groups (other except parts
  90           QModelIndex aIndex = myModel->findParent(aObject);
  91           int aStart = myModel->rowCount(aIndex) - 1;
  92           if (aStart < 0)
  93             aStart = 0;
  94           aIndex = createIndex(aIndex.row(), aIndex.column(), (void*) getModelIndex(aIndex));
  95           insertRow(aStart, aIndex);
  96         }
  97       } else {  // if sub-objects of first level nodes
  98         PartSet_PartModel* aPartModel = 0;
  99         foreach (PartSet_PartModel* aPart, myPartModels) {
 100           if (aPart->hasDocument(aDoc)) {
 101             aPartModel = aPart;
 102             break;
 103           }
 104         }
 105         if (aPartModel) {
 106           QModelIndex aIndex = aPartModel->findParent(aObject);
 107           int aStart = aPartModel->rowCount(aIndex);  // check this index
 108           aIndex = createIndex(aIndex.row(), aIndex.column(), (void*) getModelIndex(aIndex));
 109           insertRow(aStart, aIndex);
 110         } else
 111           reset();
 112       }
 113     }
 114     // Deleted object event ***********************
 115   } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED)) {
 116     std::shared_ptr<ModelAPI_ObjectDeletedMessage> aUpdMsg =
 117         std::dynamic_pointer_cast<ModelAPI_ObjectDeletedMessage>(theMessage);
 118     DocumentPtr aDoc = aUpdMsg->document();
 119     std::set<std::string> aGroups = aUpdMsg->groups();
 120
 121     std::set<std::string>::const_iterator aIt;
 122     for (aIt = aGroups.begin(); aIt != aGroups.end(); ++aIt) {
 123       std::string aGroup = (*aIt);
 124       if (aDoc == aRootDoc) {  // If root objects
 125         if (aGroup == ModelAPI_ResultPart::group()) {  // Update only Parts group
 126           PartSet_PartModel* aDelPartModel = 0;
 127           foreach (PartSet_PartModel* aPartModel, myPartModels) {
 128             if (aPartModel->position() == -1) {
 129               aDelPartModel = aPartModel;
 130               break;
 131             }
 132           }
 133           if (aDelPartModel) {
 134             deactivatePart();
 135             int aStart = myPartModels.size() - 1;
 136             removeSubModel(aDelPartModel);
 137             removeRow(aStart, partFolderNode(0));
 138           }
 139         } else {  // Update top groups (other except parts
 140           QModelIndex aIndex = myModel->findGroup(aGroup);
 141           int aStart = myModel->rowCount(aIndex);
 142           aIndex = createIndex(aIndex.row(), aIndex.column(), (void*) getModelIndex(aIndex));
 143           removeRow(aStart, aIndex);
 144         }
 145       } else {
 146         PartSet_PartModel* aPartModel = 0;
 147         foreach (PartSet_PartModel* aPart, myPartModels) {
 148           if (aPart->hasDocument(aDoc)) {
 149             aPartModel = aPart;
 150             break;
 151           }
 152         }
 153         if (aPartModel) {
 154           QModelIndex aIndex = aPartModel->findGroup(aGroup);
 155           int aStart = aPartModel->rowCount(aIndex);
 156           aIndex = createIndex(aIndex.row(), aIndex.column(), (void*) getModelIndex(aIndex));
 157           removeRow(aStart, aIndex);
 158         }
 159       }
 160     }
 161     // Deleted object event ***********************
 162   } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED)) {
 163     //std::shared_ptr<ModelAPI_ObjectUpdatedMessage> aUpdMsg = std::dynamic_pointer_cast<ModelAPI_ObjectUpdatedMessage>(theMessage);
 164     //ObjectPtr aFeature = aUpdMsg->feature();
 165     //DocumentPtr aDoc = aFeature->document();
 166
 167     // TODO: Identify the necessary index by the modified feature
 168     QModelIndex aIndex;
 169     emit dataChanged(aIndex, aIndex);
 170
 171     // Reset whole tree **************************
 172   } else if (theMessage->eventID() == Events_Loop::loop()->eventByName(Config_FeatureMessage::GUI_EVENT())) {
 173     std::shared_ptr<Config_FeatureMessage> aFeatureMsg =
 174        std::dynamic_pointer_cast<Config_FeatureMessage>(theMessage);
 175     if (!aFeatureMsg->isInternal()) {
 176       ActionInfo aFeatureInfo;
 177       aFeatureInfo.initFrom(aFeatureMsg);
 178       // Remember features icons
 179       myIcons[QString::fromStdString(aFeatureMsg->id())] = aFeatureInfo.iconFile;
 180     }
 181   } else {
 182     rebuildDataTree();
 183   }
 184 }
 185
 186 void PartSet_DocumentDataModel::rebuildDataTree()
 187 {
 188   DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
 189
 190   beginResetModel();
 191   clearModelIndexes();
 192
 193   // Delete extra models
 194   ObjectPtr aObj;
 195   FeaturePtr aFeature;
 196   QList<PartSet_PartModel*> aDelList;
 197   foreach (PartSet_PartModel* aPartModel, myPartModels) {
 198     if (aPartModel->position() == -1)
 199       aDelList.append(aPartModel);
 200   }
 201   foreach (PartSet_PartModel* aPartModel, aDelList) {
 202     removeSubModel(aPartModel);
 203   }
 204   // Add non existing models
 205   int aHistNb = aRootDoc->size(ModelAPI_Feature::group());
 206   for (int i = 0; i < aHistNb; i++) {
 207     aObj = aRootDoc->object(ModelAPI_Feature::group(), i);
 208     aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
 209     if (aFeature->getKind() == PartSetPlugin_Part::ID()) {
 210       if (!findPartModel(aFeature)) {
 211         PartSet_PartDataModel* aModel = new PartSet_PartDataModel(this);
 212         aModel->setPart(aFeature);
 213         myPartModels.append(aModel);
 214       }
 215     }
 216   }
 217   endResetModel();
 218 }
 219
 220 QVariant PartSet_DocumentDataModel::data(const QModelIndex& theIndex, int theRole) const
 221 {
 222   if (!theIndex.isValid())
 223     return QVariant();
 224
 225   DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
 226   QModelIndex aParent = theIndex.parent();
 227   if ((theIndex.column() == 1) ) {
 228     if ((theIndex.internalId() >= PartsFolder) && (theIndex.internalId() <= PartResult)) {
 229       if (ModelAPI_Session::get()->activeDocument() == aRootDoc) {
 230         if (!aParent.isValid()) {
 231           switch (theRole) {
 232           case Qt::DecorationRole:
 233             if (theIndex.row() == lastHistoryRow())
 234               return QIcon(":pictures/arrow.png");
 235           }
 236         }
 237       }
 238     } else {
 239       QModelIndex* aIndex = toSourceModelIndex(theIndex);
 240       const QAbstractItemModel* aModel = aIndex->model();
 241       if (isPartSubModel(aModel)) {
 242         return aModel->data(*aIndex, theRole);
 243       }
 244     }
 245     return QVariant();
 246   }
 247
 248   switch (theIndex.internalId()) {
 249     case PartsFolder:
 250       switch (theRole) {
 251         case Qt::DisplayRole:
 252           return tr("Parts") + QString(" (%1)").arg(rowCount(theIndex));
 253         case Qt::DecorationRole:
 254           return QIcon(":pictures/constr_folder.png");
 255         case Qt::ToolTipRole:
 256           return tr("Parts folder");
 257         case Qt::ForegroundRole:
 258           if (myActivePartIndex.isValid())
 259               return QBrush(PASSIVE_COLOR);
 260             else
 261               return QBrush(ACTIVE_COLOR);
 262         default:
 263           return QVariant();
 264       }
 265       break;
 266     case HistoryNode:
 267       {
 268         int aOffset = historyOffset();
 269         ObjectPtr aObj = aRootDoc->object(ModelAPI_Feature::group(), theIndex.row() - aOffset);
 270         FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
 271         if (!aFeature)
 272           return QVariant();
 273         switch (theRole) {
 274           case Qt::DisplayRole:
 275             if (aFeature)
 276               return aFeature->data()->name().c_str();
 277             else
 278               return QVariant();
 279           case Qt::DecorationRole:
 280             return featureIcon(aFeature);
 281           case Qt::ToolTipRole:
 282             return tr("Feature object");
 283           case Qt::ForegroundRole:
 284             if (theIndex.row() > lastHistoryRow())
 285               return QBrush(Qt::lightGray);
 286             else {
 287               if (myActivePartIndex.isValid())
 288                 return QBrush(PASSIVE_COLOR);
 289               else
 290                 return QBrush(ACTIVE_COLOR);
 291             }
 292           default:
 293             return QVariant();
 294         }
 295       }
 296       break;
 297     case PartResult:
 298       {
 299         ObjectPtr aObject = aRootDoc->object(ModelAPI_ResultPart::group(), theIndex.row());
 300         if (aObject) {
 301           switch (theRole) {
 302             case Qt::DisplayRole:
 303               return std::dynamic_pointer_cast<ModelAPI_Object>(aObject)->data()->name().c_str();
 304             case Qt::DecorationRole:
 305               return QIcon(":pictures/part_ico.png");
 306             case Qt::ForegroundRole:
 307               {
 308                 if (theIndex == myActivePartIndex)
 309                   return QBrush(ACTIVE_COLOR);
 310                 else
 311                   return QBrush(PASSIVE_COLOR);
 312               }
 313             default:
 314               return QVariant();
 315           }
 316         }
 317       }
 318       break;
 319   }
 320   if (aParent.internalId() == HistoryNode) {
 321     int aId = aParent.row() - historyOffset();
 322     QModelIndex* aIndex = toSourceModelIndex(theIndex);
 323     return findPartModel(aId)->data(*aIndex, theRole);
 324   }
 325   return toSourceModelIndex(theIndex)->data(theRole);
 326 }
 327
 328 QVariant PartSet_DocumentDataModel::headerData(int theSection, Qt::Orientation theOrient,
 329                                             int theRole) const
 330 {
 331   return QVariant();
 332 }
 333
 334 int PartSet_DocumentDataModel::rowCount(const QModelIndex& theParent) const
 335 {
 336   SessionPtr aSession = ModelAPI_Session::get();
 337   if (!aSession->hasModuleDocument())
 338     return 0;
 339   DocumentPtr aRootDoc = aSession->moduleDocument();
 340   if (!theParent.isValid()) {
 341     // Size of external models
 342     int aVal = historyOffset();
 343     // Plus history size
 344     aVal += aRootDoc->size(ModelAPI_Feature::group());
 345     return aVal;
 346   }
 347   if (theParent.internalId() == PartsFolder) {
 348     return aRootDoc->size(ModelAPI_ResultPart::group());
 349     //int aSize = myPartModels.size();
 350     //return myPartModels.size();
 351   }
 352   if (theParent.internalId() == HistoryNode) {
 353     int aId = theParent.row() - historyOffset();
 354     PartSet_PartModel* aModel = findPartModel(aId);
 355     if (aModel)
 356       return aModel->rowCount(QModelIndex());
 357     return 0;
 358   }
 359   if (theParent.internalId() == PartResult)
 360     return 0;
 361
 362   QModelIndex* aParent = toSourceModelIndex(theParent);
 363   const QAbstractItemModel* aModel = aParent->model();
 364   if (!isSubModel(aModel))
 365     return 0;
 366
 367   /*if (isPartSubModel(aModel)) {
 368    if (aModel != myActivePart)
 369    return 0;
 370    }*/
 371   return aModel->rowCount(*aParent);
 372 }
 373
 374 int PartSet_DocumentDataModel::columnCount(const QModelIndex& theParent) const
 375 {
 376   return 2;
 377 }
 378
 379 QModelIndex PartSet_DocumentDataModel::index(int theRow, int theColumn,
 380                                           const QModelIndex& theParent) const
 381 {
 382   QModelIndex aIndex;
 383   if (!theParent.isValid()) {
 384     int aOffs = myModel->rowCount();
 385     if (theRow < aOffs) {
 386       aIndex = myModel->index(theRow, theColumn, theParent);
 387       aIndex = createIndex(theRow, theColumn, (void*) getModelIndex(aIndex));
 388     } else {
 389       if (theRow == aOffs)  // Create Parts node
 390         aIndex = partFolderNode(theColumn);
 391       else {
 392         // create history node
 393         aIndex = createIndex(theRow, theColumn, HistoryNode);
 394       }
 395     }
 396   } else {
 397     if (theParent.internalId() == PartsFolder) {
 398        aIndex = createIndex(theRow, theColumn, PartResult);
 399     } else {
 400       if (theParent.internalId() == HistoryNode) {
 401         int aId = theParent.row() - historyOffset();
 402         aIndex = findPartModel(aId)->index(theRow, theColumn, QModelIndex());
 403       } else {
 404         QModelIndex* aParent = (QModelIndex*) theParent.internalPointer();
 405         aIndex = aParent->model()->index(theRow, theColumn, (*aParent));
 406       }
 407       aIndex = createIndex(theRow, theColumn, (void*) getModelIndex(aIndex));
 408     }
 409   }
 410   return aIndex;
 411 }
 412
 413 QModelIndex PartSet_DocumentDataModel::parent(const QModelIndex& theIndex) const
 414 {
 415   if ((theIndex.internalId() == PartsFolder) || (theIndex.internalId() == HistoryNode))
 416     return QModelIndex();
 417
 418   if (theIndex.internalId() == PartResult)
 419     return partFolderNode(0);
 420
 421   QModelIndex* aIndex = toSourceModelIndex(theIndex);
 422   const QAbstractItemModel* aModel = aIndex->model();
 423   if (!isSubModel(aModel))
 424     return QModelIndex();
 425
 426   QModelIndex aIndex1 = aModel->parent(*aIndex);
 427   const PartSet_PartModel* aPartModel = dynamic_cast<const PartSet_PartModel*>(aModel);
 428   if (aPartModel && (!aIndex1.isValid())) {
 429     int aId = aPartModel->position();
 430     int aRow = aId + historyOffset();
 431     return createIndex(aRow, 0, (qint32) HistoryNode);
 432   }
 433
 434   if (aIndex1.isValid())
 435     return createIndex(aIndex1.row(), 0, (void*) getModelIndex(aIndex1));
 436   return aIndex1;
 437 }
 438
 439 bool PartSet_DocumentDataModel::hasChildren(const QModelIndex& theParent) const
 440 {
 441   if (!theParent.isValid())
 442     return true;
 443   return rowCount(theParent) > 0;
 444 }
 445
 446 QModelIndex* PartSet_DocumentDataModel::toSourceModelIndex(const QModelIndex& theProxy) const
 447 {
 448   QModelIndex* aIndexPtr = static_cast<QModelIndex*>(theProxy.internalPointer());
 449   return aIndexPtr;
 450 }
 451
 452 QModelIndex* PartSet_DocumentDataModel::findModelIndex(const QModelIndex& theIndex) const
 453 {
 454   QList<QModelIndex*>::const_iterator aIt;
 455   for (aIt = myIndexes.constBegin(); aIt != myIndexes.constEnd(); ++aIt) {
 456     QModelIndex* aIndex = (*aIt);
 457     if ((*aIndex) == theIndex)
 458       return aIndex;
 459   }
 460   return 0;
 461 }
 462
 463 QModelIndex* PartSet_DocumentDataModel::getModelIndex(const QModelIndex& theIndex) const
 464 {
 465   QModelIndex* aIndexPtr = findModelIndex(theIndex);
 466   if (!aIndexPtr) {
 467     aIndexPtr = new QModelIndex(theIndex);
 468     PartSet_DocumentDataModel* that = (PartSet_DocumentDataModel*) this;
 469     that->myIndexes.append(aIndexPtr);
 470   }
 471   return aIndexPtr;
 472 }
 473
 474 void PartSet_DocumentDataModel::clearModelIndexes()
 475 {
 476   foreach (QModelIndex* aIndex, myIndexes)
 477     delete aIndex;
 478   myIndexes.clear();
 479 }
 480
 481 void PartSet_DocumentDataModel::clearSubModels()
 482 {
 483   foreach (PartSet_PartModel* aPart, myPartModels)
 484     delete aPart;
 485   myPartModels.clear();
 486 }
 487
 488 ObjectPtr PartSet_DocumentDataModel::object(const QModelIndex& theIndex) const
 489 {
 490   if (theIndex.internalId() == PartsFolder)
 491     return ObjectPtr();
 492   DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
 493   if (theIndex.internalId() == HistoryNode) {
 494     int aOffset = historyOffset();
 495     return aRootDoc->object(ModelAPI_Feature::group(), theIndex.row() - aOffset);
 496   }
 497   if (theIndex.internalId() == PartResult) {
 498     return aRootDoc->object(ModelAPI_ResultPart::group(), theIndex.row());
 499   }
 500   QModelIndex* aIndex = toSourceModelIndex(theIndex);
 501   if (!isSubModel(aIndex->model()))
 502     return ObjectPtr();
 503
 504   const PartSet_FeaturesModel* aModel = dynamic_cast<const PartSet_FeaturesModel*>(aIndex->model());
 505   return aModel->object(*aIndex);
 506 }
 507
 508 bool PartSet_DocumentDataModel::insertRows(int theRow, int theCount, const QModelIndex& theParent)
 509 {
 510   beginInsertRows(theParent, theRow, theRow + theCount - 1);
 511   //endInsertRows();
 512
 513   // Update history
 514   QModelIndex aRoot;
 515   int aRow = rowCount(aRoot);
 516   beginInsertRows(aRoot, aRow, aRow);
 517   endInsertRows();
 518
 519   return true;
 520 }
 521
 522 bool PartSet_DocumentDataModel::removeRows(int theRow, int theCount, const QModelIndex& theParent)
 523 {
 524   beginRemoveRows(theParent, theRow, theRow + theCount - 1);
 525   endRemoveRows();
 526   return true;
 527 }
 528
 529 void PartSet_DocumentDataModel::removeSubModel(int theModelId)
 530 {
 531   PartSet_PartModel* aModel = myPartModels.at(theModelId);
 532   removeSubModel(aModel);
 533 }
 534
 535 void PartSet_DocumentDataModel::removeSubModel(PartSet_PartModel* theModel)
 536 {
 537   QIntList aToRemove;
 538   for (int i = 0; i < myIndexes.size(); i++) {
 539     if (myIndexes.at(i)->model() == theModel)
 540       aToRemove.append(i);
 541   }
 542   int aId;
 543   while (aToRemove.size() > 0) {
 544     aId = aToRemove.last();
 545     delete myIndexes.at(aId);
 546     myIndexes.removeAt(aId);
 547     aToRemove.removeLast();
 548   }
 549   delete theModel;
 550   myPartModels.removeAll(theModel);
 551 }
 552
 553
 554 bool PartSet_DocumentDataModel::isSubModel(const QAbstractItemModel* theModel) const
 555 {
 556   if (theModel == myModel)
 557     return true;
 558   return isPartSubModel(theModel);
 559 }
 560
 561 bool PartSet_DocumentDataModel::isPartSubModel(const QAbstractItemModel* theModel) const
 562 {
 563   return myPartModels.contains((PartSet_PartModel*) theModel);
 564 }
 565
 566 QModelIndex PartSet_DocumentDataModel::partFolderNode(int theColumn) const
 567 {
 568   int aPos = myModel->rowCount(QModelIndex());
 569   return createIndex(aPos, theColumn, PartsFolder);
 570 }
 571
 572 int PartSet_DocumentDataModel::historyOffset() const
 573 {
 574   // Nb of rows of top model + Parts folder
 575   return myModel->rowCount(QModelIndex()) + 1;
 576 }
 577
 578 bool PartSet_DocumentDataModel::activatePart(const QModelIndex& theIndex)
 579 {
 580   if ((theIndex.internalId() == PartsFolder) || (theIndex.internalId() == HistoryNode))
 581     return false;
 582
 583   if (theIndex.isValid() && (theIndex.internalId() == PartResult)) {
 584     myActivePartIndex = theIndex;
 585     myModel->setItemsColor(PASSIVE_COLOR);
 586     if (myActivePartModel)
 587       myActivePartModel->setItemsColor(PASSIVE_COLOR);
 588
 589     // Find activated part feature by its ID
 590     ResultPartPtr aPartRes = activePart();
 591     FeaturePtr aFeature = ModelAPI_Feature::feature(aPartRes);
 592     if (aFeature.get()) {
 593       myActivePartModel = findPartModel(aFeature);
 594       myActivePartModel->setItemsColor(ACTIVE_COLOR);
 595     }
 596   }
 597   return true;
 598 }
 599
 600 ResultPartPtr PartSet_DocumentDataModel::activePart() const
 601 {
 602   if (myActivePartIndex.isValid()) {
 603     DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
 604     ObjectPtr aObj = aRootDoc->object(ModelAPI_ResultPart::group(), myActivePartIndex.row());
 605     return std::dynamic_pointer_cast<ModelAPI_ResultPart>(aObj);
 606   }
 607   return ResultPartPtr();
 608 }
 609
 610 QModelIndex PartSet_DocumentDataModel::activePartTree() const
 611 {
 612   if (myActivePartModel) {
 613     return createIndex(myActivePartModel->position() + historyOffset(), 0, HistoryNode);
 614   }
 615   return QModelIndex();
 616 }
 617
 618 void PartSet_DocumentDataModel::deactivatePart()
 619 {
 620   if (myActivePartIndex.isValid()) {
 621     if (myActivePartModel)
 622       myActivePartModel->setItemsColor(PASSIVE_COLOR);
 623     myActivePartModel = 0;
 624     myActivePartIndex = QModelIndex();
 625     myModel->setItemsColor(ACTIVE_COLOR);
 626   }
 627 }
 628
 629 Qt::ItemFlags PartSet_DocumentDataModel::flags(const QModelIndex& theIndex) const
 630 {
 631   if ((theIndex.internalId() >= PartsFolder) && (theIndex.internalId() <= PartResult)) {
 632     Qt::ItemFlags aFlags = Qt::ItemIsSelectable;
 633     if (object(theIndex)) {
 634       aFlags |= Qt::ItemIsEditable;
 635     }
 636     // Disable items which are below of last history row
 637     // Do not disable second column
 638     if (theIndex.internalId() == HistoryNode) {
 639       if (theIndex.row() <= lastHistoryRow() || (theIndex.column() == 1))
 640         aFlags |= Qt::ItemIsEnabled;
 641     } else
 642       aFlags |= Qt::ItemIsEnabled;
 643     return aFlags;
 644   } else {
 645     QModelIndex* aIndex = toSourceModelIndex(theIndex);
 646     const QAbstractItemModel* aModel = aIndex->model();
 647     return aModel->flags(*aIndex);
 648   }
 649 }
 650
 651 QModelIndex PartSet_DocumentDataModel::partIndex(const ResultPartPtr& theObject) const
 652 {
 653   DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
 654   int aNb = aRootDoc->size(ModelAPI_ResultPart::group());
 655   for (int aId = 0; aId < aNb; aId++) {
 656     if (theObject == aRootDoc->object(ModelAPI_ResultPart::group(), aId))
 657       return createIndex(aId, 0, PartResult);
 658   }
 659   return QModelIndex();
 660 }
 661
 662 QModelIndex PartSet_DocumentDataModel::objectIndex(const ObjectPtr theObject) const
 663 {
 664   // Check that this feature belongs to root document
 665   DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
 666   DocumentPtr aDoc = theObject->document();
 667   if (aDoc == aRootDoc) {
 668     // This feature belongs to histrory or top model
 669     FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(theObject);
 670     if (aFeature) {
 671       int aId;
 672       int aNb = aRootDoc->size(ModelAPI_Feature::group());
 673       for (aId = 0; aId < aNb; aId++) {
 674         if (theObject == aRootDoc->object(ModelAPI_Feature::group(), aId))
 675           break;
 676       }
 677       if (aId < aNb)
 678         return index(aId + historyOffset(), 0, QModelIndex());
 679     } else {
 680       QModelIndex aIndex = myModel->objectIndex(theObject);
 681       return
 682           aIndex.isValid() ?
 683               createIndex(aIndex.row(), 0, (void*) getModelIndex(aIndex)) :
 684               QModelIndex();
 685     }
 686   } else {
 687     PartSet_PartModel* aPartModel = 0;
 688     foreach(PartSet_PartModel* aModel, myPartModels) {
 689       if (aModel->hasDocument(aDoc)) {
 690         aPartModel = aModel;
 691         break;
 692       }
 693     }
 694     if (aPartModel) {
 695       QModelIndex aIndex = aPartModel->objectIndex(theObject);
 696       return aIndex.isValid() ?
 697               createIndex(aIndex.row(), 0, (void*) getModelIndex(aIndex)) :
 698               QModelIndex();
 699     }
 700   }
 701   return QModelIndex();
 702 }
 703
 704
 705 void PartSet_DocumentDataModel::clear()
 706 {
 707   clearModelIndexes();
 708   clearSubModels();
 709   //myActivePart = 0;
 710   myActivePartIndex = QModelIndex();
 711   myModel->setItemsColor(ACTIVE_COLOR);
 712 }
 713
 714 int PartSet_DocumentDataModel::lastHistoryRow() const
 715 {
 716   DocumentPtr aRootDoc = ModelAPI_Session::get()->moduleDocument();
 717   FeaturePtr aFeature = aRootDoc->currentFeature(true);
 718   if (aFeature.get())
 719     return historyOffset() + aRootDoc->index(aFeature);
 720   else
 721     return historyOffset() - 1;
 722 }
 723
 724 void PartSet_DocumentDataModel::setLastHistoryItem(const QModelIndex& theIndex)
 725 {
 726   SessionPtr aMgr = ModelAPI_Session::get();
 727   DocumentPtr aRootDoc = aMgr->moduleDocument();
 728   std::string aOpName = tr("History change").toStdString();
 729   if (theIndex.internalId() == HistoryNode) {
 730     ObjectPtr aObject = object(theIndex);
 731     aMgr->startOperation(aOpName);
 732     aRootDoc->setCurrentFeature(std::dynamic_pointer_cast<ModelAPI_Feature>(aObject), true);
 733     aMgr->finishOperation();
 734   } else {
 735     aMgr->startOperation(aOpName);
 736     aRootDoc->setCurrentFeature(FeaturePtr(), true);
 737     aMgr->finishOperation();
 738   }
 739 }
 740
 741 QModelIndex PartSet_DocumentDataModel::lastHistoryItem() const
 742 {
 743   return index(lastHistoryRow(), 1);
 744 }
 745
 746
 747 QIcon PartSet_DocumentDataModel::featureIcon(const FeaturePtr& theFeature)
 748 {
 749   QIcon anIcon;
 750
 751   std::string aKind = theFeature->getKind();
 752   QString aId(aKind.c_str());
 753   if (!myIcons.contains(aId))
 754     return anIcon;
 755
 756   QString anIconString = myIcons[aId];
 757
 758   ModelAPI_ExecState aState = theFeature->data()->execState();
 759   switch(aState) {
 760     case ModelAPI_StateDone:
 761     case ModelAPI_StateNothing: {
 762       anIcon = QIcon(anIconString);
 763     }
 764     break;
 765     case ModelAPI_StateMustBeUpdated: {
 766       anIcon = ModuleBase_Tools::lighter(anIconString);
 767     }
 768     break;
 769     case ModelAPI_StateExecFailed: {
 770       anIcon = ModuleBase_Tools::composite(":icons/exec_state_failed.png", anIconString);
 771     }
 772     break;
 773     case ModelAPI_StateInvalidArgument: {
 774       anIcon = ModuleBase_Tools::composite(":icons/exec_state_invalid_parameters.png",
 775                                            anIconString);
 776     }
 777     break;
 778     default: break;
 779   }
 780   return anIcon;
 781 }
 782
 783 void PartSet_DocumentDataModel::onMouseDoubleClick(const QModelIndex& theIndex)
 784 {
 785   if (theIndex.column() != 1)
 786     return;
 787   QTreeView* aTreeView = dynamic_cast<QTreeView*>(sender());
 788   if ((theIndex.internalId() >= PartsFolder) && (theIndex.internalId() <= PartResult)) {
 789     if (myActivePartModel)
 790       // It means that the root document is not active
 791       return;
 792     QModelIndex aNewIndex;
 793     if (theIndex.internalId() == HistoryNode)
 794       aNewIndex = theIndex;
 795     int aOldId = lastHistoryRow();
 796     setLastHistoryItem(theIndex);
 797     int aStartRow = std::min(aOldId, theIndex.row());
 798     int aEndRow = std::max(aOldId, theIndex.row());
 799     for (int i = aStartRow; i <= aEndRow; i++) {
 800       aTreeView->update(createIndex(i, 0, HistoryNode));
 801       aTreeView->update(createIndex(i, 1, HistoryNode));
 802     }
 803
 804   } else {
 805     QModelIndex* aIndex = toSourceModelIndex(theIndex);
 806     const QAbstractItemModel* aModel = aIndex->model();
 807     if (isPartSubModel(aModel)) {
 808       PartSet_PartDataModel* aPartModel = (PartSet_PartDataModel*)aModel;
 809       QModelIndex aOldItem = aPartModel->lastHistoryItem();
 810       aPartModel->setLastHistoryItem(*aIndex);
 811       QModelIndex aOldIndex = createIndex(aOldItem.row(), aOldItem.column(), (void*) getModelIndex(aOldItem));
 812       int aStartRow = std::min(aOldItem.row(), aIndex->row());
 813       int aEndRow = std::max(aOldItem.row(), aIndex->row());
 814       for (int i = aStartRow; i <= aEndRow; i++) {
 815         QModelIndex aInd1 = aPartModel->index(i, 0);
 816         QModelIndex aInd2 = createIndex(i, 0, (void*) getModelIndex(aInd1));
 817         aTreeView->update(aInd2);
 818         aInd1 = aPartModel->index(i, 1);
 819         aInd2 = createIndex(i, 1, (void*) getModelIndex(aInd1));
 820         aTreeView->update(aInd2);
 821       }
 822     }
 823   }
 824 }
 825
 826
 827 PartSet_PartModel* PartSet_DocumentDataModel::findPartModel(FeaturePtr thePart) const
 828 {
 829   foreach (PartSet_PartModel* aModel, myPartModels) {
 830     if (aModel->part() == thePart)
 831       return aModel;
 832   }
 833   return 0;
 834 }
 835
 836 PartSet_PartModel* PartSet_DocumentDataModel::findPartModel(int thePosition) const
 837 {
 838   foreach (PartSet_PartModel* aModel, myPartModels) {
 839     if (aModel->position() == thePosition)
 840       return aModel;
 841   }
 842   return 0;
 843 }