1 // Copyright (C) 2005 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
2 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
4 // This library is free software; you can redistribute it and/or
5 // modify it under the terms of the GNU Lesser General Public
6 // License as published by the Free Software Foundation; either
7 // version 2.1 of the License.
9 // This library is distributed in the hope that it will be useful
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 // Lesser General Public License for more details.
14 // You should have received a copy of the GNU Lesser General Public
15 // License along with this library; if not, write to the Free Software
16 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 // See http://www.salome-platform.org/
20 // File : SALOMEDSImpl_StudyBuilder.cxx
21 // Author : Sergey RUIN
26 #include "SALOMEDSImpl_Attributes.hxx"
30 #include "SALOMEDSImpl_Study.hxx"
31 #include "SALOMEDSImpl_StudyBuilder.hxx"
32 #include "SALOMEDSImpl_SObject.hxx"
33 #include "SALOMEDSImpl_SComponent.hxx"
34 #include "SALOMEDSImpl_Tool.hxx"
36 #include <TDF_ChildIterator.hxx>
37 #include <TDF_Label.hxx>
38 #include <TDF_Tool.hxx>
39 #include <TDF_Data.hxx>
40 #include <SALOMEDSImpl_ChildNodeIterator.hxx>
41 #include <TDF_ListIteratorOfAttributeList.hxx>
46 IMPLEMENT_STANDARD_HANDLE( SALOMEDSImpl_StudyBuilder, MMgt_TShared )
47 IMPLEMENT_STANDARD_RTTIEXT( SALOMEDSImpl_StudyBuilder, MMgt_TShared )
49 #define USE_CASE_LABEL_TAG 2
50 #define DIRECTORYID 16661
51 #define FILELOCALID 26662
53 static void Translate_persistentID_to_IOR(TDF_Label& Lab, SALOMEDSImpl_Driver* driver, bool isMultiFile, bool isASCII);
55 //============================================================================
56 /*! Function : constructor
59 //============================================================================
60 SALOMEDSImpl_StudyBuilder::SALOMEDSImpl_StudyBuilder(const Handle(Standard_Transient)& theOwner)
64 _doc = Handle(SALOMEDSImpl_Study)::DownCast(theOwner)->GetDocument();
67 //============================================================================
68 /*! Function : destructor
71 //============================================================================
72 SALOMEDSImpl_StudyBuilder::~SALOMEDSImpl_StudyBuilder()
75 //============================================================================
76 /*! Function : NewComponent
77 * Purpose : Create a new component (Scomponent)
79 //============================================================================
80 Handle(SALOMEDSImpl_SComponent) SALOMEDSImpl_StudyBuilder::NewComponent(const TCollection_AsciiString& DataType)
84 //Always create component under main label.
85 TDF_Label L = _doc->Main();
88 for (TDF_ChildIterator it(L); it.More(); it.Next()) {
89 if (it.Value().Tag() > imax)
90 imax = it.Value().Tag();
93 TDF_Label NL = L.FindChild(imax);
95 SALOMEDSImpl_AttributeComment::Set(NL, DataType);
97 Handle(SALOMEDSImpl_SComponent) so = Handle(SALOMEDSImpl_Study)::DownCast(_study)->GetSComponent (NL);
99 if(!_callbackOnAdd.IsNull()) _callbackOnAdd->OnAddSObject(so);
106 //============================================================================
107 /*! Function : DefineComponentInstance
108 * Purpose : Add IOR attribute of a Scomponent
110 //============================================================================
111 bool SALOMEDSImpl_StudyBuilder::DefineComponentInstance(const Handle(SALOMEDSImpl_SComponent)& aComponent,
112 const TCollection_AsciiString& IOR)
117 if(aComponent.IsNull() || IOR.IsEmpty()) {
118 _errorCode = "Invalid arguments";
122 SALOMEDSImpl_AttributeIOR::Set(aComponent->GetLabel(), IOR);
129 //============================================================================
130 /*! Function : RemoveComponent
131 * Purpose : Delete a Scomponent
133 //============================================================================
134 bool SALOMEDSImpl_StudyBuilder::RemoveComponent(const Handle(SALOMEDSImpl_SComponent)& aComponent)
138 return RemoveObject(aComponent);
141 //============================================================================
142 /*! Function : NewObject
143 * Purpose : Create a new SObject
145 //============================================================================
146 Handle(SALOMEDSImpl_SObject) SALOMEDSImpl_StudyBuilder::NewObject(const Handle(SALOMEDSImpl_SObject)& theFatherObject)
151 //Find label of father
152 TDF_Label Lab = theFatherObject->GetLabel();
156 for (TDF_ChildIterator it(Lab); it.More(); it.Next()) {
157 if (it.Value().Tag() > imax)
158 imax = it.Value().Tag();
161 TDF_Label NewLab = Lab.FindChild(imax);
163 Handle(SALOMEDSImpl_SObject) so = Handle(SALOMEDSImpl_Study)::DownCast(_study)->GetSObject(NewLab);
164 if(!_callbackOnAdd.IsNull()) _callbackOnAdd->OnAddSObject(so);
170 //============================================================================
171 /*! Function : NewObjectToTag
174 //============================================================================
175 Handle(SALOMEDSImpl_SObject) SALOMEDSImpl_StudyBuilder::NewObjectToTag(const Handle(SALOMEDSImpl_SObject)& theFatherObject,
180 //Find label of father
181 TDF_Label Lab = theFatherObject->GetLabel();
183 //Create or find label
184 TDF_Label NewLab = Lab.FindChild(theTag, 1);
186 Handle(SALOMEDSImpl_SObject) so = Handle(SALOMEDSImpl_Study)::DownCast(_study)->GetSObject(NewLab);
188 if(!_callbackOnAdd.IsNull()) _callbackOnAdd->OnAddSObject(so);
194 //============================================================================
195 /*! Function : RemoveObject
198 //============================================================================
199 bool SALOMEDSImpl_StudyBuilder::RemoveObject(const Handle(SALOMEDSImpl_SObject)& anObject)
203 if(anObject.IsNull()) {
204 _errorCode = "Null object";
208 if(!_callbackOnRemove.IsNull()) _callbackOnRemove->OnRemoveSObject(anObject);
210 TDF_Label Lab = anObject->GetLabel();
212 Handle(SALOMEDSImpl_AttributeReference) aReference;
213 if (Lab.FindAttribute(SALOMEDSImpl_AttributeReference::GetID(), aReference)) {
214 Handle(SALOMEDSImpl_AttributeTarget) aTarget;
215 if (aReference->Get().FindAttribute(SALOMEDSImpl_AttributeTarget::GetID(),aTarget))
216 aTarget->Remove(SALOMEDSImpl_Study::SObject(Lab));
219 Handle(SALOMEDSImpl_AttributeIOR) anAttr; // postponed removing of CORBA objects
220 if (Lab.FindAttribute(SALOMEDSImpl_AttributeIOR::GetID(), anAttr))
221 SALOMEDSImpl_Study::GetStudy(_doc->Main())->AddPostponed(TCollection_AsciiString(anAttr->Value()).ToCString());
223 Lab.ForgetAllAttributes();
230 //============================================================================
231 /*! Function : RemoveObjectWithChildren
234 //============================================================================
235 bool SALOMEDSImpl_StudyBuilder::RemoveObjectWithChildren(const Handle(SALOMEDSImpl_SObject)& anObject)
239 if(anObject.IsNull()) {
240 _errorCode = "Null object";
244 if(!_callbackOnRemove.IsNull()) _callbackOnRemove->OnRemoveSObject(anObject);
246 TDF_Label Lab = anObject->GetLabel();
248 Handle(SALOMEDSImpl_AttributeReference) aReference;
249 if (Lab.FindAttribute(SALOMEDSImpl_AttributeReference::GetID(), aReference)) {
250 Handle(SALOMEDSImpl_AttributeTarget) aTarget;
251 if (aReference->Get().FindAttribute(SALOMEDSImpl_AttributeTarget::GetID(),aTarget))
252 aTarget->Remove(SALOMEDSImpl_Study::SObject(Lab));
254 Handle(SALOMEDSImpl_AttributeIOR) anAttr; // postponed removing of CORBA objects
255 if (Lab.FindAttribute(SALOMEDSImpl_AttributeIOR::GetID(), anAttr))
256 SALOMEDSImpl_Study::GetStudy(_doc->Main())->AddPostponed(TCollection_AsciiString(anAttr->Value()).ToCString());
258 TDF_ChildIterator it(Lab, Standard_True);
259 for(;it.More();it.Next()) {
260 TDF_Label aLabel = it.Value();
261 if (aLabel.FindAttribute(SALOMEDSImpl_AttributeReference::GetID(), aReference)) {
262 Handle(SALOMEDSImpl_AttributeTarget) aTarget;
263 if (aReference->Get().FindAttribute(SALOMEDSImpl_AttributeTarget::GetID(),aTarget))
264 aTarget->Remove(SALOMEDSImpl_Study::SObject(aLabel));
266 Handle(SALOMEDSImpl_AttributeIOR) anAttr; // postponed removing of CORBA objects
267 if (aLabel.FindAttribute(SALOMEDSImpl_AttributeIOR::GetID(), anAttr))
268 SALOMEDSImpl_Study::GetStudy(_doc->Main())->AddPostponed(TCollection_AsciiString(anAttr->Value()).ToCString());
271 Lab.ForgetAllAttributes(Standard_True);
278 //============================================================================
279 /*! Function : LoadWith
282 //============================================================================
283 bool SALOMEDSImpl_StudyBuilder::LoadWith(const Handle(SALOMEDSImpl_SComponent)& anSCO, SALOMEDSImpl_Driver* aDriver)
287 TDF_Label Lab = anSCO->GetLabel();
288 Handle(SALOMEDSImpl_AttributePersistentRef) Att;
290 //Find the current Url of the study
291 if (_doc->Main().FindAttribute(SALOMEDSImpl_AttributePersistentRef::GetID(),Att)) {
292 int aLocked = anSCO->GetStudy()->GetProperties()->IsLocked();
293 if (aLocked) anSCO->GetStudy()->GetProperties()->SetLocked(false);
295 TCollection_ExtendedString Res(Att->Value());
297 Handle(SALOMEDSImpl_AttributeComment) type;
298 TCollection_ExtendedString DataType;
299 if (Lab.FindAttribute(SALOMEDSImpl_AttributeComment::GetID(),type))
300 DataType = type->Value();
302 // associate the driver to the SComponent
303 if(aDriver == NULL) {
304 _errorCode = "Driver is null";
308 // mpv 06.03.2003: SAL1927 - if component data if already loaded, it is not necessary to do it again
309 Handle(SALOMEDSImpl_AttributeIOR) attrIOR;
310 if (Lab.FindAttribute(SALOMEDSImpl_AttributeIOR::GetID(), attrIOR)) {
311 if (aLocked) anSCO->GetStudy()->GetProperties()->SetLocked(true);
315 DefineComponentInstance (anSCO, aDriver->GetIOR());
317 TCollection_AsciiString aHDFPath(Res);
320 bool isASCII = false;
321 if (HDFascii::isASCII(aHDFPath.ToCString())) {
323 char* aResultPath = HDFascii::ConvertFromASCIIToHDF(aHDFPath.ToCString());
324 aHDFUrl = new char[strlen(aResultPath) + 19];
325 sprintf(aHDFUrl, "%shdf_from_ascii.hdf", aResultPath);
328 aHDFUrl = aHDFPath.ToCString();
331 //Open the Study HDF file
332 HDFfile *hdf_file = new HDFfile(aHDFUrl);
334 char aMultifileState[2];
335 char ASCIIfileState[2];
337 TCollection_AsciiString scoid = anSCO->GetID();
338 hdf_file->OpenOnDisk(HDF_RDONLY);
339 HDFgroup *hdf_group = new HDFgroup("DATACOMPONENT",hdf_file);
340 hdf_group->OpenOnDisk();
341 HDFgroup *hdf_sco_group = new HDFgroup(scoid.ToCString(), hdf_group);
342 hdf_sco_group->OpenOnDisk();
344 unsigned char* aStreamFile = NULL;
347 if (hdf_sco_group->ExistInternalObject("FILE_STREAM")) {
348 HDFdataset *hdf_dataset = new HDFdataset("FILE_STREAM", hdf_sco_group);
349 hdf_dataset->OpenOnDisk();
350 aStreamSize = hdf_dataset->GetSize();
351 aStreamFile = new unsigned char[aStreamSize];
352 if(aStreamFile == NULL) throw HDFexception("Unable to open dataset FILE_STREAM");
353 hdf_dataset->ReadFromDisk(aStreamFile);
354 hdf_dataset->CloseOnDisk();
356 } else aStreamFile = NULL;
358 HDFdataset *multifile_hdf_dataset = new HDFdataset("MULTIFILE_STATE", hdf_sco_group);
359 multifile_hdf_dataset->OpenOnDisk();
360 multifile_hdf_dataset->ReadFromDisk(aMultifileState);
362 HDFdataset *ascii_hdf_dataset = new HDFdataset("ASCII_STATE", hdf_sco_group);
363 ascii_hdf_dataset->OpenOnDisk();
364 ascii_hdf_dataset->ReadFromDisk(ASCIIfileState);
366 // set path without file name from URL
367 int aFileNameSize = Res.Length();
368 char* aDir = new char[aFileNameSize];
369 memcpy(aDir, TCollection_AsciiString(Res).ToCString(), aFileNameSize);
370 for(int aCounter = aFileNameSize-1; aCounter>=0; aCounter--)
371 if (aDir[aCounter] == '/') {
372 aDir[aCounter+1] = 0;
376 bool aResult = (ASCIIfileState[0]=='A')?
377 aDriver->LoadASCII(anSCO, aStreamFile, aStreamSize, aDir, aMultifileState[0]=='M'):
378 aDriver->Load(anSCO, aStreamFile, aStreamSize, aDir, aMultifileState[0]=='M');
380 if(aStreamFile != NULL) delete []aStreamFile;
383 RemoveAttribute( anSCO, "AttributeIOR" );
385 _errorCode = "Can't load component";
386 throw HDFexception("Unable to load component");
389 if(aDir != NULL) delete []aDir;
391 multifile_hdf_dataset->CloseOnDisk();
392 multifile_hdf_dataset = 0;
393 ascii_hdf_dataset->CloseOnDisk();
394 ascii_hdf_dataset = 0;
396 hdf_sco_group->CloseOnDisk();
398 hdf_group->CloseOnDisk();
400 hdf_file->CloseOnDisk();
404 Handle(TColStd_HSequenceOfAsciiString) aFilesToRemove = new TColStd_HSequenceOfAsciiString;
405 aFilesToRemove->Append(aHDFUrl);
406 SALOMEDSImpl_Tool::RemoveTemporaryFiles(SALOMEDSImpl_Tool::GetDirFromPath(aHDFUrl), aFilesToRemove, true);
411 catch (HDFexception) {
415 Handle(TColStd_HSequenceOfAsciiString) aFilesToRemove = new TColStd_HSequenceOfAsciiString;
416 aFilesToRemove->Append(aHDFUrl);
417 SALOMEDSImpl_Tool::RemoveTemporaryFiles(SALOMEDSImpl_Tool::GetDirFromPath(aHDFUrl), aFilesToRemove, true);
421 if (aLocked) anSCO->GetStudy()->GetProperties()->SetLocked(true);
422 _errorCode = "No persistent file";
427 Translate_persistentID_to_IOR (Lab, aDriver, aMultifileState[0]=='M', ASCIIfileState[0] == 'A');
429 _errorCode = "Can not convert persistent IDs to IORs";
433 if (aLocked) anSCO->GetStudy()->GetProperties()->SetLocked(true);
435 _errorCode = "No persistent file";
442 //============================================================================
446 //============================================================================
447 bool SALOMEDSImpl_StudyBuilder::Load(const Handle(SALOMEDSImpl_SObject)& sco)
449 _errorCode = "Not implemented";
453 //============================================================================
454 /*! Function : FindOrCreateAttribute
455 * Purpose : Add attribute of given type to SObject, if there is attribute of such type, returns
458 //============================================================================
459 Handle(TDF_Attribute) SALOMEDSImpl_StudyBuilder::FindOrCreateAttribute(const Handle(SALOMEDSImpl_SObject)& anObject,
460 const TCollection_AsciiString& aTypeOfAttribute)
463 if(anObject.IsNull()) {
464 _errorCode = "Invalid arguments";
468 TDF_Label Lab = anObject->GetLabel();
470 _errorCode = "Null label";
476 //The macro adds all necessary checks for standardly behaiving attributes
477 __FindOrCreateAttributeForBuilder
479 //Add checks for TreeNode and UserID attributes
480 if (strncmp(aTypeOfAttribute.ToCString(), "AttributeTreeNode",17) == 0 ) {
481 Standard_GUID aTreeNodeGUID;
482 if (strcmp(aTypeOfAttribute.ToCString(), "AttributeTreeNode") == 0) {
483 aTreeNodeGUID = SALOMEDSImpl_AttributeTreeNode::GetDefaultTreeID();
485 char* aGUIDString = new char[40];
486 char* aType = (char*)aTypeOfAttribute.ToCString();
487 sprintf(aGUIDString, &(aType[21]));
488 aTreeNodeGUID = Standard_GUID(aGUIDString); // create tree node GUID by name
491 Handle(SALOMEDSImpl_AttributeTreeNode) anAttr;
492 if (!Lab.FindAttribute(aTreeNodeGUID, anAttr)) {
494 anAttr = SALOMEDSImpl_AttributeTreeNode::Set(Lab, aTreeNodeGUID);
499 if (strncmp(aTypeOfAttribute.ToCString(), "AttributeUserID",15) == 0 ) {
500 Handle(SALOMEDSImpl_AttributeUserID) anAttr;
501 if (!Lab.FindAttribute(SALOMEDSImpl_AttributeUserID::DefaultID(), anAttr)) {
503 anAttr = SALOMEDSImpl_AttributeUserID::Set(Lab, SALOMEDSImpl_AttributeUserID::DefaultID());
507 _errorCode = "Can not create an attribute";
512 //============================================================================
513 /*! Function : FindAttribute
514 * Purpose : Find attribute of given type assigned SObject, returns Standard_True if it is found
516 //============================================================================
518 bool SALOMEDSImpl_StudyBuilder::FindAttribute(const Handle(SALOMEDSImpl_SObject)& anObject,
519 Handle(TDF_Attribute)& anAttribute,
520 const TCollection_AsciiString& aTypeOfAttribute)
523 if(anObject.IsNull()) {
524 _errorCode = "Invalid arguments";
527 TDF_Label Lab = anObject->GetLabel();
528 if (Lab.FindAttribute(SALOMEDSImpl_SObject::GetGUID(aTypeOfAttribute), anAttribute)) {
530 return Standard_True;
532 return Standard_False;
535 //============================================================================
536 /*! Function : RemoveAttribute
537 * Purpose : Remove attribute of given type assigned SObject
539 //============================================================================
541 bool SALOMEDSImpl_StudyBuilder::RemoveAttribute(const Handle(SALOMEDSImpl_SObject)& anObject,
542 const TCollection_AsciiString& aTypeOfAttribute)
546 if(anObject.IsNull()) {
547 _errorCode = "Invalid arguments";
550 TDF_Label Lab = anObject->GetLabel();
552 if (aTypeOfAttribute == "AttributeIOR") { // postponed removing of CORBA objects
553 Handle(SALOMEDSImpl_AttributeIOR) anAttr;
554 if (Lab.FindAttribute(SALOMEDSImpl_AttributeIOR::GetID(), anAttr))
555 SALOMEDSImpl_Study::GetStudy(_doc->Main())->AddPostponed(anAttr->Value());
558 Lab.ForgetAttribute (SALOMEDSImpl_SObject::GetGUID(aTypeOfAttribute));
565 //============================================================================
566 /*! Function : Addreference
569 //============================================================================
570 bool SALOMEDSImpl_StudyBuilder::Addreference(const Handle(SALOMEDSImpl_SObject)& me,
571 const Handle(SALOMEDSImpl_SObject)& theReferencedObject)
574 if(me.IsNull() || theReferencedObject.IsNull()) {
575 _errorCode = "Invalid arguments";
579 TDF_Label Lab = me->GetLabel();
580 TDF_Label RefLab = theReferencedObject->GetLabel();
581 SALOMEDSImpl_AttributeReference::Set(Lab,RefLab);
583 SALOMEDSImpl_AttributeTarget::Set(RefLab)->Add(SALOMEDSImpl_Study::SObject(Lab));
585 if(!_callbackOnRemove.IsNull() && Lab.IsDescendant(_doc->Main())) _callbackOnRemove->OnRemoveSObject(me);
590 //============================================================================
591 /*! Function : RemoveReference
594 //============================================================================
595 bool SALOMEDSImpl_StudyBuilder::RemoveReference(const Handle(SALOMEDSImpl_SObject)& me)
598 Handle(SALOMEDSImpl_SObject) theReferencedObject;
599 if(!me->ReferencedObject(theReferencedObject)) return false; //No reference is found
602 TDF_Label Lab = me->GetLabel();
604 Lab.ForgetAttribute(SALOMEDSImpl_AttributeReference::GetID());
606 //SRN: 30 Aug, 2004 : fix from Ecole l'ete version
608 TDF_Label RefLab = theReferencedObject->GetLabel();
610 Handle(SALOMEDSImpl_AttributeTarget) aTarget;
611 if(RefLab.FindAttribute(SALOMEDSImpl_AttributeTarget::GetID(), aTarget))
612 aTarget->Remove(SALOMEDSImpl_Study::SObject(Lab));
621 //============================================================================
622 /*! Function : AddDirectory
623 * Purpose : adds a new directory with a path = thePath
625 //============================================================================
626 bool SALOMEDSImpl_StudyBuilder::AddDirectory(const TCollection_AsciiString& thePath)
630 if(thePath.IsEmpty() || thePath == "") {
631 _errorCode = "Invalid path";
635 TCollection_AsciiString aPath(thePath), aContext(""), aFatherPath;
637 Handle(SALOMEDSImpl_Study) aStudy = SALOMEDSImpl_Study::GetStudy(_doc->Main());
638 Handle(SALOMEDSImpl_SObject) anObject;
641 anObject = aStudy->FindObjectByPath(thePath); //Check if the directory already exists
645 if(!anObject.IsNull()) {
646 _errorCode = "StudyNameAlreadyUsed";
650 if(aPath.Value(1) != '/') { //Relative path
652 aPath = aStudy->GetContext() + aPath;
655 TCollection_AsciiString aToken = aPath.Token("/", 1);
656 if(aToken.Length() == 0) aFatherPath = "/";
659 while(aToken.Length() != 0) {
660 if(aPath.Token("/", i+1).Length() > 0) {
662 aFatherPath += aToken;
664 aToken = aPath.Token("/", ++i);
669 anObject = aStudy->FindObjectByPath(aFatherPath); //Check if the father directory exists
672 if(anObject.IsNull()) {
673 _errorCode = "StudyInvalidDirectory";
677 Handle(SALOMEDSImpl_SObject) aNewObject = NewObject(anObject);
678 aLabel = aNewObject->GetLabel();
679 if(aLabel.IsNull()) {
680 _errorCode = "StudyInvalidComponent";
684 SALOMEDSImpl_AttributeName::Set(aLabel, aPath.Token("/", i-1));
686 //Set LocalID attribute to identify the directory object
687 Handle(SALOMEDSImpl_AttributeLocalID) aLocalID = SALOMEDSImpl_AttributeLocalID::Set(aLabel, DIRECTORYID);
695 //============================================================================
696 /*! Function : SetGUID
699 //============================================================================
700 bool SALOMEDSImpl_StudyBuilder::SetGUID(const Handle(SALOMEDSImpl_SObject)& anObject,
701 const TCollection_AsciiString& theGUID)
705 if(anObject.IsNull()) {
706 _errorCode = "Invalid arguments";
710 TDF_Label aLabel = anObject->GetLabel();
711 SALOMEDSImpl_AttributeUserID::Set(aLabel, theGUID.ToCString());
718 //============================================================================
719 /*! Function : IsGUID
722 //============================================================================
723 bool SALOMEDSImpl_StudyBuilder::IsGUID(const Handle(SALOMEDSImpl_SObject)& anObject,
724 const TCollection_AsciiString& theGUID)
727 if(anObject.IsNull()) {
728 _errorCode = "Invalid arguments";
731 TDF_Label aLabel = anObject->GetLabel();
732 return aLabel.IsAttribute(theGUID.ToCString());
736 //============================================================================
737 /*! Function : NewCommand
740 //============================================================================
741 void SALOMEDSImpl_StudyBuilder::NewCommand()
744 // mpv: for SAL2114 - unset "lock changed" flag at the operation start
745 Handle(SALOMEDSImpl_AttributeStudyProperties) anAttr;
746 if (!_doc->Main().FindAttribute(SALOMEDSImpl_AttributeStudyProperties::GetID(), anAttr)) {
747 anAttr = new SALOMEDSImpl_AttributeStudyProperties;
748 _doc->Main().AddAttribute(anAttr);
750 anAttr->IsLockChanged(true);
755 //============================================================================
756 /*! Function : CommitCommand
759 //============================================================================
760 void SALOMEDSImpl_StudyBuilder::CommitCommand()
763 Handle(SALOMEDSImpl_AttributeStudyProperties) anAttr;
764 if (!_doc->Main().FindAttribute(SALOMEDSImpl_AttributeStudyProperties::GetID(), anAttr)) {
765 anAttr = new SALOMEDSImpl_AttributeStudyProperties;
766 _doc->Main().AddAttribute(anAttr);
768 if (anAttr->IsLocked() && !anAttr->IsLockChanged(true)) {
770 _errorCode = "LockProtection";
771 throw LockProtection("LockProtection");
773 SALOMEDSImpl_Study::GetStudy(_doc->Main())->RemovePostponed(_doc->GetUndoLimit());
775 int aModif = anAttr->GetModified();
776 if (aModif < 0) aModif = 1000; // if user make undo and then - new transaction "modify" will never be zero
777 anAttr->SetModified(aModif+1);
778 _doc->CommitCommand();
784 //============================================================================
785 /*! Function : HasOpenCommand
788 //============================================================================
789 bool SALOMEDSImpl_StudyBuilder::HasOpenCommand()
792 return _doc->HasOpenCommand();
795 //============================================================================
796 /*! Function : AbortCommand
799 //============================================================================
800 void SALOMEDSImpl_StudyBuilder::AbortCommand()
803 SALOMEDSImpl_Study::GetStudy(_doc->Main())->UndoPostponed(0);
805 _doc->AbortCommand();
808 //============================================================================
812 //============================================================================
813 void SALOMEDSImpl_StudyBuilder::Undo()
816 Handle(SALOMEDSImpl_AttributeStudyProperties) anAttr;
817 if (!_doc->Main().FindAttribute(SALOMEDSImpl_AttributeStudyProperties::GetID(), anAttr)) {
818 anAttr = new SALOMEDSImpl_AttributeStudyProperties;
819 _doc->Main().AddAttribute(anAttr);
821 if (anAttr->IsLocked()) {
822 _errorCode = "LockProtection";
823 throw LockProtection("LockProtection");
825 SALOMEDSImpl_Study::GetStudy(_doc->Main())->UndoPostponed(1);
827 anAttr->SetModified(anAttr->GetModified()-1);
833 //============================================================================
837 //============================================================================
838 void SALOMEDSImpl_StudyBuilder::Redo()
841 Handle(SALOMEDSImpl_AttributeStudyProperties) anAttr;
842 if (!_doc->Main().FindAttribute(SALOMEDSImpl_AttributeStudyProperties::GetID(), anAttr)) {
843 anAttr = new SALOMEDSImpl_AttributeStudyProperties;
844 _doc->Main().AddAttribute(anAttr);
847 if (anAttr->IsLocked()) {
848 _errorCode = "LockProtection";
849 throw LockProtection("LockProtection");
852 SALOMEDSImpl_Study::GetStudy(_doc->Main())->UndoPostponed(-1);
853 anAttr->SetModified(anAttr->GetModified()+1);
859 //============================================================================
860 /*! Function : GetAvailableUndos
863 //============================================================================
864 bool SALOMEDSImpl_StudyBuilder::GetAvailableUndos()
867 return _doc->GetAvailableUndos();
870 //============================================================================
871 /*! Function : GetAvailableRedos
874 //============================================================================
875 bool SALOMEDSImpl_StudyBuilder::GetAvailableRedos()
878 return _doc->GetAvailableRedos();
881 //============================================================================
882 /*! Function : UndoLimit
885 //============================================================================
886 int SALOMEDSImpl_StudyBuilder::UndoLimit()
889 return _doc->GetUndoLimit();
892 //============================================================================
893 /*! Function : UndoLimit
896 //============================================================================
897 void SALOMEDSImpl_StudyBuilder::UndoLimit(int n)
901 _doc->SetUndoLimit (n);
904 //============================================================================
905 /*! Function : SetOnAddSObject
908 //============================================================================
909 Handle(SALOMEDSImpl_Callback)
910 SALOMEDSImpl_StudyBuilder::SetOnAddSObject(const Handle(SALOMEDSImpl_Callback)& theCallback)
913 Handle(SALOMEDSImpl_Callback) aRet = _callbackOnAdd;
914 _callbackOnAdd = theCallback;
918 //============================================================================
919 /*! Function : SetOnNewSObject
922 //============================================================================
923 Handle(SALOMEDSImpl_Callback)
924 SALOMEDSImpl_StudyBuilder::SetOnRemoveSObject(const Handle(SALOMEDSImpl_Callback)& theCallback)
927 Handle(SALOMEDSImpl_Callback) aRet = _callbackOnRemove;
928 _callbackOnRemove = theCallback;
932 //============================================================================
933 /*! Function : CheckLocked
936 //============================================================================
937 void SALOMEDSImpl_StudyBuilder::CheckLocked()
940 if (_doc->HasOpenCommand()) return;
941 Handle(SALOMEDSImpl_AttributeStudyProperties) anAttr;
942 if (!_doc->Main().FindAttribute(SALOMEDSImpl_AttributeStudyProperties::GetID(), anAttr)) {
943 anAttr = new SALOMEDSImpl_AttributeStudyProperties;
944 _doc->Main().AddAttribute(anAttr);
946 if (anAttr->IsLocked()) {
947 _errorCode = "LockProtection";
948 throw LockProtection("LockProtection");
952 //============================================================================
953 /*! Function : SetName
956 //============================================================================
957 bool SALOMEDSImpl_StudyBuilder::SetName(const Handle(SALOMEDSImpl_SObject)& theSO,
958 const TCollection_AsciiString& theValue)
963 _errorCode = "Invalid arguments";
966 SALOMEDSImpl_AttributeName::Set(theSO->GetLabel(), theValue);
973 //============================================================================
974 /*! Function : SetComment
977 //============================================================================
978 bool SALOMEDSImpl_StudyBuilder::SetComment(const Handle(SALOMEDSImpl_SObject)& theSO,
979 const TCollection_AsciiString& theValue)
984 _errorCode = "Invalid arguments";
987 SALOMEDSImpl_AttributeComment::Set(theSO->GetLabel(), theValue);
994 //============================================================================
995 /*! Function : SetIOR
998 //============================================================================
999 bool SALOMEDSImpl_StudyBuilder::SetIOR(const Handle(SALOMEDSImpl_SObject)& theSO,
1000 const TCollection_AsciiString& theValue)
1004 if(theSO.IsNull()) {
1005 _errorCode = "Invalid arguments";
1008 SALOMEDSImpl_AttributeIOR::Set(theSO->GetLabel(), theValue);
1016 //============================================================================
1017 /*! Function : Translate_persistentID_to_IOR
1020 //============================================================================
1021 static void Translate_persistentID_to_IOR(TDF_Label& Lab, SALOMEDSImpl_Driver* driver, bool isMultiFile, bool isASCII)
1023 if(driver == NULL) return;
1024 TDF_ChildIterator itchild (Lab);
1026 for (; itchild.More(); itchild.Next()) {
1027 TDF_Label current = itchild.Value();
1028 Handle(SALOMEDSImpl_AttributePersistentRef) Att;
1029 if (current.FindAttribute(SALOMEDSImpl_AttributePersistentRef::GetID(),Att)) {
1031 Handle(SALOMEDSImpl_AttributeLocalID) anID;
1032 if (current.FindAttribute(SALOMEDSImpl_AttributeLocalID::GetID(), anID))
1033 if (anID->Value() == FILELOCALID) continue; //SRN: This attribute store a file name, skip it
1035 TCollection_AsciiString persist_ref(Att->Value());
1036 Handle(SALOMEDSImpl_SObject) so = SALOMEDSImpl_Study::SObject(current);
1037 TCollection_AsciiString ior_string = driver->LocalPersistentIDToIOR(so,
1041 SALOMEDSImpl_AttributeIOR::Set (current, ior_string);
1044 Translate_persistentID_to_IOR (current, driver, isMultiFile, isASCII);