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       setUniqueName(aFeature);
 362       myObjs[ModelAPI_Feature::group()].push_back(aFeature);
 363       // store feature in the history of features array
 364       if (aFeature->isInHistory()) {
 365         AddToRefArray(aFeaturesLab, aFeatureLab);
 366       }
 367     }
 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 /// Appenad to the array of references a new referenced label.
 378 /// If theIndex is not -1, removes element at thisindex, not theReferenced.
 379 /// \returns the index of removed element
 380 static int RemoveFromRefArray(
 381   TDF_Label theArrayLab, TDF_Label theReferenced, const int theIndex = -1) {
 382   int aResult = -1; // no returned
 383   Handle(TDataStd_ReferenceArray) aRefs;
 384   if (theArrayLab.FindAttribute(TDataStd_ReferenceArray::GetID(), aRefs)) {
 385     if (aRefs->Length() == 1) { // just erase an array
 386       if ((theIndex == -1 && aRefs->Value(0) == theReferenced) || theIndex == 0) {
 387         theArrayLab.ForgetAttribute(TDataStd_ReferenceArray::GetID());
 388       }
 389       aResult = 0;
 390     } else { // reduce the array
 391       Handle(TDataStd_HLabelArray1) aNewArray =
 392         new TDataStd_HLabelArray1(aRefs->Lower(), aRefs->Upper() - 1);
 393       int aCount = aRefs->Lower();
 394       for(int a = aCount; a <= aRefs->Upper(); a++, aCount++) {
 395         if ((theIndex == -1 && aRefs->Value(a) == theReferenced) || theIndex == a) {
 396           aCount--;
 397           aResult = a;
 398         } else {
 399           aNewArray->SetValue(aCount, aRefs->Value(a));
 400         }
 401       }
 402       aRefs->SetInternalArray(aNewArray);
 403     }
 404   }
 405   return aResult;
 406 }
 407
 408 void Model_Document::removeFeature(FeaturePtr theFeature)
 409 {
 410   boost::shared_ptr<Model_Data> aData = boost::static_pointer_cast<Model_Data>(theFeature->data());
 411   TDF_Label aFeatureLabel = aData->label().Father();
 412   // remove feature from the myObjects list
 413   std::vector<ObjectPtr>& aVec = myObjs[ModelAPI_Feature::group()];
 414   std::vector<ObjectPtr>::iterator anIter = aVec.begin();
 415   while(anIter != aVec.end()) {
 416     if (*anIter == theFeature) {
 417       anIter = aVec.erase(anIter);
 418     } else {
 419       anIter++;
 420     }
 421   }
 422   // erase all attributes under the label of feature
 423   aFeatureLabel.ForgetAllAttributes();
 424   // remove it from the references array
 425   RemoveFromRefArray(groupLabel(ModelAPI_Feature::group()), aData->label());
 426
 427   // event: feature is deleted
 428   ModelAPI_EventCreator::get()->sendDeleted(theFeature->document(), ModelAPI_Feature::group());
 429 }
 430
 431 /// returns the object group name by the object label
 432 static std::string groupName(TDF_Label theObjectLabel) {
 433   TDF_Label aGroupLab = theObjectLabel.Father();
 434   Handle(TDataStd_Comment) aComment;
 435   if (aGroupLab.FindAttribute(TDataStd_Comment::GetID(), aComment))
 436     return std::string(TCollection_AsciiString(aComment->Get()).ToCString());
 437   return ""; // not found
 438 }
 439
 440 FeaturePtr Model_Document::feature(TDF_Label& theLabel)
 441 {
 442   // iterate all features, may be optimized later by keeping labels-map
 443   std::vector<ObjectPtr>& aVec = myObjs[ModelAPI_Feature::group()];
 444   std::vector<ObjectPtr>::iterator aFIter = aVec.begin();
 445   for(; aFIter != aVec.end(); aFIter++) {
 446     boost::shared_ptr<Model_Data> aData =
 447       boost::dynamic_pointer_cast<Model_Data>((*aFIter)->data());
 448     if (aData->label().IsEqual(theLabel))
 449       return boost::dynamic_pointer_cast<ModelAPI_Feature>(*aFIter);
 450   }
 451   return FeaturePtr(); // not found
 452 }
 453
 454 ObjectPtr Model_Document::object(TDF_Label& theLabel)
 455 {
 456   // iterate all features, may be optimized later by keeping labels-map
 457   std::vector<ObjectPtr>& aVec = myObjs[ModelAPI_Feature::group()];
 458   std::vector<ObjectPtr>::iterator aFIter = aVec.begin();
 459   for(; aFIter != aVec.end(); aFIter++) {
 460     boost::shared_ptr<Model_Data> aData =
 461       boost::dynamic_pointer_cast<Model_Data>((*aFIter)->data());
 462     if (aData->label().IsEqual(theLabel))
 463       return *aFIter;
 464     std::list<boost::shared_ptr<ModelAPI_Result> >& aResults =
 465       boost::dynamic_pointer_cast<ModelAPI_Feature>(*aFIter)->results();
 466     std::list<boost::shared_ptr<ModelAPI_Result> >::iterator aRIter = aResults.begin();
 467     for(; aRIter != aResults.end(); aRIter++) {
 468       boost::shared_ptr<Model_Data> aResData =
 469         boost::dynamic_pointer_cast<Model_Data>((*aRIter)->data());
 470       if (aResData->label().IsEqual(theLabel))
 471         return *aRIter;
 472     }
 473   }
 474   return FeaturePtr(); // not found
 475 }
 476
 477 boost::shared_ptr<ModelAPI_Document> Model_Document::subDocument(std::string theDocID)
 478 {
 479   // just store sub-document identifier here to manage it later
 480   if (mySubs.find(theDocID) == mySubs.end())
 481     mySubs.insert(theDocID);
 482   return Model_Application::getApplication()->getDocument(theDocID);
 483 }
 484
 485 ObjectPtr Model_Document::object(const std::string& theGroupID, const int theIndex)
 486 {
 487   if (theGroupID == ModelAPI_Feature::group()) {
 488     std::map<std::string, std::vector<ObjectPtr> >::iterator aFind = myObjs.find(theGroupID);
 489     if (aFind != myObjs.end()) {
 490       int aSize = aFind->second.size();
 491       if (theIndex >= 0 && theIndex < aSize)
 492         return aFind->second[theIndex];
 493     }
 494   } else {
 495     // iterate all features in order to find the needed result
 496     std::map<std::string, std::vector<ObjectPtr> >::iterator aFind =
 497       myObjs.find(ModelAPI_Feature::group());
 498     if (aFind != myObjs.end()) {
 499       std::vector<ObjectPtr>::iterator aFIter = aFind->second.begin();
 500       for(int anIndex = 0; aFIter != aFind->second.end(); aFIter++) {
 501         const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults =
 502           boost::dynamic_pointer_cast<ModelAPI_Feature>(*aFIter)->results();
 503         std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
 504         for(; aRIter != aResults.cend(); aRIter++) {
 505           if ((*aRIter)->isInHistory() && (*aRIter)->groupName() == theGroupID) {
 506             if (anIndex == theIndex)
 507               return *aRIter;
 508             anIndex++;
 509           }
 510         }
 511       }
 512     }
 513   }
 514   // not found
 515   return ObjectPtr();
 516 }
 517
 518 int Model_Document::size(const std::string& theGroupID)
 519 {
 520   int aResult = 0;
 521   if (theGroupID == ModelAPI_Feature::group()) {
 522     std::map<std::string, std::vector<ObjectPtr> >::iterator aFind = myObjs.find(theGroupID);
 523     if (aFind != myObjs.end()) {
 524       aResult = aFind->second.size();
 525     }
 526   } else {
 527     // iterate all features in order to find the needed result
 528     std::map<std::string, std::vector<ObjectPtr> >::iterator aFind =
 529       myObjs.find(ModelAPI_Feature::group());
 530     if (aFind != myObjs.end()) {
 531       std::vector<ObjectPtr>::iterator aFIter = aFind->second.begin();
 532       for(; aFIter != aFind->second.end(); aFIter++) {
 533         const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults =
 534           boost::dynamic_pointer_cast<ModelAPI_Feature>(*aFIter)->results();
 535         std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
 536         for(; aRIter != aResults.cend(); aRIter++) {
 537           if ((*aRIter)->isInHistory() && (*aRIter)->groupName() == theGroupID) {
 538             aResult++;
 539           }
 540         }
 541       }
 542     }
 543   }
 544   // group is not found
 545   return aResult;
 546 }
 547
 548 Model_Document::Model_Document(const std::string theID)
 549     : myID(theID), myDoc(new TDocStd_Document("BinOcaf")) // binary OCAF format
 550 {
 551   myDoc->SetUndoLimit(UNDO_LIMIT);
 552   myTransactionsAfterSave = 0;
 553   myNestedNum = -1;
 554   //myDoc->SetNestedTransactionMode();
 555   // to have something in the document and avoid empty doc open/save problem
 556   // in transaction for nesting correct working
 557   myDoc->NewCommand();
 558   TDataStd_Integer::Set(myDoc->Main().Father(), 0);
 559   myDoc->CommitCommand();
 560 }
 561
 562 TDF_Label Model_Document::groupLabel(const std::string theGroup)
 563 {
 564   // searching for existing
 565   TCollection_ExtendedString aGroup(theGroup.c_str());
 566   TDF_ChildIDIterator aGroupIter(myDoc->Main().FindChild(TAG_OBJECTS), TDataStd_Comment::GetID());
 567   for(; aGroupIter.More(); aGroupIter.Next()) {
 568     Handle(TDataStd_Comment) aName = Handle(TDataStd_Comment)::DownCast(aGroupIter.Value());
 569     if (aName->Get() == aGroup)
 570       return aGroupIter.Value()->Label();
 571   }
 572   // create a new
 573   TDF_Label aNew = myDoc->Main().FindChild(TAG_OBJECTS).NewChild();
 574   TDataStd_Comment::Set(aNew, aGroup);
 575   return aNew;
 576 }
 577
 578 void Model_Document::setUniqueName(FeaturePtr theFeature)
 579 {
 580   std::string aName; // result
 581   // first count all objects of such kind to start with index = count + 1
 582   int a, aNumObjects = 0;
 583   int aSize = size(ModelAPI_Feature::group());
 584   for(a = 0; a < aSize; a++) {
 585     if (boost::dynamic_pointer_cast<ModelAPI_Feature>(object(ModelAPI_Feature::group(), a))->
 586         getKind() == theFeature->getKind())
 587       aNumObjects++;
 588   }
 589   // generate candidate name
 590   std::stringstream aNameStream;
 591   aNameStream<<theFeature->getKind()<<"_"<<aNumObjects + 1;
 592   aName = aNameStream.str();
 593   // check this is unique, if not, increase index by 1
 594   for(a = 0; a < aSize;) {
 595     if (boost::dynamic_pointer_cast<ModelAPI_Feature>(object(ModelAPI_Feature::group(), a))
 596           ->data()->name() == aName) {
 597       aNumObjects++;
 598       std::stringstream aNameStream;
 599       aNameStream<<theFeature->getKind()<<"_"<<aNumObjects + 1;
 600       aName = aNameStream.str();
 601       // reinitialize iterator to make sure a new name is unique
 602       a = 0;
 603     } else a++;
 604   }
 605   theFeature->data()->setName(aName);
 606 }
 607
 608 void Model_Document::initData(ObjectPtr theObj, TDF_Label& theLab, const int theTag) {
 609   boost::shared_ptr<ModelAPI_Document> aThis =
 610     Model_Application::getApplication()->getDocument(myID);
 611   boost::shared_ptr<Model_Data> aData(new Model_Data);
 612   aData->setLabel(theLab.FindChild(theTag + 1));
 613   aData->setObject(theObj);
 614   theObj->setDoc(aThis);
 615   theObj->setData(aData);
 616   FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(theObj);
 617   if (aFeature) aFeature->initAttributes();
 618 }
 619
 620 void Model_Document::synchronizeFeatures(const bool theMarkUpdated)
 621 {
 622   boost::shared_ptr<ModelAPI_Document> aThis =
 623     Model_Application::getApplication()->getDocument(myID);
 624   // update all objects: iterate from the end: as they appeared in the list
 625   std::map<std::string, std::vector<ObjectPtr> >::reverse_iterator aGroupIter = myObjs.rbegin();
 626   for(; aGroupIter != myObjs.rend(); aGroupIter++) {
 627     std::vector<ObjectPtr>::iterator anObjIter = aGroupIter->second.begin();
 628     // and in parallel iterate labels of features
 629     const std::string& aGroupName = aGroupIter->first;
 630     TDF_ChildIDIterator aLabIter(groupLabel(aGroupName), TDataStd_Comment::GetID());
 631     while(anObjIter != aGroupIter->second.end() || aLabIter.More()) {
 632       static const int INFINITE_TAG = INT_MAX; // no label means that it exists somwhere in infinite
 633       int aFeatureTag = INFINITE_TAG;
 634       if (anObjIter != aGroupIter->second.end()) { // existing tag for feature
 635         boost::shared_ptr<Model_Data> aData =
 636           boost::dynamic_pointer_cast<Model_Data>((*anObjIter)->data());
 637         aFeatureTag = aData->label().Tag();
 638       }
 639       int aDSTag = INFINITE_TAG;
 640       if (aLabIter.More()) { // next label in DS is existing
 641         aDSTag = aLabIter.Value()->Label().Tag();
 642       }
 643       if (aDSTag > aFeatureTag) { // feature is removed
 644         ObjectPtr anObj = *anObjIter;
 645         anObjIter = aGroupIter->second.erase(anObjIter);
 646         // event: model is updated
 647         if (anObj->isInHistory()) {
 648           ModelAPI_EventCreator::get()->sendDeleted(aThis, aGroupName);
 649         }
 650         ModelAPI_EventCreator::get()->sendDeleted(aThis, aGroupName);
 651       } else if (aDSTag < aFeatureTag) { // a new feature is inserted
 652         // create a feature
 653         TDF_Label aLab = aLabIter.Value()->Label();
 654         ObjectPtr aNewObj = ModelAPI_PluginManager::get()->createFeature(
 655           TCollection_AsciiString(Handle(TDataStd_Comment)::DownCast(aLabIter.Value())->Get())
 656           .ToCString());
 657         // this must be before "setData" to redo the sketch line correctly
 658         if (anObjIter == aGroupIter->second.end()) {
 659           aGroupIter->second.push_back(aNewObj);
 660           anObjIter = aGroupIter->second.end();
 661         } else {
 662           anObjIter++;
 663           aGroupIter->second.insert(anObjIter, aNewObj);
 664         }
 665         initData(aNewObj, aLab, TAG_FEATURE_ARGUMENTS);
 666
 667         // event: model is updated
 668         static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_CREATED);
 669         ModelAPI_EventCreator::get()->sendUpdated(aNewObj, anEvent);
 670         // feature for this label is added, so go to the next label
 671         aLabIter.Next();
 672       } else { // nothing is changed, both iterators are incremented
 673         if (theMarkUpdated) {
 674           static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_UPDATED);
 675           ModelAPI_EventCreator::get()->sendUpdated(*anObjIter, anEvent);
 676         }
 677         anObjIter++;
 678         aLabIter.Next();
 679       }
 680     }
 681   }
 682   // after all updates, sends a message that groups of features were created or updated
 683   boost::static_pointer_cast<Model_PluginManager>(Model_PluginManager::get())->
 684     setCheckTransactions(false);
 685   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
 686   if (theMarkUpdated)
 687     Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
 688   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_DELETED));
 689   boost::static_pointer_cast<Model_PluginManager>(Model_PluginManager::get())->
 690     setCheckTransactions(true);
 691 }
 692
 693 void Model_Document::storeResult(boost::shared_ptr<ModelAPI_Data> theFeatureData,
 694   boost::shared_ptr<ModelAPI_Result> theResult, const int theResultIndex)
 695 {
 696   boost::shared_ptr<ModelAPI_Document> aThis =
 697     Model_Application::getApplication()->getDocument(myID);
 698   theResult->setDoc(aThis);
 699   initData(theResult, boost::dynamic_pointer_cast<Model_Data>(theFeatureData)->
 700     label().Father().FindChild(TAG_FEATURE_RESULTS), theResultIndex);
 701   if (theResult->data()->name().empty()) { // if was not initialized, generate event and set a name
 702     static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_CREATED);
 703     ModelAPI_EventCreator::get()->sendUpdated(theResult, anEvent);
 704     theResult->data()->setName(theFeatureData->name());
 705   }
 706 }
 707
 708 boost::shared_ptr<ModelAPI_ResultConstruction> Model_Document::createConstruction(
 709   const boost::shared_ptr<ModelAPI_Data>& theFeatureData, const int theIndex)
 710 {
 711   boost::shared_ptr<ModelAPI_ResultConstruction> aResult(new Model_ResultConstruction());
 712   storeResult(theFeatureData, aResult);
 713   return aResult;
 714 }
 715
 716 boost::shared_ptr<ModelAPI_ResultBody> Model_Document::createBody(
 717   const boost::shared_ptr<ModelAPI_Data>& theFeatureData, const int theIndex)
 718 {
 719   boost::shared_ptr<ModelAPI_ResultBody> aResult(new Model_ResultBody());
 720   storeResult(theFeatureData, aResult);
 721   return aResult;
 722 }
 723
 724 boost::shared_ptr<ModelAPI_ResultPart> Model_Document::createPart(
 725   const boost::shared_ptr<ModelAPI_Data>& theFeatureData, const int theIndex)
 726 {
 727   boost::shared_ptr<ModelAPI_ResultPart> aResult(new Model_ResultPart());
 728   storeResult(theFeatureData, aResult);
 729   return aResult;
 730 }
 731
 732 boost::shared_ptr<ModelAPI_Feature> Model_Document::feature(
 733   const boost::shared_ptr<ModelAPI_Result>& theResult)
 734 {
 735   // iterate all features in order to find the needed result
 736   std::map<std::string, std::vector<ObjectPtr> >::iterator aFind =
 737     myObjs.find(ModelAPI_Feature::group());
 738   if (aFind != myObjs.end()) {
 739     std::vector<ObjectPtr>::iterator aFIter = aFind->second.begin();
 740     for(; aFIter != aFind->second.end(); aFIter++) {
 741       FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(*aFIter);
 742       const std::list<boost::shared_ptr<ModelAPI_Result> >& aResults = aFeature->results();
 743       std::list<boost::shared_ptr<ModelAPI_Result> >::const_iterator aRIter = aResults.begin();
 744       for(; aRIter != aResults.cend(); aRIter++) {
 745         if (*aRIter == theResult) {
 746           return aFeature;
 747         }
 748       }
 749     }
 750   }
 751   return FeaturePtr();
 752 }