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