Salome HOME
Merge from V6_main_20120808 08Aug12
[modules/geom.git] / src / GEOM_I / GEOM_Object_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 #include <Standard_OStream.hxx>
24
25 #include <GEOM_Object_i.hh>
26 #include <GEOM_ISubShape.hxx>
27 #include <GEOMImpl_Types.hxx>
28
29 #include "utilities.h"
30 #include <fstream>
31 #include <sstream>
32
33 #include <OpUtil.hxx>
34 #include <Utils_ExceptHandlers.hxx>
35 #include <TDF_Tool.hxx>
36 #include <TDF_Label.hxx>
37 #include <TCollection_AsciiString.hxx>
38
39 #include <BRepTools_ShapeSet.hxx>
40 #include <BRepTools.hxx>
41 #include <TopAbs.hxx>
42 #include <TopoDS_Iterator.hxx>
43
44 #ifdef WNT
45 #pragma warning( disable:4786 )
46 #endif
47
48 //=============================================================================
49 /*!
50  *   constructor:
51  */
52 //=============================================================================
53
54 GEOM_Object_i::GEOM_Object_i (PortableServer::POA_ptr thePOA, GEOM::GEOM_Gen_ptr theEngine,
55                               Handle(GEOM_Object) theImpl)
56 : SALOME::GenericObj_i( thePOA ), _engine(theEngine), _impl(theImpl)
57 {
58 }
59
60 //=============================================================================
61 /*!
62  *  destructor
63  */
64 //=============================================================================
65
66 GEOM_Object_i::~GEOM_Object_i()
67 {
68   MESSAGE("GEOM_Object_i::~GEOM_Object_i");
69   GEOM_Engine::GetEngine()->RemoveObject(_impl);
70 }
71
72
73 //=============================================================================
74 /*!
75  *  GetEntry
76  */
77 //=============================================================================
78 char* GEOM_Object_i::GetEntry()
79 {
80   const TDF_Label& aLabel = _impl->GetEntry();
81   TCollection_AsciiString anEntry;
82   TDF_Tool::Entry(aLabel, anEntry);
83   const char* anEntstr = anEntry.ToCString();
84   return CORBA::string_dup(anEntstr);
85 }
86
87 //=============================================================================
88 /*!
89  *  GetStudyID
90  */
91 //=============================================================================
92 CORBA::Long GEOM_Object_i::GetStudyID()
93 {
94    return _impl->GetDocID();
95 }
96
97
98 //=============================================================================
99 /*!
100  *  GetType
101  */
102 //=============================================================================
103 CORBA::Long GEOM_Object_i::GetType()
104 {
105   return _impl->GetType();
106 }
107
108 //=============================================================================
109 /*!
110  *  GetShapeType
111  */
112 //=============================================================================
113 GEOM::shape_type GEOM_Object_i::GetShapeType()
114 {
115   TopoDS_Shape _geom = _impl->GetValue();
116   if(_geom.IsNull()) return GEOM::SHAPE;
117   return (GEOM::shape_type)_geom.ShapeType();
118 }
119
120 //=============================================================================
121 /*!
122  *  GetTopologyType
123  */
124 //=============================================================================
125 GEOM::shape_type GEOM_Object_i::GetTopologyType()
126 {
127   TopoDS_Shape shape = _impl->GetValue();
128   if(shape.IsNull()) return GEOM::SHAPE;
129
130   if ( shape.ShapeType() == TopAbs_COMPOUND || shape.ShapeType() == TopAbs_COMPSOLID ) {
131     TopoDS_Shape shape_i;
132     TopoDS_Iterator It (shape, Standard_True, Standard_False);
133     for (; It.More(); It.Next()) {
134       if ( !shape_i.IsNull() ) return (GEOM::shape_type)shape.ShapeType();
135       shape_i = It.Value();
136     }
137     if ( !shape_i.IsNull() )
138       return (GEOM::shape_type) shape_i.ShapeType();
139   }
140
141   return (GEOM::shape_type)shape.ShapeType();
142 }
143
144 static GEOM::shape_type getMinMaxShapeType( const TopoDS_Shape& shape, bool ismin )
145 {
146   if ( shape.IsNull() )
147     return GEOM::SHAPE;
148
149   GEOM::shape_type ret = (GEOM::shape_type)shape.ShapeType();
150
151   if ( shape.ShapeType() == TopAbs_COMPOUND || shape.ShapeType() == TopAbs_COMPSOLID ) {
152     TopoDS_Iterator it(shape, Standard_True, Standard_False);
153     for (; it.More(); it.Next()) {
154       TopoDS_Shape sub_shape = it.Value();
155       if ( sub_shape.IsNull() ) continue;
156       GEOM::shape_type stype = (GEOM::shape_type)getMinMaxShapeType( sub_shape, ismin );
157       if ( stype == GEOM::SHAPE ) continue;
158       if ( ismin && stype > ret )
159         ret = stype;
160       else if ( !ismin && ( ret < GEOM::SOLID || stype < ret ) )
161         ret = stype;
162     }
163   }
164
165   return ret;
166 }
167
168 //=============================================================================
169 /*!
170  *  GetMinShapeType
171  */
172 //=============================================================================
173 GEOM::shape_type GEOM_Object_i::GetMinShapeType()
174 {
175   return getMinMaxShapeType( _impl->GetValue(), true );
176 }
177
178 //=============================================================================
179 /*!
180  *  GetMaxShapeType
181  */
182 //=============================================================================
183 GEOM::shape_type GEOM_Object_i::GetMaxShapeType()
184 {
185   return getMinMaxShapeType( _impl->GetValue(), false );
186 }
187
188 //=============================================================================
189 /*!
190  *  SetName
191  */
192 //=============================================================================
193 void GEOM_Object_i::SetName(const char* theName)
194 {
195   _impl->SetName(theName);
196 }
197
198
199 //=============================================================================
200 /*!
201  *  GetName
202  */
203 //=============================================================================
204 char* GEOM_Object_i::GetName()
205 {
206   char* aName = _impl->GetName();
207   if (aName)
208     return aName; // this is already copy of pointer (see implementation of _impl)
209   return strdup("");
210 }
211
212 //=============================================================================
213 /*!
214  *  SetColor
215  */
216 //=============================================================================
217 void GEOM_Object_i::SetColor(const SALOMEDS::Color& theColor)
218 {
219   ::GEOM_Object::Color aColor;
220   aColor.R = theColor.R;
221   aColor.G = theColor.G;
222   aColor.B = theColor.B;
223   _impl->SetColor(aColor);
224 }
225
226
227 //=============================================================================
228 /*!
229  *  GetColor
230  */
231 //=============================================================================
232 SALOMEDS::Color GEOM_Object_i::GetColor()
233 {
234   SALOMEDS::Color aColor;
235   aColor.R = _impl->GetColor().R;
236   aColor.G = _impl->GetColor().G;
237   aColor.B = _impl->GetColor().B;
238   return aColor;
239 }
240
241
242 //=============================================================================
243 /*!
244  *  SetAutoColor
245  */
246 //=============================================================================
247 void GEOM_Object_i::SetAutoColor(CORBA::Boolean theAutoColor)
248 {
249   _impl->SetAutoColor(theAutoColor);
250 }
251
252
253 //=============================================================================
254 /*!
255  *  GetAutoColor
256  */
257 //=============================================================================
258 CORBA::Boolean GEOM_Object_i::GetAutoColor()
259 {
260   return _impl->GetAutoColor();
261 }
262
263
264 //=============================================================================
265 /*!
266  *  SetMarkerStd
267  */
268 //=============================================================================
269 void GEOM_Object_i::SetMarkerStd(GEOM::marker_type theType, GEOM::marker_size theSize)
270 {
271   if ( theType == GEOM::MT_NONE || theSize == GEOM::MS_NONE ) {
272     _impl->UnsetMarker();
273   }
274   else {
275     Aspect_TypeOfMarker aType = (Aspect_TypeOfMarker)( (int)theType-1 );
276     double aSize = ((int)theSize+1)*0.5;
277     _impl->SetMarkerStd( aType, aSize );
278   }
279 }
280
281
282 //=============================================================================
283 /*!
284  *  SetMarkerTexture
285  */
286 //=============================================================================
287 void GEOM_Object_i::SetMarkerTexture(CORBA::Long theTextureId)
288 {
289   _impl->SetMarkerTexture( theTextureId );
290 }
291
292
293 //=============================================================================
294 /*!
295  *  GetMarkerType
296  */
297 //=============================================================================
298 GEOM::marker_type GEOM_Object_i::GetMarkerType()
299 {
300   return (GEOM::marker_type)( (int)_impl->GetMarkerType()+1 );
301 }
302
303
304 //=============================================================================
305 /*!
306  *  GetMarkerSize
307  */
308 //=============================================================================
309 GEOM::marker_size GEOM_Object_i::GetMarkerSize()
310 {
311   int aSize = (int)( _impl->GetMarkerSize()/0.5 ) - 1;
312   return aSize < GEOM::MS_10 || aSize > GEOM::MS_70 ? GEOM::MS_NONE : (GEOM::marker_size)aSize;
313 }
314
315
316 //=============================================================================
317 /*!
318  *  GetMarkerTexture
319  */
320 //=============================================================================
321 CORBA::Long GEOM_Object_i::GetMarkerTexture()
322 {
323   return _impl->GetMarkerTexture();
324 }
325
326
327 //=============================================================================
328 /*!
329  *  SetStudyEntry
330  */
331 //=============================================================================
332 void GEOM_Object_i::SetStudyEntry(const char* theEntry)
333 {
334   _impl->SetAuxData(theEntry);
335 }
336
337
338 //=============================================================================
339 /*!
340  *  GetStudyEntry
341  */
342 //=============================================================================
343 char* GEOM_Object_i::GetStudyEntry()
344 {
345   TCollection_AsciiString anEntry = _impl->GetAuxData();
346   if(!anEntry.IsEmpty()) return CORBA::string_dup(anEntry.ToCString());
347   return CORBA::string_dup("");
348 }
349
350
351 //=============================================================================
352 /*!
353  *  GetDependency
354  */
355 //=============================================================================
356 GEOM::ListOfGO* GEOM_Object_i::GetDependency()
357 {
358   GEOM::ListOfGO_var aList = new GEOM::ListOfGO();
359   aList->length(0);
360
361   Handle(TColStd_HSequenceOfTransient) aSeq = _impl->GetAllDependency();
362   if (aSeq.IsNull()) return aList._retn();
363   int aLength = aSeq->Length();
364   if (aLength == 0) return aList._retn();
365
366   aList->length(aLength);
367
368   TCollection_AsciiString anEntry;
369
370   for (int i = 1; i<=aLength; i++) {
371     Handle(GEOM_Object) anObj = Handle(GEOM_Object)::DownCast(aSeq->Value(i));
372     if (anObj.IsNull()) continue;
373     TDF_Tool::Entry(anObj->GetEntry(), anEntry);
374     GEOM::GEOM_Object_var obj = _engine->GetObject(anObj->GetDocID(), anEntry.ToCString());
375     aList[i-1] = obj;
376   }
377
378   return aList._retn();
379 }
380
381 //=============================================================================
382 /*!
383  * GetLastDependency
384  */
385 //=============================================================================
386 GEOM::ListOfGO* GEOM_Object_i::GetLastDependency()
387 {
388   GEOM::ListOfGO_var aList = new GEOM::ListOfGO();
389   aList->length(0);
390
391   Handle(TColStd_HSequenceOfTransient) aSeq = _impl->GetLastDependency();
392   if (aSeq.IsNull()) return aList._retn();
393   int aLength = aSeq->Length();
394   if (aLength == 0) return aList._retn();
395
396   aList->length(aLength);
397
398   TCollection_AsciiString anEntry;
399
400   for (int i = 1; i<=aLength; i++) {
401      Handle(GEOM_Object) anObj = Handle(GEOM_Object)::DownCast(aSeq->Value(i));
402      if (anObj.IsNull()) continue;
403      TDF_Tool::Entry(anObj->GetEntry(), anEntry);
404      GEOM::GEOM_Object_var obj = GEOM::GEOM_Object::_duplicate(_engine->GetObject(anObj->GetDocID(), anEntry.ToCString()));
405      aList[i-1] = obj;
406   }
407
408   return aList._retn();
409 }
410
411 //=================================================================================
412 // function : GetShapeStream
413 // Transfer resulting shape to client as sequence of bytes
414 //client can extract shape from stream using BrepTools::Read function
415 //=================================================================================
416 SALOMEDS::TMPFile* GEOM_Object_i::GetShapeStream()
417 {
418   TopoDS_Shape aShape = _impl->GetValue();
419
420   if(aShape.IsNull()) return NULL;
421
422   std::ostringstream streamShape;
423   //Write TopoDS_Shape in ASCII format to the stream
424   BRepTools::Write(aShape, streamShape);
425   //Returns the number of bytes that have been stored in the stream's buffer.
426   int size = streamShape.str().size();
427   //Allocate octect buffer of required size
428   CORBA::Octet* OctetBuf = SALOMEDS::TMPFile::allocbuf(size);
429   //Copy ostrstream content to the octect buffer
430   memcpy(OctetBuf, streamShape.str().c_str(), size);
431   //Create and return TMPFile
432   SALOMEDS::TMPFile_var SeqFile = new SALOMEDS::TMPFile(size,size,OctetBuf,1);
433   return SeqFile._retn();
434 }
435
436
437 //=======================================================================
438 //function : getShape
439 //purpose  : return the TopoDS_Shape when client and servant are colocated, be careful
440 //=======================================================================
441 CORBA::LongLong GEOM_Object_i::getShape() {
442   _geom = _impl->GetValue();
443   return ((CORBA::LongLong)(&_geom));
444 }
445
446 //=============================================================================
447 /*!
448  *  GetSubShapeIndices
449  */
450 //=============================================================================
451 GEOM::ListOfLong* GEOM_Object_i::GetSubShapeIndices()
452 {
453   GEOM::ListOfLong_var anIndices = new GEOM::ListOfLong;
454
455   if(!_impl->IsMainShape()) {
456     Handle(GEOM_Function) aFunction = _impl->GetLastFunction(); //Get Sub-shape function (always the first (and last)  one)
457     if(aFunction.IsNull()) return anIndices._retn();
458     GEOM_ISubShape ISS(aFunction);
459     Handle(TColStd_HArray1OfInteger) anArray = ISS.GetIndices();
460     if(anArray.IsNull() || anArray->Length() < 1) return anIndices._retn();
461     anIndices->length(anArray->Length());
462     for(int i=1; i<=anArray->Length(); i++) anIndices[i-1] = anArray->Value(i);
463   }
464   else {
465     anIndices->length(0);
466   }
467
468   return anIndices._retn();
469 }
470
471
472 //=============================================================================
473 /*!
474  *  GetMainShape
475  */
476 //=============================================================================
477 GEOM::GEOM_Object_ptr GEOM_Object_i::GetMainShape()
478 {
479   GEOM::GEOM_Object_var obj;
480   if(!_impl->IsMainShape()) {
481     Handle(GEOM_Function) aFunction = _impl->GetFunction(1); //Get Sub-shape function (always the first (and last)  one)
482     if(aFunction.IsNull()) return obj._retn();
483     GEOM_ISubShape ISS(aFunction);
484
485     aFunction = ISS.GetMainShape();
486     if(aFunction.IsNull()) return obj._retn();
487     TDF_Label aLabel  = aFunction->GetOwnerEntry();
488     if(aLabel.IsNull()) return obj._retn();
489     TCollection_AsciiString anEntry;
490     TDF_Tool::Entry(aLabel, anEntry);
491     return _engine->GetObject(_impl->GetDocID(), anEntry.ToCString());
492   }
493
494   return obj._retn();
495 }
496
497 bool GEOM_Object_i::IsShape()
498 {
499   return !_impl->GetValue().IsNull() && _impl->GetType() != GEOM_MARKER;
500 }
501
502 bool GEOM_Object_i::IsSame(GEOM::GEOM_Object_ptr other)
503 {
504   TopoDS_Shape thisShape  = _impl->GetValue();
505   TopoDS_Shape otherShape;
506   if ( !CORBA::is_nil( other ) ) {
507     Handle(GEOM_Object) otherObject = GEOM_Engine::GetEngine()->GetObject( other->GetStudyID(), other->GetEntry(), false );
508     if ( !otherObject.IsNull() )
509       otherShape = otherObject->GetValue();
510   }
511   return thisShape.IsSame( otherShape );
512 }
513
514 void GEOM_Object_i::SetParameters(const char* theParameters)
515 {
516   _impl->SetParameters((char*)theParameters);
517 }
518
519 char* GEOM_Object_i::GetParameters()
520 {
521   return CORBA::string_dup(_impl->GetParameters().ToCString());
522 }
523