1 // Copyright (C) 2007-2013 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 // GEOM GEOMGUI : GUI for Geometry component
24 // File : GEOMBase.cxx
25 // Author : Damien COQUERET, Open CASCADE S.A.S.
29 #include <GeometryGUI.h>
30 #include <GEOM_Client.hxx>
33 #include <GEOM_Actor.h>
34 #include <SVTK_ViewModel.h>
35 #include <SVTK_ViewWindow.h>
36 #include <OCCViewer_ViewPort3d.h>
37 #include <OCCViewer_ViewModel.h>
38 #include <OCCViewer_ViewWindow.h>
39 #include <SOCC_ViewModel.h>
42 #include <SALOME_ListIO.hxx>
43 #include <SALOME_ListIteratorOfListIO.hxx>
45 #include <SUIT_Desktop.h>
46 #include <SUIT_Session.h>
47 #include <SUIT_ViewManager.h>
48 #include <SUIT_ViewWindow.h>
49 #include <SUIT_MessageBox.h>
50 #include <SalomeApp_Application.h>
51 #include <SalomeApp_Study.h>
53 // // Open CASCADE Includes
54 #include <BRep_Tool.hxx>
55 #include <BRepAdaptor_Curve.hxx>
56 #include <BRepAdaptor_Surface.hxx>
57 #include <BRepPrimAPI_MakeCone.hxx>
59 #include <AIS_ListIteratorOfListOfInteractive.hxx>
60 #include <AIS_ListOfInteractive.hxx>
64 #include <TopTools_IndexedMapOfShape.hxx>
67 #include <Precision.hxx>
69 #include <vtkRenderer.h>
73 //=====================================================================================
74 // function : GetShapeFromIOR()
75 // purpose : Get shape data by the specified IOR
76 //=====================================================================================
77 TopoDS_Shape GEOMBase::GetShapeFromIOR( const QString& IOR )
79 GEOM::GEOM_Object_var geomObj = GEOMBase::GetObjectFromIOR( IOR );
81 GetShape( geomObj, shape, TopAbs_SHAPE );
86 //=====================================================================================
87 // function : GetIndex()
88 // purpose : Get the index of a sub-shape in a main shape : index start at 1
89 //=====================================================================================
90 int GEOMBase::GetIndex( const TopoDS_Shape& subshape, const TopoDS_Shape& shape )
93 if ( !shape.IsNull() && !subshape.IsNull() ) {
94 TopTools_IndexedMapOfShape anIndices;
95 TopExp::MapShapes( shape, anIndices );
96 if ( anIndices.Contains( subshape ) )
97 idx = anIndices.FindIndex( subshape );
103 //=======================================================================
104 // function : GetTopoFromSelection()
105 // purpose : Define tds from a single selection and retuen true
106 //=======================================================================
107 TopoDS_Shape GEOMBase::GetTopoFromSelection( const SALOME_ListIO& IObjects )
110 if ( IObjects.Extent() == 1 ){
111 Handle(SALOME_InteractiveObject) IO = IObjects.First();
112 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
113 if ( IO->hasEntry() && study ) {
114 _PTR(Study) studyDS = study->studyDS();
115 _PTR(SObject) obj( studyDS->FindObjectID( IO->getEntry() ) );
116 _PTR(GenericAttribute) anAttr;
117 if ( obj && obj->FindAttribute( anAttr, "AttributeIOR" ) ) {
118 _PTR(AttributeIOR) anIOR( anAttr );
119 shape = GetShapeFromIOR( anIOR->Value().c_str() );
126 //=======================================================================
127 // function : GetNameOfSelectedIObjects()
128 // purpose : Define the name geom++ or other name of mono or multi sel.
129 //=======================================================================
130 int GEOMBase::GetNameOfSelectedIObjects( const SALOME_ListIO& IObjects,
132 const bool shapesOnly )
135 name = ""; // clear output name
138 nbSel = IObjects.Extent();
140 Handle(SALOME_InteractiveObject) anIObj = IObjects.First();
141 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
142 if ( anIObj->hasEntry() && study ) {
143 _PTR(Study) studyDS = study->studyDS();
144 _PTR(SObject) obj( studyDS->FindObjectID( anIObj->getEntry() ) );
145 _PTR(GenericAttribute) anAttr;
146 if ( obj && obj->FindAttribute( anAttr, "AttributeName" ) ) {
147 _PTR(AttributeName) aNameAttr ( anAttr );
148 name = aNameAttr->Value().c_str();
154 GEOM::ListOfGO anObjs;
155 ConvertListOfIOInListOfGO( IObjects, anObjs, shapesOnly );
156 nbSel = anObjs.length();
158 name = GetName( anObjs[ 0 ] );
162 name = QObject::tr( "%1_objects" ).arg( nbSel );
168 //=================================================================================
169 // function : GetShapeTypeString()
170 // purpose : for a single shape
171 //=================================================================================
172 QString GEOMBase::GetShapeTypeString(const TopoDS_Shape& shape)
175 if ( !shape.IsNull() ) {
176 switch ( shape.ShapeType() ) {
177 case TopAbs_COMPOUND:
179 aTypeString = QObject::tr( "GEOM_COMPOUND" );
182 case TopAbs_COMPSOLID:
184 aTypeString = QObject::tr( "GEOM_COMPOUNDSOLID" );
189 aTypeString = QObject::tr( "GEOM_SOLID" );
194 aTypeString = QObject::tr( "GEOM_SHELL" );
199 BRepAdaptor_Surface surf( TopoDS::Face( shape ) );
200 switch ( surf.GetType() ) {
203 aTypeString = QObject::tr( "GEOM_PLANE" );
206 case GeomAbs_Cylinder:
208 aTypeString = QObject::tr( "GEOM_SURFCYLINDER" );
213 aTypeString = QObject::tr( "GEOM_SURFSPHERE" );
218 aTypeString = QObject::tr( "GEOM_SURFTORUS" );
223 aTypeString = QObject::tr( "GEOM_SURFCONE" );
228 aTypeString = QObject::tr( "GEOM_FACE" );
236 aTypeString = QObject::tr( "GEOM_WIRE" );
241 BRepAdaptor_Curve curv( TopoDS::Edge( shape ) );
242 switch ( curv.GetType() ) {
245 aTypeString = ( qAbs( curv.FirstParameter() ) >= 1E6 || qAbs( curv.LastParameter() ) >= 1E6 ) ?
246 QObject::tr( "GEOM_LINE" ) : QObject::tr( "GEOM_EDGE" );
251 aTypeString = curv.IsClosed() ? QObject::tr( "GEOM_CIRCLE" ) : QObject::tr( "GEOM_ARC" );
256 aTypeString = QObject::tr( "GEOM_EDGE" );
264 aTypeString = QObject::tr( "GEOM_VERTEX" );
269 aTypeString = QObject::tr( "GEOM_SHAPE" );
282 //=======================================================================
283 // function : ConvertIORinGEOMAISShape()
285 //=======================================================================
286 Handle(GEOM_AISShape) GEOMBase::ConvertIORinGEOMAISShape(const QString& IOR, bool onlyInActiveView)
288 Handle(GEOM_AISShape) shape;
290 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
292 _PTR(Study) studyDS = study->studyDS();
293 _PTR(SObject) obj( studyDS->FindObjectIOR( IOR.toLatin1().constData() ) );
295 QList<SUIT_ViewWindow*> views;
296 if ( onlyInActiveView )
297 views.append( SUIT_Session::session()->activeApplication()->desktop()->activeWindow() );
299 views = SUIT_Session::session()->activeApplication()->desktop()->windows();
300 foreach ( SUIT_ViewWindow* view, views ) {
301 if ( view && view->getViewManager()->getType() == OCCViewer_Viewer::Type() ) {
302 Handle(AIS_InteractiveContext) ic = ((OCCViewer_Viewer*)view->getViewManager()->getViewModel())->getAISContext();
304 AIS_ListOfInteractive displayed;
305 ic->DisplayedObjects( displayed );
306 ic->ObjectsInCollector( displayed );
308 AIS_ListIteratorOfListOfInteractive it( displayed );
309 while ( it.More() && shape.IsNull() ) {
310 if ( it.Value()->IsInstance( STANDARD_TYPE(GEOM_AISShape) ) ) {
311 Handle(GEOM_AISShape) sh = Handle(GEOM_AISShape)::DownCast( it.Value() );
312 if ( !sh.IsNull() && sh->hasIO() ) {
313 Handle(SALOME_InteractiveObject) IO = Handle(SALOME_InteractiveObject)::DownCast( sh->getIO() );
314 if ( !IO.IsNull() && IO->hasEntry() && obj->GetID() == IO->getEntry() )
321 if ( !shape.IsNull() ) break;
329 //=======================================================================
330 // function : ConvertIORinGEOMActor()
332 //=======================================================================
333 GEOM_Actor* GEOMBase::ConvertIORinGEOMActor(const QString& IOR, bool onlyInActiveView)
335 GEOM_Actor* actor = 0;
337 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
339 _PTR(Study) studyDS = study->studyDS();
340 _PTR(SObject) obj( studyDS->FindObjectIOR( IOR.toLatin1().constData() ) );
342 QList<SUIT_ViewWindow*> views;
343 if ( onlyInActiveView )
344 views.append( SUIT_Session::session()->activeApplication()->desktop()->activeWindow() );
346 views = SUIT_Session::session()->activeApplication()->desktop()->windows();
347 foreach ( SUIT_ViewWindow* view, views ) {
348 if ( view && view->getViewManager()->getType() == SVTK_Viewer::Type() ) {
349 SVTK_ViewWindow* aVTKViewWindow = dynamic_cast<SVTK_ViewWindow*>( view );
350 if( !aVTKViewWindow )
352 vtkRenderer* Renderer = aVTKViewWindow->getRenderer();
353 vtkActorCollection* theActors = Renderer->GetActors();
354 theActors->InitTraversal();
355 vtkActor* a = theActors->GetNextActor();
356 while( a && !actor ) {
357 if ( a->IsA( "GEOM_Actor" ) ) {
358 GEOM_Actor* ga = GEOM_Actor::SafeDownCast( a );
359 if ( ga && ga->hasIO() ) {
360 Handle(SALOME_InteractiveObject) IO = Handle(SALOME_InteractiveObject)::DownCast( ga->getIO() );
361 if ( !IO.IsNull() && IO->hasEntry() && obj->GetID() == IO->getEntry() )
365 a = theActors->GetNextActor();
375 //=======================================================================
376 // function : GetAIS()
378 //=======================================================================
379 Handle(AIS_InteractiveObject) GEOMBase::GetAIS( const Handle(SALOME_InteractiveObject)& IO,
380 bool onlyInActiveView, bool onlyGeom )
382 Handle(AIS_InteractiveObject) aisObject;
384 if ( !IO.IsNull() && IO->hasEntry() ) {
385 QList<SUIT_ViewWindow*> views;
386 if ( onlyInActiveView )
387 views.append( SUIT_Session::session()->activeApplication()->desktop()->activeWindow() );
389 views = SUIT_Session::session()->activeApplication()->desktop()->windows();
391 foreach ( SUIT_ViewWindow* view, views ) {
392 if ( view && view->getViewManager()->getType() == OCCViewer_Viewer::Type() ) {
393 OCCViewer_Viewer* occViewer=(OCCViewer_Viewer*)view->getViewManager()->getViewModel();
394 SOCC_Viewer* soccViewer = dynamic_cast<SOCC_Viewer*>(occViewer);
396 SOCC_Prs* occPrs = dynamic_cast<SOCC_Prs*>( soccViewer->CreatePrs( IO->getEntry() ) );
397 if ( occPrs && !occPrs->IsNull() ) {
398 AIS_ListOfInteractive shapes; occPrs->GetObjects( shapes );
399 if( !shapes.Extent() ) continue;
400 aisObject=shapes.First();
405 if ( !aisObject.IsNull() ) break;
413 //=======================================================================
414 // function : ConvertIOinGEOMAISShape()
416 //=======================================================================
417 Handle(GEOM_AISShape) GEOMBase::ConvertIOinGEOMAISShape( const Handle(SALOME_InteractiveObject)& IO, bool onlyInActiveView )
419 return Handle(GEOM_AISShape)::DownCast( GetAIS( IO, onlyInActiveView, true ) );
423 //=======================================================================
424 // function : ConvertListOfIOInListOfIOR()
426 //=======================================================================
427 QStringList GEOMBase::ConvertListOfIOInListOfIOR( const SALOME_ListIO& IObjects )
430 SALOME_ListIteratorOfListIO it( IObjects );
431 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
433 _PTR(Study) studyDS = study->studyDS();
434 for ( ; it.More(); it.Next() ) {
435 GEOM::GEOM_Object_var geomObj = ConvertIOinGEOMObject( it.Value() );
436 if ( !CORBA::is_nil( geomObj ) )
437 iors.append( GetIORFromObject( geomObj ) );
444 //=======================================================================
445 // function : ConvertIOinGEOMObject()
447 //=======================================================================
448 GEOM::GEOM_Object_ptr GEOMBase::ConvertIOinGEOMObject( const Handle(SALOME_InteractiveObject)& IO )
450 GEOM::GEOM_Object_var object;
452 if ( !IO.IsNull() && IO->hasEntry() ) {
453 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
455 _PTR(Study) studyDS = study->studyDS();
456 _PTR(SObject) obj( studyDS->FindObjectID( IO->getEntry() ) );
458 CORBA::Object_var corbaObj = GeometryGUI::ClientSObjectToObject( obj );
459 if ( !CORBA::is_nil( corbaObj ) )
460 object = GEOM::GEOM_Object::_narrow( corbaObj );
464 return object._retn();
468 //=======================================================================
469 // function : ConvertListOfIOInListOfGO()
471 //=======================================================================
472 void GEOMBase::ConvertListOfIOInListOfGO( const SALOME_ListIO& IObjects,
473 GEOM::ListOfGO& geomObjects,
476 geomObjects.length( 0 );
478 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
480 _PTR(Study) studyDS = study->studyDS();
482 geomObjects.length( IObjects.Extent() );
483 SALOME_ListIteratorOfListIO it( IObjects );
486 for ( ; it.More(); it.Next() ) {
487 GEOM::GEOM_Object_var geomObj = ConvertIOinGEOMObject( it.Value() );
488 if ( !CORBA::is_nil( geomObj ) && ( !shapesOnly || IsShape( geomObj ) ) )
489 geomObjects[ i++ ] = geomObj;
491 geomObjects.length( i );
496 //=================================================================================
497 // function : CreateArrowForLinearEdge()
498 // purpose : Create a cone topology to be used to display an arrow in the middle
499 // : of an edge showing its orientation. (For simulation and Viewer OCC only)
500 //=================================================================================
501 TopoDS_Shape GEOMBase::CreateArrowForLinearEdge( const TopoDS_Shape& shape )
503 TopoDS_Shape ArrowCone;
505 SUIT_ViewWindow* view = SUIT_Session::session()->activeApplication()->desktop()->activeWindow();
506 if ( view && view->getViewManager()->getType() == OCCViewer_Viewer::Type() && shape.ShapeType() == TopAbs_EDGE ) {
507 Handle(V3d_View) view3d = ((OCCViewer_ViewWindow*)view)->getViewPort()->getView();
508 Standard_Real Width, Height;
509 view3d->Size( Width, Height );
510 const Standard_Real aHeight = (Width + Height) / 50.0;
513 Standard_Real first, last;
514 Handle(Geom_Curve) curv = BRep_Tool::Curve( TopoDS::Edge( shape ), first, last );
515 if ( !curv.IsNull() && curv->IsCN(1) ) {
516 const Standard_Real param = ( first+last ) / 2.0;
517 gp_Pnt middleParamPoint;
519 curv->D1( param, middleParamPoint, V1 );
520 if ( V1.Magnitude() > Precision::Confusion() ) {
521 /* Topology orientation not geom orientation */
522 if ( shape.Orientation() == TopAbs_REVERSED )
525 gp_Ax2 anAxis( middleParamPoint, gp_Dir( V1 ) );
526 const Standard_Real radius1 = aHeight / 5.0;
527 if ( radius1 > 10.0 * Precision::Confusion() && aHeight > 10.0 * Precision::Confusion() )
528 ArrowCone = BRepPrimAPI_MakeCone( anAxis, radius1, 0.0, aHeight ).Shape();
532 catch ( Standard_Failure ) {
533 // OCC failures are hard to catch in GUI.
534 // This is because of the position for #include <Standard_ErrorHandler.hxx> that is very critical to find
535 // in SALOME environment : compilation error !
543 //=================================================================================
544 // function : VertexToPoint()
545 // purpose : If S can be converted in a gp_Pnt returns true and the result is P
546 //=================================================================================
547 bool GEOMBase::VertexToPoint( const TopoDS_Shape& shape, gp_Pnt& point )
549 if ( shape.IsNull() || shape.ShapeType() != TopAbs_VERTEX )
551 point = BRep_Tool::Pnt( TopoDS::Vertex( shape ) );
556 //=================================================================================
557 // function : GetBipointDxDyDz()
559 //=================================================================================
560 void GEOMBase::GetBipointDxDyDz( const gp_Pnt& point1, const gp_Pnt& point2, double& dx, double& dy, double& dz )
562 dx = point2.X() - point1.X();
563 dy = point2.Y() - point1.Y();
564 dz = point2.Z() - point1.Z();
568 //=================================================================================
569 // function : LinearEdgeExtremities()
570 // purpose : If S can be converted in a linear edge and if initial an final points
571 // : distance is sufficient, returns true else returns false.
572 // : Resulting points are respectively P1 and P2
573 //=================================================================================
574 bool GEOMBase::LinearEdgeExtremities( const TopoDS_Shape& shape, gp_Pnt& point1, gp_Pnt& point2 )
576 if ( shape.IsNull() || shape.ShapeType() != TopAbs_EDGE )
579 BRepAdaptor_Curve curv( TopoDS::Edge( shape ) );
580 if ( curv.GetType() != GeomAbs_Line )
585 curv.D0( curv.FirstParameter(), p1 );
586 curv.D0( curv.LastParameter(), p2 );
588 if ( p1.Distance( p2 ) <= Precision::Confusion() )
597 //=======================================================================
598 // function : SelectionByNameInDialogs()
599 // purpose : Called when user has entered a name of object in a LineEdit.
600 // : The selection is changed. Dialog box will receive the
601 // : corresponding signal to manage this event.
602 //=======================================================================
603 bool GEOMBase::SelectionByNameInDialogs( QWidget* widget, const QString& objectUserName, const SALOME_ListIO& /*IObjects*/ )
605 /* Find SObject with name in component GEOM */
606 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
607 if ( !study ) return false;
608 _PTR(Study) studyDS = study->studyDS();
610 std::vector<_PTR(SObject)> listSO = studyDS->FindObjectByName( objectUserName.toStdString(), "GEOM" );
612 if ( listSO.size() < 1 ) {
613 SUIT_MessageBox::critical( widget,
614 QObject::tr( "GEOM_WRN_WARNING" ),
615 QObject::tr( "GEOM_NAME_INCORRECT" ),
616 QObject::tr( "GEOM_BUT_OK" ) );
620 /* More than one object with same name */
621 if ( listSO.size() > 1 ) {
622 SUIT_MessageBox::critical( widget,
623 QObject::tr("GEOM_WRN_WARNING"),
624 QObject::tr("GEOM_IDENTICAL_NAMES_SELECT_BY_MOUSE"),
625 QObject::tr("GEOM_BUT_OK") );
629 /* Create a SALOME_InteractiveObject with a SALOME::SObject */
630 Handle(SALOME_InteractiveObject) IO = new SALOME_InteractiveObject( listSO[0]->GetID().c_str(),
632 objectUserName.toLatin1().constData() );
634 /* Add as a selected object */
635 /* Clear any previous selection : */
636 /* Warning the LineEdit is purged because of signal currentSelectionChanged ! */
637 // Sel->ClearIObjects(); //mzn
638 // Sel->AddIObject(SI); //mzn
643 //=======================================================================
644 // function : DefineDlgPosition()
645 // purpose : Define x and y the default position for a dialog box
646 //=======================================================================
647 void GEOMBase::DefineDlgPosition( QWidget* dlg, int& x, int& y )
649 /* Here the position is on the bottom right corner - 10 */
650 SUIT_Desktop* d = SUIT_Session::session()->activeApplication()->desktop();
651 x = abs( d->x() + d->size().width() - dlg->size().width() - 10 );
652 y = abs( d->y() + d->size().height() - dlg->size().height() - 10 );
656 //=======================================================================
657 // function : GetDefaultName()
658 // purpose : Generates default names
659 //=======================================================================
660 QString GEOMBase::GetDefaultName( const QString& operation, bool extractPrefix )
664 // collect all object names of GEOM component
665 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
667 _PTR(Study) studyDS = study->studyDS();
669 std::set<std::string> names;
670 _PTR(SComponent) component( studyDS->FindComponent( "GEOM" ) );
672 _PTR(ChildIterator) it( studyDS->NewChildIterator( component ) );
673 for ( it->InitEx( true ); it->More(); it->Next() ) {
674 names.insert( it->Value()->GetName() );
678 // build a unique name
680 bool isUnique = false;
681 QString prefix = operation;
683 if ( extractPrefix ) {
684 QStringList parts = prefix.split( "_", QString::KeepEmptyParts );
685 if ( parts.count() > 1 ) {
687 aNumber = parts.last().toLong( &ok );
690 prefix = parts.join( "_" );
696 while ( !isUnique ) {
697 aName = prefix + "_" + QString::number( ++aNumber );
698 isUnique = ( names.count( aName.toStdString()) == 0 );
705 //=======================================================================
706 // function : ShowErrorMessage()
707 // purpose : Shows message box with error code and comment
708 //=======================================================================
709 void GEOMBase::ShowErrorMessage( const QString& errorCode, const QString& comment )
712 text << QObject::tr( "GEOM_PRP_ABORT" );
713 if ( !errorCode.isEmpty() )
714 text << QObject::tr( errorCode.toLatin1().constData() );
715 if ( !comment.isEmpty() )
716 text << QObject::tr( comment.toLatin1().constData() );
718 SUIT_MessageBox::critical( SUIT_Session::session()->activeApplication()->desktop(),
719 QObject::tr( "GEOM_ERROR" ),
721 QObject::tr( "GEOM_BUT_OK" ) );
725 //=======================================================================
726 // function : GetObjectFromIOR()
727 // purpose : returns a GEOM_Object by given IOR (string)
728 //=======================================================================
729 GEOM::GEOM_Object_ptr GEOMBase::GetObjectFromIOR( const QString& IOR )
731 return GeometryGUI::GetObjectFromIOR (IOR);
734 //=======================================================================
735 // function : GetIORFromObject()
736 // purpose : returns IOR of a given GEOM_Object
737 //=======================================================================
738 QString GEOMBase::GetIORFromObject( GEOM::GEOM_Object_ptr object )
740 return GeometryGUI::GetIORFromObject (object);
743 //=======================================================================
744 // function : GetShape()
745 // purpose : returns a TopoDS_Shape stored in GEOM_Object
746 //=======================================================================
747 bool GEOMBase::GetShape( GEOM::GEOM_Object_ptr object, TopoDS_Shape& shape, const TopAbs_ShapeEnum type )
749 shape = TopoDS_Shape();
750 if ( !CORBA::is_nil( object ) ) {
751 TopAbs_ShapeEnum stype = (TopAbs_ShapeEnum)( object->GetShapeType() );
752 if ( type == TopAbs_SHAPE || type == stype )
753 shape = GEOM_Client::get_client().GetShape( GeometryGUI::GetGeomGen(), object );
755 return !shape.IsNull();
758 //=======================================================================
759 // function : GetName()
760 // purpose : Get name of object
761 //=======================================================================
762 QString GEOMBase::GetName( GEOM::GEOM_Object_ptr object )
765 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
767 if ( !CORBA::is_nil( object ) ) {
768 // 1. search if object is already published in the study
769 CORBA::String_var IOR = SalomeApp_Application::orb()->object_to_string( object );
770 if ( study && strcmp( IOR.in(), "" ) != 0 ) {
771 _PTR(SObject) aSObj( study->studyDS()->FindObjectIOR( std::string( IOR.in() ) ) );
772 _PTR(GenericAttribute) anAttr;
773 if ( aSObj && aSObj->FindAttribute( anAttr, "AttributeName") ) {
774 _PTR(AttributeName) aNameAttr( anAttr );
775 name = aNameAttr->Value().c_str();
779 // 2. if object is not found in the study, try default name
780 if ( name.isEmpty() ) {
781 if ( object->IsMainShape() ) {
782 name = GetDefaultName( "geomObj" );
785 GEOM::GEOM_Object_var mainShape = object->GetMainShape();
786 if ( !CORBA::is_nil( mainShape ) ) {
787 GEOM::ListOfLong_var indices = object->GetSubShapeIndices();
788 if ( indices->length() > 0 ) {
789 TopAbs_ShapeEnum type = (TopAbs_ShapeEnum)( object->GetShapeType() );
790 name = QString( "%1:%2_%3" ).arg( GetName( mainShape.in() ) )
791 .arg( TypeName( type ) ).arg( indices[0] );
801 //=======================================================================
802 // function : IsShape()
803 // purpose : Return TRUE if object is valid and has shape
804 //=======================================================================
805 bool GEOMBase::IsShape( GEOM::GEOM_Object_ptr object )
807 return !object->_is_nil() && object->IsShape();
810 //=======================================================================
811 // function : TypeName()
812 // purpose : Get string representation for the shape type
813 //=======================================================================
814 QString GEOMBase::TypeName( TopAbs_ShapeEnum type )
816 QString name = "shape";
818 case TopAbs_COMPSOLID:
819 name = "compsolid"; break;
820 case TopAbs_COMPOUND:
821 name = "compound"; break;
823 name = "solid"; break;
825 name = "shell"; break;
827 name = "face"; break;
829 name = "wire"; break;
831 name = "edge"; break;
833 name = "vertex"; break;
840 //================================================================
841 // Function : GetEntry
842 // Purpose : Get study entry for the given object (if it is published)
843 //================================================================
844 QString GEOMBase::GetEntry( GEOM::GEOM_Object_ptr object )
847 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
848 if ( study && !CORBA::is_nil( object ) ) {
849 QString objIOR = GetIORFromObject( object );
850 if ( !objIOR.isEmpty() ) {
851 _PTR(SObject) SO( study->studyDS()->FindObjectIOR( objIOR.toLatin1().constData() ) );
853 entry = SO->GetID().c_str();
859 //================================================================
860 // Function : PublishSubObject
861 // Purpose : Publish sub-shape under the main object
862 //================================================================
863 void GEOMBase::PublishSubObject( GEOM::GEOM_Object_ptr object )
865 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
866 if ( study && !CORBA::is_nil( object ) ) {
867 _PTR(Study) studyDS = study->studyDS();
868 QString entry = GetEntry( object );
869 GEOM::GEOM_Object_var father = object->GetMainShape();
870 QString fatherEntry = GetEntry( father );
871 if ( entry.isEmpty() && !CORBA::is_nil( father ) && !fatherEntry.isEmpty() ) {
872 QString name = GetName( object );
873 GeometryGUI::GetGeomGen()->AddInStudy( GeometryGUI::ClientStudyToStudy( studyDS ),
874 object, name.toLatin1().data(), father.in() );