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