src/Model/Model_Document.cpp

   1 // File:        Model_Document.cxx
   2 // Created:     28 Feb 2014
   3 // Author:      Mikhail PONIKAROV
   4
   5 #include <Model_Document.h>
   6 #include <Model_Data.h>
   7 #include <Model_Application.h>
   8 #include <Model_PluginManager.h>
   9 #include <Model_Events.h>
  10 #include <Model_ResultPart.h>
  11 #include <Model_ResultConstruction.h>
  12 #include <Model_ResultBody.h>
  13 #include <Events_Loop.h>
  14 #include <Events_Error.h>
  15
  16 #include <TDataStd_Integer.hxx>
  17 #include <TDataStd_Comment.hxx>
  18 #include <TDF_ChildIDIterator.hxx>
  19 #include <TDataStd_ReferenceArray.hxx>
  20 #include <TDataStd_HLabelArray1.hxx>
  21 #include <TDataStd_Name.hxx>
  22 #include <TDF_Reference.hxx>
  23 #include <TDF_ChildIDIterator.hxx>
  24 #include <TDF_LabelMapHasher.hxx>
  25
  26 #include <climits>
  27 #ifndef WIN32
  28 #include <sys/stat.h>
  29 #endif
  30
  31 #ifdef WIN32
  32 # define _separator_ '\\'
  33 #else
  34 # define _separator_ '/'
  35 #endif
  36
  37 static const int UNDO_LIMIT = 10; // number of possible undo operations
  38
  39 static const int TAG_GENERAL = 1; // general properties tag
  40 static const int TAG_OBJECTS = 2; // tag of the objects sub-tree (features, results)
  41 static const int TAG_HISTORY = 3; // tag of the history sub-tree (python dump)
  42
  43 // feature sub-labels
  44 static const int TAG_FEATURE_ARGUMENTS = 1; ///< where the arguments are located
  45 static const int TAG_FEATURE_RESULTS = 2; ///< where the results are located
  46
  47
  48 Model_Document::Model_Document(const std::string theID)
  49     : myID(theID), myDoc(new TDocStd_Document("BinOcaf")) // binary OCAF format
  50 {
  51   myDoc->SetUndoLimit(UNDO_LIMIT);
  52   myTransactionsAfterSave = 0;
  53   myNestedNum = -1;
  54   //myDoc->SetNestedTransactionMode();
  55   // to have something in the document and avoid empty doc open/save problem
  56   // in transaction for nesting correct working
  57   myDoc->NewCommand();
  58   TDataStd_Integer::Set(myDoc->Main().Father(), 0);
  59   myDoc->CommitCommand();
  60 }
  61
  62 /// Returns the file name of this document by the nameof directory and identifuer of a document
  63 static TCollection_ExtendedString DocFileName(const char* theFileName, const std::string& theID)
  64 {
  65   TCollection_ExtendedString aPath ((const Standard_CString)theFileName);
  66   aPath += _separator_;
  67   aPath += theID.c_str();
  68   aPath += ".cbf"; // standard binary file extension
  69   return aPath;
  70 }
  71
  72 bool Model_Document::load(const char* theFileName)
  73 {
  74   Handle(Model_Application) anApp = Model_Application::getApplication();
  75   if (this == Model_PluginManager::get()->rootDocument().get()) {
  76     anApp->setLoadPath(theFileName);
  77   }
  78   TCollection_ExtendedString aPath (DocFileName(theFileName, myID));
  79   PCDM_ReaderStatus aStatus = (PCDM_ReaderStatus) -1;
  80   try
  81   {
  82     aStatus = anApp->Open(aPath, myDoc);
  83   }
  84   catch (Standard_Failure)
  85   {
  86     Handle(Standard_Failure) aFail = Standard_Failure::Caught();
  87     Events_Error::send(std::string("Exception in opening of document: ") + aFail->GetMessageString());
  88     return false;
  89   }
  90   bool isError = aStatus != PCDM_RS_OK;
  91   if (isError)
  92   {
  93     switch (aStatus)
  94     {
  95     case PCDM_RS_UnknownDocument:
  96       Events_Error::send(std::string("Can not open document: PCDM_RS_UnknownDocument")); break;
  97     case PCDM_RS_AlreadyRetrieved:
  98       Events_Error::send(std::string("Can not open document: PCDM_RS_AlreadyRetrieved")); break;
  99     case PCDM_RS_AlreadyRetrievedAndModified:
 100       Events_Error::send(
 101         std::string("Can not open document: PCDM_RS_AlreadyRetrievedAndModified"));
 102       break;
 103     case PCDM_RS_NoDriver:
 104       Events_Error::send(std::string("Can not open document: PCDM_RS_NoDriver")); break;
 105     case PCDM_RS_UnknownFileDriver:
 106       Events_Error::send(std::string("Can not open document: PCDM_RS_UnknownFileDriver")); break;
 107     case PCDM_RS_OpenError:
 108       Events_Error::send(std::string("Can not open document: PCDM_RS_OpenError")); break;
 109     case PCDM_RS_NoVersion:
 110       Events_Error::send(std::string("Can not open document: PCDM_RS_NoVersion")); break;
 111     case PCDM_RS_NoModel:
 112       Events_Error::send(std::string("Can not open document: PCDM_RS_NoModel")); break;
 113     case PCDM_RS_NoDocument:
 114       Events_Error::send(std::string("Can not open document: PCDM_RS_NoDocument")); break;
 115     case PCDM_RS_FormatFailure:
 116       Events_Error::send(std::string("Can not open document: PCDM_RS_FormatFailure")); break;
 117     case PCDM_RS_TypeNotFoundInSchema:
 118       Events_Error::send(std::string("Can not open document: PCDM_RS_TypeNotFoundInSchema"));
 119       break;
 120     case PCDM_RS_UnrecognizedFileFormat:
 121       Events_Error::send(std::string("Can not open document: PCDM_RS_UnrecognizedFileFormat"));
 122       break;
 123     case PCDM_RS_MakeFailure:
 124       Events_Error::send(std::string("Can not open document: PCDM_RS_MakeFailure")); break;
 125     case PCDM_RS_PermissionDenied:
 126       Events_Error::send(std::string("Can not open document: PCDM_RS_PermissionDenied")); break;
 127     case PCDM_RS_DriverFailure:
 128       Events_Error::send(std::string("Can not open document: PCDM_RS_DriverFailure")); break;
 129     default:
 130       Events_Error::send(std::string("Can not open document: unknown error")); break;
 131     }
 132   }
 133   if (!isError) {
 134     myDoc->SetUndoLimit(UNDO_LIMIT);
 135     synchronizeFeatures();
 136   }
 137   return !isError;
 138 }
 139
 140 bool Model_Document::save(const char* theFileName)
 141 {
 142   // create a directory in the root document if it is not yet exist
 143   if (this == Model_PluginManager::get()->rootDocument().get()) {
 144 #ifdef WIN32
 145     CreateDirectory(theFileName, NULL);
 146 #else
 147     mkdir(theFileName, 0x1ff);
 148 #endif
 149   }
 150   // filename in the dir is id of document inside of the given directory
 151   TCollection_ExtendedString aPath(DocFileName(theFileName, myID));
 152   PCDM_StoreStatus aStatus;
 153   try {
 154     aStatus = Model_Application::getApplication()->SaveAs(myDoc, aPath);
 155   }
 156   catch (Standard_Failure) {
 157     Handle(Standard_Failure) aFail = Standard_Failure::Caught();
 158     Events_Error::send(std::string("Exception in saving of document: ") + aFail->GetMessageString());
 159     return false;
 160   }
 161   bool isDone = aStatus == PCDM_SS_OK || aStatus == PCDM_SS_No_Obj;
 162   if (!isDone)
 163   {
 164     switch (aStatus)
 165     {
 166     case PCDM_SS_DriverFailure:
 167       Events_Error::send(std::string("Can not save document: PCDM_SS_DriverFailure"));
 168       break;
 169     case PCDM_SS_WriteFailure:
 170       Events_Error::send(std::string("Can not save document: PCDM_SS_WriteFailure"));
 171       break;
 172     case PCDM_SS_Failure:
 173     default:
 174       Events_Error::send(std::string("Can not save document: PCDM_SS_Failure"));
 175       break;
 176     }
 177   }
 178   myTransactionsAfterSave = 0;
 179   if (isDone) { // save also sub-documents if any
 180     std::set<std::string>::iterator aSubIter = mySubs.begin();
 181     for(; aSubIter != mySubs.end() && isDone; aSubIter++)
 182       isDone = subDocument(*aSubIter)->save(theFileName);
 183   }
 184   return isDone;
 185 }
 186
 187 void Model_Document::close()
 188 {
 189   boost::shared_ptr<ModelAPI_PluginManager> aPM = Model_PluginManager::get();
 190   if (this != aPM->rootDocument().get() &&
 191       this == aPM->currentDocument().get()) {
 192     aPM->setCurrentDocument(aPM->rootDocument());
 193   }
 194   // close all subs
 195   std::set<std::string>::iterator aSubIter = mySubs.begin();
 196   for(; aSubIter != mySubs.end(); aSubIter++)
 197     subDocument(*aSubIter)->close();
 198   mySubs.clear();
 199   // close this
 200   /* do not close because it can be undoed
 201   if (myDoc->CanClose() == CDM_CCS_OK)
 202     myDoc->Close();
 203   Model_Application::getApplication()->deleteDocument(myID);
 204   */
 205 }
 206
 207 void Model_Document::startOperation()
 208 {
 209   if (myDoc->HasOpenCommand()) { // start of nested command
 210     if (myNestedNum == -1) {
 211       myNestedNum = 0;
 212       myDoc->InitDeltaCompaction();
 213     }
 214     myIsEmptyTr[myTransactionsAfterSave] = false;
 215     myTransactionsAfterSave++;
 216     myDoc->NewCommand();
 217   } else { // start of simple command
 218     myDoc->NewCommand();
 219   }
 220   // new command for all subs
 221   std::set<std::string>::iterator aSubIter = mySubs.begin();
 222   for(; aSubIter != mySubs.end(); aSubIter++)
 223     subDocument(*aSubIter)->startOperation();
 224 }
 225
 226 void Model_Document::compactNested() {
 227   while(myNestedNum != -1) {
 228     myTransactionsAfterSave--;
 229     myIsEmptyTr.erase(myTransactionsAfterSave);
 230     myNestedNum--;
 231   }
 232   myIsEmptyTr[myTransactionsAfterSave] = false;
 233   myTransactionsAfterSave++;
 234   myDoc->PerformDeltaCompaction();
 235 }
 236
 237 void Model_Document::finishOperation()
 238 {
 239   // just to be sure that everybody knows that changes were performed
 240
 241   if (!myDoc->HasOpenCommand() && myNestedNum != -1)
 242     boost::static_pointer_cast<Model_PluginManager>(Model_PluginManager::get())->
 243       setCheckTransactions(false); // for nested transaction commit
 244   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
 245   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
 246   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
 247   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_DELETED));
 248   if (!myDoc->HasOpenCommand() && myNestedNum != -1)
 249     boost::static_pointer_cast<Model_PluginManager>(Model_PluginManager::get())->
 250       setCheckTransactions(true); // for nested transaction commit
 251
 252   if (myNestedNum != -1) // this nested transaction is owervritten
 253     myNestedNum++;
 254   if (!myDoc->HasOpenCommand()) {
 255     if (myNestedNum != -1) {
 256       myNestedNum--;
 257       compactNested();
 258     }
 259   } else {
 260     // returns false if delta is empty and no transaction was made
 261     myIsEmptyTr[myTransactionsAfterSave] = !myDoc->CommitCommand() && (myNestedNum == -1);
 262     myTransactionsAfterSave++;
 263   }
 264
 265   // finish for all subs
 266   std::set<std::string>::iterator aSubIter = mySubs.begin();
 267   for(; aSubIter != mySubs.end(); aSubIter++)
 268     subDocument(*aSubIter)->finishOperation();
 269 }
 270
 271 void Model_Document::abortOperation()
 272 {
 273   if (myNestedNum > 0 && !myDoc->HasOpenCommand()) { // abort all what was done in nested
 274     // first compact all nested
 275     compactNested();
 276     // for nested it is undo and clear redos
 277     myDoc->Undo();
 278     myDoc->ClearRedos();
 279     myTransactionsAfterSave--;
 280     myIsEmptyTr.erase(myTransactionsAfterSave);
 281   } else {
 282     if (myNestedNum == 0) // abort only high-level
 283       myNestedNum = -1;
 284     myDoc->AbortCommand();
 285   }
 286   synchronizeFeatures(true);
 287   // abort for all subs
 288   std::set<std::string>::iterator aSubIter = mySubs.begin();
 289     for(; aSubIter != mySubs.end(); aSubIter++)
 290     subDocument(*aSubIter)->abortOperation();
 291 }
 292
 293 bool Model_Document::isOperation()
 294 {
 295   // operation is opened for all documents: no need to check subs
 296   return myDoc->HasOpenCommand() == Standard_True;
 297 }
 298
 299 bool Model_Document::isModified()
 300 {
 301   // is modified if at least one operation was commited and not undoed
 302   return myTransactionsAfterSave > 0;
 303 }
 304
 305 bool Model_Document::canUndo()
 306 {
 307   if (myDoc->GetAvailableUndos() > 0 && myNestedNum != 0 && myTransactionsAfterSave != 0 /* for omitting the first useless transaction */)
 308     return true;
 309   // check other subs contains operation that can be undoed
 310   std::set<std::string>::iterator aSubIter = mySubs.begin();
 311   for(; aSubIter != mySubs.end(); aSubIter++)
 312     if (subDocument(*aSubIter)->canUndo())
 313       return true;
 314   return false;
 315 }
 316
 317 void Model_Document::undo()
 318 {
 319   myTransactionsAfterSave--;
 320   if (myNestedNum > 0) myNestedNum--;
 321   if (!myIsEmptyTr[myTransactionsAfterSave])
 322     myDoc->Undo();
 323   synchronizeFeatures(true);
 324   // undo for all subs
 325   std::set<std::string>::iterator aSubIter = mySubs.begin();
 326   for(; aSubIter != mySubs.end(); aSubIter++)
 327     subDocument(*aSubIter)->undo();
 328 }
 329
 330 bool Model_Document::canRedo()
 331 {
 332   if (myDoc->GetAvailableRedos() > 0)
 333     return true;
 334   // check other subs contains operation that can be redoed
 335   std::set<std::string>::iterator aSubIter = mySubs.begin();
 336   for(; aSubIter != mySubs.end(); aSubIter++)
 337     if (subDocument(*aSubIter)->canRedo())
 338       return true;
 339   return false;
 340 }
 341
 342 void Model_Document::redo()
 343 {
 344   if (myNestedNum != -1) myNestedNum++;
 345   if (!myIsEmptyTr[myTransactionsAfterSave])
 346     myDoc->Redo();
 347   myTransactionsAfterSave++;
 348   synchronizeFeatures(true);
 349   // redo for all subs
 350   std::set<std::string>::iterator aSubIter = mySubs.begin();
 351   for(; aSubIter != mySubs.end(); aSubIter++)
 352     subDocument(*aSubIter)->redo();
 353 }
 354
 355 /// Appenad to the array of references a new referenced label
 356 static void AddToRefArray(TDF_Label& theArrayLab, TDF_Label& theReferenced) {
 357   Handle(TDataStd_ReferenceArray) aRefs;
 358   if (!theArrayLab.FindAttribute(TDataStd_ReferenceArray::GetID(), aRefs)) {
 359     aRefs = TDataStd_ReferenceArray::Set(theArrayLab, 0, 0);
 360     aRefs->SetValue(0, theReferenced);
 361   } else { // extend array by one more element
 362     Handle(TDataStd_HLabelArray1) aNewArray =
 363       new TDataStd_HLabelArray1(aRefs->Lower(), aRefs->Upper() + 1);
 364     for(int a = aRefs->Lower(); a <= aRefs->Upper(); a++) {
 365       aNewArray->SetValue(a, aRefs->Value(a));
 366     }
 367     aNewArray->SetValue(aRefs->Upper() + 1, theReferenced);
 368     aRefs->SetInternalArray(aNewArray);
 369   }
 370 }
 371
 372 FeaturePtr Model_Document::addFeature(std::string theID)
 373 {
 374   TDF_Label anEmptyLab;
 375   FeaturePtr anEmptyFeature;
 376   FeaturePtr aFeature = ModelAPI_PluginManager::get()->createFeature(theID);
 377   boost::shared_ptr<Model_Document> aDocToAdd =
 378     boost::dynamic_pointer_cast<Model_Document>(aFeature->documentToAdd());
 379   if (aFeature) {
 380     TDF_Label aFeatureLab;
 381     if (!aFeature->isAction()) {// do not add action to the data model
 382       TDF_Label aFeaturesLab = aDocToAdd->featuresLabel();
 383       aFeatureLab = aFeaturesLab.NewChild();
 384       aDocToAdd->initData(aFeature, aFeatureLab, TAG_FEATURE_ARGUMENTS);
 385       // keep the feature ID to restore document later correctly
 386       TDataStd_Comment::Set(aFeatureLab, aFeature->getKind().c_str());
 387       aDocToAdd->myObjs.Bind(aFeatureLab, aFeature);
 388       // store feature in the history of features array
 389       if (aFeature->isInHistory()) {
 390         AddToRefArray(aFeaturesLab, aFeatureLab);
 391       }
 392     }
 393     if (!aFeature->isAction()) {// do not add action to the data model
 394       // event: feature is added
 395       static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_CREATED);
 396       ModelAPI_EventCreator::get()->sendUpdated(aFeature, anEvent);
 397     }
 398   }
 399   return aFeature;
 400 }
 401
 402 /// Appenad to the array of references a new referenced label.
 403 /// If theIndex is not -1, removes element at thisindex, not theReferenced.
 404 /// \returns the index of removed element
 405 static int RemoveFromRefArray(
 406   TDF_Label theArrayLab, TDF_Label theReferenced, const int theIndex = -1) {
 407   int aResult = -1; // no returned
 408   Handle(TDataStd_ReferenceArray) aRefs;
 409   if (theArrayLab.FindAttribute(TDataStd_ReferenceArray::GetID(), aRefs)) {
 410     if (aRefs->Length() == 1) { // just erase an array
 411       if ((theIndex == -1 && aRefs->Value(0) == theReferenced) || theIndex == 0) {
 412         theArrayLab.ForgetAttribute(TDataStd_ReferenceArray::GetID());
 413       }
 414       aResult = 0;
 415     } else { // reduce the array
 416       Handle(TDataStd_HLabelArray1) aNewArray =
 417         new TDataStd_HLabelArray1(aRefs->Lower(), aRefs->Upper() - 1);
 418       int aCount = aRefs->Lower();
 419       for(int a = aCount; a <= aRefs->Upper(); a++, aCount++) {
 420         if ((theIndex == -1 && aRefs->Value(a) == theReferenced) || theIndex == a) {
 421           aCount--;
 422           aResult = a;
 423         } else {
 424           aNewArray->SetValue(aCount, aRefs->Value(a));
 425         }
 426       }
 427       aRefs->SetInternalArray(aNewArray);
 428     }
 429   }
 430   return aResult;
 431 }
 432
 433 void Model_Document::removeFeature(FeaturePtr theFeature)
 434 {
 435   boost::shared_ptr<Model_Data> aData = boost::static_pointer_cast<Model_Data>(theFeature->data());
 436   TDF_Label aFeatureLabel = aData->label().Father();
 437   if (myObjs.IsBound(aFeatureLabel))
 438     myObjs.UnBind(aFeatureLabel);
 439   else return; // not found feature => do not remove
 440
 441   // erase all attributes under the label of feature
 442   aFeatureLabel.ForgetAllAttributes();
 443   // remove it from the references array
 444   RemoveFromRefArray(featuresLabel(), aFeatureLabel);
 445
 446   // event: feature is deleted
 447   ModelAPI_EventCreator::get()->sendDeleted(theFeature->document(), ModelAPI_Feature::group());
 448   // results of this feature must be redisplayed
 449   static Events_ID EVENT_DISP = Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY);
 450   const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = theFeature->results();
 451   std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
 452   for(; aRIter != aResults.cend(); aRIter++) {
 453     boost::shared_ptr<ModelAPI_Result> aRes = *aRIter;
 454     aRes->setData(boost::shared_ptr<ModelAPI_Data>()); // deleted flag
 455     ModelAPI_EventCreator::get()->sendUpdated(aRes, EVENT_DISP);
 456     ModelAPI_EventCreator::get()->sendDeleted(theFeature->document(), aRes->groupName());
 457   }
 458 }
 459
 460 FeaturePtr Model_Document::feature(TDF_Label& theLabel)
 461 {
 462   if (myObjs.IsBound(theLabel))
 463     return myObjs.Find(theLabel);
 464   return FeaturePtr(); // not found
 465 }
 466
 467 ObjectPtr Model_Document::object(TDF_Label theLabel)
 468 {
 469   // try feature by label
 470   FeaturePtr aFeature = feature(theLabel);
 471   if (aFeature)
 472     return feature(theLabel);
 473   TDF_Label aFeatureLabel = theLabel.Father().Father(); // let's suppose it is result
 474   aFeature = feature(aFeatureLabel);
 475   if (aFeature) {
 476     const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
 477     std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.cbegin();
 478     for(; aRIter != aResults.cend(); aRIter++) {
 479       boost::shared_ptr<Model_Data> aResData =
 480         boost::dynamic_pointer_cast<Model_Data>((*aRIter)->data());
 481       if (aResData->label().Father().IsEqual(theLabel))
 482         return *aRIter;
 483     }
 484   }
 485   return FeaturePtr(); // not found
 486 }
 487
 488 boost::shared_ptr<ModelAPI_Document> Model_Document::subDocument(std::string theDocID)
 489 {
 490   // just store sub-document identifier here to manage it later
 491   if (mySubs.find(theDocID) == mySubs.end())
 492     mySubs.insert(theDocID);
 493   return Model_Application::getApplication()->getDocument(theDocID);
 494 }
 495
 496 ObjectPtr Model_Document::object(const std::string& theGroupID,
 497   const int theIndex, const bool theHidden)
 498 {
 499   if (theGroupID == ModelAPI_Feature::group()) {
 500     if (theHidden) {
 501       int anIndex = 0;
 502       TDF_ChildIDIterator aLabIter(featuresLabel(), TDataStd_Comment::GetID());
 503       for(; aLabIter.More(); aLabIter.Next()) {
 504         if (theIndex == anIndex) {
 505           TDF_Label aFLabel = aLabIter.Value()->Label();
 506           return feature(aFLabel);
 507         }
 508         anIndex++;
 509       }
 510     } else {
 511       Handle(TDataStd_ReferenceArray) aRefs;
 512       if (!featuresLabel().FindAttribute(TDataStd_ReferenceArray::GetID(), aRefs))
 513         return ObjectPtr();
 514       if (aRefs->Lower() > theIndex || aRefs->Upper() < theIndex)
 515         return ObjectPtr();
 516       TDF_Label aFeatureLabel = aRefs->Value(theIndex);
 517       return feature(aFeatureLabel);
 518     }
 519   } else {
 520     // comment must be in any feature: it is kind
 521     int anIndex = 0;
 522     TDF_ChildIDIterator aLabIter(featuresLabel(), TDataStd_Comment::GetID());
 523     for(; aLabIter.More(); aLabIter.Next()) {
 524       TDF_Label aFLabel = aLabIter.Value()->Label();
 525       FeaturePtr aFeature = feature(aFLabel);
 526       const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
 527       std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
 528       for(; aRIter != aResults.cend(); aRIter++) {
 529         if ((theHidden || (*aRIter)->isInHistory()) && (*aRIter)->groupName() == theGroupID) {
 530           if (anIndex == theIndex)
 531             return *aRIter;
 532           anIndex++;
 533         }
 534       }
 535     }
 536   }
 537   // not found
 538   return ObjectPtr();
 539 }
 540
 541 int Model_Document::size(const std::string& theGroupID, const bool theHidden)
 542 {
 543   int aResult = 0;
 544   if (theGroupID == ModelAPI_Feature::group()) {
 545     if (theHidden) {
 546       return myObjs.Size();
 547     } else {
 548       Handle(TDataStd_ReferenceArray) aRefs;
 549       if (featuresLabel().FindAttribute(TDataStd_ReferenceArray::GetID(), aRefs))
 550         return aRefs->Length();
 551     }
 552   } else {
 553     // comment must be in any feature: it is kind
 554     TDF_ChildIDIterator aLabIter(featuresLabel(), TDataStd_Comment::GetID());
 555     for(; aLabIter.More(); aLabIter.Next()) {
 556       TDF_Label aFLabel = aLabIter.Value()->Label();
 557       FeaturePtr aFeature = feature(aFLabel);
 558       const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
 559       std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
 560       for(; aRIter != aResults.cend(); aRIter++) {
 561         if ((theHidden || (*aRIter)->isInHistory()) && (*aRIter)->groupName() == theGroupID) {
 562           aResult++;
 563         }
 564       }
 565     }
 566   }
 567   // group is not found
 568   return aResult;
 569 }
 570
 571 TDF_Label Model_Document::featuresLabel()
 572 {
 573   return myDoc->Main().FindChild(TAG_OBJECTS);
 574 }
 575
 576 void Model_Document::setUniqueName(FeaturePtr theFeature)
 577 {
 578   if (!theFeature->data()->name().empty()) return; // not needed, name is already defined
 579   std::string aName; // result
 580   // first count all objects of such kind to start with index = count + 1
 581   int aNumObjects = 0;
 582   NCollection_DataMap<TDF_Label, FeaturePtr>::Iterator aFIter(myObjs);
 583   for(; aFIter.More(); aFIter.Next()) {
 584     if (aFIter.Value()->getKind() == theFeature->getKind())
 585       aNumObjects++;
 586   }
 587   // generate candidate name
 588   std::stringstream aNameStream;
 589   aNameStream<<theFeature->getKind()<<"_"<<aNumObjects + 1;
 590   aName = aNameStream.str();
 591   // check this is unique, if not, increase index by 1
 592   for(aFIter.Initialize(myObjs); aFIter.More(); ) {
 593     FeaturePtr aFeature = aFIter.Value();
 594     bool isSameName = aFeature->isInHistory() && aFeature->data()->name() == aName;
 595     if (!isSameName) { // check also results to avoid same results names (actual for Parts)
 596       const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
 597       std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
 598       for(; aRIter != aResults.cend(); aRIter++) {
 599         isSameName = (*aRIter)->isInHistory() && (*aRIter)->data()->name() == aName;
 600       }
 601     }
 602     if (isSameName) {
 603       aNumObjects++;
 604       std::stringstream aNameStream;
 605       aNameStream<<theFeature->getKind()<<"_"<<aNumObjects + 1;
 606       aName = aNameStream.str();
 607       // reinitialize iterator to make sure a new name is unique
 608       aFIter.Initialize(myObjs);
 609     } else aFIter.Next();
 610   }
 611   theFeature->data()->setName(aName);
 612 }
 613
 614 void Model_Document::initData(ObjectPtr theObj, TDF_Label theLab, const int theTag) {
 615   boost::shared_ptr<ModelAPI_Document> aThis =
 616     Model_Application::getApplication()->getDocument(myID);
 617   boost::shared_ptr<Model_Data> aData(new Model_Data);
 618   aData->setLabel(theLab.FindChild(theTag));
 619   aData->setObject(theObj);
 620   theObj->setDoc(aThis);
 621   theObj->setData(aData);
 622   FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(theObj);
 623   if (aFeature) {
 624     setUniqueName(aFeature); // must be before "initAttributes" because duplicate part uses name
 625     aFeature->initAttributes();
 626   }
 627 }
 628
 629 void Model_Document::synchronizeFeatures(const bool theMarkUpdated)
 630 {
 631   boost::shared_ptr<ModelAPI_Document> aThis =
 632     Model_Application::getApplication()->getDocument(myID);
 633   // update all objects by checking are they of labels or not
 634   std::set<FeaturePtr> aNewFeatures, aKeptFeatures;
 635   TDF_ChildIDIterator aLabIter(featuresLabel(), TDataStd_Comment::GetID());
 636   for(; aLabIter.More(); aLabIter.Next()) {
 637     TDF_Label aFeatureLabel = aLabIter.Value()->Label();
 638     if (!myObjs.IsBound(aFeatureLabel)) { // a new feature is inserted
 639       // create a feature
 640       FeaturePtr aNewObj = ModelAPI_PluginManager::get()->createFeature(
 641         TCollection_AsciiString(Handle(TDataStd_Comment)::DownCast(aLabIter.Value())->Get())
 642         .ToCString());
 643       // this must be before "setData" to redo the sketch line correctly
 644       myObjs.Bind(aFeatureLabel, aNewObj);
 645       aNewFeatures.insert(aNewObj);
 646       initData(aNewObj, aFeatureLabel, TAG_FEATURE_ARGUMENTS);
 647
 648       // event: model is updated
 649       static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_CREATED);
 650       ModelAPI_EventCreator::get()->sendUpdated(aNewObj, anEvent);
 651       // feature for this label is added, so go to the next label
 652     } else { // nothing is changed, both iterators are incremented
 653       aKeptFeatures.insert(myObjs.Find(aFeatureLabel));
 654       if (theMarkUpdated) {
 655         static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_UPDATED);
 656         ModelAPI_EventCreator::get()->sendUpdated(myObjs.Find(aFeatureLabel), anEvent);
 657       }
 658     }
 659   }
 660   // execute new features to restore results: after features creation to make all references valid
 661   /*std::set<FeaturePtr>::iterator aNewIter = aNewFeatures.begin();
 662   for(; aNewIter != aNewFeatures.end(); aNewIter++) {
 663       (*aNewIter)->execute();
 664   }*/
 665   // check all features are checked: if not => it was removed
 666   NCollection_DataMap<TDF_Label, FeaturePtr>::Iterator aFIter(myObjs);
 667   while(aFIter.More()) {
 668     if (aKeptFeatures.find(aFIter.Value()) == aKeptFeatures.end() &&
 669         aNewFeatures.find(aFIter.Value()) == aNewFeatures.end()) {
 670       FeaturePtr aFeature = aFIter.Value();
 671       TDF_Label aLab = aFIter.Key();
 672       aFIter.Next();
 673       myObjs.UnBind(aLab);
 674       // event: model is updated
 675       if (aFeature->isInHistory()) {
 676         ModelAPI_EventCreator::get()->sendDeleted(aThis, ModelAPI_Feature::group());
 677       }
 678       // results of this feature must be redisplayed (hided)
 679       static Events_ID EVENT_DISP = Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY);
 680       const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
 681       std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
 682       for(; aRIter != aResults.cend(); aRIter++) {
 683         boost::shared_ptr<ModelAPI_Result> aRes = *aRIter;
 684         aRes->setData(boost::shared_ptr<ModelAPI_Data>()); // deleted flag
 685         ModelAPI_EventCreator::get()->sendUpdated(aRes, EVENT_DISP);
 686         ModelAPI_EventCreator::get()->sendDeleted(aThis, aRes->groupName());
 687       }
 688     } else aFIter.Next();
 689   }
 690
 691   // after all updates, sends a message that groups of features were created or updated
 692   boost::static_pointer_cast<Model_PluginManager>(Model_PluginManager::get())->
 693     setCheckTransactions(false);
 694   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
 695   if (theMarkUpdated)
 696     Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
 697   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_DELETED));
 698   boost::static_pointer_cast<Model_PluginManager>(Model_PluginManager::get())->
 699     setCheckTransactions(true);
 700 }
 701
 702 void Model_Document::storeResult(boost::shared_ptr<ModelAPI_Data> theFeatureData,
 703   boost::shared_ptr<ModelAPI_Result> theResult, const int theResultIndex)
 704 {
 705   boost::shared_ptr<ModelAPI_Document> aThis =
 706     Model_Application::getApplication()->getDocument(myID);
 707   theResult->setDoc(aThis);
 708   initData(theResult, boost::dynamic_pointer_cast<Model_Data>(theFeatureData)->label().
 709     Father().FindChild(TAG_FEATURE_RESULTS).FindChild(theResultIndex + 1), TAG_FEATURE_ARGUMENTS);
 710   if (theResult->data()->name().empty()) { // if was not initialized, generate event and set a name
 711     theResult->data()->setName(theFeatureData->name());
 712   }
 713 }
 714
 715 boost::shared_ptr<ModelAPI_ResultConstruction> Model_Document::createConstruction(
 716   const boost::shared_ptr<ModelAPI_Data>& theFeatureData, const int theIndex)
 717 {
 718   ObjectPtr anOldObject = object(boost::dynamic_pointer_cast<Model_Data>(theFeatureData)->
 719     label().Father().FindChild(TAG_FEATURE_RESULTS).FindChild(theIndex + 1));
 720   boost::shared_ptr<ModelAPI_ResultConstruction> aResult;
 721   if (anOldObject) {
 722     aResult = boost::dynamic_pointer_cast<ModelAPI_ResultConstruction>(anOldObject);
 723   }
 724   if (!aResult) {
 725     aResult = boost::shared_ptr<ModelAPI_ResultConstruction>(new Model_ResultConstruction);
 726     storeResult(theFeatureData, aResult);
 727   }
 728   return aResult;
 729 }
 730
 731 boost::shared_ptr<ModelAPI_ResultBody> Model_Document::createBody(
 732   const boost::shared_ptr<ModelAPI_Data>& theFeatureData, const int theIndex)
 733 {
 734   ObjectPtr anOldObject = object(boost::dynamic_pointer_cast<Model_Data>(theFeatureData)->
 735     label().Father().FindChild(TAG_FEATURE_RESULTS).FindChild(theIndex + 1));
 736   boost::shared_ptr<ModelAPI_ResultBody> aResult;
 737   if (anOldObject) {
 738     aResult = boost::dynamic_pointer_cast<ModelAPI_ResultBody>(anOldObject);
 739   }
 740   if (!aResult) {
 741     aResult = boost::shared_ptr<ModelAPI_ResultBody>(new Model_ResultBody);
 742     storeResult(theFeatureData, aResult);
 743   }
 744   return aResult;
 745 }
 746
 747 boost::shared_ptr<ModelAPI_ResultPart> Model_Document::createPart(
 748   const boost::shared_ptr<ModelAPI_Data>& theFeatureData, const int theIndex)
 749 {
 750   ObjectPtr anOldObject = object(boost::dynamic_pointer_cast<Model_Data>(theFeatureData)->
 751     label().Father().FindChild(TAG_FEATURE_RESULTS).FindChild(theIndex + 1));
 752   boost::shared_ptr<ModelAPI_ResultPart> aResult;
 753   if (anOldObject) {
 754     aResult = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(anOldObject);
 755   }
 756   if (!aResult) {
 757     aResult = boost::shared_ptr<ModelAPI_ResultPart>(new Model_ResultPart);
 758     storeResult(theFeatureData, aResult);
 759   }
 760   return aResult;
 761 }
 762
 763 boost::shared_ptr<ModelAPI_Feature> Model_Document::feature(
 764   const boost::shared_ptr<ModelAPI_Result>& theResult)
 765 {
 766   boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(theResult->data());
 767   if (aData) {
 768     TDF_Label aFeatureLab = aData->label().Father().Father().Father();
 769     return feature(aFeatureLab);
 770   }
 771   return FeaturePtr();
 772 }
 773
 774 Standard_Integer HashCode(const TDF_Label& theLab,const Standard_Integer theUpper)
 775 {
 776   return TDF_LabelMapHasher::HashCode(theLab, theUpper);
 777
 778 }
 779 Standard_Boolean IsEqual(const TDF_Label& theLab1,const TDF_Label& theLab2)
 780 {
 781   return TDF_LabelMapHasher::IsEqual(theLab1, theLab2);
 782 }