Salome HOME
Merge from V6_main 11/02/2013
[modules/geom.git] / src / GEOM_I / GEOM_Gen_i.cc
1 // Copyright (C) 2007-2012  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
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.
10 //
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.
15 //
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
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 #ifdef WNT
24 #pragma warning( disable:4786 )
25 #endif
26
27 #include <Standard_Stream.hxx>
28
29 #include "GEOM_Gen_i.hh"
30 #include "GEOM_Object_i.hh"
31 #include "GEOM_version.h"
32
33 #include <set>
34 #include <sstream>
35
36 #include "Utils_CorbaException.hxx"
37 #include "OpUtil.hxx"
38 #include "Utils_ExceptHandlers.hxx"
39 #include "utilities.h"
40
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"
48
49 // Cascade headers
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>
61 #include <TopExp.hxx>
62 #include <OSD.hxx>
63
64 #include <SALOMEDS_Tool.hxx>
65
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)
76 {
77   _thisObj = this;
78   _id = _poa->activate_object(_thisObj);
79   name_service = new SALOME_NamingService(_orb);
80
81   _impl = new ::GEOMImpl_Gen;
82
83   //PAL10867: disable signals catching with "noexcepthandler" option
84   char* envNoCatchSignals = getenv("NOT_INTERCEPT_SIGNALS");
85   if (!envNoCatchSignals || !atoi(envNoCatchSignals))
86   {
87     //work around PAL12004, PAL12628
88     //OSD::SetSignal( true );
89     bool raiseFPE;
90 #ifdef _DEBUG_
91     raiseFPE = true;
92     char* envDisableFPE = getenv("DISABLE_FPE");
93     if (envDisableFPE && atoi(envDisableFPE))
94       raiseFPE = false;
95 #else
96     raiseFPE = false;
97 #endif
98     OSD::SetSignal( raiseFPE );
99   }
100 }
101
102 //============================================================================
103 // function : ~GEOM_Gen_i()
104 // purpose  : destructor
105 //============================================================================
106 GEOM_Gen_i::~GEOM_Gen_i() {
107   delete name_service;
108   delete _impl;
109 }
110
111
112 //============================================================================
113 // function : IORToLocalPersistentID()
114 // purpose  :
115 //============================================================================
116 char* GEOM_Gen_i::IORToLocalPersistentID(SALOMEDS::SObject_ptr theSObject,
117                                          const char* IORString,
118                                          CORBA::Boolean isMultiFile,
119                                          CORBA::Boolean isASCII)
120 {
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());
124   }
125   return 0;
126 }
127
128
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)
139 {
140   SALOMEDS::Study_var aStudy = theSObject->GetStudy();
141
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());
146
147   CORBA::String_var aPersRefString = _orb->object_to_string(obj);
148   return CORBA::string_dup(aPersRefString);
149 }
150
151 //============================================================================
152 // function : CanPublishInStudy
153 // purpose  :
154 //============================================================================
155 bool GEOM_Gen_i::CanPublishInStudy(CORBA::Object_ptr theIOR)
156 {
157   GEOM::GEOM_Object_var anObject = GEOM::GEOM_Object::_narrow(theIOR);
158   return !(anObject->_is_nil());
159 }
160
161
162 //============================================================================
163 // function : PublishInStudy
164 // purpose  :
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)
170 {
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;
176
177   SALOMEDS::GenericAttribute_var anAttr;
178   SALOMEDS::StudyBuilder_var     aStudyBuilder = theStudy->NewBuilder();
179
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");
186     aName->UnRegister();
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());
192   }
193   if (aFather->_is_nil()) return aResultSO;
194
195   if (CORBA::is_nil(theSObject)) {
196     aResultSO = aStudyBuilder->NewObject(aFather);
197   } else {
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);
201   }
202   CORBA::String_var aGeomObjIOR = _orb->object_to_string(theObject);
203   aResultSO->SetAttrString("AttributeIOR",aGeomObjIOR);
204
205   TCollection_AsciiString aShapeName("Shape_");
206
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 ) ) {
211     case TopAbs_VERTEX:
212       aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_GROUP_PNT" );
213       aShapeName = "Group_Of_Vertices_";
214       break;
215     case TopAbs_EDGE:
216       aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_GROUP_EDGE");
217       aShapeName = "Group_Of_Edges_";
218       break;
219     case TopAbs_FACE:
220       aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_GROUP_FACE");
221       aShapeName = "Group_Of_Faces_";
222       break;
223     case TopAbs_SOLID:
224       aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_GROUP_SOLID");
225       aShapeName = "Group_Of_Solids_";
226       break;
227     }
228   } else if ( mytype == GEOM_MARKER ) {
229     aResultSO->SetAttrString("AttributePixMap","ICON_OBJBROWSER_LCS");
230     aShapeName = "LocalCS_";
231   } else if ( mytype > ADVANCED_BASE ) {
232     char buf[20];
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_";
237   } else {
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_";
263   }
264   }
265   //if (strlen(theName) == 0) aShapeName += TCollection_AsciiString(aResultSO->Tag());
266   //else aShapeName = TCollection_AsciiString(CORBA::string_dup(theName));
267
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();
278   }
279   entry = aMainSh->GetEntry();
280   Handle(GEOM_Object) anObj = _impl->GetObject(aMainSh->GetStudyID(), entry);
281   TDF_Label aMainLbl = anObj->GetFunction(1)->GetNamingEntry();
282
283   // check all named shapes using iterator
284   TDF_ChildIDIterator anIt (aMainLbl, TNaming_NamedShape::GetID(), Standard_True);
285
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();
296         HasName = true;
297       }
298     }
299   }
300   // END: try to find existed name for current shape
301
302   if (!HasName) {
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;
311       do {
312         aNewShapeName = aShapeName + TCollection_AsciiString(++i);
313         obj = theStudy->FindObject( aNewShapeName.ToCString() );
314       }
315       while ( !obj->_is_nil() );
316       aShapeName = aNewShapeName;
317     }
318     else // MOST PROBABLY CALLED FROM GEOM GUI (ALREADY WITH VALID NAME)
319       aShapeName = TCollection_AsciiString((char*)theName);
320   }
321
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());
325
326   //Set a name of the added shape
327   aResultSO->SetAttrString("AttributeName",aShapeName.ToCString());
328
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());
338       if(j != m-1)
339         aVars += ":";
340     }
341     if(i != n-1)
342       aVars += "|";
343   }
344   aResultSO->SetAttrString("AttributeString",aVars.ToCString());
345
346   aFather->UnRegister();
347
348   //Set a name of the GEOM object
349   aShape->SetName(aShapeName.ToCString());
350
351   return aResultSO._retn();
352 }
353
354
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)
366 {
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) {
371     // create a group
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);
378     // add named objects
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);
400     }
401   }
402 }
403
404
405 //============================================================================
406 // function : PublishNamedShapesInStudy
407 // purpose  :
408 //============================================================================
409 GEOM::ListOfGO* GEOM_Gen_i::
410             PublishNamedShapesInStudy(SALOMEDS::Study_ptr theStudy,
411                                       //SALOMEDS::SObject_ptr theSObject,
412                                       CORBA::Object_ptr theObject)
413 {
414   //Unexpect aCatch(SALOME_SalomeException);
415   GEOM::ListOfGO_var aResList = new GEOM::ListOfGO;
416
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();
420
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();
425
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) {
441         SolidSeqS.Append(S);
442         SolidSeqN.Append(aName->Get());
443       }
444       else if(S.ShapeType()==TopAbs_FACE) {
445         FaceSeqS.Append(S);
446         FaceSeqN.Append(aName->Get());
447       }
448       else if(S.ShapeType()==TopAbs_EDGE) {
449         EdgeSeqS.Append(S);
450         EdgeSeqN.Append(aName->Get());
451       }
452       else if(S.ShapeType()==TopAbs_VERTEX) {
453         VertSeqS.Append(S);
454         VertSeqN.Append(aName->Get());
455       }
456     }
457   }
458
459   TopTools_IndexedMapOfShape anIndices;
460   TopExp::MapShapes(MainSh, anIndices);
461
462   CreateAndPublishGroup(theStudy, theMainShape, anIndices, SolidSeqS, SolidSeqN,
463                         "Group_Of_Named_Solids", aResList);
464
465   CreateAndPublishGroup(theStudy, theMainShape, anIndices, FaceSeqS, FaceSeqN,
466                         "Group_Of_Named_Faces", aResList);
467
468   CreateAndPublishGroup(theStudy, theMainShape, anIndices, EdgeSeqS, EdgeSeqN,
469                         "Group_Of_Named_Edges", aResList);
470
471   CreateAndPublishGroup(theStudy, theMainShape, anIndices, VertSeqS, VertSeqN,
472                         "Group_Of_Named_Vertices", aResList);
473
474   return aResList._retn();
475 }
476
477
478 //============================================================================
479 // function : Save()
480 // purpose  : save OCAF/Geom document
481 //============================================================================
482 SALOMEDS::TMPFile* GEOM_Gen_i::Save(SALOMEDS::SComponent_ptr theComponent,
483                                     const char* theURL,
484                                     bool isMultiFile) {
485   SALOMEDS::TMPFile_var aStreamFile;
486   // Get a temporary directory to store a file
487   std::string aTmpDir = (isMultiFile)?theURL:SALOMEDS_Tool::GetTmpDir();
488
489   // OCCT BUG: cannot save a document (in current folder)
490   // if directory name is empty
491   if (aTmpDir.size() == 0) {
492 #ifdef WNT
493     aTmpDir = ".\\";
494 #else
495     aTmpDir = "./";
496 #endif
497   }
498
499   // Create a list to store names of created files
500   SALOMEDS::ListOfFileNames_var aSeq = new SALOMEDS::ListOfFileNames;
501   aSeq->length(1);
502   // Prepare a file name to open
503   TCollection_AsciiString aNameWithExt("");
504   if (isMultiFile)
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);
517
518   // Return the created byte stream
519   return aStreamFile._retn();
520 }
521
522
523 //============================================================================
524 // function : SaveASCII()
525 // purpose  :
526 //============================================================================
527 SALOMEDS::TMPFile* GEOM_Gen_i::SaveASCII(SALOMEDS::SComponent_ptr theComponent,
528                                          const char* theURL,
529                                          bool isMultiFile) {
530   SALOMEDS::TMPFile_var aStreamFile = Save(theComponent, theURL, isMultiFile);
531   return aStreamFile._retn();
532 }
533
534
535 //============================================================================
536 // function : Load()
537 // purpose  :
538 //============================================================================
539 CORBA::Boolean GEOM_Gen_i::Load(SALOMEDS::SComponent_ptr theComponent,
540                                 const SALOMEDS::TMPFile& theStream,
541                                 const char* theURL,
542                                 bool isMultiFile) {
543
544   if (theStream.length() <= 9) {
545     MESSAGE("The TMPFile is too short : " << theStream.length() << " bytes ");
546     return false;
547   }
548
549   // Get a temporary directory for a file
550   std::string aTmpDir = isMultiFile?theURL:SALOMEDS_Tool::GetTmpDir();
551
552   // OCCT BUG: cannot load a document (from current folder)
553   // if directory name is empty
554   if (aTmpDir.size() == 0) {
555 #ifdef WNT
556     aTmpDir = ".\\";
557 #else
558     aTmpDir = "./";
559 #endif
560   }
561
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);
565
566   // Prepare a file name to open
567   TCollection_AsciiString aNameWithExt("");
568   if (isMultiFile)
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);
573
574   // Open document
575   if (!_impl->Load(theComponent->GetStudy()->StudyId(),(char*) aFullName.ToCString())) return false;
576
577   // Remove the created file and tmp directory
578   if (!isMultiFile) SALOMEDS_Tool::RemoveTemporaryFiles(aTmpDir.c_str(), aSeq.in(), true);
579
580   return true;
581 }
582
583
584 //============================================================================
585 // function : LoadASCII()
586 // purpose  :
587 //============================================================================
588 CORBA::Boolean GEOM_Gen_i::LoadASCII(SALOMEDS::SComponent_ptr theComponent,
589                                      const SALOMEDS::TMPFile& theStream,
590                                      const char* theURL,
591                                      bool isMultiFile) {
592   return Load(theComponent, theStream, theURL, isMultiFile);
593 }
594
595
596 //============================================================================
597 // function : Close()
598 // purpose  :
599 //============================================================================
600 void GEOM_Gen_i::Close(SALOMEDS::SComponent_ptr theComponent)
601 {
602   SALOMEDS::Study_var aStudy= theComponent->GetStudy();
603   _impl->Close(aStudy->StudyId());
604 }
605
606 //============================================================================
607 // function : CanCopy()
608 // purpose  :
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;
614
615   SALOMEDS::AttributeIOR_var anIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
616
617   CORBA::String_var aString=anIOR->Value();
618   anIOR->UnRegister();
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;
623   return true;
624 }
625
626 //============================================================================
627 // function : CopyFrom()
628 // purpose  :
629 //============================================================================
630 SALOMEDS::TMPFile* GEOM_Gen_i::CopyFrom(SALOMEDS::SObject_ptr theObject, CORBA::Long& theObjectID)
631 {
632   // Declare a sequence of the byte to store the copied object
633   SALOMEDS::TMPFile_var aStreamFile = new SALOMEDS::TMPFile;
634
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();
641
642   aStreamFile = anObject->GetShapeStream();
643
644   // Assign an ID  the type of  GEOM_Object
645   theObjectID = anObject->GetType();
646
647   // Return created TMPFile
648   return aStreamFile._retn();
649 }
650
651 //============================================================================
652 // function : CanPaste()
653 // purpose  :
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;
659   return true;
660 }
661
662 //============================================================================
663 // function : PasteInto()
664 // purpose  :
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();
672
673   // Retrieve a TopoDS_Shape from byte stream
674   TopoDS_Shape aTopology;
675   std::istringstream aStreamedBrep((char*) &theStream[0]);
676   BRep_Builder aBuilder;
677   try {
678     BRepTools::Read(aTopology, aStreamedBrep, aBuilder);
679   } catch (Standard_Failure) {
680     return false;
681   }
682
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);
688
689
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);
694
695   TCollection_AsciiString anEntry;
696   TDF_Tool::Entry(anObj->GetEntry(), anEntry);
697   GEOM::GEOM_Object_var obj = GetObject(anObj->GetDocID(), anEntry.ToCString());
698
699   //Set the study entry of the published GEOM_Object
700   obj->SetStudyEntry(aNewSO->GetID());
701
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());
707   anIOR->UnRegister();
708
709   // Return the created in the Study SObject
710   return aNewSO._retn();
711 }
712
713 //============================================================================
714 // function : ComponentDataType()
715 // purpose  :
716 //============================================================================
717 char* GEOM_Gen_i::ComponentDataType()
718 {
719   return CORBA::string_dup("GEOM");
720 }
721
722 //============================================================================
723 // function : AddInStudy
724 // purpose  :
725 //============================================================================
726 SALOMEDS::SObject_ptr GEOM_Gen_i::AddInStudy (SALOMEDS::Study_ptr theStudy,
727                                               GEOM::GEOM_Object_ptr theObject,
728                                               const char* theName,
729                                               GEOM::GEOM_Object_ptr theFather)
730 {
731   SALOMEDS::SObject_var aResultSO;
732   if(theObject->_is_nil() || theStudy->_is_nil()) return aResultSO;
733
734   SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
735   CORBA::String_var IOR;
736
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);
744   }
745
746   aResultSO = PublishInStudy(theStudy, aResultSO, theObject, theName);
747   if(aResultSO->_is_nil()) return aResultSO._retn();
748
749   GEOM::ListOfGO_var aList = theObject->GetDependency();
750   Standard_Integer aLength = aList->length();
751   if(aLength < 1) return aResultSO._retn();
752
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;
763     aPrevID = anID.in();
764     SALOMEDS::SObject_var aSubSO = aStudyBuilder->NewObject(aResultSO);
765     aStudyBuilder->Addreference(aSubSO, aSO);
766     aSO->UnRegister();
767     aSubSO->UnRegister();
768   }
769
770   return aResultSO._retn();
771 }
772
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)
784 {
785   GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
786   if (CORBA::is_nil(theStudy) || CORBA::is_nil(theObject))
787     return aParts._retn();
788
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();
796
797   aParts = RestoreSubShapes(theStudy, theObject, aSO, theArgs,
798                             theFindMethod, theInheritFirstArg, theAddPrefix);
799   if (!CORBA::is_nil(aSO)) aSO->UnRegister();
800   return aParts._retn();
801 }
802
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)
814 {
815   GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
816   if (CORBA::is_nil(theStudy) || CORBA::is_nil(theObject))
817     return aParts._retn();
818
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();
826
827   aParts = RestoreGivenSubShapes(theStudy, theObject, aSO, theArgs,
828                                  theFindMethod, theInheritFirstArg, theAddPrefix);
829   if (!CORBA::is_nil(aSO)) aSO->UnRegister();
830   return aParts._retn();
831 }
832
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)
844 {
845   GEOM::ListOfGO_var aParts = new GEOM::ListOfGO;
846   if (CORBA::is_nil(theStudy) || CORBA::is_nil(theSObject))
847     return aParts._retn();
848
849   SALOMEDS::GenericAttribute_var anAttr;
850   if (!theSObject->FindAttribute(anAttr, "AttributeIOR"))
851     return aParts._retn();
852
853   SALOMEDS::AttributeIOR_var anAttrIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
854   CORBA::String_var anIORso = anAttrIOR->Value();
855
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();
860
861   aParts = RestoreSubShapes(theStudy, anO, theSObject, theArgs,
862                             theFindMethod, theInheritFirstArg, theAddPrefix);
863   return aParts._retn();
864 }
865
866 //============================================================================
867 // function : addToListOfGO
868 // purpose  : static local function
869 //============================================================================
870 static void addToListOfGO( GEOM::GEOM_Object_ptr theObject,
871                            GEOM::ListOfGO& theList )
872 {
873   const int oldLen = theList.length();
874   theList.length(oldLen + 1);
875   theList[ oldLen ] = GEOM::GEOM_Object::_duplicate( theObject );
876 }
877
878 //============================================================================
879 // function : addToListOfGO
880 // purpose  : static local function
881 //============================================================================
882 static void addToListOfGO( const GEOM::ListOfGO& theSrcList,
883                            GEOM::ListOfGO& theTrgList )
884 {
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 ] );
890 }
891
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)
904 {
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();
910
911   // For Dump Python (mantis issue 0020768)
912   GEOM::ListOfGO_var anOutArgs = new GEOM::ListOfGO;
913
914   // Arguments to be published
915   GEOM::ListOfGO_var aList;
916
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();
921   if (aLength > 0) {
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] );
926     }
927   }
928   else {
929     // Get all arguments
930     aList = theObject->GetDependency();
931     aLength = aList->length();
932     nbArgsActual = aLength;
933   }
934
935   if (aLength < 1)
936     return aParts._retn();
937
938   if (theInheritFirstArg || (nbArgsActual == 1)) {
939     // Do not publish argument's reflection,
940     // but only reconstruct its published sub-shapes
941
942     CORBA::String_var anIOR = _orb->object_to_string(aList[0]);
943     SALOMEDS::SObject_var anArgSO = theStudy->FindObjectIOR(anIOR.in());
944
945     // remember restored objects for Python Dump
946     addToListOfGO(aList[0], anOutArgs);
947
948     aParts = RestoreSubShapesOneLevel(theStudy, anArgSO, theSObject, theObject,
949                                       anOutArgs, theFindMethod, theAddPrefix);
950
951     // set the color of the transformed shape to the color of initial shape
952     theObject->SetColor(aList[0]->GetColor());
953     // set the texture
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());
958     }
959
960     anArgSO->UnRegister();
961   }
962   else {
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());
967
968     PortableServer::Servant aServant = _poa->reference_to_servant(aTrsfOp.in());
969     GEOM_ITransformOperations_i*      aTrsfOpSv = dynamic_cast<GEOM_ITransformOperations_i*>(aServant);
970
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++)
975     {
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)) {
982           anArgName = "arg_";
983           anArgName += TCollection_AsciiString(i);
984         }
985         else {
986           anArgName = anArgSO->GetName();
987         }
988
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:
993           {
994             // Use GetInPlace
995             aSubO = aShapesOp->GetInPlace(theObject, anArgO);
996           }
997           break;
998         case GEOM::FSM_MultiTransformed:
999           {
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);
1008             }
1009             /*
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()) {
1019                 case TRANSLATE_1D:
1020                   {
1021                     GEOMImpl_ITranslate aTI (anOFun);
1022                     aMultiArgShape = GEOMImpl_ITransformOperations::TranslateShape1D(anArgOShape, &aTI);
1023                     //anArgOMulti = aTrsfOpSv->Translate1D(anArgO, , , );
1024                   }
1025                   break;
1026                 case TRANSLATE_2D:
1027                   {
1028                     GEOMImpl_ITranslate aTI (anOFun);
1029                     aMultiArgShape = GEOMImpl_ITransformOperations::TranslateShape2D(anArgOShape, &aTI);
1030                   }
1031                   break;
1032                 case ROTATE_1D:
1033                   {
1034                     GEOMImpl_IRotate aTI (anOFun);
1035                     //aMultiArgShape = GEOMImpl_ITransformOperations::TranslateShape2D(anArgOShape, &aTI);
1036                   }
1037                   break;
1038                 case ROTATE_2D:
1039                   {
1040                     GEOMImpl_IRotate aTI (anOFun);
1041                     //aMultiArgShape = GEOMImpl_ITransformOperations::TranslateShape2D(anArgOShape, &aTI);
1042                   }
1043                   break;
1044                 default:
1045                   {}
1046                 }
1047                 GEOM::GEOM_Object_var anArgOMulti = (aMultiArgShape); // TODO
1048                 Handle(GEOM_Function) anArgOMultiFun = anArgOMulti->GetLastFunction();
1049                 anArgOMultiFun->SetDescription("");
1050                 aSubO = aShapesOp->GetInPlace(theObject, anArgOMulti);
1051               }
1052             }
1053             */
1054           }
1055           break;
1056         case GEOM::FSM_Transformed:
1057           {
1058             // transformation, cannot use GetInPlace, operate with indices
1059             GEOM::ListOfLong_var anIDs = anArgO->GetSubShapeIndices();
1060             if (anIDs->length() > 1) {
1061               // group
1062               aSubO = aGroupOp->CreateGroup(theObject, aGroupOp->GetType(anArgO));
1063               if (!CORBA::is_nil(aSubO))
1064                 aGroupOp->UnionIDs(aSubO, anIDs);
1065             }
1066             else if (anIDs->length() > 0) {
1067               // single sub-shape
1068               aSubO = aShapesOp->GetSubShape(theObject, anIDs[0]);
1069             }
1070           }
1071           break;
1072         case GEOM::FSM_GetSame:
1073           {
1074             // Use GetSame
1075             aSubO = aShapesOp->GetSame(theObject, anArgO);
1076           }
1077           break;
1078         case GEOM::FSM_GetShapesOnShape:
1079           {
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);
1083           }
1084           break;
1085         case GEOM::FSM_GetInPlaceByHistory:
1086           {
1087             // Use GetInPlaceByHistory
1088             aSubO = aShapesOp->GetInPlaceByHistory(theObject, anArgO);
1089           }
1090           break;
1091         default:
1092           {}
1093         }
1094
1095         if (!CORBA::is_nil(aSubO)) {
1096           // remember restored objects for Python Dump
1097           addToListOfGO(anArgO, anOutArgs);
1098
1099           // add to parts list
1100           addToListOfGO( aSubO, aParts );
1101
1102           // Publish the sub-shape
1103           SALOMEDS::SObject_var aSubSO;
1104           if (!CORBA::is_nil(theSObject)) {
1105             TCollection_AsciiString aSubName;
1106             if (theAddPrefix) {
1107               aSubName = "from_";
1108             }
1109             aSubName += anArgName;
1110             aSubSO = aStudyBuilder->NewObject(theSObject);
1111             aSubSO = PublishInStudy(theStudy, aSubSO, aSubO, aSubName.ToCString());
1112             // Restore color
1113             aSubO->SetColor(anArgO->GetColor());
1114             // set the texture
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());
1119             }
1120           }
1121
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);
1129             else
1130               aSubParts = RestoreSubShapesOneLevel(theStudy, anArgSO, aSubSO,
1131                                                    aSubO, anOutArgs, theFindMethod, theAddPrefix);
1132             // add to parts list
1133             addToListOfGO( aSubParts, aParts );
1134           }
1135         }
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);
1141
1142             // Restore published sub-shapes of the argument
1143             GEOM::ListOfGO_var aSubParts =
1144               RestoreSubShapesOneLevel(theStudy, anArgSO, aSubSO,
1145                                        theObject, anOutArgs, theFindMethod, theAddPrefix);
1146
1147             // add to parts list
1148             addToListOfGO( aSubParts, aParts );
1149
1150             if (aSubParts->length() > 0) {
1151               // remember restored objects for Python Dump
1152               addToListOfGO(anArgO, anOutArgs);
1153
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 );
1160               }
1161               else {
1162                 aSubO = aSubParts[0];
1163               }
1164               if (!CORBA::is_nil(aSubO) && !CORBA::is_nil(aSubSO)) {
1165                 // Publish the sub-shape
1166                 TCollection_AsciiString aSubName;
1167                 if (theAddPrefix) {
1168                   aSubName = "from_parts_of_";
1169                 }
1170                 aSubName += anArgName;
1171                 aSubSO = PublishInStudy(theStudy, aSubSO, aSubO, aSubName.ToCString());
1172                 // Restore color
1173                 aSubO->SetColor(anArgO->GetColor());
1174                 // set the texture
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());
1179                 }
1180               }
1181             }
1182             else if (!CORBA::is_nil(aSubSO)) {
1183               // remove created aSubSO, because no parts have been found
1184               aStudyBuilder->RemoveObject(aSubSO);
1185             }
1186           }
1187         } // try to build from published parts
1188         anArgSO->UnRegister();
1189       }
1190     } // process arguments
1191   }
1192   std::set<std::string> anObjEntryMap;
1193   GEOM::ListOfGO_var aResParts = new GEOM::ListOfGO;
1194   int nbRes = 0;
1195   int nb = aParts->length();
1196   aResParts->length(nb);
1197   if (nb > 0)
1198   {
1199     Handle(GEOM_Object) aMainObj = _impl->GetObject(theObject->GetStudyID(), theObject->GetEntry());
1200     Handle(GEOM_Function) aFunction = aMainObj->GetLastFunction();
1201     GEOM::TPythonDump pd (aFunction, true);
1202     pd <<"[";
1203     int i = 0, j = 0;
1204     for ( ; i < nb; i++ )
1205     {
1206       GEOM::GEOM_Object_var anObj = GEOM::GEOM_Object::_duplicate( aParts[ i ] );
1207       if (CORBA::is_nil(anObj))
1208         continue;
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( "" );
1219       if ( j > 0 )
1220         pd << ", ";
1221       pd << aGeomObj;
1222       j++;
1223     }
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++ )
1228     {
1229       //GEOM::GEOM_Object_var anObj = theArgs[ i ];
1230       GEOM::GEOM_Object_var anObj = anOutArgs[ i ];
1231       if (CORBA::is_nil(anObj))
1232         continue;
1233       Handle(GEOM_Object) aGeomObj = _impl->GetObject(anObj->GetStudyID(), anObj->GetEntry());
1234       if ( j > 0 )
1235         pd << ", ";
1236       pd << aGeomObj;
1237       j++;
1238     }
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:
1252     default:
1253       pd << "FSM_GetInPlaceByHistory"; break;
1254     }
1255     pd << ", " << theInheritFirstArg << ", " << theAddPrefix << ")";
1256   }
1257   aResParts->length(nbRes);
1258   return aResParts._retn();
1259 }
1260
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)
1272 {
1273   int i = 0;
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();
1279
1280   SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
1281
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());
1286
1287   PortableServer::Servant aServant = _poa->reference_to_servant(aTrsfOp.in());
1288   GEOM_ITransformOperations_i*      aTrsfOpSv = dynamic_cast<GEOM_ITransformOperations_i*>(aServant);
1289
1290   // Reconstruct published sub-shapes
1291   SALOMEDS::ChildIterator_var it = theStudy->NewChildIterator(theOldSO);
1292
1293   int aLen = 0;
1294   for (it->Init(); it->More(); it->Next()) {
1295     aLen++;
1296   }
1297   aParts->length(aLen);
1298
1299   for (it->Init(); it->More(); it->Next()) {
1300     SALOMEDS::SObject_var anOldSubSO = it->Value();
1301
1302     TCollection_AsciiString anArgName = anOldSubSO->GetName();
1303
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:
1314           {
1315             // Use GetInPlace
1316             aNewSubO = aShapesOp->GetInPlace(theNewO, anOldSubO);
1317           }
1318           break;
1319         case GEOM::FSM_MultiTransformed:
1320           {
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);
1329             }
1330           }
1331           break;
1332         case GEOM::FSM_Transformed:
1333           {
1334             // transformation, cannot use GetInPlace, operate with indices
1335             GEOM::ListOfLong_var anIDs = anOldSubO->GetSubShapeIndices();
1336             if (anIDs->length() > 1) {
1337               // group
1338               aNewSubO = aGroupOp->CreateGroup(theNewO, aGroupOp->GetType(anOldSubO));
1339               if (!CORBA::is_nil(aNewSubO))
1340                 aGroupOp->UnionIDs(aNewSubO, anIDs);
1341             }
1342             else {
1343               // single sub-shape
1344               aNewSubO = aShapesOp->GetSubShape(theNewO, anIDs[0]);
1345             }
1346           }
1347           break;
1348         case GEOM::FSM_GetSame:
1349           {
1350             // Use GetSame
1351             aNewSubO = aShapesOp->GetSame(theNewO, anOldSubO);
1352           }
1353           break;
1354         case GEOM::FSM_GetShapesOnShape:
1355           {
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);
1359           }
1360           break;
1361         case GEOM::FSM_GetInPlaceByHistory:
1362           {
1363             // Use GetInPlaceByHistory
1364             aNewSubO = aShapesOp->GetInPlaceByHistory(theNewO, anOldSubO);
1365           }
1366           break;
1367         default:
1368           {}
1369         }
1370
1371         if (!CORBA::is_nil(aNewSubO)) {
1372           // remember restored objects for Python Dump
1373           addToListOfGO(anOldSubO, theOutArgs);
1374
1375           // add the part to the list
1376           aParts[i] = aNewSubO;
1377           i++;
1378           // add to parts list
1379           addToListOfGO( aNewSubO, aNewParts );
1380
1381           SALOMEDS::SObject_var aNewSubSO;
1382           if (!CORBA::is_nil(theNewSO)) {
1383               // Publish the sub-shape
1384             TCollection_AsciiString aSubName;
1385             if (theAddPrefix) {
1386               aSubName = "from_";
1387             }
1388             aSubName += anArgName;
1389             aNewSubSO = aStudyBuilder->NewObject(theNewSO);
1390             aNewSubSO = PublishInStudy(theStudy, aNewSubSO, aNewSubO, aSubName.ToCString());
1391             // Restore color
1392             aNewSubO->SetColor(anOldSubO->GetColor());
1393             // set the texture
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());
1398             }
1399           }
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);
1406           else
1407             aSubParts = RestoreSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO,
1408                                                  aNewSubO, theOutArgs, theFindMethod, theAddPrefix);
1409           // add to parts list
1410           addToListOfGO( aSubParts, aNewParts );
1411         }
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);
1416
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 );
1423
1424           if (aSubParts->length() > 0) {
1425             // remember restored objects for Python Dump
1426             addToListOfGO(anOldSubO, theOutArgs);
1427
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 );
1434             }
1435             else {
1436               aNewSubO = aSubParts[0];
1437             }
1438
1439             if (!CORBA::is_nil(aNewSubO)) {
1440               // add the part to the list
1441               aSubParts[i] = aNewSubO;
1442               i++;
1443
1444               // Publish the sub-shape
1445               if (!CORBA::is_nil(aNewSubSO)) {
1446                 TCollection_AsciiString aSubName;
1447                 if (theAddPrefix) {
1448                   aSubName = "from_parts_of_";
1449                 }
1450                 aSubName += anArgName;
1451                 aNewSubSO = PublishInStudy(theStudy, aNewSubSO, aNewSubO, aSubName.ToCString());
1452                 // Restore color
1453                 aNewSubO->SetColor(anOldSubO->GetColor());
1454                 // set the texture
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());
1459                 }
1460               }
1461             }
1462           }
1463           else if (!CORBA::is_nil(aNewSubSO)) {
1464             // remove created aSubSO, because no parts have been found
1465             aStudyBuilder->RemoveObject(aNewSubSO);
1466           }
1467         } // try to build from published parts
1468       }
1469     }
1470   } // iterate on published sub-shapes
1471
1472   aParts->length(i);
1473   // add to parts list
1474   addToListOfGO( aNewParts, aParts );
1475   return aParts._retn();
1476 }
1477
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)
1492 {
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();
1498
1499   // If theArgs list is empty, nothing to do
1500   Standard_Integer aLength = theArgs.length();
1501   if (aLength == 0)
1502     return aParts._retn();
1503
1504   // Get all arguments
1505   GEOM::ListOfGO_var anOpArgsList = theObject->GetDependency();
1506   Standard_Integer nbArgsActual = anOpArgsList->length();
1507
1508   // If anOpArgsList list is empty, nothing to do
1509   if (nbArgsActual == 0)
1510     return aParts._retn();
1511
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());
1517   }
1518
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);
1523
1524   int k = 0;
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]);
1529       k++;
1530     }
1531   }
1532   nbArgsActual = k;
1533   //aList->length(nbArgsActual);
1534
1535   if (nbArgsActual < 1)
1536     return aParts._retn();
1537
1538   if (theInheritFirstArg || (nbArgsActual == 1)) {
1539     // Do not publish argument's reflection,
1540     // but only reconstruct its published sub-shapes
1541
1542     CORBA::String_var anIOR = _orb->object_to_string(aList[0]);
1543     SALOMEDS::SObject_var anArgSO = theStudy->FindObjectIOR(anIOR.in());
1544
1545     aParts = RestoreGivenSubShapesOneLevel(theStudy, anArgSO, theSObject, theObject,
1546                                            anArgs, theFindMethod, theAddPrefix);
1547
1548     // set the color of the transformed shape to the color of initial shape
1549     theObject->SetColor(aList[0]->GetColor());
1550     // set the texture
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());
1555     }
1556
1557     anArgSO->UnRegister();
1558   }
1559   else {
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());
1564
1565     PortableServer::Servant aServant = _poa->reference_to_servant(aTrsfOp.in());
1566     GEOM_ITransformOperations_i*      aTrsfOpSv = dynamic_cast<GEOM_ITransformOperations_i*>(aServant);
1567
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++)
1572     {
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)) {
1579           anArgName = "arg_";
1580           anArgName += TCollection_AsciiString(i);
1581         }
1582         else {
1583           anArgName = anArgSO->GetName();
1584         }
1585
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:
1590           {
1591             // Use GetInPlace
1592             aSubO = aShapesOp->GetInPlace(theObject, anArgO);
1593           }
1594           break;
1595         case GEOM::FSM_MultiTransformed:
1596           {
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);
1605             }
1606           }
1607           break;
1608         case GEOM::FSM_Transformed:
1609           {
1610             // transformation, cannot use GetInPlace, operate with indices
1611             GEOM::ListOfLong_var anIDs = anArgO->GetSubShapeIndices();
1612             if (anIDs->length() > 1) {
1613               // group
1614               aSubO = aGroupOp->CreateGroup(theObject, aGroupOp->GetType(anArgO));
1615               if (!CORBA::is_nil(aSubO))
1616                 aGroupOp->UnionIDs(aSubO, anIDs);
1617             }
1618             else if (anIDs->length() > 0) {
1619               // single sub-shape
1620               aSubO = aShapesOp->GetSubShape(theObject, anIDs[0]);
1621             }
1622           }
1623           break;
1624         case GEOM::FSM_GetSame:
1625           {
1626             // Use GetSame
1627             aSubO = aShapesOp->GetSame(theObject, anArgO);
1628           }
1629           break;
1630         case GEOM::FSM_GetShapesOnShape:
1631           {
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);
1635           }
1636           break;
1637         case GEOM::FSM_GetInPlaceByHistory:
1638           {
1639             // Use GetInPlaceByHistory
1640             aSubO = aShapesOp->GetInPlaceByHistory(theObject, anArgO);
1641           }
1642           break;
1643         default:
1644           {}
1645         }
1646
1647         if (!CORBA::is_nil(aSubO)) {
1648           // add to parts list
1649           addToListOfGO( aSubO, aParts );
1650
1651           // Publish the sub-shape
1652           SALOMEDS::SObject_var aSubSO;
1653           if (!CORBA::is_nil(theSObject)) {
1654             TCollection_AsciiString aSubName;
1655             if (theAddPrefix) {
1656               aSubName = "from_";
1657             }
1658             aSubName += anArgName;
1659             aSubSO = aStudyBuilder->NewObject(theSObject);
1660             aSubSO = PublishInStudy(theStudy, aSubSO, aSubO, aSubName.ToCString());
1661             // Restore color
1662             aSubO->SetColor(anArgO->GetColor());
1663             // set the texture
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());
1668             }
1669           }
1670
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);
1678             else
1679               aSubParts = RestoreGivenSubShapesOneLevel(theStudy, anArgSO, aSubSO,
1680                                                         aSubO, anArgs, theFindMethod, theAddPrefix);
1681             // add to parts list
1682             addToListOfGO( aSubParts, aParts );
1683           }
1684         }
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);
1690
1691             // Restore published sub-shapes of the argument
1692             GEOM::ListOfGO_var aSubParts =
1693               RestoreGivenSubShapesOneLevel(theStudy, anArgSO, aSubSO,
1694                                             theObject, anArgs, theFindMethod, theAddPrefix);
1695
1696             // add to parts list
1697             addToListOfGO( aSubParts, aParts );
1698
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 );
1706               }
1707               else {
1708                 aSubO = aSubParts[0];
1709               }
1710               if (!CORBA::is_nil(aSubO) && !CORBA::is_nil(aSubSO)) {
1711                 // Publish the sub-shape
1712                 TCollection_AsciiString aSubName;
1713                 if (theAddPrefix) {
1714                   aSubName = "from_parts_of_";
1715                 }
1716                 aSubName += anArgName;
1717                 aSubSO = PublishInStudy(theStudy, aSubSO, aSubO, aSubName.ToCString());
1718                 // Restore color
1719                 aSubO->SetColor(anArgO->GetColor());
1720                 // set the texture
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());
1725                 }
1726               }
1727             }
1728             else if (!CORBA::is_nil(aSubSO)) {
1729               // remove created aSubSO, because no parts have been found
1730               aStudyBuilder->RemoveObject(aSubSO);
1731             }
1732           }
1733         } // try to build from published parts
1734         anArgSO->UnRegister();
1735       }
1736     } // process arguments
1737   }
1738   std::set<std::string> anObjEntryMap;
1739   GEOM::ListOfGO_var aResParts = new GEOM::ListOfGO;
1740   int nbRes = 0;
1741   int nb = aParts->length();
1742   aResParts->length(nb);
1743   if (nb > 0)
1744   {
1745     Handle(GEOM_Object) aMainObj = _impl->GetObject(theObject->GetStudyID(), theObject->GetEntry());
1746     Handle(GEOM_Function) aFunction = aMainObj->GetLastFunction();
1747     GEOM::TPythonDump pd (aFunction, true);
1748     pd <<"[";
1749     int i = 0, j = 0;
1750     for ( ; i < nb; i++ )
1751     {
1752       GEOM::GEOM_Object_var anObj = GEOM::GEOM_Object::_duplicate( aParts[ i ] );
1753       if (CORBA::is_nil(anObj))
1754         continue;
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( "" );
1765       if ( j > 0 )
1766         pd << ", ";
1767       pd << aGeomObj;
1768       j++;
1769     }
1770     pd <<"]" << " = geompy.RestoreGivenSubShapes(" << aMainObj << ", " << "[";
1771     i = 0; nb = theArgs.length(); j = 0;
1772     for ( ; i < nb; i++ )
1773     {
1774       GEOM::GEOM_Object_var anObj = theArgs[ i ];
1775       if (CORBA::is_nil(anObj))
1776         continue;
1777       Handle(GEOM_Object) aGeomObj = _impl->GetObject(anObj->GetStudyID(), anObj->GetEntry());
1778       if ( j > 0 )
1779         pd << ", ";
1780       pd << aGeomObj;
1781       j++;
1782     }
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:
1796     default:
1797       pd << "FSM_GetInPlaceByHistory"; break;
1798     }
1799     pd << ", " << theInheritFirstArg << ", " << theAddPrefix << ")";
1800   }
1801   aResParts->length(nbRes);
1802   return aResParts._retn();
1803 }
1804
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)
1816 {
1817   int i = 0;
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();
1823
1824   SALOMEDS::StudyBuilder_var aStudyBuilder = theStudy->NewBuilder();
1825
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());
1830
1831   PortableServer::Servant aServant = _poa->reference_to_servant(aTrsfOp.in());
1832   GEOM_ITransformOperations_i*      aTrsfOpSv = dynamic_cast<GEOM_ITransformOperations_i*>(aServant);
1833
1834   // Reconstruct published sub-shapes
1835   SALOMEDS::ChildIterator_var it = theStudy->NewChildIterator(theOldSO);
1836
1837   int aLen = 0;
1838   for (it->Init(); it->More(); it->Next()) {
1839     aLen++;
1840   }
1841   aParts->length(aLen);
1842
1843   for (it->Init(); it->More(); it->Next()) {
1844     SALOMEDS::SObject_var anOldSubSO = it->Value();
1845
1846     TCollection_AsciiString anArgName = anOldSubSO->GetName();
1847
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()));
1853
1854       bool okToContinue = false;
1855
1856       if (!CORBA::is_nil(anOldSubO)) {
1857         CORBA::String_var anEntry = anOldSubO->GetEntry();
1858         okToContinue = theArgs.count(anEntry.in());
1859       }
1860
1861       if (okToContinue) {
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:
1866           {
1867             // Use GetInPlace
1868             aNewSubO = aShapesOp->GetInPlace(theNewO, anOldSubO);
1869           }
1870           break;
1871         case GEOM::FSM_MultiTransformed:
1872           {
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);
1881             }
1882           }
1883           break;
1884         case GEOM::FSM_Transformed:
1885           {
1886             // transformation, cannot use GetInPlace, operate with indices
1887             GEOM::ListOfLong_var anIDs = anOldSubO->GetSubShapeIndices();
1888             if (anIDs->length() > 1) {
1889               // group
1890               aNewSubO = aGroupOp->CreateGroup(theNewO, aGroupOp->GetType(anOldSubO));
1891               if (!CORBA::is_nil(aNewSubO))
1892                 aGroupOp->UnionIDs(aNewSubO, anIDs);
1893             }
1894             else {
1895               // single sub-shape
1896               aNewSubO = aShapesOp->GetSubShape(theNewO, anIDs[0]);
1897             }
1898           }
1899           break;
1900         case GEOM::FSM_GetSame:
1901           {
1902             // Use GetSame
1903             aNewSubO = aShapesOp->GetSame(theNewO, anOldSubO);
1904           }
1905           break;
1906         case GEOM::FSM_GetShapesOnShape:
1907           {
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);
1911           }
1912           break;
1913         case GEOM::FSM_GetInPlaceByHistory:
1914           {
1915             // Use GetInPlaceByHistory
1916             aNewSubO = aShapesOp->GetInPlaceByHistory(theNewO, anOldSubO);
1917           }
1918           break;
1919         default:
1920           {}
1921         }
1922
1923         if (!CORBA::is_nil(aNewSubO)) {
1924           // add the part to the list
1925           aParts[i] = aNewSubO;
1926           i++;
1927           // add to parts list
1928           addToListOfGO( aNewSubO, aNewParts );
1929
1930           SALOMEDS::SObject_var aNewSubSO;
1931           if (!CORBA::is_nil(theNewSO)) {
1932               // Publish the sub-shape
1933             TCollection_AsciiString aSubName;
1934             if (theAddPrefix) {
1935               aSubName = "from_";
1936             }
1937             aSubName += anArgName;
1938             aNewSubSO = aStudyBuilder->NewObject(theNewSO);
1939             aNewSubSO = PublishInStudy(theStudy, aNewSubSO, aNewSubO, aSubName.ToCString());
1940             // Restore color
1941             aNewSubO->SetColor(anOldSubO->GetColor());
1942             // set the texture
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());
1947             }
1948           }
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);
1955           else
1956             aSubParts = RestoreGivenSubShapesOneLevel(theStudy, anOldSubSO, aNewSubSO,
1957                                                       aNewSubO, theArgs, theFindMethod, theAddPrefix);
1958           // add to parts list
1959           addToListOfGO( aSubParts, aNewParts );
1960         }
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);
1965
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 );
1972
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 );
1980             }
1981             else {
1982               aNewSubO = aSubParts[0];
1983             }
1984
1985             if (!CORBA::is_nil(aNewSubO)) {
1986               // add the part to the list
1987               aSubParts[i] = aNewSubO;
1988               i++;
1989
1990               // Publish the sub-shape
1991               if (!CORBA::is_nil(aNewSubSO)) {
1992                 TCollection_AsciiString aSubName;
1993                 if (theAddPrefix) {
1994                   aSubName = "from_parts_of_";
1995                 }
1996                 aSubName += anArgName;
1997                 aNewSubSO = PublishInStudy(theStudy, aNewSubSO, aNewSubO, aSubName.ToCString());
1998                 // Restore color
1999                 aNewSubO->SetColor(anOldSubO->GetColor());
2000                 // set the texture
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());
2005                 }
2006               }
2007             }
2008           }
2009           else if (!CORBA::is_nil(aNewSubSO)) {
2010             // remove created aSubSO, because no parts have been found
2011             aStudyBuilder->RemoveObject(aNewSubSO);
2012           }
2013         } // try to build from published parts
2014       }
2015     }
2016   } // iterate on published sub-shapes
2017
2018   aParts->length(i);
2019   // add to parts list
2020   addToListOfGO( aNewParts, aParts );
2021   return aParts._retn();
2022 }
2023
2024 //============================================================================
2025 // function : register()
2026 // purpose  : register 'name' in 'name_service'
2027 //============================================================================
2028 void GEOM_Gen_i::register_name(char * name)
2029 {
2030   GEOM::GEOM_Gen_var g = _this();
2031   name_service->Register(g, name);
2032 }
2033
2034 //============================================================================
2035 // function : Undo
2036 // purpose  :
2037 //============================================================================
2038 void GEOM_Gen_i::Undo(CORBA::Long theStudyID)
2039 {
2040   _impl->Undo(theStudyID);
2041 }
2042
2043 //============================================================================
2044 // function : Redo
2045 // purpose  :
2046 //============================================================================
2047 void GEOM_Gen_i::Redo(CORBA::Long theStudyID)
2048 {
2049   _impl->Redo(theStudyID);
2050 }
2051
2052 //============================================================================
2053 // function : GetIBasicOperations
2054 // purpose  :
2055 //============================================================================
2056 GEOM::GEOM_IBasicOperations_ptr GEOM_Gen_i::GetIBasicOperations(CORBA::Long theStudyID)
2057      throw ( SALOME::SALOME_Exception )
2058 {
2059   Unexpect aCatch(SALOME_SalomeException);
2060   MESSAGE( "GEOM_Gen_i::GetIBasicOperations" );
2061
2062   GEOM::GEOM_Gen_ptr engine = _this();
2063
2064   //transfer reference on engine
2065   GEOM_IBasicOperations_i* aServant =
2066     new GEOM_IBasicOperations_i(_poa, engine, _impl->GetIBasicOperations(theStudyID));
2067
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();
2072 }
2073
2074 //============================================================================
2075 // function : GetITransformOperations
2076 // purpose  :
2077 //============================================================================
2078 GEOM::GEOM_ITransformOperations_ptr GEOM_Gen_i::GetITransformOperations(CORBA::Long theStudyID)
2079      throw ( SALOME::SALOME_Exception )
2080 {
2081   Unexpect aCatch(SALOME_SalomeException);
2082   MESSAGE( "GEOM_Gen_i::GetITransformOperations" );
2083
2084   GEOM::GEOM_Gen_ptr engine = _this();
2085
2086   GEOM_ITransformOperations_i* aServant =
2087     new GEOM_ITransformOperations_i(_poa, engine, _impl->GetITransformOperations(theStudyID));
2088
2089   // activate the CORBA servant
2090   GEOM::GEOM_ITransformOperations_var operations = aServant->_this();
2091   return operations._retn();
2092 }
2093
2094 //============================================================================
2095 // function : GetI3DPrimOperations
2096 // purpose  :
2097 //============================================================================
2098 GEOM::GEOM_I3DPrimOperations_ptr GEOM_Gen_i::GetI3DPrimOperations(CORBA::Long theStudyID)
2099      throw ( SALOME::SALOME_Exception )
2100 {
2101   Unexpect aCatch(SALOME_SalomeException);
2102   MESSAGE( "GEOM_Gen_i::GetI3DPrimOperations" );
2103
2104   GEOM::GEOM_Gen_ptr engine = _this();
2105
2106   GEOM_I3DPrimOperations_i* aServant =
2107     new GEOM_I3DPrimOperations_i(_poa, engine, _impl->GetI3DPrimOperations(theStudyID));
2108   PortableServer::ObjectId_var id = _poa->activate_object(aServant);
2109
2110   // activate the CORBA servant
2111   GEOM::GEOM_I3DPrimOperations_var operations = aServant->_this();
2112   return operations._retn();
2113 }
2114
2115 //============================================================================
2116 // function : GetIShapesOperations
2117 // purpose  :
2118 //============================================================================
2119 GEOM::GEOM_IShapesOperations_ptr GEOM_Gen_i::GetIShapesOperations(CORBA::Long theStudyID)
2120      throw ( SALOME::SALOME_Exception )
2121 {
2122   Unexpect aCatch(SALOME_SalomeException);
2123   MESSAGE( "GEOM_Gen_i::GetIShapesOperations" );
2124
2125   GEOM::GEOM_Gen_ptr engine = _this();
2126
2127   GEOM_IShapesOperations_i* aServant =
2128     new GEOM_IShapesOperations_i(_poa, engine, _impl->GetIShapesOperations(theStudyID));
2129
2130   // activate the CORBA servant
2131   GEOM::GEOM_IShapesOperations_var operations = aServant->_this();
2132   return operations._retn();
2133 }
2134
2135 //============================================================================
2136 // function : GetIBlocksOperations
2137 // purpose  :
2138 //============================================================================
2139 GEOM::GEOM_IBlocksOperations_ptr GEOM_Gen_i::GetIBlocksOperations(CORBA::Long theStudyID)
2140      throw ( SALOME::SALOME_Exception )
2141 {
2142   Unexpect aCatch(SALOME_SalomeException);
2143   MESSAGE( "GEOM_Gen_i::GetIBlocksOperations" );
2144
2145   GEOM::GEOM_Gen_ptr engine = _this();
2146
2147   GEOM_IBlocksOperations_i* aServant =
2148     new GEOM_IBlocksOperations_i(_poa, engine, _impl->GetIBlocksOperations(theStudyID));
2149
2150   // activate the CORBA servant
2151   GEOM::GEOM_IBlocksOperations_var operations = aServant->_this();
2152   return operations._retn();
2153 }
2154
2155 //============================================================================
2156 // function : GetIBooleanOperations
2157 // purpose  :
2158 //============================================================================
2159 GEOM::GEOM_IBooleanOperations_ptr GEOM_Gen_i::GetIBooleanOperations(CORBA::Long theStudyID)
2160      throw ( SALOME::SALOME_Exception )
2161 {
2162   Unexpect aCatch(SALOME_SalomeException);
2163   MESSAGE( "GEOM_Gen_i::GetIBooleanOperations" );
2164
2165   GEOM::GEOM_Gen_ptr engine = _this();
2166
2167   GEOM_IBooleanOperations_i* aServant =
2168     new GEOM_IBooleanOperations_i(_poa, engine, _impl->GetIBooleanOperations(theStudyID));
2169
2170   // activate the CORBA servant
2171   GEOM::GEOM_IBooleanOperations_var operations = aServant->_this();
2172   return operations._retn();
2173 }
2174
2175 //============================================================================
2176 // function : GetICurvesOperations
2177 // purpose  :
2178 //============================================================================
2179 GEOM::GEOM_ICurvesOperations_ptr GEOM_Gen_i::GetICurvesOperations(CORBA::Long theStudyID)
2180      throw ( SALOME::SALOME_Exception )
2181 {
2182   Unexpect aCatch(SALOME_SalomeException);
2183   MESSAGE( "GEOM_Gen_i::GetICurvesOperations" );
2184
2185   GEOM::GEOM_Gen_ptr engine = _this();
2186
2187   GEOM_ICurvesOperations_i* aServant =
2188     new GEOM_ICurvesOperations_i(_poa, engine, _impl->GetICurvesOperations(theStudyID));
2189
2190   // activate the CORBA servant
2191   GEOM::GEOM_ICurvesOperations_var operations = aServant->_this();
2192   return operations._retn();
2193 }
2194
2195 //============================================================================
2196 // function : GetILocalOperations
2197 // purpose  :
2198 //============================================================================
2199 GEOM::GEOM_ILocalOperations_ptr GEOM_Gen_i::GetILocalOperations(CORBA::Long theStudyID)
2200      throw ( SALOME::SALOME_Exception )
2201 {
2202   Unexpect aCatch(SALOME_SalomeException);
2203   MESSAGE( "GEOM_Gen_i::GetILocalOperations" );
2204
2205   GEOM::GEOM_Gen_ptr engine = _this();
2206
2207   GEOM_ILocalOperations_i* aServant =
2208     new GEOM_ILocalOperations_i(_poa, engine, _impl->GetILocalOperations(theStudyID));
2209
2210   // activate the CORBA servant
2211   GEOM::GEOM_ILocalOperations_var operations = aServant->_this();
2212   return operations._retn();
2213 }
2214
2215 //============================================================================
2216 // function : GetIHealingOperations
2217 // purpose  :
2218 //============================================================================
2219 GEOM::GEOM_IHealingOperations_ptr GEOM_Gen_i::GetIHealingOperations(CORBA::Long theStudyID)
2220      throw ( SALOME::SALOME_Exception )
2221 {
2222   Unexpect aCatch(SALOME_SalomeException);
2223   MESSAGE( "GEOM_Gen_i::IHealingOperations" );
2224
2225   GEOM::GEOM_Gen_ptr engine = _this();
2226
2227   GEOM_IHealingOperations_i* aServant =
2228     new GEOM_IHealingOperations_i(_poa, engine, _impl->GetIHealingOperations(theStudyID));
2229
2230   // activate the CORBA servant
2231   GEOM::GEOM_IHealingOperations_var operations = aServant->_this();
2232   return operations._retn();
2233 }
2234
2235 //============================================================================
2236 // function : GetIInsertOperations
2237 // purpose  :
2238 //============================================================================
2239 GEOM::GEOM_IInsertOperations_ptr GEOM_Gen_i::GetIInsertOperations(CORBA::Long theStudyID)
2240      throw ( SALOME::SALOME_Exception )
2241 {
2242   Unexpect aCatch(SALOME_SalomeException);
2243   MESSAGE( "GEOM_Gen_i::GetIInsertOperations" );
2244
2245   GEOM::GEOM_Gen_ptr engine = _this();
2246
2247   GEOM_IInsertOperations_i* aServant =
2248     new GEOM_IInsertOperations_i(_poa, engine, _impl->GetIInsertOperations(theStudyID));
2249
2250   // activate the CORBA servant
2251   GEOM::GEOM_IInsertOperations_var operations = aServant->_this();
2252   return operations._retn();
2253 }
2254
2255 //============================================================================
2256 // function : GetIMeasureOperations
2257 // purpose  :
2258 //============================================================================
2259 GEOM::GEOM_IMeasureOperations_ptr GEOM_Gen_i::GetIMeasureOperations(CORBA::Long theStudyID)
2260      throw ( SALOME::SALOME_Exception )
2261 {
2262   Unexpect aCatch(SALOME_SalomeException);
2263   MESSAGE( "GEOM_Gen_i::GetIMeasureOperations" );
2264
2265   GEOM::GEOM_Gen_ptr engine = _this();
2266
2267   GEOM_IMeasureOperations_i* aServant =
2268     new GEOM_IMeasureOperations_i(_poa, engine, _impl->GetIMeasureOperations(theStudyID));
2269
2270   // activate the CORBA servant
2271   GEOM::GEOM_IMeasureOperations_var operations = aServant->_this();
2272   return operations._retn();
2273 }
2274
2275 //============================================================================
2276 // function : GetIGroupOperations
2277 // purpose  :
2278 //============================================================================
2279 GEOM::GEOM_IGroupOperations_ptr GEOM_Gen_i::GetIGroupOperations(CORBA::Long theStudyID)
2280      throw ( SALOME::SALOME_Exception )
2281 {
2282   Unexpect aCatch(SALOME_SalomeException);
2283   MESSAGE( "GEOM_Gen_i::GetIGroupOperations" );
2284
2285   GEOM::GEOM_Gen_ptr engine = _this();
2286
2287   GEOM_IGroupOperations_i* aServant =
2288     new GEOM_IGroupOperations_i(_poa, engine, _impl->GetIGroupOperations(theStudyID));
2289
2290   // activate the CORBA servant
2291   GEOM::GEOM_IGroupOperations_var operations = aServant->_this();
2292   return operations._retn();
2293 }
2294
2295 //============================================================================
2296 // function : GetIAdvancedOperations
2297 // purpose  :
2298 //============================================================================
2299 GEOM::GEOM_IAdvancedOperations_ptr GEOM_Gen_i::GetIAdvancedOperations(CORBA::Long theStudyID)
2300      throw ( SALOME::SALOME_Exception )
2301 {
2302   Unexpect aCatch(SALOME_SalomeException);
2303   MESSAGE( "GEOM_Gen_i::GetIAdvancedOperations" );
2304
2305   GEOM::GEOM_Gen_ptr engine = _this();
2306
2307   GEOM_IAdvancedOperations_i* aServant =
2308     new GEOM_IAdvancedOperations_i(_poa, engine, _impl->GetIAdvancedOperations(theStudyID));
2309
2310   // activate the CORBA servant
2311   GEOM::GEOM_IAdvancedOperations_var operations = aServant->_this();
2312   return operations._retn();
2313 }
2314
2315 //=============================================================================
2316 /*!
2317  *  AddSubShape
2318  */
2319 //=============================================================================
2320 GEOM::GEOM_Object_ptr GEOM_Gen_i::AddSubShape (GEOM::GEOM_Object_ptr theMainShape,
2321                                                const GEOM::ListOfLong& theIndices)
2322 {
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();
2328
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]);
2331
2332   Handle(GEOM_Object) anObject = _impl->AddSubShape(aMainShape, anArray, true);
2333   if(anObject.IsNull()) return GEOM::GEOM_Object::_nil();
2334
2335   TCollection_AsciiString anEntry;
2336   TDF_Tool::Entry(anObject->GetEntry(), anEntry);
2337   return GetObject(anObject->GetDocID(), anEntry.ToCString());
2338 }
2339
2340 //=============================================================================
2341 /*!
2342  *  RemoveObject
2343  */
2344 //=============================================================================
2345 void GEOM_Gen_i::RemoveObject(GEOM::GEOM_Object_ptr theObject)
2346 {
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);
2351   return;
2352 }
2353
2354
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);
2361 }
2362
2363
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());
2374   }
2375   return aGeomObject._retn();
2376 }
2377
2378 //=================================================================================
2379 // function : GetObject()
2380 // purpose  :
2381 //=================================================================================
2382 GEOM::GEOM_Object_ptr GEOM_Gen_i::GetObject (CORBA::Long theStudyID, const char* theEntry)
2383 {
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();
2387
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);
2392     return obj._retn();
2393   }
2394
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);
2399
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 );
2404   return obj._retn();
2405 }
2406
2407 //=================================================================================
2408 // function : hasObjectInfo()
2409 // purpose  : shows if module provides information for its objects
2410 //=================================================================================
2411 bool GEOM_Gen_i::hasObjectInfo()
2412 {
2413   return true;
2414 }
2415
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)
2421 {
2422   GEOM::GEOM_Object_var aGeomObject;
2423
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;
2431
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);
2439   }
2440   if (!aSObj->_is_nil() )
2441     aSObj->UnRegister();
2442
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;
2448
2449     GEOM::GEOM_IMeasureOperations_var anOp = GetIMeasureOperations( studyId );
2450     aKind = anOp->KindOfShape( aGeomObject, anInts, aDbls );
2451
2452     if ( anOp->IsDone() ) {
2453       switch ( aKind ) {
2454       case GEOM::GEOM_IKindOfShape::COMPOUND:
2455         aTypeInfo = "Compound";
2456         break;
2457       case GEOM::GEOM_IKindOfShape::COMPSOLID:
2458         aTypeInfo = "CompSolid";
2459         break;
2460       case GEOM::GEOM_IKindOfShape::SHELL:
2461         aTypeInfo = "Shell";
2462         break;
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";
2468         else
2469           aTypeInfo = "Wire";
2470         break;
2471         // SOLIDs
2472       case GEOM::GEOM_IKindOfShape::SPHERE:
2473         aTypeInfo = "Sphere";
2474         break;
2475       case GEOM::GEOM_IKindOfShape::CYLINDER:
2476         aTypeInfo = "Cylinder";
2477         break;
2478       case GEOM::GEOM_IKindOfShape::BOX:
2479       case GEOM::GEOM_IKindOfShape::ROTATED_BOX:
2480         aTypeInfo = "Box";
2481         break;
2482       case GEOM::GEOM_IKindOfShape::TORUS:
2483         aTypeInfo = "Torus";
2484         break;
2485       case GEOM::GEOM_IKindOfShape::CONE:
2486         aTypeInfo = "Cone";
2487         break;
2488       case GEOM::GEOM_IKindOfShape::POLYHEDRON:
2489         aTypeInfo = "Polyhedron";
2490         break;
2491       case GEOM::GEOM_IKindOfShape::SOLID:
2492         aTypeInfo = "Solid";
2493         break;
2494         // FACEs
2495       case GEOM::GEOM_IKindOfShape::SPHERE2D:
2496         aTypeInfo = "Spherical Face";
2497         break;
2498       case GEOM::GEOM_IKindOfShape::CYLINDER2D:
2499         aTypeInfo = "Cylindrical Face";
2500         break;
2501       case GEOM::GEOM_IKindOfShape::TORUS2D:
2502         aTypeInfo = "Toroidal Face";
2503         break;
2504       case GEOM::GEOM_IKindOfShape::CONE2D:
2505         aTypeInfo = "Conical Face";
2506         break;
2507       case GEOM::GEOM_IKindOfShape::DISK_CIRCLE:
2508         aTypeInfo = "Disk";
2509         break;
2510       case GEOM::GEOM_IKindOfShape::DISK_ELLIPSE:
2511         aTypeInfo = "Elliptical Face";
2512         break;
2513       case GEOM::GEOM_IKindOfShape::POLYGON:
2514         aTypeInfo = "Polygon";
2515         break;
2516       case GEOM::GEOM_IKindOfShape::PLANE:
2517         aTypeInfo = "Plane";
2518         break;
2519       case GEOM::GEOM_IKindOfShape::PLANAR:
2520         aTypeInfo = "Planar Face";
2521         break;
2522       case GEOM::GEOM_IKindOfShape::FACE:
2523         aTypeInfo = "Face";
2524         break;
2525         // EDGEs
2526       case GEOM::GEOM_IKindOfShape::CIRCLE:
2527         aTypeInfo = "Circle";
2528         break;
2529       case GEOM::GEOM_IKindOfShape::ARC_CIRCLE:
2530         aTypeInfo = "Arc Circle";
2531         break;
2532       case GEOM::GEOM_IKindOfShape::ELLIPSE:
2533         aTypeInfo = "Ellipse";
2534         break;
2535       case GEOM::GEOM_IKindOfShape::ARC_ELLIPSE:
2536         aTypeInfo = "Arc Ellipse";
2537         break;
2538       case GEOM::GEOM_IKindOfShape::LINE:
2539         aTypeInfo = "Line";
2540         break;
2541       case GEOM::GEOM_IKindOfShape::SEGMENT:
2542         aTypeInfo = "Segment";
2543         break;
2544       case GEOM::GEOM_IKindOfShape::EDGE:
2545         aTypeInfo = "Edge";
2546         break;
2547       case GEOM::GEOM_IKindOfShape::VERTEX:
2548         aTypeInfo = "Vertex";
2549         break;
2550       default:
2551         break;
2552       }
2553     }
2554   }
2555
2556   char* anInfo = new char[strlen("Module ") + strlen(ComponentDataType()) + strlen(", ") + strlen(aTypeInfo) + 3];
2557   sprintf(anInfo, "Module %s, %s", ComponentDataType(), aTypeInfo);
2558
2559   char* ret = CORBA::string_dup(anInfo);
2560   delete [] anInfo;
2561   return ret;
2562 }
2563
2564 // Version information
2565 char* GEOM_Gen_i::getVersion()
2566 {
2567 #if GEOM_DEVELOPMENT
2568   return CORBA::string_dup(GEOM_VERSION_STR"dev");
2569 #else
2570   return CORBA::string_dup(GEOM_VERSION_STR);
2571 #endif
2572 }
2573
2574 //=====================================================================================
2575 // EXPORTED METHODS
2576 //=====================================================================================
2577 extern "C"
2578 {
2579   /*
2580   GEOM_I_EXPORT
2581   PortableServer::ObjectId* GEOMEngine_factory(CORBA::ORB*, PortableServer::POA*, PortableServer::ObjectId*, const char*, const char*);
2582   */
2583
2584   GEOM_I_EXPORT
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)
2590   {
2591     GEOM_Gen_i* myGEOM_Gen_i = new GEOM_Gen_i(orb, poa, contId, instanceName, interfaceName);
2592     return myGEOM_Gen_i->getId();
2593   }
2594 }