1 // Copyright (C) 2007-2012 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
24 #pragma warning( disable:4786 )
27 #include <Standard_Stream.hxx>
29 #include "GEOM_Gen_i.hh"
30 #include "GEOM_Object_i.hh"
31 #include "GEOM_version.h"
36 #include "Utils_CorbaException.hxx"
38 #include "Utils_ExceptHandlers.hxx"
39 #include "utilities.h"
41 #include "GEOM_Object_i.hh"
42 #include "GEOM_Object.hxx"
43 #include "GEOM_Function.hxx"
44 #include "GEOM_ISubShape.hxx"
45 #include <GEOM_PythonDump.hxx>
46 #include "GEOMImpl_Types.hxx"
47 #include "GEOMImpl_CopyDriver.hxx"
50 #include <BRep_Builder.hxx>
51 #include <BRepTools.hxx>
52 #include <TDF_Label.hxx>
53 #include <TDF_Tool.hxx>
54 #include <TDF_ChildIDIterator.hxx>
55 #include <TNaming_NamedShape.hxx>
56 #include <TDataStd_Name.hxx>
57 #include <TCollection_AsciiString.hxx>
58 #include <TColStd_HArray1OfInteger.hxx>
59 #include <TopAbs_ShapeEnum.hxx>
60 //#include <TopTools_IndexedMapOfShape.hxx>
64 #include <SALOMEDS_Tool.hxx>
66 //============================================================================
67 // function : GEOM_Gen_i()
68 // purpose : constructor to be called for servant creation.
69 //============================================================================
70 GEOM_Gen_i::GEOM_Gen_i(CORBA::ORB_ptr orb,
71 PortableServer::POA_ptr poa,
72 PortableServer::ObjectId* contId,
73 const char* instanceName,
74 const char* interfaceName) :
75 Engines_Component_i(orb, poa, contId, instanceName, interfaceName)
78 _id = _poa->activate_object(_thisObj);
79 name_service = new SALOME_NamingService(_orb);
81 _impl = new ::GEOMImpl_Gen;
83 //PAL10867: disable signals catching with "noexcepthandler" option
84 char* envNoCatchSignals = getenv("NOT_INTERCEPT_SIGNALS");
85 if (!envNoCatchSignals || !atoi(envNoCatchSignals))
87 //work around PAL12004, PAL12628
88 //OSD::SetSignal( true );
92 char* envDisableFPE = getenv("DISABLE_FPE");
93 if (envDisableFPE && atoi(envDisableFPE))
98 OSD::SetSignal( raiseFPE );
102 //============================================================================
103 // function : ~GEOM_Gen_i()
104 // purpose : destructor
105 //============================================================================
106 GEOM_Gen_i::~GEOM_Gen_i() {
112 //============================================================================
113 // function : IORToLocalPersistentID()
115 //============================================================================
116 char* GEOM_Gen_i::IORToLocalPersistentID(SALOMEDS::SObject_ptr theSObject,
117 const char* IORString,
118 CORBA::Boolean isMultiFile,
119 CORBA::Boolean isASCII)
121 GEOM::GEOM_Object_var anObject = GEOM::GEOM_Object::_narrow(_orb->string_to_object(IORString));
122 if (!CORBA::is_nil(anObject)) {
123 return CORBA::string_dup(anObject->GetEntry());
129 //============================================================================
130 // function : LocalPersistentIDToIOR()
131 // purpose : Create/Load CORBA object from a persistent ref (an entry)
132 // : Used when a study is loaded
133 // : The IOR (IORName) of object created is returned
134 //============================================================================
135 char* GEOM_Gen_i::LocalPersistentIDToIOR(SALOMEDS::SObject_ptr theSObject,
136 const char* aLocalPersistentID,
137 CORBA::Boolean isMultiFile,
138 CORBA::Boolean isASCII)
140 SALOMEDS::Study_var aStudy = theSObject->GetStudy();
142 Handle(GEOM_Object) anObject = _impl->GetObject(aStudy->StudyId(), const_cast<char*>(aLocalPersistentID));
143 TCollection_AsciiString anEntry;
144 TDF_Tool::Entry(anObject->GetEntry(), anEntry);
145 GEOM::GEOM_Object_var obj = GetObject(anObject->GetDocID(), anEntry.ToCString());
147 CORBA::String_var aPersRefString = _orb->object_to_string(obj);
148 return CORBA::string_dup(aPersRefString);
151 //============================================================================
152 // function : CanPublishInStudy
154 //============================================================================
155 bool GEOM_Gen_i::CanPublishInStudy(CORBA::Object_ptr theIOR)
157 GEOM::GEOM_Object_var anObject = GEOM::GEOM_Object::_narrow(theIOR);
158 return !(anObject->_is_nil());
162 //============================================================================
163 // function : PublishInStudy
165 //============================================================================
166 SALOMEDS::SObject_ptr GEOM_Gen_i::PublishInStudy(SALOMEDS::Study_ptr theStudy,
167 SALOMEDS::SObject_ptr theSObject,
168 CORBA::Object_ptr theObject,
169 const char* theName) throw (SALOME::SALOME_Exception)
171 Unexpect aCatch(SALOME_SalomeException);
172 SALOMEDS::SObject_var aResultSO;
173 if(CORBA::is_nil(theObject) || theStudy->_is_nil()) return aResultSO;
174 GEOM::GEOM_Object_var aShape = GEOM::GEOM_Object::_narrow(theObject);
175 if(aShape->_is_nil()) return aResultSO;
177 SALOMEDS::GenericAttribute_var anAttr;
178 SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
180 SALOMEDS::SComponent_var aFather = theStudy->FindComponent("GEOM");
181 if (aFather->_is_nil()) {
182 aFather = aStudyBuilder->NewComponent("GEOM");
183 anAttr = aStudyBuilder->FindOrCreateAttribute(aFather, "AttributeName");
184 SALOMEDS::AttributeName_var aName = SALOMEDS::AttributeName::_narrow(anAttr);
185 aName->SetValue("Geometry");
187 anAttr = aStudyBuilder->FindOrCreateAttribute(aFather, "AttributePixMap");
188 SALOMEDS::AttributePixMap_var aPixMap=SALOMEDS::AttributePixMap::_narrow(anAttr);
189 aPixMap->SetPixMap("ICON_OBJBROWSER_Geometry");
190 aPixMap->UnRegister();
191 aStudyBuilder->DefineComponentInstance(aFather, (GEOM::GEOM_Gen_var)GEOM_Gen::_this());
193 if (aFather->_is_nil()) return aResultSO;
195 if (CORBA::is_nil(theSObject)) {
196 aResultSO = aStudyBuilder->NewObject(aFather);
198 if (!theSObject->ReferencedObject(aResultSO))
199 aResultSO = SALOMEDS::SObject::_duplicate(theSObject); //SRN: Added Aug 24,2004 : for the method AddInStudy with theFather argumenet != NULL
200 //THROW_SALOME_CORBA_EXCEPTION("Publish in study supervision graph error",SALOME::BAD_PARAM);
202 CORBA::String_var aGeomObjIOR = _orb->object_to_string(theObject);
203 aResultSO->SetAttrString("AttributeIOR",aGeomObjIOR);
205 TCollection_AsciiString aShapeName("Shape_");
207 CORBA::Long mytype=aShape->GetType();
208 if ( mytype == GEOM_GROUP ) {
209 GEOM::GEOM_IGroupOperations_var anOp = GetIGroupOperations( theStudy->StudyId() );
210 switch ( (TopAbs_ShapeEnum)anOp->GetType( aShape ) ) {
212 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_GROUP_PNT" );
213 aShapeName = "Group_Of_Vertices_";
216 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_GROUP_EDGE");
217 aShapeName = "Group_Of_Edges_";
220 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_GROUP_FACE");
221 aShapeName = "Group_Of_Faces_";
224 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_GROUP_SOLID");
225 aShapeName = "Group_Of_Solids_";
228 } else if ( mytype == GEOM_MARKER ) {
229 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_LCS");
230 aShapeName = "LocalCS_";
231 } else if ( mytype > ADVANCED_BASE ) {
233 sprintf( buf, "%d", aShape->GetType() );
234 std::string advId = "ICON_OBJBROWSER_ADVANCED_"; advId += buf;
235 aResultSO->SetAttrString("AttributePixMap",advId.c_str());
236 aShapeName = "Advanced_";
238 GEOM::shape_type myshapetype=aShape->GetShapeType();
239 if ( myshapetype == GEOM::COMPOUND ) {
240 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_COMPOUND" );
241 aShapeName = "Compound_";
242 } else if ( myshapetype == GEOM::COMPSOLID ) {
243 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_COMPSOLID");
244 aShapeName = "Compsolid_";
245 } else if ( myshapetype == GEOM::SOLID ) {
246 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_SOLID");
247 aShapeName = "Solid_";
248 } else if ( myshapetype == GEOM::SHELL ) {
249 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_SHELL");
250 aShapeName = "Shell_";
251 } else if ( myshapetype == GEOM::FACE ) {
252 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_FACE");
253 aShapeName = "Face_";
254 } else if ( myshapetype == GEOM::WIRE ) {
255 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_WIRE");
256 aShapeName = "Wire_";
257 } else if ( myshapetype == GEOM::EDGE ) {
258 aResultSO->SetAttrString("AttributePixMap", "ICON_OBJBROWSER_EDGE");
259 aShapeName = "Edge_";
260 } else if ( myshapetype == GEOM::VERTEX ) {
261 aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_VERTEX" );
262 aShapeName = "Vertex_";
265 //if (strlen(theName) == 0) aShapeName += TCollection_AsciiString(aResultSO->Tag());
266 //else aShapeName = TCollection_AsciiString(CORBA::string_dup(theName));
268 // BEGIN: try to find existed name for current shape
269 bool HasName = false;
270 // recieve current TopoDS shape
271 CORBA::String_var entry = aShape->GetEntry();
272 Handle(GEOM_Object) aGShape = _impl->GetObject(aShape->GetStudyID(), entry);
273 TopoDS_Shape TopoSh = aGShape->GetValue();
274 // find label of main shape
275 GEOM::GEOM_Object_var aMainSh = aShape;
276 while( !aMainSh->IsMainShape() ) {
277 aMainSh = aMainSh->GetMainShape();
279 entry = aMainSh->GetEntry();
280 Handle(GEOM_Object) anObj = _impl->GetObject(aMainSh->GetStudyID(), entry);
281 TDF_Label aMainLbl = anObj->GetFunction(1)->GetNamingEntry();
283 // check all named shapes using iterator
284 TDF_ChildIDIterator anIt (aMainLbl, TNaming_NamedShape::GetID(), Standard_True);
286 for (; anIt.More() && !HasName; anIt.Next()) {
287 Handle(TNaming_NamedShape) anAttr =
288 Handle(TNaming_NamedShape)::DownCast(anIt.Value());
289 if (anAttr.IsNull()) continue;
290 TopoDS_Shape S = anAttr->Get();
291 if (S.IsEqual(TopoSh)) {
292 TDF_Label L = anAttr->Label();
293 Handle(TDataStd_Name) aName;
294 if (L.FindAttribute(TDataStd_Name::GetID(), aName)) {
295 aShapeName = aName->Get();
300 // END: try to find existed name for current shape
303 // asv : 11.11.04 Introducing a more sofisticated method of name creation, just as
304 // it is done in GUI in GEOMBase::GetDefaultName() - not just add a Tag() == number
305 // of objects in the study, but compute a number of objects with the same prefix
306 // and build a new name as Prefix_N+1
307 if ( strlen( theName ) == 0 ) { // MOST PROBABLY CALLED FROM BATCHMODE OR SUPERVISOR
308 int i = 0; // (WITH EMPTY NEW NAME)
309 SALOMEDS::SObject_var obj;
310 TCollection_AsciiString aNewShapeName;
312 aNewShapeName = aShapeName + TCollection_AsciiString(++i);
313 obj = theStudy->FindObject( aNewShapeName.ToCString() );
315 while ( !obj->_is_nil() );
316 aShapeName = aNewShapeName;
318 else // MOST PROBABLY CALLED FROM GEOM GUI (ALREADY WITH VALID NAME)
319 aShapeName = TCollection_AsciiString((char*)theName);
322 //Set the study entry as a name of the published GEOM_Object
323 CORBA::String_var anID =aResultSO->GetID();
324 aShape->SetStudyEntry(anID.in());
326 //Set a name of the added shape
327 aResultSO->SetAttrString("AttributeName",aShapeName.ToCString());
329 //Set NoteBook variables used in the object creation
330 TCollection_AsciiString aVars;
331 CORBA::String_var aString=aShape->GetParameters();
332 SALOMEDS::ListOfListOfStrings_var aSections = theStudy->ParseVariables(aString);
333 for(int i = 0, n = aSections->length(); i < n; i++) {
334 SALOMEDS::ListOfStrings aListOfVars = aSections[i];
335 for(int j = 0, m = aListOfVars.length(); j < m; j++) {
336 if(theStudy->IsVariable(aListOfVars[j].in()))
337 aVars += TCollection_AsciiString(aListOfVars[j].in());
344 aResultSO->SetAttrString("AttributeString",aVars.ToCString());
346 aFather->UnRegister();
348 //Set a name of the GEOM object
349 aShape->SetName(aShapeName.ToCString());
351 return aResultSO._retn();
355 //============================================================================
356 // function : CreateAndPublishGroup
357 // purpose : auxilary for PublishNamedShapesInStudy
358 //============================================================================
359 void GEOM_Gen_i::CreateAndPublishGroup(SALOMEDS::Study_ptr theStudy,
360 GEOM::GEOM_Object_var theMainShape,
361 const TopTools_IndexedMapOfShape& anIndices,
362 const TopTools_SequenceOfShape& SeqS,
363 const TColStd_SequenceOfAsciiString& SeqN,
364 const Standard_CString& GrName,
365 GEOM::ListOfGO_var aResList)
367 CORBA::String_var entry = theMainShape->GetEntry();
368 Handle(GEOM_Object) aMainShape = _impl->GetObject(theMainShape->GetStudyID(), entry);
369 Handle(TColStd_HArray1OfInteger) anArray;
370 if(SeqS.Length()>0) {
372 GEOM::GEOM_IGroupOperations_var GOp = GetIGroupOperations(theStudy->StudyId());
373 GEOM::GEOM_Object_ptr GrObj =
374 GOp->CreateGroup( theMainShape, SeqS.Value(1).ShapeType() );
375 AddInStudy(theStudy, GrObj, GrName, theMainShape._retn());
376 CORBA::String_var GrEntry = GrObj->GetEntry();
377 Handle(GEOM_Object) HGrObj = _impl->GetObject(GrObj->GetStudyID(), GrEntry);
379 //Handle(GEOM_Object) anObj;
380 for(int i=1; i<=SeqS.Length(); i++) {
381 TopoDS_Shape aValue = SeqS.Value(i);
382 //anArray = new TColStd_HArray1OfInteger(1,1);
383 Standard_Integer anIndex = anIndices.FindIndex(aValue);
384 //anArray->SetValue(1, anIndex);
385 GOp->AddObject(GrObj,anIndex);
386 //anObj = GEOM_Engine::GetEngine()->AddObject(aMainShape->GetDocID(), GEOM_SUBSHAPE);
387 //if (anObj.IsNull()) continue;
388 //Handle(GEOM_Function) aFunction = anObj->AddFunction(GEOM_Object::GetSubShapeID(), 1);
389 //if (aFunction.IsNull()) continue;
390 //GEOM_ISubShape aSSI(aFunction);
391 //aSSI.SetMainShape(aMainShape->GetLastFunction());
392 //aSSI.SetIndices(anArray);
393 //aFunction->SetValue(aValue);
394 //GOp->UnionIDs(GrObj, anIndex);
395 //SALOMEDS::SObject_var aResultSO;
396 //TCollection_AsciiString anEntry;
397 //TDF_Tool::Entry(anObj->GetEntry(),anEntry);
398 //GEOM::GEOM_Object_var aGObj = GetObject(anObj->GetDocID(), anEntry.ToCString());
399 //AddInStudy(theStudy, aGObj._retn(), SeqN.Value(i).ToCString(), GrObj);
405 //============================================================================
406 // function : PublishNamedShapesInStudy
408 //============================================================================
409 GEOM::ListOfGO* GEOM_Gen_i::
410 PublishNamedShapesInStudy(SALOMEDS::Study_ptr theStudy,
411 //SALOMEDS::SObject_ptr theSObject,
412 CORBA::Object_ptr theObject)
414 //Unexpect aCatch(SALOME_SalomeException);
415 GEOM::ListOfGO_var aResList = new GEOM::ListOfGO;
417 //CORBA::Object_var theObject = theSObject->GetObject();
418 GEOM::GEOM_Object_var theMainShape = GEOM::GEOM_Object::_narrow(theObject);
419 if(theMainShape->_is_nil()) return aResList._retn();
421 CORBA::String_var entry = theMainShape->GetEntry();
422 Handle(GEOM_Object) aMainShape = _impl->GetObject(theMainShape->GetStudyID(), entry);
423 if (aMainShape.IsNull()) return aResList._retn();
424 TopoDS_Shape MainSh = aMainShape->GetValue();
426 TDF_Label aMainLbl = aMainShape->GetEntry();
427 TopTools_SequenceOfShape SolidSeqS, FaceSeqS, EdgeSeqS, VertSeqS;
428 TColStd_SequenceOfAsciiString SolidSeqN, FaceSeqN, EdgeSeqN, VertSeqN;
429 TDF_ChildIDIterator anIt(aMainLbl, TNaming_NamedShape::GetID(), Standard_True);
430 for(; anIt.More(); anIt.Next()) {
431 Handle(TNaming_NamedShape) anAttr =
432 Handle(TNaming_NamedShape)::DownCast(anIt.Value());
433 if(anAttr.IsNull()) continue;
434 TopoDS_Shape S = anAttr->Get();
435 TDF_Label L = anAttr->Label();
436 //if(S.IsEqual(MainSh)) continue;
437 Handle(TDataStd_Name) aName;
438 if(L.FindAttribute(TDataStd_Name::GetID(),aName)) {
439 TCollection_ExtendedString EName = aName->Get();
440 if(S.ShapeType()==TopAbs_SOLID) {
442 SolidSeqN.Append(aName->Get());
444 else if(S.ShapeType()==TopAbs_FACE) {
446 FaceSeqN.Append(aName->Get());
448 else if(S.ShapeType()==TopAbs_EDGE) {
450 EdgeSeqN.Append(aName->Get());
452 else if(S.ShapeType()==TopAbs_VERTEX) {
454 VertSeqN.Append(aName->Get());
459 TopTools_IndexedMapOfShape anIndices;
460 TopExp::MapShapes(MainSh, anIndices);
462 CreateAndPublishGroup(theStudy, theMainShape, anIndices, SolidSeqS, SolidSeqN,
463 "Group_Of_Named_Solids", aResList);
465 CreateAndPublishGroup(theStudy, theMainShape, anIndices, FaceSeqS, FaceSeqN,
466 "Group_Of_Named_Faces", aResList);
468 CreateAndPublishGroup(theStudy, theMainShape, anIndices, EdgeSeqS, EdgeSeqN,
469 "Group_Of_Named_Edges", aResList);
471 CreateAndPublishGroup(theStudy, theMainShape, anIndices, VertSeqS, VertSeqN,
472 "Group_Of_Named_Vertices", aResList);
474 return aResList._retn();
478 //============================================================================
480 // purpose : save OCAF/Geom document
481 //============================================================================
482 SALOMEDS::TMPFile* GEOM_Gen_i::Save(SALOMEDS::SComponent_ptr theComponent,
485 SALOMEDS::TMPFile_var aStreamFile;
486 // Get a temporary directory to store a file
487 std::string aTmpDir = (isMultiFile)?theURL:SALOMEDS_Tool::GetTmpDir();
489 // OCCT BUG: cannot save a document (in current folder)
490 // if directory name is empty
491 if (aTmpDir.size() == 0) {
499 // Create a list to store names of created files
500 SALOMEDS::ListOfFileNames_var aSeq = new SALOMEDS::ListOfFileNames;
502 // Prepare a file name to open
503 TCollection_AsciiString aNameWithExt("");
505 aNameWithExt = TCollection_AsciiString((char*)(SALOMEDS_Tool::GetNameFromPath
506 (theComponent->GetStudy()->URL())).c_str());
507 aNameWithExt += TCollection_AsciiString("_GEOM.sgd");
508 aSeq[0] = CORBA::string_dup(aNameWithExt.ToCString());
509 // Build a full file name of temporary file
510 TCollection_AsciiString aFullName = TCollection_AsciiString((char*)aTmpDir.c_str()) + aNameWithExt;
511 // Save GEOM component in this file
512 _impl->Save(theComponent->GetStudy()->StudyId(),(char*) aFullName.ToCString());
513 // Conver a file to the byte stream
514 aStreamFile = SALOMEDS_Tool::PutFilesToStream(aTmpDir.c_str(), aSeq.in(), isMultiFile);
515 // Remove the created file and tmp directory
516 if (!isMultiFile) SALOMEDS_Tool::RemoveTemporaryFiles(aTmpDir.c_str(), aSeq.in(), true);
518 // Return the created byte stream
519 return aStreamFile._retn();
523 //============================================================================
524 // function : SaveASCII()
526 //============================================================================
527 SALOMEDS::TMPFile* GEOM_Gen_i::SaveASCII(SALOMEDS::SComponent_ptr theComponent,
530 SALOMEDS::TMPFile_var aStreamFile = Save(theComponent, theURL, isMultiFile);
531 return aStreamFile._retn();
535 //============================================================================
538 //============================================================================
539 CORBA::Boolean GEOM_Gen_i::Load(SALOMEDS::SComponent_ptr theComponent,
540 const SALOMEDS::TMPFile& theStream,
544 if (theStream.length() <= 9) {
545 MESSAGE("The TMPFile is too short : " << theStream.length() << " bytes ");
549 // Get a temporary directory for a file
550 std::string aTmpDir = isMultiFile?theURL:SALOMEDS_Tool::GetTmpDir();
552 // OCCT BUG: cannot load a document (from current folder)
553 // if directory name is empty
554 if (aTmpDir.size() == 0) {
562 // Conver the byte stream theStream to a file and place it in tmp directory
563 SALOMEDS::ListOfFileNames_var aSeq =
564 SALOMEDS_Tool::PutStreamToFiles(theStream, aTmpDir.c_str(), isMultiFile);
566 // Prepare a file name to open
567 TCollection_AsciiString aNameWithExt("");
569 aNameWithExt = TCollection_AsciiString((char*)(SALOMEDS_Tool::GetNameFromPath
570 (theComponent->GetStudy()->URL())).c_str());
571 aNameWithExt += TCollection_AsciiString("_GEOM.sgd");
572 TCollection_AsciiString aFullName = (TCollection_AsciiString((char*)aTmpDir.c_str()) + aNameWithExt);
575 if (!_impl->Load(theComponent->GetStudy()->StudyId(),(char*) aFullName.ToCString())) return false;
577 // Remove the created file and tmp directory
578 if (!isMultiFile) SALOMEDS_Tool::RemoveTemporaryFiles(aTmpDir.c_str(), aSeq.in(), true);
584 //============================================================================
585 // function : LoadASCII()
587 //============================================================================
588 CORBA::Boolean GEOM_Gen_i::LoadASCII(SALOMEDS::SComponent_ptr theComponent,
589 const SALOMEDS::TMPFile& theStream,
592 return Load(theComponent, theStream, theURL, isMultiFile);
596 //============================================================================
597 // function : Close()
599 //============================================================================
600 void GEOM_Gen_i::Close(SALOMEDS::SComponent_ptr theComponent)
602 SALOMEDS::Study_var aStudy= theComponent->GetStudy();
603 _impl->Close(aStudy->StudyId());
606 //============================================================================
607 // function : CanCopy()
609 //============================================================================
610 CORBA::Boolean GEOM_Gen_i::CanCopy(SALOMEDS::SObject_ptr theObject) {
611 // Try to retrieve known by Geometry component GEOM_Object by given IOR
612 SALOMEDS::GenericAttribute_var anAttr;
613 if (!theObject->FindAttribute(anAttr, "AttributeIOR")) return false;
615 SALOMEDS::AttributeIOR_var anIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
617 CORBA::String_var aString=anIOR->Value();
619 CORBA::Object_var anObj = _orb->string_to_object(aString);
620 GEOM::GEOM_Object_var anObject = GEOM::GEOM_Object::_narrow(anObj);
621 // If the object is null one it can't be copied: return false
622 if (anObject->_is_nil()) return false;
626 //============================================================================
627 // function : CopyFrom()
629 //============================================================================
630 SALOMEDS::TMPFile* GEOM_Gen_i::CopyFrom(SALOMEDS::SObject_ptr theObject, CORBA::Long& theObjectID)
632 // Declare a sequence of the byte to store the copied object
633 SALOMEDS::TMPFile_var aStreamFile = new SALOMEDS::TMPFile;
635 // Try to get GEOM_Object object by given SObject
636 SALOMEDS::GenericAttribute_var anAttr;
637 if (!theObject->FindAttribute(anAttr, "AttributeIOR")) return aStreamFile._retn();
638 GEOM::GEOM_Object_var anObject = GEOM::GEOM_Object::_narrow
639 (_orb->string_to_object(SALOMEDS::AttributeIOR::_narrow(anAttr)->Value()));
640 if (anObject->_is_nil()) return aStreamFile._retn();
642 aStreamFile = anObject->GetShapeStream();
644 // Assign an ID the type of GEOM_Object
645 theObjectID = anObject->GetType();
647 // Return created TMPFile
648 return aStreamFile._retn();
651 //============================================================================
652 // function : CanPaste()
654 //============================================================================
655 CORBA::Boolean GEOM_Gen_i::CanPaste(const char* theComponentName, CORBA::Long theObjectID) {
656 // The Geometry component can paste only objects copied by Geometry component
657 // and with the object type = 1
658 if (strcmp(theComponentName, ComponentDataType()) != 0) return false;
662 //============================================================================
663 // function : PasteInto()
665 //============================================================================
666 SALOMEDS::SObject_ptr GEOM_Gen_i::PasteInto(const SALOMEDS::TMPFile& theStream,
667 CORBA::Long theObjectID,
668 SALOMEDS::SObject_ptr theObject) {
669 // Find the current Study and StudyBuilder
670 SALOMEDS::Study_var aStudy = theObject->GetStudy();
671 SALOMEDS::StudyBuilder_var aStudyBuilder = aStudy->NewBuilder();
673 // Retrieve a TopoDS_Shape from byte stream
674 TopoDS_Shape aTopology;
675 std::istringstream aStreamedBrep((char*) &theStream[0]);
676 BRep_Builder aBuilder;
678 BRepTools::Read(aTopology, aStreamedBrep, aBuilder);
679 } catch (Standard_Failure) {
683 // SObject of the created shape is theObject or new Child of Component if theObject == geom component
684 SALOMEDS::SObject_var aNewSO;
685 if (strcmp(theObject->GetFatherComponent()->GetID(),theObject->GetID()) == 0) {
686 aNewSO = aStudyBuilder->NewObject(theObject);
687 } else aNewSO = SALOMEDS::SObject::_duplicate(theObject);
690 //Create a new GEOM_Object
691 Handle(GEOM_Object) anObj = _impl->AddObject(aNewSO->GetStudy()->StudyId(), theObjectID);
692 Handle(GEOM_Function) aFunction = anObj->AddFunction(GEOMImpl_CopyDriver::GetID(), COPY_WITHOUT_REF);
693 aFunction->SetValue(aTopology);
695 TCollection_AsciiString anEntry;
696 TDF_Tool::Entry(anObj->GetEntry(), anEntry);
697 GEOM::GEOM_Object_var obj = GetObject(anObj->GetDocID(), anEntry.ToCString());
699 //Set the study entry of the published GEOM_Object
700 obj->SetStudyEntry(aNewSO->GetID());
702 // Add IORAttribute to the Study and set IOR of the created GEOM_Object to it
703 SALOMEDS::GenericAttribute_var anAttr = aStudyBuilder->FindOrCreateAttribute(aNewSO, "AttributeIOR");
704 SALOMEDS::AttributeIOR_var anIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
705 CORBA::String_var objStr = _orb->object_to_string(obj);
706 anIOR->SetValue(objStr.in());
709 // Return the created in the Study SObject
710 return aNewSO._retn();
713 //============================================================================
714 // function : ComponentDataType()
716 //============================================================================
717 char* GEOM_Gen_i::ComponentDataType()
719 return CORBA::string_dup("GEOM");
722 //============================================================================
723 // function : AddInStudy
725 //============================================================================
726 SALOMEDS::SObject_ptr GEOM_Gen_i::AddInStudy (SALOMEDS::Study_ptr theStudy,
727 GEOM::GEOM_Object_ptr theObject,
729 GEOM::GEOM_Object_ptr theFather)
731 SALOMEDS::SObject_var aResultSO;
732 if(theObject->_is_nil() || theStudy->_is_nil()) return aResultSO;
734 SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
735 CORBA::String_var IOR;
737 if(!theFather->_is_nil()) {
738 IOR = _orb->object_to_string(theFather);
739 SALOMEDS::SObject_var aFatherSO = theStudy->FindObjectIOR(IOR.in());
740 if(aFatherSO->_is_nil()) return aResultSO._retn();
741 aResultSO = aStudyBuilder->NewObject(aFatherSO);
742 aFatherSO->UnRegister();
743 //aStudyBuilder->Addreference(aResultSO, aResultSO);
746 aResultSO = PublishInStudy(theStudy, aResultSO, theObject, theName);
747 if(aResultSO->_is_nil()) return aResultSO._retn();
749 GEOM::ListOfGO_var aList = theObject->GetDependency();
750 Standard_Integer aLength = aList->length();
751 if(aLength < 1) return aResultSO._retn();
753 //Publish the arguments
754 TCollection_AsciiString aPrevID; // to avoid multiple references to same object
755 for(Standard_Integer i = 0; i< aLength; i++) {
756 GEOM::GEOM_Object_var anObject = aList[i];
757 if(anObject->_is_nil()) continue;
758 IOR = _orb->object_to_string(anObject);
759 SALOMEDS::SObject_var aSO = theStudy->FindObjectIOR(IOR.in());
760 if(aSO->_is_nil()) continue;
761 CORBA::String_var anID = aSO->GetID();
762 if ( aPrevID == anID.in() ) continue;
764 SALOMEDS::SObject_var aSubSO = aStudyBuilder->NewObject(aResultSO);
765 aStudyBuilder->Addreference(aSubSO, aSO);
767 aSubSO->UnRegister();
770 return aResultSO._retn();
773 //============================================================================
774 // function : RestoreSubShapesO
775 // purpose : Publish sub-shapes, standing for arguments and sub-shapes of arguments.
776 // To be used from python scripts out of geompy.addToStudy (non-default usage)
777 //============================================================================
778 GEOM::ListOfGO* GEOM_Gen_i::RestoreSubShapesO (SALOMEDS::Study_ptr theStudy,
779 GEOM::GEOM_Object_ptr theObject,
780 const GEOM::ListOfGO& theArgs,
781 GEOM::find_shape_method theFindMethod,
782 CORBA::Boolean theInheritFirstArg,
783 CORBA::Boolean theAddPrefix)
785 GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
786 if (CORBA::is_nil(theStudy) || CORBA::is_nil(theObject))
787 return aParts._retn();
789 // find SObject in the study if it is already published
790 CORBA::String_var anIORo = _orb->object_to_string(theObject);
791 SALOMEDS::SObject_var aSO = theStudy->FindObjectIOR(anIORo.in());
792 //PTv, IMP 0020001, The salome object <aSO>
793 // is not obligatory in case of invokation from script
794 // if (CORBA::is_nil(aSO))
795 // return aParts._retn();
797 aParts = RestoreSubShapes(theStudy, theObject, aSO, theArgs,
798 theFindMethod, theInheritFirstArg, theAddPrefix);
799 if (!CORBA::is_nil(aSO)) aSO->UnRegister();
800 return aParts._retn();
803 //============================================================================
804 // function : RestoreGivenSubShapesO
805 // purpose : Publish sub-shapes, standing for arguments and sub-shapes of arguments.
806 // To be used from python scripts, generated by Dump Python.
807 //============================================================================
808 GEOM::ListOfGO* GEOM_Gen_i::RestoreGivenSubShapesO (SALOMEDS::Study_ptr theStudy,
809 GEOM::GEOM_Object_ptr theObject,
810 const GEOM::ListOfGO& theArgs,
811 GEOM::find_shape_method theFindMethod,
812 CORBA::Boolean theInheritFirstArg,
813 CORBA::Boolean theAddPrefix)
815 GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
816 if (CORBA::is_nil(theStudy) || CORBA::is_nil(theObject))
817 return aParts._retn();
819 // find SObject in the study if it is already published
820 CORBA::String_var anIORo = _orb->object_to_string(theObject);
821 SALOMEDS::SObject_var aSO = theStudy->FindObjectIOR(anIORo.in());
822 //PTv, IMP 0020001, The salome object <aSO>
823 // is not obligatory in case of invokation from script
824 // if (CORBA::is_nil(aSO))
825 // return aParts._retn();
827 aParts = RestoreGivenSubShapes(theStudy, theObject, aSO, theArgs,
828 theFindMethod, theInheritFirstArg, theAddPrefix);
829 if (!CORBA::is_nil(aSO)) aSO->UnRegister();
830 return aParts._retn();
833 //============================================================================
834 // function : RestoreSubShapesSO
835 // purpose : Publish sub-shapes, standing for arguments and sub-shapes of arguments.
836 // To be used from GUI and from geompy.addToStudy
837 //============================================================================
838 GEOM::ListOfGO* GEOM_Gen_i::RestoreSubShapesSO (SALOMEDS::Study_ptr theStudy,
839 SALOMEDS::SObject_ptr theSObject,
840 const GEOM::ListOfGO& theArgs,
841 GEOM::find_shape_method theFindMethod,
842 CORBA::Boolean theInheritFirstArg,
843 CORBA::Boolean theAddPrefix)
845 GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
846 if (CORBA::is_nil(theStudy) || CORBA::is_nil(theSObject))
847 return aParts._retn();
849 SALOMEDS::GenericAttribute_var anAttr;
850 if (!theSObject->FindAttribute(anAttr, "AttributeIOR"))
851 return aParts._retn();
853 SALOMEDS::AttributeIOR_var anAttrIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
854 CORBA::String_var anIORso = anAttrIOR->Value();
856 // get Object from SObject
857 GEOM::GEOM_Object_var anO = GEOM::GEOM_Object::_narrow(_orb->string_to_object(anIORso));
858 if (CORBA::is_nil(anO))
859 return aParts._retn();
861 aParts = RestoreSubShapes(theStudy, anO, theSObject, theArgs,
862 theFindMethod, theInheritFirstArg, theAddPrefix);
863 return aParts._retn();
866 //============================================================================
867 // function : addToListOfGO
868 // purpose : static local function
869 //============================================================================
870 static void addToListOfGO( GEOM::GEOM_Object_ptr theObject,
871 GEOM::ListOfGO& theList )
873 const int oldLen = theList.length();
874 theList.length(oldLen + 1);
875 theList[ oldLen ] = GEOM::GEOM_Object::_duplicate( theObject );
878 //============================================================================
879 // function : addToListOfGO
880 // purpose : static local function
881 //============================================================================
882 static void addToListOfGO( const GEOM::ListOfGO& theSrcList,
883 GEOM::ListOfGO& theTrgList )
885 const int oldLen = theTrgList.length();
886 const int srcLen = theSrcList.length();
887 theTrgList.length(oldLen + srcLen);
888 for( int i = 0; i < srcLen; i++ )
889 theTrgList[ oldLen + i ] = GEOM::GEOM_Object::_duplicate( theSrcList[ i ] );
892 //============================================================================
893 // function : RestoreSubShapes
894 // purpose : Private method. Works only if both theObject and theSObject
895 // are defined, and does not check, if they correspond to each other.
896 //============================================================================
897 GEOM::ListOfGO* GEOM_Gen_i::RestoreSubShapes(SALOMEDS::Study_ptr theStudy,
898 GEOM::GEOM_Object_ptr theObject,
899 SALOMEDS::SObject_ptr theSObject,
900 const GEOM::ListOfGO& theArgs,
901 GEOM::find_shape_method theFindMethod,
902 CORBA::Boolean theInheritFirstArg,
903 CORBA::Boolean theAddPrefix)
905 GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
906 //PTv, IMP 0020001, The salome object <theSObject>
907 // is not obligatory in case of invokation from script
908 if (CORBA::is_nil(theStudy) || CORBA::is_nil(theObject) /*|| CORBA::is_nil(theSObject)*/)
909 return aParts._retn();
911 // For Dump Python (mantis issue 0020768)
912 GEOM::ListOfGO_var anOutArgs = new GEOM::ListOfGO;
914 // Arguments to be published
915 GEOM::ListOfGO_var aList;
917 // If theArgs list is empty, we try to publish all arguments,
918 // otherwise publish only passed args
919 Standard_Integer nbArgsActual = -1; // -1 means unknown
920 Standard_Integer aLength = theArgs.length();
922 aList = new GEOM::ListOfGO;
923 aList->length(aLength);
924 for (int i = 0; i < aLength; i++) {
925 aList[i] = GEOM::GEOM_Object::_duplicate( theArgs[i] );
930 aList = theObject->GetDependency();
931 aLength = aList->length();
932 nbArgsActual = aLength;
936 return aParts._retn();
938 if (theInheritFirstArg || (nbArgsActual == 1)) {
939 // Do not publish argument's reflection,
940 // but only reconstruct its published sub-shapes
942 CORBA::String_var anIOR = _orb->object_to_string(aList[0]);
943 SALOMEDS::SObject_var anArgSO = theStudy->FindObjectIOR(anIOR.in());
945 // remember restored objects for Python Dump
946 addToListOfGO(aList[0], anOutArgs);
948 aParts = RestoreSubShapesOneLevel(theStudy, anArgSO, theSObject, theObject,
949 anOutArgs, theFindMethod, theAddPrefix);
951 // set the color of the transformed shape to the color of initial shape
952 theObject->SetColor(aList[0]->GetColor());
954 if (theObject->GetShapeType() == GEOM::VERTEX) {
955 theObject->SetMarkerStd(aList[0]->GetMarkerType(), aList[0]->GetMarkerSize());
956 if (aList[0]->GetMarkerType() == GEOM::MT_USER)
957 theObject->SetMarkerTexture(aList[0]->GetMarkerTexture());
960 anArgSO->UnRegister();
963 // Get interface, containing method, which we will use to reconstruct sub-shapes
964 GEOM::GEOM_IShapesOperations_var aShapesOp = GetIShapesOperations(theStudy->StudyId());
965 GEOM::GEOM_IGroupOperations_var aGroupOp = GetIGroupOperations(theStudy->StudyId());
966 GEOM::GEOM_ITransformOperations_var aTrsfOp = GetITransformOperations(theStudy->StudyId());
968 PortableServer::Servant aServant = _poa->reference_to_servant(aTrsfOp.in());
969 GEOM_ITransformOperations_i* aTrsfOpSv = dynamic_cast<GEOM_ITransformOperations_i*>(aServant);
971 // Reconstruct arguments and tree of sub-shapes of the arguments
972 CORBA::String_var anIOR;
973 SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
974 for (Standard_Integer i = 0; i < aLength; i++)
976 GEOM::GEOM_Object_var anArgO = aList[i];
977 if (!CORBA::is_nil(anArgO)) {
978 anIOR = _orb->object_to_string(anArgO);
979 SALOMEDS::SObject_var anArgSO = theStudy->FindObjectIOR(anIOR.in());
980 TCollection_AsciiString anArgName;
981 if (CORBA::is_nil(anArgSO)) {
983 anArgName += TCollection_AsciiString(i);
986 anArgName = anArgSO->GetName();
989 // Find a sub-shape of theObject in place of the argument
990 GEOM::GEOM_Object_var aSubO;
991 switch (theFindMethod) {
992 case GEOM::FSM_GetInPlace:
995 aSubO = aShapesOp->GetInPlace(theObject, anArgO);
998 case GEOM::FSM_MultiTransformed:
1000 // Only for Multi-transformations
1001 GEOM::GEOM_Object_var anArgOTrsf = aTrsfOpSv->TransformLikeOtherCopy(anArgO, theObject);
1002 if (!CORBA::is_nil(anArgOTrsf)) {
1003 CORBA::String_var anArgOTrsfEntry = anArgOTrsf->GetEntry();
1004 Handle(GEOM_Object) anArgOTrsfImpl = _impl->GetObject(anArgOTrsf->GetStudyID(), anArgOTrsfEntry);
1005 Handle(GEOM_Function) anArgOTrsfFun = anArgOTrsfImpl->GetLastFunction();
1006 anArgOTrsfFun->SetDescription("");
1007 aSubO = aShapesOp->GetInPlace(theObject, anArgOTrsf);
1010 Handle(GEOM_Function) anOFun = theObject->GetLastFunction();
1011 if (!anOFun.IsNull()) {
1012 CORBA::String_var entryArg = anArgO->GetEntry();
1013 Handle(GEOM_Object) anArgOImpl = _impl->GetObject(anArgO->GetStudyID(), entryArg);
1014 if (!anArgOImpl.IsNull()) {
1015 TopoDS_Shape anArgOShape = anArgOImpl->GetValue();
1016 TopoDS_Shape aMultiArgShape;
1017 //GEOM::GEOM_Object_var anArgOMulti; // ???
1018 switch (anOFun->GetType()) {
1021 GEOMImpl_ITranslate aTI (anOFun);
1022 aMultiArgShape = GEOMImpl_ITransformOperations::TranslateShape1D(anArgOShape, &aTI);
1023 //anArgOMulti = aTrsfOpSv->Translate1D(anArgO, , , );
1028 GEOMImpl_ITranslate aTI (anOFun);
1029 aMultiArgShape = GEOMImpl_ITransformOperations::TranslateShape2D(anArgOShape, &aTI);
1034 GEOMImpl_IRotate aTI (anOFun);
1035 //aMultiArgShape = GEOMImpl_ITransformOperations::TranslateShape2D(anArgOShape, &aTI);
1040 GEOMImpl_IRotate aTI (anOFun);
1041 //aMultiArgShape = GEOMImpl_ITransformOperations::TranslateShape2D(anArgOShape, &aTI);
1047 GEOM::GEOM_Object_var anArgOMulti = (aMultiArgShape); // TODO
1048 Handle(GEOM_Function) anArgOMultiFun = anArgOMulti->GetLastFunction();
1049 anArgOMultiFun->SetDescription("");
1050 aSubO = aShapesOp->GetInPlace(theObject, anArgOMulti);
1056 case GEOM::FSM_Transformed:
1058 // transformation, cannot use GetInPlace, operate with indices
1059 GEOM::ListOfLong_var anIDs = anArgO->GetSubShapeIndices();
1060 if (anIDs->length() > 1) {
1062 aSubO = aGroupOp->CreateGroup(theObject, aGroupOp->GetType(anArgO));
1063 if (!CORBA::is_nil(aSubO))
1064 aGroupOp->UnionIDs(aSubO, anIDs);
1066 else if (anIDs->length() > 0) {
1068 aSubO = aShapesOp->GetSubShape(theObject, anIDs[0]);
1072 case GEOM::FSM_GetSame:
1075 aSubO = aShapesOp->GetSame(theObject, anArgO);
1078 case GEOM::FSM_GetShapesOnShape:
1080 // Use GetShapesOnShape. Can work only on solids, so it has sense to search only solids
1081 aSubO = aShapesOp->GetShapesOnShapeAsCompound(anArgO, theObject,
1082 (short)GEOM::SOLID, GEOM::ST_ONIN);
1085 case GEOM::FSM_GetInPlaceByHistory:
1087 // Use GetInPlaceByHistory
1088 aSubO = aShapesOp->GetInPlaceByHistory(theObject, anArgO);
1095 if (!CORBA::is_nil(aSubO)) {
1096 // remember restored objects for Python Dump
1097 addToListOfGO(anArgO, anOutArgs);
1099 // add to parts list
1100 addToListOfGO( aSubO, aParts );
1102 // Publish the sub-shape
1103 SALOMEDS::SObject_var aSubSO;
1104 if (!CORBA::is_nil(theSObject)) {
1105 TCollection_AsciiString aSubName;
1109 aSubName += anArgName;
1110 aSubSO = aStudyBuilder->NewObject(theSObject);
1111 aSubSO = PublishInStudy(theStudy, aSubSO, aSubO, aSubName.ToCString());
1113 aSubO->SetColor(anArgO->GetColor());
1115 if (aSubO->GetShapeType() == GEOM::VERTEX) {
1116 aSubO->SetMarkerStd(anArgO->GetMarkerType(), anArgO->GetMarkerSize());
1117 if (anArgO->GetMarkerType() == GEOM::MT_USER)
1118 aSubO->SetMarkerTexture(anArgO->GetMarkerTexture());
1122 if (!CORBA::is_nil(anArgSO)) {
1123 // Restore published sub-shapes of the argument
1124 GEOM::ListOfGO_var aSubParts;
1125 if (theFindMethod == GEOM::FSM_GetInPlaceByHistory)
1126 // pass theObject, because only it has the history
1127 aSubParts = RestoreSubShapesOneLevel(theStudy, anArgSO, aSubSO,
1128 theObject, anOutArgs, theFindMethod, theAddPrefix);
1130 aSubParts = RestoreSubShapesOneLevel(theStudy, anArgSO, aSubSO,
1131 aSubO, anOutArgs, theFindMethod, theAddPrefix);
1132 // add to parts list
1133 addToListOfGO( aSubParts, aParts );
1136 else { // GetInPlace failed, try to build from published parts
1137 if (!CORBA::is_nil(anArgSO)) {
1138 SALOMEDS::SObject_var aSubSO;
1139 if (!CORBA::is_nil(theSObject))
1140 aSubSO = aStudyBuilder->NewObject(theSObject);
1142 // Restore published sub-shapes of the argument
1143 GEOM::ListOfGO_var aSubParts =
1144 RestoreSubShapesOneLevel(theStudy, anArgSO, aSubSO,
1145 theObject, anOutArgs, theFindMethod, theAddPrefix);
1147 // add to parts list
1148 addToListOfGO( aSubParts, aParts );
1150 if (aSubParts->length() > 0) {
1151 // remember restored objects for Python Dump
1152 addToListOfGO(anArgO, anOutArgs);
1154 // try to build an argument from a set of its sub-shapes,
1155 // that published and will be reconstructed
1156 if (aSubParts->length() > 1) {
1157 aSubO = aShapesOp->MakeCompound(aSubParts);
1158 // add to parts list
1159 addToListOfGO( aSubO, aParts );
1162 aSubO = aSubParts[0];
1164 if (!CORBA::is_nil(aSubO) && !CORBA::is_nil(aSubSO)) {
1165 // Publish the sub-shape
1166 TCollection_AsciiString aSubName;
1168 aSubName = "from_parts_of_";
1170 aSubName += anArgName;
1171 aSubSO = PublishInStudy(theStudy, aSubSO, aSubO, aSubName.ToCString());
1173 aSubO->SetColor(anArgO->GetColor());
1175 if (aSubO->GetShapeType() == GEOM::VERTEX) {
1176 aSubO->SetMarkerStd(anArgO->GetMarkerType(), anArgO->GetMarkerSize());
1177 if (anArgO->GetMarkerType() == GEOM::MT_USER)
1178 aSubO->SetMarkerTexture(anArgO->GetMarkerTexture());
1182 else if (!CORBA::is_nil(aSubSO)) {
1183 // remove created aSubSO, because no parts have been found
1184 aStudyBuilder->RemoveObject(aSubSO);
1187 } // try to build from published parts
1188 anArgSO->UnRegister();
1190 } // process arguments
1192 std::set<std::string> anObjEntryMap;
1193 GEOM::ListOfGO_var aResParts = new GEOM::ListOfGO;
1195 int nb = aParts->length();
1196 aResParts->length(nb);
1199 Handle(GEOM_Object) aMainObj = _impl->GetObject(theObject->GetStudyID(), theObject->GetEntry());
1200 Handle(GEOM_Function) aFunction = aMainObj->GetLastFunction();
1201 GEOM::TPythonDump pd (aFunction, true);
1204 for ( ; i < nb; i++ )
1206 GEOM::GEOM_Object_var anObj = GEOM::GEOM_Object::_duplicate( aParts[ i ] );
1207 if (CORBA::is_nil(anObj))
1209 char* anEntry = anObj->GetEntry();
1210 if (anObjEntryMap.count(anEntry))
1211 continue; // already treated
1212 anObjEntryMap.insert(anEntry);
1213 aResParts[nbRes++] = anObj;
1214 // clear python dump of object
1215 Handle(GEOM_Object) aGeomObj = _impl->GetObject(anObj->GetStudyID(), anEntry);
1216 Handle(GEOM_Function) anObjFun = aGeomObj->GetLastFunction();
1217 if ( !anObjFun.IsNull() )
1218 anObjFun->SetDescription( "" );
1224 pd <<"]" << " = geompy.RestoreGivenSubShapes(" << aMainObj << ", " << "[";
1225 //i = 0; nb = theArgs.length(); j = 0;
1226 i = 0; nb = anOutArgs->length(); j = 0;
1227 for ( ; i < nb; i++ )
1229 //GEOM::GEOM_Object_var anObj = theArgs[ i ];
1230 GEOM::GEOM_Object_var anObj = anOutArgs[ i ];
1231 if (CORBA::is_nil(anObj))
1233 Handle(GEOM_Object) aGeomObj = _impl->GetObject(anObj->GetStudyID(), anObj->GetEntry());
1239 pd <<"]" << ", " <<"geompy.GEOM.";
1240 switch (theFindMethod) {
1241 case GEOM::FSM_GetInPlace:
1242 pd << "FSM_GetInPlace"; break;
1243 case GEOM::FSM_MultiTransformed:
1244 pd << "FSM_MultiTransformed"; break;
1245 case GEOM::FSM_Transformed:
1246 pd << "FSM_Transformed"; break;
1247 case GEOM::FSM_GetSame:
1248 pd << "FSM_GetSame"; break;
1249 case GEOM::FSM_GetShapesOnShape:
1250 pd << "FSM_GetShapesOnShape"; break;
1251 case GEOM::FSM_GetInPlaceByHistory:
1253 pd << "FSM_GetInPlaceByHistory"; break;
1255 pd << ", " << theInheritFirstArg << ", " << theAddPrefix << ")";
1257 aResParts->length(nbRes);
1258 return aResParts._retn();
1261 //============================================================================
1262 // function : RestoreSubShapesOneLevel
1263 // purpose : Private method
1264 //============================================================================
1265 GEOM::ListOfGO* GEOM_Gen_i::RestoreSubShapesOneLevel (SALOMEDS::Study_ptr theStudy,
1266 SALOMEDS::SObject_ptr theOldSO,
1267 SALOMEDS::SObject_ptr theNewSO,
1268 GEOM::GEOM_Object_ptr theNewO,
1269 GEOM::ListOfGO& theOutArgs,
1270 GEOM::find_shape_method theFindMethod,
1271 CORBA::Boolean theAddPrefix)
1274 GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
1275 GEOM::ListOfGO_var aNewParts = new GEOM::ListOfGO;
1276 if (CORBA::is_nil(theStudy) || CORBA::is_nil(theOldSO) ||
1277 CORBA::is_nil(theNewO) /*|| CORBA::is_nil(theNewSO)*/)
1278 return aParts._retn();
1280 SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
1282 // Get interface, containing method, which we will use to reconstruct sub-shapes
1283 GEOM::GEOM_IShapesOperations_var aShapesOp = GetIShapesOperations(theStudy->StudyId());
1284 GEOM::GEOM_IGroupOperations_var aGroupOp = GetIGroupOperations(theStudy->StudyId());
1285 GEOM::GEOM_ITransformOperations_var aTrsfOp = GetITransformOperations(theStudy->StudyId());
1287 PortableServer::Servant aServant = _poa->reference_to_servant(aTrsfOp.in());
1288 GEOM_ITransformOperations_i* aTrsfOpSv = dynamic_cast<GEOM_ITransformOperations_i*>(aServant);
1290 // Reconstruct published sub-shapes
1291 SALOMEDS::ChildIterator_var it = theStudy->NewChildIterator(theOldSO);
1294 for (it->Init(); it->More(); it->Next()) {
1297 aParts->length(aLen);
1299 for (it->Init(); it->More(); it->Next()) {
1300 SALOMEDS::SObject_var anOldSubSO = it->Value();
1302 TCollection_AsciiString anArgName = anOldSubSO->GetName();
1304 SALOMEDS::GenericAttribute_var anAttr;
1305 if (anOldSubSO->FindAttribute(anAttr, "AttributeIOR")) {
1306 SALOMEDS::AttributeIOR_var anAttrIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
1307 GEOM::GEOM_Object_var anOldSubO =
1308 GEOM::GEOM_Object::_narrow(_orb->string_to_object(anAttrIOR->Value()));
1309 if (!CORBA::is_nil(anOldSubO)) {
1310 // Find a sub-shape of theNewO in place of anOldSubO
1311 GEOM::GEOM_Object_var aNewSubO;
1312 switch (theFindMethod) {
1313 case GEOM::FSM_GetInPlace:
1316 aNewSubO = aShapesOp->GetInPlace(theNewO, anOldSubO);
1319 case GEOM::FSM_MultiTransformed:
1321 // Only for Multi-transformations
1322 GEOM::GEOM_Object_var anArgOTrsf = aTrsfOpSv->TransformLikeOtherCopy(anOldSubO, theNewO);
1323 if (!CORBA::is_nil(anArgOTrsf)) {
1324 CORBA::String_var anArgOTrsfEntry = anArgOTrsf->GetEntry();
1325 Handle(GEOM_Object) anArgOTrsfImpl = _impl->GetObject(anArgOTrsf->GetStudyID(), anArgOTrsfEntry);
1326 Handle(GEOM_Function) anArgOTrsfFun = anArgOTrsfImpl->GetLastFunction();
1327 anArgOTrsfFun->SetDescription("");
1328 aNewSubO = aShapesOp->GetInPlace(theNewO, anArgOTrsf);
1332 case GEOM::FSM_Transformed:
1334 // transformation, cannot use GetInPlace, operate with indices
1335 GEOM::ListOfLong_var anIDs = anOldSubO->GetSubShapeIndices();
1336 if (anIDs->length() > 1) {
1338 aNewSubO = aGroupOp->CreateGroup(theNewO, aGroupOp->GetType(anOldSubO));
1339 if (!CORBA::is_nil(aNewSubO))
1340 aGroupOp->UnionIDs(aNewSubO, anIDs);
1344 aNewSubO = aShapesOp->GetSubShape(theNewO, anIDs[0]);
1348 case GEOM::FSM_GetSame:
1351 aNewSubO = aShapesOp->GetSame(theNewO, anOldSubO);
1354 case GEOM::FSM_GetShapesOnShape:
1356 // Use GetShapesOnShape. Can work only on solids, so it has sense to search only solids
1357 aNewSubO = aShapesOp->GetShapesOnShapeAsCompound(anOldSubO, theNewO,
1358 (short)GEOM::SOLID, GEOM::ST_ONIN);
1361 case GEOM::FSM_GetInPlaceByHistory:
1363 // Use GetInPlaceByHistory
1364 aNewSubO = aShapesOp->GetInPlaceByHistory(theNewO, anOldSubO);
1371 if (!CORBA::is_nil(aNewSubO)) {
1372 // remember restored objects for Python Dump
1373 addToListOfGO(anOldSubO, theOutArgs);
1375 // add the part to the list
1376 aParts[i] = aNewSubO;
1378 // add to parts list
1379 addToListOfGO( aNewSubO, aNewParts );
1381 SALOMEDS::SObject_var aNewSubSO;
1382 if (!CORBA::is_nil(theNewSO)) {
1383 // Publish the sub-shape
1384 TCollection_AsciiString aSubName;
1388 aSubName += anArgName;
1389 aNewSubSO = aStudyBuilder->NewObject(theNewSO);
1390 aNewSubSO = PublishInStudy(theStudy, aNewSubSO, aNewSubO, aSubName.ToCString());
1392 aNewSubO->SetColor(anOldSubO->GetColor());
1394 if (aNewSubO->GetShapeType() == GEOM::VERTEX) {
1395 aNewSubO->SetMarkerStd(anOldSubO->GetMarkerType(), anOldSubO->GetMarkerSize());
1396 if (anOldSubO->GetMarkerType() == GEOM::MT_USER)
1397 aNewSubO->SetMarkerTexture(anOldSubO->GetMarkerTexture());
1400 // Restore published sub-shapes of the argument
1401 GEOM::ListOfGO_var aSubParts;
1402 if (theFindMethod == GEOM::FSM_GetInPlaceByHistory)
1403 // pass the main shape as Object, because only it has the history
1404 aSubParts = RestoreSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO,
1405 theNewO, theOutArgs, theFindMethod, theAddPrefix);
1407 aSubParts = RestoreSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO,
1408 aNewSubO, theOutArgs, theFindMethod, theAddPrefix);
1409 // add to parts list
1410 addToListOfGO( aSubParts, aNewParts );
1412 else { // GetInPlace failed, try to build from published parts
1413 SALOMEDS::SObject_var aNewSubSO;
1414 if (!CORBA::is_nil(theNewSO))
1415 aNewSubSO = aStudyBuilder->NewObject(theNewSO);
1417 // Restore published sub-shapes of the argument
1418 GEOM::ListOfGO_var aSubParts =
1419 RestoreSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO,
1420 theNewO, theOutArgs, theFindMethod, theAddPrefix);
1421 // add to parts list
1422 addToListOfGO( aSubParts, aNewParts );
1424 if (aSubParts->length() > 0) {
1425 // remember restored objects for Python Dump
1426 addToListOfGO(anOldSubO, theOutArgs);
1428 // try to build an object from a set of its sub-shapes,
1429 // that published and will be reconstructed
1430 if (aSubParts->length() > 1) {
1431 aNewSubO = aShapesOp->MakeCompound(aSubParts);
1432 // add to parts list
1433 addToListOfGO( aNewSubO, aNewParts );
1436 aNewSubO = aSubParts[0];
1439 if (!CORBA::is_nil(aNewSubO)) {
1440 // add the part to the list
1441 aSubParts[i] = aNewSubO;
1444 // Publish the sub-shape
1445 if (!CORBA::is_nil(aNewSubSO)) {
1446 TCollection_AsciiString aSubName;
1448 aSubName = "from_parts_of_";
1450 aSubName += anArgName;
1451 aNewSubSO = PublishInStudy(theStudy, aNewSubSO, aNewSubO, aSubName.ToCString());
1453 aNewSubO->SetColor(anOldSubO->GetColor());
1455 if (aNewSubO->GetShapeType() == GEOM::VERTEX) {
1456 aNewSubO->SetMarkerStd(anOldSubO->GetMarkerType(), anOldSubO->GetMarkerSize());
1457 if (anOldSubO->GetMarkerType() == GEOM::MT_USER)
1458 aNewSubO->SetMarkerTexture(anOldSubO->GetMarkerTexture());
1463 else if (!CORBA::is_nil(aNewSubSO)) {
1464 // remove created aSubSO, because no parts have been found
1465 aStudyBuilder->RemoveObject(aNewSubSO);
1467 } // try to build from published parts
1470 } // iterate on published sub-shapes
1473 // add to parts list
1474 addToListOfGO( aNewParts, aParts );
1475 return aParts._retn();
1478 //============================================================================
1479 // function : RestoreGivenSubShapes
1480 // purpose : Private method. Works only if both theObject and theSObject
1481 // are defined, and does not check, if they correspond to each other.
1482 // List theArgs in this case contains not only operation arguments,
1483 // but also all subshapes, which must be published.
1484 //============================================================================
1485 GEOM::ListOfGO* GEOM_Gen_i::RestoreGivenSubShapes(SALOMEDS::Study_ptr theStudy,
1486 GEOM::GEOM_Object_ptr theObject,
1487 SALOMEDS::SObject_ptr theSObject,
1488 const GEOM::ListOfGO& theArgs,
1489 GEOM::find_shape_method theFindMethod,
1490 CORBA::Boolean theInheritFirstArg,
1491 CORBA::Boolean theAddPrefix)
1493 GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
1494 //PTv, IMP 0020001, The salome object <theSObject>
1495 // is not obligatory in case of invokation from script
1496 if (CORBA::is_nil(theStudy) || CORBA::is_nil(theObject) /*|| CORBA::is_nil(theSObject)*/)
1497 return aParts._retn();
1499 // If theArgs list is empty, nothing to do
1500 Standard_Integer aLength = theArgs.length();
1502 return aParts._retn();
1504 // Get all arguments
1505 GEOM::ListOfGO_var anOpArgsList = theObject->GetDependency();
1506 Standard_Integer nbArgsActual = anOpArgsList->length();
1508 // If anOpArgsList list is empty, nothing to do
1509 if (nbArgsActual == 0)
1510 return aParts._retn();
1512 // Entries of arguments and subshapes
1513 std::set<std::string> anArgs;
1514 for (int i = 0; i < aLength; i++) {
1515 CORBA::String_var anEntry = theArgs[i]->GetEntry();
1516 anArgs.insert(anEntry.in());
1519 // Arguments to be published
1520 // We try to publish all arguments, that are in theArgs list
1521 GEOM::ListOfGO_var aList = new GEOM::ListOfGO;
1522 aList->length(nbArgsActual);
1525 for (int j = 0; j < nbArgsActual; j++) {
1526 CORBA::String_var anEntry = anOpArgsList[j]->GetEntry();
1527 if (anArgs.count(anEntry.in())) {
1528 aList[k] = GEOM::GEOM_Object::_duplicate(anOpArgsList[j]);
1533 //aList->length(nbArgsActual);
1535 if (nbArgsActual < 1)
1536 return aParts._retn();
1538 if (theInheritFirstArg || (nbArgsActual == 1)) {
1539 // Do not publish argument's reflection,
1540 // but only reconstruct its published sub-shapes
1542 CORBA::String_var anIOR = _orb->object_to_string(aList[0]);
1543 SALOMEDS::SObject_var anArgSO = theStudy->FindObjectIOR(anIOR.in());
1545 aParts = RestoreGivenSubShapesOneLevel(theStudy, anArgSO, theSObject, theObject,
1546 anArgs, theFindMethod, theAddPrefix);
1548 // set the color of the transformed shape to the color of initial shape
1549 theObject->SetColor(aList[0]->GetColor());
1551 if (theObject->GetShapeType() == GEOM::VERTEX) {
1552 theObject->SetMarkerStd(aList[0]->GetMarkerType(), aList[0]->GetMarkerSize());
1553 if (aList[0]->GetMarkerType() == GEOM::MT_USER)
1554 theObject->SetMarkerTexture(aList[0]->GetMarkerTexture());
1557 anArgSO->UnRegister();
1560 // Get interface, containing method, which we will use to reconstruct sub-shapes
1561 GEOM::GEOM_IShapesOperations_var aShapesOp = GetIShapesOperations(theStudy->StudyId());
1562 GEOM::GEOM_IGroupOperations_var aGroupOp = GetIGroupOperations(theStudy->StudyId());
1563 GEOM::GEOM_ITransformOperations_var aTrsfOp = GetITransformOperations(theStudy->StudyId());
1565 PortableServer::Servant aServant = _poa->reference_to_servant(aTrsfOp.in());
1566 GEOM_ITransformOperations_i* aTrsfOpSv = dynamic_cast<GEOM_ITransformOperations_i*>(aServant);
1568 // Reconstruct arguments and tree of sub-shapes of the arguments
1569 CORBA::String_var anIOR;
1570 SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
1571 for (Standard_Integer i = 0; i < nbArgsActual; i++)
1573 GEOM::GEOM_Object_var anArgO = aList[i];
1574 if (!CORBA::is_nil(anArgO)) {
1575 anIOR = _orb->object_to_string(anArgO);
1576 SALOMEDS::SObject_var anArgSO = theStudy->FindObjectIOR(anIOR.in());
1577 TCollection_AsciiString anArgName;
1578 if (CORBA::is_nil(anArgSO)) {
1580 anArgName += TCollection_AsciiString(i);
1583 anArgName = anArgSO->GetName();
1586 // Find a sub-shape of theObject in place of the argument
1587 GEOM::GEOM_Object_var aSubO;
1588 switch (theFindMethod) {
1589 case GEOM::FSM_GetInPlace:
1592 aSubO = aShapesOp->GetInPlace(theObject, anArgO);
1595 case GEOM::FSM_MultiTransformed:
1597 // Only for Multi-transformations
1598 GEOM::GEOM_Object_var anArgOTrsf = aTrsfOpSv->TransformLikeOtherCopy(anArgO, theObject);
1599 if (!CORBA::is_nil(anArgOTrsf)) {
1600 CORBA::String_var anArgOTrsfEntry = anArgOTrsf->GetEntry();
1601 Handle(GEOM_Object) anArgOTrsfImpl = _impl->GetObject(anArgOTrsf->GetStudyID(), anArgOTrsfEntry);
1602 Handle(GEOM_Function) anArgOTrsfFun = anArgOTrsfImpl->GetLastFunction();
1603 anArgOTrsfFun->SetDescription("");
1604 aSubO = aShapesOp->GetInPlace(theObject, anArgOTrsf);
1608 case GEOM::FSM_Transformed:
1610 // transformation, cannot use GetInPlace, operate with indices
1611 GEOM::ListOfLong_var anIDs = anArgO->GetSubShapeIndices();
1612 if (anIDs->length() > 1) {
1614 aSubO = aGroupOp->CreateGroup(theObject, aGroupOp->GetType(anArgO));
1615 if (!CORBA::is_nil(aSubO))
1616 aGroupOp->UnionIDs(aSubO, anIDs);
1618 else if (anIDs->length() > 0) {
1620 aSubO = aShapesOp->GetSubShape(theObject, anIDs[0]);
1624 case GEOM::FSM_GetSame:
1627 aSubO = aShapesOp->GetSame(theObject, anArgO);
1630 case GEOM::FSM_GetShapesOnShape:
1632 // Use GetShapesOnShape. Can work only on solids, so it has sense to search only solids
1633 aSubO = aShapesOp->GetShapesOnShapeAsCompound(anArgO, theObject,
1634 (short)GEOM::SOLID, GEOM::ST_ONIN);
1637 case GEOM::FSM_GetInPlaceByHistory:
1639 // Use GetInPlaceByHistory
1640 aSubO = aShapesOp->GetInPlaceByHistory(theObject, anArgO);
1647 if (!CORBA::is_nil(aSubO)) {
1648 // add to parts list
1649 addToListOfGO( aSubO, aParts );
1651 // Publish the sub-shape
1652 SALOMEDS::SObject_var aSubSO;
1653 if (!CORBA::is_nil(theSObject)) {
1654 TCollection_AsciiString aSubName;
1658 aSubName += anArgName;
1659 aSubSO = aStudyBuilder->NewObject(theSObject);
1660 aSubSO = PublishInStudy(theStudy, aSubSO, aSubO, aSubName.ToCString());
1662 aSubO->SetColor(anArgO->GetColor());
1664 if (aSubO->GetShapeType() == GEOM::VERTEX) {
1665 aSubO->SetMarkerStd(anArgO->GetMarkerType(), anArgO->GetMarkerSize());
1666 if (anArgO->GetMarkerType() == GEOM::MT_USER)
1667 aSubO->SetMarkerTexture(anArgO->GetMarkerTexture());
1671 if (!CORBA::is_nil(anArgSO)) {
1672 // Restore published sub-shapes of the argument
1673 GEOM::ListOfGO_var aSubParts;
1674 if (theFindMethod == GEOM::FSM_GetInPlaceByHistory)
1675 // pass theObject, because only it has the history
1676 aSubParts = RestoreGivenSubShapesOneLevel(theStudy, anArgSO, aSubSO,
1677 theObject, anArgs, theFindMethod, theAddPrefix);
1679 aSubParts = RestoreGivenSubShapesOneLevel(theStudy, anArgSO, aSubSO,
1680 aSubO, anArgs, theFindMethod, theAddPrefix);
1681 // add to parts list
1682 addToListOfGO( aSubParts, aParts );
1685 else { // GetInPlace failed, try to build from published parts
1686 if (!CORBA::is_nil(anArgSO)) {
1687 SALOMEDS::SObject_var aSubSO;
1688 if (!CORBA::is_nil(theSObject))
1689 aSubSO = aStudyBuilder->NewObject(theSObject);
1691 // Restore published sub-shapes of the argument
1692 GEOM::ListOfGO_var aSubParts =
1693 RestoreGivenSubShapesOneLevel(theStudy, anArgSO, aSubSO,
1694 theObject, anArgs, theFindMethod, theAddPrefix);
1696 // add to parts list
1697 addToListOfGO( aSubParts, aParts );
1699 if (aSubParts->length() > 0) {
1700 // try to build an argument from a set of its sub-shapes,
1701 // that published and will be reconstructed
1702 if (aSubParts->length() > 1) {
1703 aSubO = aShapesOp->MakeCompound(aSubParts);
1704 // add to parts list
1705 addToListOfGO( aSubO, aParts );
1708 aSubO = aSubParts[0];
1710 if (!CORBA::is_nil(aSubO) && !CORBA::is_nil(aSubSO)) {
1711 // Publish the sub-shape
1712 TCollection_AsciiString aSubName;
1714 aSubName = "from_parts_of_";
1716 aSubName += anArgName;
1717 aSubSO = PublishInStudy(theStudy, aSubSO, aSubO, aSubName.ToCString());
1719 aSubO->SetColor(anArgO->GetColor());
1721 if (aSubO->GetShapeType() == GEOM::VERTEX) {
1722 aSubO->SetMarkerStd(anArgO->GetMarkerType(), anArgO->GetMarkerSize());
1723 if (anArgO->GetMarkerType() == GEOM::MT_USER)
1724 aSubO->SetMarkerTexture(anArgO->GetMarkerTexture());
1728 else if (!CORBA::is_nil(aSubSO)) {
1729 // remove created aSubSO, because no parts have been found
1730 aStudyBuilder->RemoveObject(aSubSO);
1733 } // try to build from published parts
1734 anArgSO->UnRegister();
1736 } // process arguments
1738 std::set<std::string> anObjEntryMap;
1739 GEOM::ListOfGO_var aResParts = new GEOM::ListOfGO;
1741 int nb = aParts->length();
1742 aResParts->length(nb);
1745 Handle(GEOM_Object) aMainObj = _impl->GetObject(theObject->GetStudyID(), theObject->GetEntry());
1746 Handle(GEOM_Function) aFunction = aMainObj->GetLastFunction();
1747 GEOM::TPythonDump pd (aFunction, true);
1750 for ( ; i < nb; i++ )
1752 GEOM::GEOM_Object_var anObj = GEOM::GEOM_Object::_duplicate( aParts[ i ] );
1753 if (CORBA::is_nil(anObj))
1755 char* anEntry = anObj->GetEntry();
1756 if (anObjEntryMap.count(anEntry))
1757 continue; // already treated
1758 anObjEntryMap.insert(anEntry);
1759 aResParts[nbRes++] = anObj;
1760 // clear python dump of object
1761 Handle(GEOM_Object) aGeomObj = _impl->GetObject(anObj->GetStudyID(), anEntry);
1762 Handle(GEOM_Function) anObjFun = aGeomObj->GetLastFunction();
1763 if ( !anObjFun.IsNull() )
1764 anObjFun->SetDescription( "" );
1770 pd <<"]" << " = geompy.RestoreGivenSubShapes(" << aMainObj << ", " << "[";
1771 i = 0; nb = theArgs.length(); j = 0;
1772 for ( ; i < nb; i++ )
1774 GEOM::GEOM_Object_var anObj = theArgs[ i ];
1775 if (CORBA::is_nil(anObj))
1777 Handle(GEOM_Object) aGeomObj = _impl->GetObject(anObj->GetStudyID(), anObj->GetEntry());
1783 pd <<"]" << ", " <<"geompy.GEOM.";
1784 switch (theFindMethod) {
1785 case GEOM::FSM_GetInPlace:
1786 pd << "FSM_GetInPlace"; break;
1787 case GEOM::FSM_MultiTransformed:
1788 pd << "FSM_MultiTransformed"; break;
1789 case GEOM::FSM_Transformed:
1790 pd << "FSM_Transformed"; break;
1791 case GEOM::FSM_GetSame:
1792 pd << "FSM_GetSame"; break;
1793 case GEOM::FSM_GetShapesOnShape:
1794 pd << "FSM_GetShapesOnShape"; break;
1795 case GEOM::FSM_GetInPlaceByHistory:
1797 pd << "FSM_GetInPlaceByHistory"; break;
1799 pd << ", " << theInheritFirstArg << ", " << theAddPrefix << ")";
1801 aResParts->length(nbRes);
1802 return aResParts._retn();
1805 //============================================================================
1806 // function : RestoreGivenSubShapesOneLevel
1807 // purpose : Private method
1808 //============================================================================
1809 GEOM::ListOfGO* GEOM_Gen_i::RestoreGivenSubShapesOneLevel (SALOMEDS::Study_ptr theStudy,
1810 SALOMEDS::SObject_ptr theOldSO,
1811 SALOMEDS::SObject_ptr theNewSO,
1812 GEOM::GEOM_Object_ptr theNewO,
1813 std::set<std::string> theArgs,
1814 GEOM::find_shape_method theFindMethod,
1815 CORBA::Boolean theAddPrefix)
1818 GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
1819 GEOM::ListOfGO_var aNewParts = new GEOM::ListOfGO;
1820 if (CORBA::is_nil(theStudy) || CORBA::is_nil(theOldSO) ||
1821 CORBA::is_nil(theNewO) /*|| CORBA::is_nil(theNewSO)*/)
1822 return aParts._retn();
1824 SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
1826 // Get interface, containing method, which we will use to reconstruct sub-shapes
1827 GEOM::GEOM_IShapesOperations_var aShapesOp = GetIShapesOperations(theStudy->StudyId());
1828 GEOM::GEOM_IGroupOperations_var aGroupOp = GetIGroupOperations(theStudy->StudyId());
1829 GEOM::GEOM_ITransformOperations_var aTrsfOp = GetITransformOperations(theStudy->StudyId());
1831 PortableServer::Servant aServant = _poa->reference_to_servant(aTrsfOp.in());
1832 GEOM_ITransformOperations_i* aTrsfOpSv = dynamic_cast<GEOM_ITransformOperations_i*>(aServant);
1834 // Reconstruct published sub-shapes
1835 SALOMEDS::ChildIterator_var it = theStudy->NewChildIterator(theOldSO);
1838 for (it->Init(); it->More(); it->Next()) {
1841 aParts->length(aLen);
1843 for (it->Init(); it->More(); it->Next()) {
1844 SALOMEDS::SObject_var anOldSubSO = it->Value();
1846 TCollection_AsciiString anArgName = anOldSubSO->GetName();
1848 SALOMEDS::GenericAttribute_var anAttr;
1849 if (anOldSubSO->FindAttribute(anAttr, "AttributeIOR")) {
1850 SALOMEDS::AttributeIOR_var anAttrIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
1851 GEOM::GEOM_Object_var anOldSubO =
1852 GEOM::GEOM_Object::_narrow(_orb->string_to_object(anAttrIOR->Value()));
1854 bool okToContinue = false;
1856 if (!CORBA::is_nil(anOldSubO)) {
1857 CORBA::String_var anEntry = anOldSubO->GetEntry();
1858 okToContinue = theArgs.count(anEntry.in());
1862 // Find a sub-shape of theNewO in place of anOldSubO
1863 GEOM::GEOM_Object_var aNewSubO;
1864 switch (theFindMethod) {
1865 case GEOM::FSM_GetInPlace:
1868 aNewSubO = aShapesOp->GetInPlace(theNewO, anOldSubO);
1871 case GEOM::FSM_MultiTransformed:
1873 // Only for Multi-transformations
1874 GEOM::GEOM_Object_var anArgOTrsf = aTrsfOpSv->TransformLikeOtherCopy(anOldSubO, theNewO);
1875 if (!CORBA::is_nil(anArgOTrsf)) {
1876 CORBA::String_var anArgOTrsfEntry = anArgOTrsf->GetEntry();
1877 Handle(GEOM_Object) anArgOTrsfImpl = _impl->GetObject(anArgOTrsf->GetStudyID(), anArgOTrsfEntry);
1878 Handle(GEOM_Function) anArgOTrsfFun = anArgOTrsfImpl->GetLastFunction();
1879 anArgOTrsfFun->SetDescription("");
1880 aNewSubO = aShapesOp->GetInPlace(theNewO, anArgOTrsf);
1884 case GEOM::FSM_Transformed:
1886 // transformation, cannot use GetInPlace, operate with indices
1887 GEOM::ListOfLong_var anIDs = anOldSubO->GetSubShapeIndices();
1888 if (anIDs->length() > 1) {
1890 aNewSubO = aGroupOp->CreateGroup(theNewO, aGroupOp->GetType(anOldSubO));
1891 if (!CORBA::is_nil(aNewSubO))
1892 aGroupOp->UnionIDs(aNewSubO, anIDs);
1896 aNewSubO = aShapesOp->GetSubShape(theNewO, anIDs[0]);
1900 case GEOM::FSM_GetSame:
1903 aNewSubO = aShapesOp->GetSame(theNewO, anOldSubO);
1906 case GEOM::FSM_GetShapesOnShape:
1908 // Use GetShapesOnShape. Can work only on solids, so it has sense to search only solids
1909 aNewSubO = aShapesOp->GetShapesOnShapeAsCompound(anOldSubO, theNewO,
1910 (short)GEOM::SOLID, GEOM::ST_ONIN);
1913 case GEOM::FSM_GetInPlaceByHistory:
1915 // Use GetInPlaceByHistory
1916 aNewSubO = aShapesOp->GetInPlaceByHistory(theNewO, anOldSubO);
1923 if (!CORBA::is_nil(aNewSubO)) {
1924 // add the part to the list
1925 aParts[i] = aNewSubO;
1927 // add to parts list
1928 addToListOfGO( aNewSubO, aNewParts );
1930 SALOMEDS::SObject_var aNewSubSO;
1931 if (!CORBA::is_nil(theNewSO)) {
1932 // Publish the sub-shape
1933 TCollection_AsciiString aSubName;
1937 aSubName += anArgName;
1938 aNewSubSO = aStudyBuilder->NewObject(theNewSO);
1939 aNewSubSO = PublishInStudy(theStudy, aNewSubSO, aNewSubO, aSubName.ToCString());
1941 aNewSubO->SetColor(anOldSubO->GetColor());
1943 if (aNewSubO->GetShapeType() == GEOM::VERTEX) {
1944 aNewSubO->SetMarkerStd(anOldSubO->GetMarkerType(), anOldSubO->GetMarkerSize());
1945 if (anOldSubO->GetMarkerType() == GEOM::MT_USER)
1946 aNewSubO->SetMarkerTexture(anOldSubO->GetMarkerTexture());
1949 // Restore published sub-shapes of the argument
1950 GEOM::ListOfGO_var aSubParts;
1951 if (theFindMethod == GEOM::FSM_GetInPlaceByHistory)
1952 // pass the main shape as Object, because only it has the history
1953 aSubParts = RestoreGivenSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO,
1954 theNewO, theArgs, theFindMethod, theAddPrefix);
1956 aSubParts = RestoreGivenSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO,
1957 aNewSubO, theArgs, theFindMethod, theAddPrefix);
1958 // add to parts list
1959 addToListOfGO( aSubParts, aNewParts );
1961 else { // GetInPlace failed, try to build from published parts
1962 SALOMEDS::SObject_var aNewSubSO;
1963 if (!CORBA::is_nil(theNewSO))
1964 aNewSubSO = aStudyBuilder->NewObject(theNewSO);
1966 // Restore published sub-shapes of the argument
1967 GEOM::ListOfGO_var aSubParts =
1968 RestoreGivenSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO,
1969 theNewO, theArgs, theFindMethod, theAddPrefix);
1970 // add to parts list
1971 addToListOfGO( aSubParts, aNewParts );
1973 if (aSubParts->length() > 0) {
1974 // try to build an object from a set of its sub-shapes,
1975 // that published and will be reconstructed
1976 if (aSubParts->length() > 1) {
1977 aNewSubO = aShapesOp->MakeCompound(aSubParts);
1978 // add to parts list
1979 addToListOfGO( aNewSubO, aNewParts );
1982 aNewSubO = aSubParts[0];
1985 if (!CORBA::is_nil(aNewSubO)) {
1986 // add the part to the list
1987 aSubParts[i] = aNewSubO;
1990 // Publish the sub-shape
1991 if (!CORBA::is_nil(aNewSubSO)) {
1992 TCollection_AsciiString aSubName;
1994 aSubName = "from_parts_of_";
1996 aSubName += anArgName;
1997 aNewSubSO = PublishInStudy(theStudy, aNewSubSO, aNewSubO, aSubName.ToCString());
1999 aNewSubO->SetColor(anOldSubO->GetColor());
2001 if (aNewSubO->GetShapeType() == GEOM::VERTEX) {
2002 aNewSubO->SetMarkerStd(anOldSubO->GetMarkerType(), anOldSubO->GetMarkerSize());
2003 if (anOldSubO->GetMarkerType() == GEOM::MT_USER)
2004 aNewSubO->SetMarkerTexture(anOldSubO->GetMarkerTexture());
2009 else if (!CORBA::is_nil(aNewSubSO)) {
2010 // remove created aSubSO, because no parts have been found
2011 aStudyBuilder->RemoveObject(aNewSubSO);
2013 } // try to build from published parts
2016 } // iterate on published sub-shapes
2019 // add to parts list
2020 addToListOfGO( aNewParts, aParts );
2021 return aParts._retn();
2024 //============================================================================
2025 // function : register()
2026 // purpose : register 'name' in 'name_service'
2027 //============================================================================
2028 void GEOM_Gen_i::register_name(char * name)
2030 GEOM::GEOM_Gen_var g = _this();
2031 name_service->Register(g, name);
2034 //============================================================================
2037 //============================================================================
2038 void GEOM_Gen_i::Undo(CORBA::Long theStudyID)
2040 _impl->Undo(theStudyID);
2043 //============================================================================
2046 //============================================================================
2047 void GEOM_Gen_i::Redo(CORBA::Long theStudyID)
2049 _impl->Redo(theStudyID);
2052 //============================================================================
2053 // function : GetIBasicOperations
2055 //============================================================================
2056 GEOM::GEOM_IBasicOperations_ptr GEOM_Gen_i::GetIBasicOperations(CORBA::Long theStudyID)
2057 throw ( SALOME::SALOME_Exception )
2059 Unexpect aCatch(SALOME_SalomeException);
2060 MESSAGE( "GEOM_Gen_i::GetIBasicOperations" );
2062 GEOM::GEOM_Gen_ptr engine = _this();
2064 //transfer reference on engine
2065 GEOM_IBasicOperations_i* aServant =
2066 new GEOM_IBasicOperations_i(_poa, engine, _impl->GetIBasicOperations(theStudyID));
2068 PortableServer::ObjectId_var id = _poa->activate_object(aServant);
2069 // activate the CORBA servant
2070 GEOM::GEOM_IBasicOperations_var operations = aServant->_this();
2071 return operations._retn();
2074 //============================================================================
2075 // function : GetITransformOperations
2077 //============================================================================
2078 GEOM::GEOM_ITransformOperations_ptr GEOM_Gen_i::GetITransformOperations(CORBA::Long theStudyID)
2079 throw ( SALOME::SALOME_Exception )
2081 Unexpect aCatch(SALOME_SalomeException);
2082 MESSAGE( "GEOM_Gen_i::GetITransformOperations" );
2084 GEOM::GEOM_Gen_ptr engine = _this();
2086 GEOM_ITransformOperations_i* aServant =
2087 new GEOM_ITransformOperations_i(_poa, engine, _impl->GetITransformOperations(theStudyID));
2089 // activate the CORBA servant
2090 GEOM::GEOM_ITransformOperations_var operations = aServant->_this();
2091 return operations._retn();
2094 //============================================================================
2095 // function : GetI3DPrimOperations
2097 //============================================================================
2098 GEOM::GEOM_I3DPrimOperations_ptr GEOM_Gen_i::GetI3DPrimOperations(CORBA::Long theStudyID)
2099 throw ( SALOME::SALOME_Exception )
2101 Unexpect aCatch(SALOME_SalomeException);
2102 MESSAGE( "GEOM_Gen_i::GetI3DPrimOperations" );
2104 GEOM::GEOM_Gen_ptr engine = _this();
2106 GEOM_I3DPrimOperations_i* aServant =
2107 new GEOM_I3DPrimOperations_i(_poa, engine, _impl->GetI3DPrimOperations(theStudyID));
2108 PortableServer::ObjectId_var id = _poa->activate_object(aServant);
2110 // activate the CORBA servant
2111 GEOM::GEOM_I3DPrimOperations_var operations = aServant->_this();
2112 return operations._retn();
2115 //============================================================================
2116 // function : GetIShapesOperations
2118 //============================================================================
2119 GEOM::GEOM_IShapesOperations_ptr GEOM_Gen_i::GetIShapesOperations(CORBA::Long theStudyID)
2120 throw ( SALOME::SALOME_Exception )
2122 Unexpect aCatch(SALOME_SalomeException);
2123 MESSAGE( "GEOM_Gen_i::GetIShapesOperations" );
2125 GEOM::GEOM_Gen_ptr engine = _this();
2127 GEOM_IShapesOperations_i* aServant =
2128 new GEOM_IShapesOperations_i(_poa, engine, _impl->GetIShapesOperations(theStudyID));
2130 // activate the CORBA servant
2131 GEOM::GEOM_IShapesOperations_var operations = aServant->_this();
2132 return operations._retn();
2135 //============================================================================
2136 // function : GetIBlocksOperations
2138 //============================================================================
2139 GEOM::GEOM_IBlocksOperations_ptr GEOM_Gen_i::GetIBlocksOperations(CORBA::Long theStudyID)
2140 throw ( SALOME::SALOME_Exception )
2142 Unexpect aCatch(SALOME_SalomeException);
2143 MESSAGE( "GEOM_Gen_i::GetIBlocksOperations" );
2145 GEOM::GEOM_Gen_ptr engine = _this();
2147 GEOM_IBlocksOperations_i* aServant =
2148 new GEOM_IBlocksOperations_i(_poa, engine, _impl->GetIBlocksOperations(theStudyID));
2150 // activate the CORBA servant
2151 GEOM::GEOM_IBlocksOperations_var operations = aServant->_this();
2152 return operations._retn();
2155 //============================================================================
2156 // function : GetIBooleanOperations
2158 //============================================================================
2159 GEOM::GEOM_IBooleanOperations_ptr GEOM_Gen_i::GetIBooleanOperations(CORBA::Long theStudyID)
2160 throw ( SALOME::SALOME_Exception )
2162 Unexpect aCatch(SALOME_SalomeException);
2163 MESSAGE( "GEOM_Gen_i::GetIBooleanOperations" );
2165 GEOM::GEOM_Gen_ptr engine = _this();
2167 GEOM_IBooleanOperations_i* aServant =
2168 new GEOM_IBooleanOperations_i(_poa, engine, _impl->GetIBooleanOperations(theStudyID));
2170 // activate the CORBA servant
2171 GEOM::GEOM_IBooleanOperations_var operations = aServant->_this();
2172 return operations._retn();
2175 //============================================================================
2176 // function : GetICurvesOperations
2178 //============================================================================
2179 GEOM::GEOM_ICurvesOperations_ptr GEOM_Gen_i::GetICurvesOperations(CORBA::Long theStudyID)
2180 throw ( SALOME::SALOME_Exception )
2182 Unexpect aCatch(SALOME_SalomeException);
2183 MESSAGE( "GEOM_Gen_i::GetICurvesOperations" );
2185 GEOM::GEOM_Gen_ptr engine = _this();
2187 GEOM_ICurvesOperations_i* aServant =
2188 new GEOM_ICurvesOperations_i(_poa, engine, _impl->GetICurvesOperations(theStudyID));
2190 // activate the CORBA servant
2191 GEOM::GEOM_ICurvesOperations_var operations = aServant->_this();
2192 return operations._retn();
2195 //============================================================================
2196 // function : GetILocalOperations
2198 //============================================================================
2199 GEOM::GEOM_ILocalOperations_ptr GEOM_Gen_i::GetILocalOperations(CORBA::Long theStudyID)
2200 throw ( SALOME::SALOME_Exception )
2202 Unexpect aCatch(SALOME_SalomeException);
2203 MESSAGE( "GEOM_Gen_i::GetILocalOperations" );
2205 GEOM::GEOM_Gen_ptr engine = _this();
2207 GEOM_ILocalOperations_i* aServant =
2208 new GEOM_ILocalOperations_i(_poa, engine, _impl->GetILocalOperations(theStudyID));
2210 // activate the CORBA servant
2211 GEOM::GEOM_ILocalOperations_var operations = aServant->_this();
2212 return operations._retn();
2215 //============================================================================
2216 // function : GetIHealingOperations
2218 //============================================================================
2219 GEOM::GEOM_IHealingOperations_ptr GEOM_Gen_i::GetIHealingOperations(CORBA::Long theStudyID)
2220 throw ( SALOME::SALOME_Exception )
2222 Unexpect aCatch(SALOME_SalomeException);
2223 MESSAGE( "GEOM_Gen_i::IHealingOperations" );
2225 GEOM::GEOM_Gen_ptr engine = _this();
2227 GEOM_IHealingOperations_i* aServant =
2228 new GEOM_IHealingOperations_i(_poa, engine, _impl->GetIHealingOperations(theStudyID));
2230 // activate the CORBA servant
2231 GEOM::GEOM_IHealingOperations_var operations = aServant->_this();
2232 return operations._retn();
2235 //============================================================================
2236 // function : GetIInsertOperations
2238 //============================================================================
2239 GEOM::GEOM_IInsertOperations_ptr GEOM_Gen_i::GetIInsertOperations(CORBA::Long theStudyID)
2240 throw ( SALOME::SALOME_Exception )
2242 Unexpect aCatch(SALOME_SalomeException);
2243 MESSAGE( "GEOM_Gen_i::GetIInsertOperations" );
2245 GEOM::GEOM_Gen_ptr engine = _this();
2247 GEOM_IInsertOperations_i* aServant =
2248 new GEOM_IInsertOperations_i(_poa, engine, _impl->GetIInsertOperations(theStudyID));
2250 // activate the CORBA servant
2251 GEOM::GEOM_IInsertOperations_var operations = aServant->_this();
2252 return operations._retn();
2255 //============================================================================
2256 // function : GetIMeasureOperations
2258 //============================================================================
2259 GEOM::GEOM_IMeasureOperations_ptr GEOM_Gen_i::GetIMeasureOperations(CORBA::Long theStudyID)
2260 throw ( SALOME::SALOME_Exception )
2262 Unexpect aCatch(SALOME_SalomeException);
2263 MESSAGE( "GEOM_Gen_i::GetIMeasureOperations" );
2265 GEOM::GEOM_Gen_ptr engine = _this();
2267 GEOM_IMeasureOperations_i* aServant =
2268 new GEOM_IMeasureOperations_i(_poa, engine, _impl->GetIMeasureOperations(theStudyID));
2270 // activate the CORBA servant
2271 GEOM::GEOM_IMeasureOperations_var operations = aServant->_this();
2272 return operations._retn();
2275 //============================================================================
2276 // function : GetIGroupOperations
2278 //============================================================================
2279 GEOM::GEOM_IGroupOperations_ptr GEOM_Gen_i::GetIGroupOperations(CORBA::Long theStudyID)
2280 throw ( SALOME::SALOME_Exception )
2282 Unexpect aCatch(SALOME_SalomeException);
2283 MESSAGE( "GEOM_Gen_i::GetIGroupOperations" );
2285 GEOM::GEOM_Gen_ptr engine = _this();
2287 GEOM_IGroupOperations_i* aServant =
2288 new GEOM_IGroupOperations_i(_poa, engine, _impl->GetIGroupOperations(theStudyID));
2290 // activate the CORBA servant
2291 GEOM::GEOM_IGroupOperations_var operations = aServant->_this();
2292 return operations._retn();
2295 //============================================================================
2296 // function : GetIAdvancedOperations
2298 //============================================================================
2299 GEOM::GEOM_IAdvancedOperations_ptr GEOM_Gen_i::GetIAdvancedOperations(CORBA::Long theStudyID)
2300 throw ( SALOME::SALOME_Exception )
2302 Unexpect aCatch(SALOME_SalomeException);
2303 MESSAGE( "GEOM_Gen_i::GetIAdvancedOperations" );
2305 GEOM::GEOM_Gen_ptr engine = _this();
2307 GEOM_IAdvancedOperations_i* aServant =
2308 new GEOM_IAdvancedOperations_i(_poa, engine, _impl->GetIAdvancedOperations(theStudyID));
2310 // activate the CORBA servant
2311 GEOM::GEOM_IAdvancedOperations_var operations = aServant->_this();
2312 return operations._retn();
2315 //=============================================================================
2319 //=============================================================================
2320 GEOM::GEOM_Object_ptr GEOM_Gen_i::AddSubShape (GEOM::GEOM_Object_ptr theMainShape,
2321 const GEOM::ListOfLong& theIndices)
2323 if (CORBA::is_nil(theMainShape) || theIndices.length() < 1)
2324 return GEOM::GEOM_Object::_nil();
2325 CORBA::String_var entry = theMainShape->GetEntry();
2326 Handle(GEOM_Object) aMainShape = _impl->GetObject(theMainShape->GetStudyID(), entry);
2327 if (aMainShape.IsNull()) return GEOM::GEOM_Object::_nil();
2329 Handle(TColStd_HArray1OfInteger) anArray = new TColStd_HArray1OfInteger(1, theIndices.length());
2330 for(Standard_Integer i = 0; i<theIndices.length(); i++) anArray->SetValue(i+1, theIndices[i]);
2332 Handle(GEOM_Object) anObject = _impl->AddSubShape(aMainShape, anArray, true);
2333 if(anObject.IsNull()) return GEOM::GEOM_Object::_nil();
2335 TCollection_AsciiString anEntry;
2336 TDF_Tool::Entry(anObject->GetEntry(), anEntry);
2337 return GetObject(anObject->GetDocID(), anEntry.ToCString());
2340 //=============================================================================
2344 //=============================================================================
2345 void GEOM_Gen_i::RemoveObject(GEOM::GEOM_Object_ptr theObject)
2347 CORBA::String_var anEntry = theObject->GetEntry();
2348 Handle(GEOM_Object) anObject = _impl->GetObject(theObject->GetStudyID(), anEntry, false);
2349 if (anObject.IsNull()) return;
2350 _impl->RemoveObject(anObject);
2355 //=================================================================================
2356 // function : GetStringFromIOR()
2357 // purpose : returns a string that represents a 'GEOM::GEOM_Object_var'
2358 //=================================================================================
2359 char* GEOM_Gen_i::GetStringFromIOR(GEOM::GEOM_Object_ptr theObject) {
2360 return _orb->object_to_string(theObject);
2364 //=================================================================================
2365 // function : GetIORFromString()
2366 // purpose : returns a 'GEOM::GEOM_Object_var' from a string representing it
2367 //=================================================================================
2368 GEOM::GEOM_Object_ptr GEOM_Gen_i::GetIORFromString(const char* stringIOR) {
2369 GEOM::GEOM_Object_var aGeomObject;
2370 if(strcmp(stringIOR,"") != 0){
2371 CORBA::Object_var anObject = _orb->string_to_object(stringIOR);
2372 if(!CORBA::is_nil(anObject))
2373 aGeomObject = GEOM::GEOM_Object::_narrow(anObject.in());
2375 return aGeomObject._retn();
2378 //=================================================================================
2379 // function : GetObject()
2381 //=================================================================================
2382 GEOM::GEOM_Object_ptr GEOM_Gen_i::GetObject (CORBA::Long theStudyID, const char* theEntry)
2384 GEOM::GEOM_Object_var obj;
2385 Handle(GEOM_Object) handle_object = _impl->GetObject(theStudyID, (char*)theEntry);
2386 if (handle_object.IsNull()) return obj._retn();
2388 TCollection_AsciiString stringIOR = handle_object->GetIOR();
2389 if (stringIOR.Length() > 1) {
2390 CORBA::Object_var corba_object = _orb->string_to_object(stringIOR.ToCString());
2391 if (!CORBA::is_nil(corba_object)) obj = GEOM::GEOM_Object::_narrow(corba_object);
2395 GEOM::GEOM_Gen_ptr engine = _this();
2396 //transfer the reference to GEOM_Object_i
2397 GEOM_Object_i* servant = new GEOM_Object_i (_poa, engine, handle_object);
2398 PortableServer::ObjectId_var id = _poa->activate_object(servant);
2400 obj = servant->_this();
2401 CORBA::String_var objStr = _orb->object_to_string(obj);
2402 TCollection_AsciiString anAscii( (char *)objStr.in() );
2403 handle_object->SetIOR( anAscii );
2407 //=================================================================================
2408 // function : hasObjectInfo()
2409 // purpose : shows if module provides information for its objects
2410 //=================================================================================
2411 bool GEOM_Gen_i::hasObjectInfo()
2416 //=================================================================================
2417 // function : getObjectInfo()
2418 // purpose : returns an information for a given object by its entry
2419 //=================================================================================
2420 char* GEOM_Gen_i::getObjectInfo(CORBA::Long studyId, const char* entry)
2422 GEOM::GEOM_Object_var aGeomObject;
2424 CORBA::Object_var aSMObject = name_service->Resolve( "/myStudyManager" );
2425 SALOMEDS::StudyManager_var aStudyManager = SALOMEDS::StudyManager::_narrow( aSMObject );
2426 SALOMEDS::Study_var aStudy = aStudyManager->GetStudyByID( studyId );
2427 SALOMEDS::SObject_var aSObj = aStudy->FindObjectID( entry );
2428 SALOMEDS::SObject_var aResultSObj;
2429 if (aSObj->ReferencedObject(aResultSObj))
2430 aSObj = aResultSObj;
2432 SALOMEDS::GenericAttribute_var anAttr;
2433 if (!aSObj->_is_nil() && aSObj->FindAttribute(anAttr, "AttributeIOR")) {
2434 SALOMEDS::AttributeIOR_var anIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
2435 CORBA::String_var aVal = anIOR->Value();
2436 anIOR->UnRegister();
2437 CORBA::Object_var anObject = aStudy->ConvertIORToObject(aVal);
2438 aGeomObject = GEOM::GEOM_Object::_narrow(anObject);
2440 if (!aSObj->_is_nil() )
2441 aSObj->UnRegister();
2443 const char* aTypeInfo = "Object";
2444 if ( !aGeomObject->_is_nil() ) {
2445 GEOM::GEOM_IKindOfShape::shape_kind aKind;
2446 GEOM::ListOfLong_var anInts;
2447 GEOM::ListOfDouble_var aDbls;
2449 GEOM::GEOM_IMeasureOperations_var anOp = GetIMeasureOperations( studyId );
2450 aKind = anOp->KindOfShape( aGeomObject, anInts, aDbls );
2452 if ( anOp->IsDone() ) {
2454 case GEOM::GEOM_IKindOfShape::COMPOUND:
2455 aTypeInfo = "Compound";
2457 case GEOM::GEOM_IKindOfShape::COMPSOLID:
2458 aTypeInfo = "CompSolid";
2460 case GEOM::GEOM_IKindOfShape::SHELL:
2461 aTypeInfo = "Shell";
2463 case GEOM::GEOM_IKindOfShape::WIRE:
2464 if ( anInts[0] == 1 )
2465 aTypeInfo = "Closed Wire";
2466 else if ( anInts[0] == 2 )
2467 aTypeInfo = "Opened Wire";
2472 case GEOM::GEOM_IKindOfShape::SPHERE:
2473 aTypeInfo = "Sphere";
2475 case GEOM::GEOM_IKindOfShape::CYLINDER:
2476 aTypeInfo = "Cylinder";
2478 case GEOM::GEOM_IKindOfShape::BOX:
2479 case GEOM::GEOM_IKindOfShape::ROTATED_BOX:
2482 case GEOM::GEOM_IKindOfShape::TORUS:
2483 aTypeInfo = "Torus";
2485 case GEOM::GEOM_IKindOfShape::CONE:
2488 case GEOM::GEOM_IKindOfShape::POLYHEDRON:
2489 aTypeInfo = "Polyhedron";
2491 case GEOM::GEOM_IKindOfShape::SOLID:
2492 aTypeInfo = "Solid";
2495 case GEOM::GEOM_IKindOfShape::SPHERE2D:
2496 aTypeInfo = "Spherical Face";
2498 case GEOM::GEOM_IKindOfShape::CYLINDER2D:
2499 aTypeInfo = "Cylindrical Face";
2501 case GEOM::GEOM_IKindOfShape::TORUS2D:
2502 aTypeInfo = "Toroidal Face";
2504 case GEOM::GEOM_IKindOfShape::CONE2D:
2505 aTypeInfo = "Conical Face";
2507 case GEOM::GEOM_IKindOfShape::DISK_CIRCLE:
2510 case GEOM::GEOM_IKindOfShape::DISK_ELLIPSE:
2511 aTypeInfo = "Elliptical Face";
2513 case GEOM::GEOM_IKindOfShape::POLYGON:
2514 aTypeInfo = "Polygon";
2516 case GEOM::GEOM_IKindOfShape::PLANE:
2517 aTypeInfo = "Plane";
2519 case GEOM::GEOM_IKindOfShape::PLANAR:
2520 aTypeInfo = "Planar Face";
2522 case GEOM::GEOM_IKindOfShape::FACE:
2526 case GEOM::GEOM_IKindOfShape::CIRCLE:
2527 aTypeInfo = "Circle";
2529 case GEOM::GEOM_IKindOfShape::ARC_CIRCLE:
2530 aTypeInfo = "Arc Circle";
2532 case GEOM::GEOM_IKindOfShape::ELLIPSE:
2533 aTypeInfo = "Ellipse";
2535 case GEOM::GEOM_IKindOfShape::ARC_ELLIPSE:
2536 aTypeInfo = "Arc Ellipse";
2538 case GEOM::GEOM_IKindOfShape::LINE:
2541 case GEOM::GEOM_IKindOfShape::SEGMENT:
2542 aTypeInfo = "Segment";
2544 case GEOM::GEOM_IKindOfShape::EDGE:
2547 case GEOM::GEOM_IKindOfShape::VERTEX:
2548 aTypeInfo = "Vertex";
2556 char* anInfo = new char[strlen("Module ") + strlen(ComponentDataType()) + strlen(", ") + strlen(aTypeInfo) + 3];
2557 sprintf(anInfo, "Module %s, %s", ComponentDataType(), aTypeInfo);
2559 char* ret = CORBA::string_dup(anInfo);
2564 // Version information
2565 char* GEOM_Gen_i::getVersion()
2567 #if GEOM_DEVELOPMENT
2568 return CORBA::string_dup(GEOM_VERSION_STR"dev");
2570 return CORBA::string_dup(GEOM_VERSION_STR);
2574 //=====================================================================================
2576 //=====================================================================================
2581 PortableServer::ObjectId* GEOMEngine_factory(CORBA::ORB*, PortableServer::POA*, PortableServer::ObjectId*, const char*, const char*);
2585 PortableServer::ObjectId* GEOMEngine_factory(CORBA::ORB_ptr orb,
2586 PortableServer::POA_ptr poa,
2587 PortableServer::ObjectId* contId,
2588 const char* instanceName,
2589 const char* interfaceName)
2591 GEOM_Gen_i* myGEOM_Gen_i = new GEOM_Gen_i(orb, poa, contId, instanceName, interfaceName);
2592 return myGEOM_Gen_i->getId();