1 // Copyright (C) 2007-2008 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
22 // GEOM GEOMGUI : GUI for Geometry component
23 // File : GEOM_Displayer.cxx
24 // Author : Vadim SANDLER
28 #include "GEOM_Displayer.h"
30 #include "GeometryGUI.h"
32 #include "GEOM_TypeFilter.h"
33 #include "GEOM_EdgeFilter.h"
34 #include "GEOM_FaceFilter.h"
35 #include "GEOM_CompoundFilter.h"
36 #include "GEOM_PreviewFilter.h"
37 #include "GEOM_LogicalFilter.h"
38 #include "GEOM_OCCFilter.h"
40 #include "GEOM_Actor.h"
41 #include "GEOM_AssemblyBuilder.h"
42 #include "GEOM_AISShape.hxx"
43 #include "GEOM_AISVector.hxx"
44 #include "GEOM_AISTrihedron.hxx"
45 #include "GEOM_VTKTrihedron.hxx"
47 #include <SUIT_Desktop.h>
48 #include <SUIT_ViewWindow.h>
49 #include <SUIT_Session.h>
50 #include <SUIT_Tools.h>
51 #include <SUIT_ViewManager.h>
53 #include <SalomeApp_Study.h>
54 #include <SalomeApp_Application.h>
55 #include <LightApp_SelectionMgr.h>
56 #include <SalomeApp_TypeFilter.h>
57 #include <SalomeApp_Tools.h>
59 #include <SALOME_ListIteratorOfListIO.hxx>
62 #include <SOCC_ViewModel.h>
65 #include <SVTK_ViewModel.h>
67 #include <SALOMEDSClient.hxx>
68 #include <SALOMEDSClient_SObject.hxx>
71 #include <AIS_Drawer.hxx>
72 #include <AIS_ListIteratorOfListOfInteractive.hxx>
73 #include <Prs3d_IsoAspect.hxx>
74 #include <Prs3d_PointAspect.hxx>
75 #include <Graphic3d_AspectMarker3d.hxx>
76 #include <StdSelect_TypeOfEdge.hxx>
77 #include <StdSelect_TypeOfFace.hxx>
78 #include <TopoDS_Face.hxx>
79 #include <BRep_Tool.hxx>
80 #include <Geom_Plane.hxx>
81 #include <Geom_Axis2Placement.hxx>
83 #include <TColStd_MapOfInteger.hxx>
84 #include <TColStd_MapIteratorOfMapOfInteger.hxx>
85 #include <TopoDS_Iterator.hxx>
88 #include <vtkActorCollection.h>
89 #include <vtkProperty.h>
95 #include CORBA_CLIENT_HEADER(SALOMEDS_Attributes)
97 #include "GEOMImpl_Types.hxx"
101 //================================================================
102 // Function : getActiveStudy
103 // Purpose : Get active study, returns 0 if no open study frame
104 //================================================================
105 static inline SalomeApp_Study* getActiveStudy()
107 SUIT_Session* session = SUIT_Session::session();
108 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
110 return ( SalomeApp_Study* )app->activeStudy();
114 //================================================================
115 // Function : getTopAbsMode
116 // Purpose : Get TopAbs_ShapeEnum value corresponding to the
117 // one from GEOMImpl_Types.h
118 //================================================================
119 static inline int getTopAbsMode( const int implType )
123 case GEOM_COMPOUND : return TopAbs_COMPOUND;
124 case GEOM_SOLID : return TopAbs_SOLID;
125 case GEOM_SHELL : return TopAbs_SHELL;
126 case GEOM_FACE : return TopAbs_FACE;
127 case GEOM_WIRE : return TopAbs_WIRE;
128 case GEOM_EDGE : return TopAbs_EDGE;
129 case GEOM_POINT : return TopAbs_VERTEX;
134 //================================================================
135 // Function : getFilter
136 // Purpose : Get filter corresponding to the type of object
137 // from GEOMImpl_Types.h
138 //================================================================
139 SUIT_SelectionFilter* GEOM_Displayer::getFilter( const int theMode )
141 SUIT_SelectionFilter* aFilter;
143 int aTopAbsMode = getTopAbsMode( theMode );
144 if ( aTopAbsMode != -1 )
145 aFilter = new GEOM_TypeFilter( getStudy(), aTopAbsMode, true ); //@ aFilter = new GEOM_TypeFilter( ( TopAbs_ShapeEnum )aTopAbsMode );
149 case GEOM_LINE : aFilter = new GEOM_EdgeFilter( getStudy(), StdSelect_Line ); break;
150 case GEOM_CIRCLE : aFilter = new GEOM_EdgeFilter( getStudy(), StdSelect_Circle ); break;
152 case GEOM_PLANE : aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Plane ); break;
153 case GEOM_CYLINDER : aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Cylinder ); break;
154 case GEOM_SPHERE : aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Sphere ); break;
155 case GEOM_TORUS : aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Torus ); break;
156 case GEOM_REVOLUTION: aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Revol ); break;
157 case GEOM_CONE : aFilter = new GEOM_FaceFilter( getStudy(), StdSelect_Cone ); break;
159 case GEOM_PREVIEW : aFilter = new GEOM_PreviewFilter( getStudy() ); break;
161 case GEOM_ALLSHAPES : aFilter = new GEOM_SelectionFilter(getStudy(), true ); break;
162 case GEOM_ALLGEOM : aFilter = new SalomeApp_TypeFilter( getStudy(), "GEOM" ); break;
164 default : aFilter = new GEOM_TypeFilter( getStudy(), theMode ); break;
170 //================================================================
171 // Function : getComplexFilter
172 // Purpose : Get compound filter corresponding to the type of
173 // object from GEOMImpl_Types.h
174 //================================================================
175 SUIT_SelectionFilter* GEOM_Displayer::getComplexFilter( const QValueList<int>* aSubShapes)
177 GEOM_CompoundFilter* aFilter;
179 if(aSubShapes != NULL ) {
180 aFilter = new GEOM_CompoundFilter(getStudy());
181 QValueList<int> aTopAbsTypes;
182 QValueList<int>::const_iterator it;
183 for(it = aSubShapes->constBegin(); it != aSubShapes->constEnd(); ++it ) {
184 int topAbsMode = getTopAbsMode(*it);
185 if(topAbsMode != -1 )
186 aTopAbsTypes.append(topAbsMode);
188 aFilter->addSubTypes(aTopAbsTypes);
194 //================================================================
195 // Function : getEntry
197 //================================================================
198 static string getEntry( GEOM::GEOM_Object_ptr object )
200 SUIT_Session* session = SUIT_Session::session();
201 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
204 CORBA::String_var IOR = app->orb()->object_to_string( object );
205 if ( strcmp(IOR.in(), "") != 0 )
207 SalomeApp_Study* study = ( SalomeApp_Study* )app->activeStudy();
208 _PTR(SObject) SO ( study->studyDS()->FindObjectIOR( string(IOR) ) );
216 //================================================================
217 // Function : getName
219 //================================================================
220 static string getName( GEOM::GEOM_Object_ptr object )
222 SUIT_Session* session = SUIT_Session::session();
223 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
226 CORBA::String_var IOR = app->orb()->object_to_string( object );
227 if ( strcmp(IOR.in(), "") != 0 )
229 SalomeApp_Study* study = ( SalomeApp_Study* )app->activeStudy();
230 _PTR(SObject) aSObj ( study->studyDS()->FindObjectIOR( string(IOR) ) );
232 _PTR(GenericAttribute) anAttr;
234 if ( aSObj && aSObj->FindAttribute( anAttr, "AttributeName") )
236 _PTR(AttributeName) aNameAttr( anAttr );
237 return aNameAttr->Value();
245 //=================================================================
247 * GEOM_Displayer::GEOM_Displayer
250 //=================================================================
251 GEOM_Displayer::GEOM_Displayer( SalomeApp_Study* st )
254 myApp = dynamic_cast<SalomeApp_Application*>( st->application() );
259 SUIT_Session* session = SUIT_Session::session();
260 SUIT_ResourceMgr* resMgr = session->resourceMgr();
262 QColor col = resMgr->colorValue( "Geometry", "shading_color", QColor( 255, 0, 0 ) );
263 myShadingColor = SalomeApp_Tools::color( col );
265 myDisplayMode = resMgr->integerValue("Geometry", "display_mode", 0);
266 myTypeOfMarker = (Aspect_TypeOfMarker)resMgr->integerValue("Geometry", "type_of_marker", Aspect_TOM_PLUS);
267 myScaleOfMarker = resMgr->doubleValue("Geometry", "marker_scale", 1.);
268 if(myScaleOfMarker < 1.0)
269 myScaleOfMarker = 1.0;
270 if(myScaleOfMarker > 7.)
271 myScaleOfMarker = 7.;
275 // This color is used for shape displaying. If it is equal -1 then
276 // default color is used.
282 // This parameter is used for activisation/deactivisation of objects to be displayed
287 //=================================================================
289 * GEOM_Displayer::~GEOM_Displayer
292 //=================================================================
293 GEOM_Displayer::~GEOM_Displayer()
297 //=================================================================
299 * GEOM_Displayer::Display
300 * Display interactive object in the current viewer
302 //=================================================================
303 void GEOM_Displayer::Display( const Handle(SALOME_InteractiveObject)& theIO,
304 const bool updateViewer,
305 SALOME_View* theViewFrame )
307 SALOME_View* vf = theViewFrame ? theViewFrame : GetActiveView();
310 SALOME_Prs* prs = buildPresentation( theIO->getEntry(), vf );
314 vf->BeforeDisplay( this );
316 vf->AfterDisplay( this );
321 delete prs; // delete presentation because displayer is its owner
326 //=================================================================
328 * GEOM_Displayer::Display
329 * This overloaded Display() method can be useful for operations
330 * not using dialog boxes.
332 //=================================================================
333 void GEOM_Displayer::Display( GEOM::GEOM_Object_ptr theObj, const bool updateViewer )
335 if ( theObj->_is_nil() )
338 string entry = getEntry( theObj );
340 Display(new SALOME_InteractiveObject(entry.c_str(), "GEOM", getName(theObj).c_str()),
345 //=================================================================
347 * GEOM_Displayer::Erase
348 * Erase interactive object in the current viewer
350 //=================================================================
351 void GEOM_Displayer::Erase( const Handle(SALOME_InteractiveObject)& theIO,
353 const bool updateViewer,
354 SALOME_View* theViewFrame )
356 if ( theIO.IsNull() )
359 SALOME_View* vf = theViewFrame ? theViewFrame : GetActiveView();
362 SALOME_Prs* prs = vf->CreatePrs( theIO->getEntry() );
364 vf->Erase( prs, forced );
367 delete prs; // delete presentation because displayer is its owner
372 //=================================================================
374 * GEOM_Displayer::Erase
375 * Erase geometry object in the current viewer
377 //=================================================================
378 void GEOM_Displayer::Erase( GEOM::GEOM_Object_ptr theObj,
380 const bool updateViewer )
382 string entry = getEntry( theObj );
385 Erase(new SALOME_InteractiveObject(entry.c_str(), "GEOM", getName(theObj).c_str()),
386 forced, updateViewer);
390 //=================================================================
392 * GEOM_Displayer::Redisplay
393 * Redisplay (erase and then display again) interactive object
394 * in the current viewer
396 //=================================================================
397 void GEOM_Displayer::Redisplay( const Handle(SALOME_InteractiveObject)& theIO,
398 const bool updateViewer )
400 // Remove the object permanently (<forced> == true)
401 SUIT_Session* ses = SUIT_Session::session();
402 SUIT_Application* app = ses->activeApplication();
405 SUIT_Desktop* desk = app->desktop();
406 QPtrList<SUIT_ViewWindow> wnds = desk->windows();
407 SUIT_ViewWindow* wnd;
408 for ( wnd = wnds.first(); wnd; wnd = wnds.next() )
410 SUIT_ViewManager* vman = wnd->getViewManager();
413 SUIT_ViewModel* vmodel = vman->getViewModel();
416 SALOME_View* view = dynamic_cast<SALOME_View*>(vmodel);
419 if ( view->isVisible( theIO ) || view == GetActiveView() )
421 Erase( theIO, true, false, view );
422 Display( theIO, updateViewer, view );
431 //=================================================================
433 * GEOM_Displayer::Display
434 * Calls Display() method for each object in the given list
436 //=================================================================
437 void GEOM_Displayer::Display( const SALOME_ListIO& theIOList, const bool updateViewer )
439 SALOME_ListIteratorOfListIO Iter( theIOList );
440 for ( ; Iter.More(); Iter.Next() ) {
441 Display( Iter.Value(), false );
447 //=================================================================
449 * GEOM_Displayer::Erase
450 * Calls Erase() method for each object in the given list
452 //=================================================================
453 void GEOM_Displayer::Erase( const SALOME_ListIO& theIOList,
455 const bool updateViewer )
457 SALOME_ListIteratorOfListIO Iter( theIOList );
458 for ( ; Iter.More(); Iter.Next() )
459 Erase( Iter.Value(), forced, false );
465 //=================================================================
467 * GEOM_Displayer::Redisplay
468 * Calls Redisplay() method for each object in the given list
470 //=================================================================
471 void GEOM_Displayer::Redisplay( const SALOME_ListIO& theIOList, const bool updateViewer )
473 SALOME_ListIteratorOfListIO Iter( theIOList );
474 for ( ; Iter.More(); Iter.Next() )
475 Redisplay( Iter.Value(), false );
481 //=================================================================
483 * GEOM_Displayer::Update
484 * Update OCC presentaion
485 * [ Reimplemented from SALOME_Displayer ]
487 //=================================================================
488 void GEOM_Displayer::Update( SALOME_OCCPrs* prs )
490 SOCC_Prs* occPrs = dynamic_cast<SOCC_Prs*>( prs );
494 if ( myType == GEOM_MARKER && !myShape.IsNull() && myShape.ShapeType() == TopAbs_FACE )
496 TopoDS_Face aFace = TopoDS::Face( myShape );
497 Handle(Geom_Plane) aPlane = Handle(Geom_Plane)::DownCast( BRep_Tool::Surface( aFace ) );
498 if ( !aPlane.IsNull() )
500 gp_Ax3 aPos = aPlane->Pln().Position();
501 Handle(Geom_Axis2Placement) aPlc = new Geom_Axis2Placement( aPos.Ax2() );
503 Handle(GEOM_AISTrihedron) aTrh;
505 if ( occPrs->IsNull() )
507 aTrh = new GEOM_AISTrihedron( aPlc );
510 aTrh->SetColor( (Quantity_NameOfColor)GetColor() );
513 aTrh->SetWidth( GetWidth() );
515 if ( !myIO.IsNull() )
518 aTrh->SetOwner( myIO );
521 occPrs->AddObject( aTrh );
525 AIS_ListOfInteractive aList;
526 occPrs->GetObjects( aList );
527 AIS_ListIteratorOfListOfInteractive anIter( aList );
528 for ( ; anIter.More(); anIter.Next() )
530 aTrh = Handle(GEOM_AISTrihedron)::DownCast( anIter.Value() );
531 if ( !aTrh.IsNull() )
533 aTrh->SetComponent( aPlc );
539 occPrs->SetToActivate( ToActivate() );
544 // if presentation is empty we try to create new one
545 if ( occPrs->IsNull() )
547 if ( !myShape.IsNull() )
549 //Handle(GEOM_AISShape) AISShape = new GEOM_AISShape( myShape, "" );
550 Handle(GEOM_AISShape) AISShape;
551 if (myType == GEOM_VECTOR)
552 AISShape = new GEOM_AISVector (myShape, "");
554 if (myShape.ShapeType() != TopAbs_VERTEX && // fix pb with not displayed points
555 !TopoDS_Iterator(myShape).More())
556 return;// NPAL15983 (Bug when displaying empty groups)
557 AISShape = new GEOM_AISShape (myShape, "");
559 // Temporary staff: vertex must be infinite for correct visualization
560 AISShape->SetInfiniteState( myShape.Infinite() || myShape.ShapeType() == TopAbs_VERTEX );
562 // Setup shape properties here ..., e.g. display mode, color, transparency, etc
563 AISShape->SetDisplayMode( myDisplayMode );
564 AISShape->SetShadingColor( myShadingColor );
566 // Set color and number for iso lines
567 SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
568 QColor col = aResMgr->colorValue( "Geometry", "isos_color",
569 QColor(int(0.5*255), int(0.5*255), int(0.5*255)) );
570 Quantity_Color aColor = SalomeApp_Tools::color( col );
571 int anUIsoNumber = aResMgr->integerValue("OCCViewer", "iso_number_u", 1);
572 int aVIsoNumber = aResMgr->integerValue("OCCViewer", "iso_number_v", 1);
574 Handle(Prs3d_IsoAspect) anAspect = AISShape->Attributes()->UIsoAspect();
575 anAspect->SetNumber( anUIsoNumber );
576 anAspect->SetColor( aColor );
577 AISShape->Attributes()->SetUIsoAspect( anAspect );
579 anAspect = AISShape->Attributes()->VIsoAspect();
580 anAspect->SetNumber( aVIsoNumber );
581 anAspect->SetColor( aColor );
582 AISShape->Attributes()->SetVIsoAspect( anAspect );
586 AISShape->SetColor( (Quantity_NameOfColor)GetColor() );
587 if ( myShape.ShapeType() == TopAbs_VERTEX )
589 Handle(Prs3d_PointAspect) anAspect = AISShape->Attributes()->PointAspect();
590 anAspect->SetColor( (Quantity_NameOfColor)GetColor() );
591 anAspect->SetScale( myScaleOfMarker );
592 anAspect->SetTypeOfMarker( myTypeOfMarker );
593 AISShape->Attributes()->SetPointAspect( anAspect );
598 if ( myShape.ShapeType() == TopAbs_VERTEX )
600 col = aResMgr->colorValue( "Geometry", "point_color", QColor( 255, 255, 0 ) );
601 aColor = SalomeApp_Tools::color( col );
603 Handle(Prs3d_PointAspect) anAspect = AISShape->Attributes()->PointAspect();
604 anAspect->SetColor( aColor );
605 anAspect->SetScale( myScaleOfMarker );
606 anAspect->SetTypeOfMarker( myTypeOfMarker );
607 AISShape->Attributes()->SetPointAspect( anAspect );
612 col = aResMgr->colorValue( "Geometry", "wireframe_color", QColor( 255, 255, 0 ) );
613 aColor = SalomeApp_Tools::color( col );
615 Handle(Prs3d_LineAspect) anAspect = AISShape->Attributes()->LineAspect();
616 anAspect->SetColor( aColor );
617 AISShape->Attributes()->SetLineAspect( anAspect );
619 // Set unfree boundaries aspect
620 anAspect = AISShape->Attributes()->UnFreeBoundaryAspect();
621 anAspect->SetColor( aColor );
622 AISShape->Attributes()->SetUnFreeBoundaryAspect( anAspect );
624 // Set free boundaries aspect
625 col = aResMgr->colorValue( "Geometry", "free_bound_color", QColor( 0, 255, 0 ) );
626 aColor = SalomeApp_Tools::color( col );
628 anAspect = AISShape->Attributes()->FreeBoundaryAspect();
629 anAspect->SetColor( aColor );
630 AISShape->Attributes()->SetFreeBoundaryAspect( anAspect );
633 col = aResMgr->colorValue( "Geometry", "line_color", QColor( 255, 0, 0 ) );
634 aColor = SalomeApp_Tools::color( col );
636 anAspect = AISShape->Attributes()->WireAspect();
637 anAspect->SetColor( aColor );
638 AISShape->Attributes()->SetWireAspect( anAspect );
640 // bug [SALOME platform 0019868]
641 // Set deviation angle. Default one is 12 degrees (Prs3d_Drawer.cxx:18)
642 AISShape->SetOwnDeviationAngle( 10*PI/180 );
647 AISShape->SetWidth( GetWidth() );
649 if ( !myIO.IsNull() )
651 AISShape->setIO( myIO );
652 AISShape->SetOwner( myIO );
654 else if ( !myName.empty() )
656 // Workaround to allow selection of temporary objects
657 static int tempId = 0;
659 sprintf( buf, "TEMP_%d", tempId++ );
660 Handle( SALOME_InteractiveObject ) anObj =
661 new SALOME_InteractiveObject( buf, "GEOM", myName.c_str() );
662 AISShape->setIO( anObj );
663 AISShape->SetOwner( anObj );
666 // Get color from GEOM_Object
667 SUIT_Session* session = SUIT_Session::session();
668 SUIT_Application* app = session->activeApplication();
671 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
674 Handle( SALOME_InteractiveObject ) anIO = AISShape->getIO();
675 if ( !anIO.IsNull() )
677 _PTR(SObject) SO ( study->studyDS()->FindObjectID( anIO->getEntry() ) );
680 // get CORBA reference to data object
681 CORBA::Object_var object = GeometryGUI::ClientSObjectToObject(SO);
682 if ( !CORBA::is_nil( object ) )
684 // downcast to GEOM object
685 GEOM::GEOM_Object_var aGeomObject = GEOM::GEOM_Object::_narrow( object );
686 if ( !aGeomObject->_is_nil() )
688 SALOMEDS::Color aSColor = aGeomObject->GetColor();
689 bool hasColor = aSColor.R > 0 || aSColor.G > 0 || aSColor.B > 0;
690 if( !hasColor && aGeomObject->GetType() == GEOM_GROUP ) // auto color for group
692 GEOM::GEOM_Gen_var aGeomGen = GeometryGUI::GetGeomGen();
693 GEOM::GEOM_IGroupOperations_var anOperations = aGeomGen->GetIGroupOperations( study->id() );
694 GEOM::GEOM_Object_var aMainObject = anOperations->GetMainShape( aGeomObject );
695 if ( !aMainObject->_is_nil() && aMainObject->GetAutoColor() )
697 QValueList<SALOMEDS::Color> aReservedColors;
699 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( app );
700 CORBA::String_var IOR = app->orb()->object_to_string( aMainObject );
701 if ( strcmp(IOR.in(), "") != 0 )
703 _PTR(Study) aStudy = study->studyDS();
704 _PTR(SObject) aMainSObject( aStudy->FindObjectIOR( string(IOR) ) );
705 _PTR(ChildIterator) it( aStudy->NewChildIterator( aMainSObject ) );
706 for( ; it->More(); it->Next() )
708 _PTR(SObject) aChildSObject( it->Value() );
709 GEOM::GEOM_Object_var aChildObject =
710 GEOM::GEOM_Object::_narrow(GeometryGUI::ClientSObjectToObject(aChildSObject));
711 if( CORBA::is_nil( aChildObject ) )
714 if( aChildObject->GetType() != GEOM_GROUP )
717 SALOMEDS::Color aReservedColor = aChildObject->GetColor();
718 aReservedColors.append( aReservedColor );
722 aSColor = getUniqueColor( aReservedColors );
729 Quantity_Color aQuanColor( aSColor.R, aSColor.G, aSColor.B, Quantity_TOC_RGB );
730 AISShape->SetColor( aQuanColor );
731 AISShape->SetShadingColor( aQuanColor );
740 // AISShape->SetName(???); ??? necessary to set name ???
741 occPrs->AddObject( AISShape );
743 // In accordance with ToActivate() value object will be activated/deactivated
744 // when it will be displayed
745 occPrs->SetToActivate( ToActivate() );
748 // if presentation is found -> set again shape for it
751 if ( !myShape.IsNull() )
753 AIS_ListOfInteractive IOList;
754 occPrs->GetObjects( IOList );
755 AIS_ListIteratorOfListOfInteractive Iter( IOList );
756 for ( ; Iter.More(); Iter.Next() )
758 Handle(GEOM_AISShape) AISShape = Handle(GEOM_AISShape)::DownCast( Iter.Value() );
759 if ( AISShape.IsNull() )
761 if ( AISShape->Shape() != myShape )
763 AISShape->Set( myShape );
764 AISShape->UpdateSelection();
765 AISShape->SetToUpdate();
767 if ( !myIO.IsNull() )
769 AISShape->setIO( myIO );
770 AISShape->SetOwner( myIO );
778 //=================================================================
780 * GEOM_Displayer::Update
781 * Update VTK presentaion
782 * [ Reimplemented from SALOME_Displayer ]
784 //=================================================================
785 void GEOM_Displayer::Update( SALOME_VTKPrs* prs )
787 SVTK_Prs* vtkPrs = dynamic_cast<SVTK_Prs*>( prs );
788 if ( !vtkPrs || myShape.IsNull() )
791 vtkActorCollection* theActors = 0;
793 if ( myType == GEOM_MARKER && myShape.ShapeType() == TopAbs_FACE )
795 myToActivate = false;
796 GEOM_VTKTrihedron* aTrh = GEOM_VTKTrihedron::New();
800 Quantity_Color aColor( (Quantity_NameOfColor)GetColor() );
801 aTrh->SetColor( aColor.Red(), aColor.Green(), aColor.Blue() );
804 Handle(Geom_Plane) aPlane =
805 Handle(Geom_Plane)::DownCast( BRep_Tool::Surface( TopoDS::Face( myShape ) ) );
806 if ( aPlane.IsNull() )
809 gp_Ax2 anAx2 = aPlane->Pln().Position().Ax2();
810 aTrh->SetPlacement( new Geom_Axis2Placement( anAx2 ) );
812 // if ( SVTK_Viewer* vf = dynamic_cast<SVTK_Viewer*>( GetActiveView() ) )
813 // aTrh->SetSize( 0.5 * vf->GetTrihedronSize() );
815 vtkPrs->AddObject( aTrh );
817 theActors = vtkActorCollection::New();
818 theActors->AddItem( aTrh );
822 bool isVector = (myType == GEOM_VECTOR);
823 theActors = GEOM_AssemblyBuilder::BuildActors( myShape, 0, 0, Standard_True, isVector );
826 theActors->InitTraversal();
828 vtkActor* anActor = (vtkActor*)theActors->GetNextActor();
830 vtkProperty* aProp = 0;
832 if ( HasColor() || HasWidth() )
834 aProp = vtkProperty::New();
835 aProp->SetRepresentationToWireframe();
840 Quantity_Color aColor( (Quantity_NameOfColor)GetColor() );
841 aProp->SetColor( aColor.Red(), aColor.Green(), aColor.Blue() );
846 aProp->SetLineWidth( GetWidth() );
849 while ( anActor != NULL )
851 SALOME_Actor* GActor = SALOME_Actor::SafeDownCast( anActor );
853 GActor->setIO( myIO );
857 GActor->SetProperty( aProp );
858 GActor->SetPreviewProperty( aProp );
860 GEOM_Actor* aGeomGActor = GEOM_Actor::SafeDownCast( anActor );
861 if ( aGeomGActor != 0 )
863 aGeomGActor->SetShadingProperty( aProp );
864 aGeomGActor->SetWireframeProperty( aProp );
869 GActor->PickableOn();
871 GActor->PickableOff();
873 vtkPrs->AddObject( GActor );
875 anActor = (vtkActor*)theActors->GetNextActor();
884 //=================================================================
886 * GEOM_Displayer::BuildPrs
887 * Build presentation accordint to the current viewer type
889 //=================================================================
890 SALOME_Prs* GEOM_Displayer::BuildPrs( GEOM::GEOM_Object_ptr theObj )
892 if ( theObj->_is_nil() )
895 myViewFrame = GetActiveView();
896 if ( myViewFrame == 0 )
899 SALOME_Prs* aPrs = myViewFrame->CreatePrs();
904 setShape( GEOM_Client().GetShape( GeometryGUI::GetGeomGen(), theObj ) );
905 myType = theObj->GetType();
907 // Update presentation
913 //=================================================================
915 * GEOM_Displayer::BuildPrs
916 * Build presentation accordint to the current viewer type
918 //=================================================================
919 SALOME_Prs* GEOM_Displayer::BuildPrs( const TopoDS_Shape& theShape )
921 myViewFrame = GetActiveView();
922 if ( theShape.IsNull() || myViewFrame == 0 )
925 SALOME_Prs* aPrs = myViewFrame->CreatePrs();
930 setShape( theShape );
938 //=================================================================
940 * GEOM_Displayer::buildPresentation
941 * Builds/finds object's presentation for the current viewer
942 * Calls corresponding Update() method by means of double dispatch
945 //=================================================================
946 SALOME_Prs* GEOM_Displayer::buildPresentation( const QString& entry,
947 SALOME_View* theViewFrame )
952 myViewFrame = theViewFrame ? theViewFrame : GetActiveView();
956 prs = LightApp_Displayer::buildPresentation( entry, theViewFrame );
959 Handle( SALOME_InteractiveObject ) theIO = new SALOME_InteractiveObject();
960 theIO->setEntry( entry.latin1() );
961 if ( !theIO.IsNull() )
963 // set interactive object
965 // Find SOBject (because shape should be published previously)
966 SUIT_Session* session = SUIT_Session::session();
967 SUIT_Application* app = session->activeApplication();
970 SalomeApp_Study* study = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
973 _PTR(SObject) SO ( study->studyDS()->FindObjectID( theIO->getEntry() ) );
976 // get CORBA reference to data object
977 CORBA::Object_var object = GeometryGUI::ClientSObjectToObject(SO);
978 if ( !CORBA::is_nil( object ) )
980 // downcast to GEOM object
981 GEOM::GEOM_Object_var GeomObject = GEOM::GEOM_Object::_narrow( object );
982 if ( !GeomObject->_is_nil() )
985 setShape( GEOM_Client().GetShape( GeometryGUI::GetGeomGen(), GeomObject ) );
986 myType = GeomObject->GetType();
993 UpdatePrs( prs ); // Update presentation by using of the double dispatch
999 //=================================================================
1001 * GEOM_Displayer::internalReset
1002 * Resets internal data
1005 //=================================================================
1006 void GEOM_Displayer::internalReset()
1012 //=================================================================
1014 * GEOM_Displayer::LocalSelection
1015 * Activate selection of CAD shapes with activisation of selection
1016 * of their sub-shapes (with opened local context for OCC viewer)
1018 //=================================================================
1019 void GEOM_Displayer::LocalSelection( const Handle(SALOME_InteractiveObject)& theIO, const int theMode )
1021 SUIT_Session* session = SUIT_Session::session();
1022 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
1026 LightApp_SelectionMgr* sm = app->selectionMgr();
1028 // remove all filters from selection
1031 SALOME_View* vf = GetActiveView();
1033 if (!theIO.IsNull() && !vf->isVisible(theIO))
1035 SALOME_Prs* prs = vf->CreatePrs( theIO.IsNull() ? 0 : theIO->getEntry() );
1036 vf->LocalSelection( prs, theMode );
1037 delete prs; // delete presentation because displayer is its owner
1041 //=================================================================
1043 * GEOM_Displayer::globalSelection
1044 * Activate selection of CAD shapes without activisation of selection
1045 * of their sub-shapes (without opened local context for OCC viewer)
1047 //=================================================================
1048 void GEOM_Displayer::GlobalSelection( const int theMode, const bool update )
1050 TColStd_MapOfInteger aModes;
1051 aModes.Add( theMode );
1052 GlobalSelection( aModes, update );
1055 //=================================================================
1057 * GEOM_Displayer::globalSelection
1058 * Activate selection of CAD shapes without activisation of selection
1059 * of their sub-shapes (without opened local context for OCC viewer)
1061 //=================================================================
1062 void GEOM_Displayer::GlobalSelection( const TColStd_MapOfInteger& theModes,
1063 const bool update, const QValueList<int>* theSubShapes)
1065 SUIT_Session* session = SUIT_Session::session();
1066 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
1070 SALOME_View* vf = GetActiveView();
1074 // Close local context
1075 vf->GlobalSelection( update );
1077 // Set selection filters in accordance with current mode
1078 LightApp_SelectionMgr* sm = app->selectionMgr();
1082 // Remove from selection temporary objects if necessary
1083 if ( !theModes.Contains( GEOM_PREVIEW ) )
1084 clearTemporary( sm );
1086 //@ aSel->ClearIndex();
1090 // Remove filters from AIS_InteractiveContext
1091 Handle(AIS_InteractiveContext) ic;
1092 SOCC_Viewer* viewer = dynamic_cast<SOCC_Viewer*>( vf );
1095 ic = viewer->getAISContext();
1097 ic->RemoveFilters();
1100 if ( theModes.Contains( GEOM_ALLOBJECTS ) )
1103 SUIT_SelectionFilter* aFilter;
1104 if ( theModes.Extent() == 1 )
1106 int aMode = TColStd_MapIteratorOfMapOfInteger( theModes ).Key();
1108 if(aMode == GEOM_COMPOUNDFILTER)
1109 aFilter = getComplexFilter(theSubShapes);
1111 aFilter = getFilter( aMode );
1113 else if ( theModes.Extent() > 1 )
1115 TColStd_MapOfInteger aTopAbsModes;
1116 TColStd_MapIteratorOfMapOfInteger anIter( theModes );
1117 QPtrList<SUIT_SelectionFilter> aListOfFilters;
1118 for ( ; anIter.More(); anIter.Next() )
1120 SUIT_SelectionFilter* aFilter;
1121 int aMode = anIter.Key();
1122 if(aMode == GEOM_COMPOUNDFILTER)
1123 aFilter = getComplexFilter(theSubShapes);
1125 aFilter = getFilter( aMode );
1128 aListOfFilters.append( aFilter );
1131 aFilter = new GEOM_LogicalFilter( aListOfFilters, GEOM_LogicalFilter::LO_OR );
1138 sm->installFilter( aFilter );
1141 Handle(GEOM_OCCFilter) anOCCFilter = new GEOM_OCCFilter( sm );
1142 ic->AddFilter( anOCCFilter );
1147 //=================================================================
1149 * GEOM_Displayer::LocalSelection
1150 * Activate selection of CAD shapes with activisation of selection
1151 * of their sub-shapes (with opened local context for OCC viewer)
1153 //=================================================================
1154 void GEOM_Displayer::LocalSelection( const SALOME_ListIO& theIOList, const int theMode )
1156 SALOME_ListIteratorOfListIO Iter( theIOList );
1157 for ( ; Iter.More(); Iter.Next() )
1158 LocalSelection( Iter.Value(), theMode );
1161 //=================================================================
1163 * GEOM_Displayer::BeforeDisplay
1164 * Called before displaying of pars. Close local context
1165 * [ Reimplemented from SALOME_Displayer ]
1167 //=================================================================
1168 void GEOM_Displayer::BeforeDisplay( SALOME_View* v, const SALOME_OCCViewType& )
1170 SOCC_Viewer* vf = dynamic_cast<SOCC_Viewer*>( v );
1173 Handle(AIS_InteractiveContext) ic = vf->getAISContext();
1176 if ( ic->HasOpenedContext() )
1177 ic->CloseAllContexts();
1182 void GEOM_Displayer::AfterDisplay( SALOME_View*, const SALOME_OCCViewType& )
1187 //=================================================================
1189 * GEOM_Displayer::SetColor
1190 * Set color for shape displaying. If it is equal -1 then default color is used.
1191 * Available values are from Quantity_NameOfColor enumeration
1193 //=================================================================
1194 void GEOM_Displayer::SetColor( const int color )
1201 myShadingColor = Quantity_Color( (Quantity_NameOfColor)color );
1205 int GEOM_Displayer::GetColor() const
1210 bool GEOM_Displayer::HasColor() const
1212 return myColor != -1;
1215 void GEOM_Displayer::UnsetColor()
1219 SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
1220 QColor col = resMgr->colorValue( "Geometry", "shading_color", QColor( 255, 0, 0 ) );
1221 myShadingColor = SalomeApp_Tools::color( col );
1224 //=================================================================
1226 * GEOM_Displayer::SetWidth
1227 * Set width of shape displaying. If it is equal -1 then default width is used.
1229 //=================================================================
1230 void GEOM_Displayer::SetWidth( const double width )
1235 double GEOM_Displayer::GetWidth() const
1240 bool GEOM_Displayer::HasWidth() const
1242 return myWidth != -1;
1245 void GEOM_Displayer::UnsetWidth()
1250 //=================================================================
1252 * GEOM_Displayer::SetToActivate
1253 * This method is used for activisation/deactivisation of objects to be displayed
1255 //=================================================================
1256 void GEOM_Displayer::SetToActivate( const bool toActivate )
1258 myToActivate = toActivate;
1260 bool GEOM_Displayer::ToActivate() const
1262 return myToActivate;
1265 //=================================================================
1267 * GEOM_Displayer::clearTemporary
1268 * Removes from selection temporary objects
1270 //=================================================================
1271 void GEOM_Displayer::clearTemporary( LightApp_SelectionMgr* theSelMgr )
1273 SALOME_ListIO selected, toSelect;
1274 theSelMgr->selectedObjects( selected );
1276 for ( SALOME_ListIteratorOfListIO it( selected ) ; it.More(); it.Next() ) {
1277 Handle(SALOME_InteractiveObject) io = it.Value();
1278 if ( !io.IsNull() && io->hasEntry() && strncmp( io->getEntry(), "TEMP_", 5 ) != 0 )
1279 toSelect.Append( it.Value() );
1282 theSelMgr->setSelectedObjects( toSelect, true );
1285 void GEOM_Displayer::SetName( const char* theName )
1290 void GEOM_Displayer::UnsetName()
1295 SalomeApp_Study* GEOM_Displayer::getStudy() const
1297 return dynamic_cast<SalomeApp_Study*>( myApp->activeStudy() );
1300 void GEOM_Displayer::setIO( const Handle(SALOME_InteractiveObject)& theIO )
1305 void GEOM_Displayer::setShape( const TopoDS_Shape& theShape )
1310 bool GEOM_Displayer::canBeDisplayed( const QString& /*entry*/, const QString& viewer_type ) const
1312 return viewer_type==SOCC_Viewer::Type() || viewer_type==SVTK_Viewer::Type();
1315 int GEOM_Displayer::SetDisplayMode( const int theMode )
1317 int aPrevMode = myDisplayMode;
1318 if ( theMode != -1 )
1319 myDisplayMode = theMode;
1322 SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
1323 myDisplayMode = resMgr->integerValue( "Geometry", "display_mode", 0 );
1328 int GEOM_Displayer::GetDisplayMode() const
1330 return myDisplayMode;
1333 int GEOM_Displayer::UnsetDisplayMode()
1335 int aPrevMode = myDisplayMode;
1336 SUIT_ResourceMgr* resMgr = SUIT_Session::session()->resourceMgr();
1337 myDisplayMode = resMgr->integerValue( "Geometry", "display_mode", 0 );
1341 SALOMEDS::Color GEOM_Displayer::getUniqueColor( const QValueList<SALOMEDS::Color>& theReservedColors )
1344 int aTolerance = 64;
1345 int anIterations = 0;
1351 if( anIterations % aPeriod == 0 )
1354 if( aTolerance < 1 )
1357 //cout << "Iteration N" << anIterations << " (tolerance=" << aTolerance << ")"<< endl;
1359 aHue = (int)( 360.0 * rand() / RAND_MAX );
1360 //cout << "Hue = " << aHue << endl;
1362 //cout << "Auto colors : ";
1364 QValueList<SALOMEDS::Color>::const_iterator it = theReservedColors.constBegin();
1365 QValueList<SALOMEDS::Color>::const_iterator itEnd = theReservedColors.constEnd();
1366 for( ; it != itEnd; ++it )
1368 SALOMEDS::Color anAutoColor = *it;
1369 QColor aQColor( (int)( anAutoColor.R * 255.0 ), (int)( anAutoColor.G * 255.0 ), (int)( anAutoColor.B * 255.0 ) );
1372 aQColor.getHsv( &h, &s, &v );
1374 if( abs( h - aHue ) < aTolerance )
1377 //cout << "break (diff = " << abs( h - aHue ) << ")";
1387 //cout << "Hue of the returned color = " << aHue << endl;
1389 aColor.setHsv( aHue, 255, 255 );
1391 SALOMEDS::Color aSColor;
1392 aSColor.R = (double)aColor.red() / 255.0;
1393 aSColor.G = (double)aColor.green() / 255.0;
1394 aSColor.B = (double)aColor.blue() / 255.0;