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 // File : GEOMGUI_Selection.cxx
24 // Author : Alexander SOLOVYOV, Open CASCADE S.A.S. (alexander.solovyov@opencascade.com)
26 #include "GEOMGUI_Selection.h"
28 #include "GeometryGUI.h"
29 #include "GEOM_Displayer.h"
31 #include "Material_Model.h"
33 #include <GEOM_Constants.h>
35 #include <SalomeApp_Application.h>
36 #include <SalomeApp_Study.h>
38 #include "LightApp_DataOwner.h"
40 #include <SUIT_Session.h>
41 #include <SUIT_Desktop.h>
42 #include <SUIT_ViewWindow.h>
43 #include <SUIT_ViewManager.h>
45 #include <SALOME_Prs.h>
46 #include <SALOME_InteractiveObject.hxx>
50 #include <SALOME_Actor.h>
51 #include <GEOM_Actor.h>
53 #include <OCCViewer_ViewModel.h>
54 #include <SVTK_ViewModel.h>
56 #include <GEOMImpl_Types.hxx>
58 #include <GEOM_AISShape.hxx>
59 #include <GEOM_VTKPropertyMaterial.hxx>
63 #include <AIS_InteractiveObject.hxx>
64 #include <AIS_ListOfInteractive.hxx>
65 #include <AIS_GraphicTool.hxx>
66 #include <AIS_Drawer.hxx>
67 #include <Aspect_TypeOfFacingModel.hxx>
68 #include <Prs3d_ShadingAspect.hxx>
69 #include<Graphic3d_MaterialAspect.hxx>
72 #include <vtkActorCollection.h>
74 #define OCC_DISPLAY_MODE_TO_STRING( str, dm ) { \
75 if ( dm == AIS_WireFrame ) \
76 str = QString( "Wireframe" ); \
77 else if ( dm == AIS_Shaded ) \
78 str = QString( "Shading" ); \
79 else if ( dm == GEOM_AISShape::ShadingWithEdges ) \
80 str = QString( "ShadingWithEdges" ); \
81 else if ( dm == GEOM_AISShape::TexturedShape ) \
82 str = QString( "Texture" ); \
86 #define VTK_DISPLAY_MODE_TO_STRING( str, dm ) { \
88 str = QString( "Wireframe" ); \
90 str = QString( "Shading" ); \
92 str = QString( "ShadingWithEdges" ); \
96 #define USE_VISUAL_PROP_MAP
98 GEOMGUI_Selection::GEOMGUI_Selection()
99 : LightApp_Selection()
103 GEOMGUI_Selection::~GEOMGUI_Selection()
107 void GEOMGUI_Selection::init( const QString& context, LightApp_SelectionMgr* selMgr )
109 LightApp_Selection::init( context, selMgr );
111 myObjects.resize( count() );
113 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( study() );
115 _PTR(Study) study = appStudy->studyDS();
116 for ( int idx = 0; idx < count(); idx++ ) {
117 QString anEntry = entry( idx );
118 if ( study && !anEntry.isEmpty() ) {
119 _PTR(SObject) aSO( study->FindObjectID( anEntry.toStdString() ) );
121 CORBA::Object_var varObj = GeometryGUI::ClientSObjectToObject( aSO );
122 myObjects[idx] = GEOM::GEOM_Object::_narrow( varObj );
129 //QVariant GEOMGUI_Selection::contextParameter( const QString& p ) const
130 QVariant GEOMGUI_Selection::parameter( const QString& p ) const
134 v = activeViewType() == OCCViewer_Viewer::Type();
135 else if ( p == "selectionmode" )
137 else if ( p == "hasImported" )
139 else if ( p == "allImported" )
142 v = LightApp_Selection::parameter( p );
146 //QVariant GEOMGUI_Selection::objectParameter( const int idx, const QString& p ) const
147 QVariant GEOMGUI_Selection::parameter( const int idx, const QString& p ) const
152 else if ( p == "typeid" )
154 else if ( p == "displaymode" )
155 v = displayMode( idx );
156 else if ( p == "isAutoColor" )
157 v = isAutoColor( idx );
158 else if ( p == "isVectorsMode" )
159 v = isVectorsMode( idx );
160 else if ( p == "topLevel" )
162 else if ( p == "hasChildren" )
163 v = hasChildren( idx );
164 else if ( p == "nbChildren" )
166 else if ( p == "hasConcealedChildren" )
167 v = hasConcealedChildren( idx );
168 else if ( p == "hasDisclosedChildren" )
169 v = hasDisclosedChildren( idx );
170 else if ( p == "compoundOfVertices" )
171 v = compoundOfVertices( idx );
172 else if ( p == "imported" )
173 v = isImported( idx );
174 else if ( p == "isPhysicalMaterial" )
175 v = isPhysicalMaterial(idx);
176 else if ( p == "isFolder" )
179 v = LightApp_Selection::parameter( idx, p );
184 // the method to skip temporary objects from selection (called from LightApp)
185 bool GEOMGUI_Selection::processOwner( const LightApp_DataOwner* theOwner )
187 return !theOwner->entry().contains("_");
190 QString GEOMGUI_Selection::typeName( const int index ) const
192 if ( isComponent( index ) )
194 if ( isFolder( index ) )
197 static QString aGroup( "Group" );
198 static QString aShape( "Shape" );
199 static QString anUnknown( "Unknown" );
201 GEOM::GEOM_Object_var anObj = getObject( index );
202 if ( !CORBA::is_nil( anObj ) ) {
203 const int aGeomType = anObj->GetType();
204 if ( aGeomType == GEOM_GROUP )
212 int GEOMGUI_Selection::typeId( const int index ) const
215 GEOM::GEOM_Object_var anObj = getObject( index );
216 if ( !CORBA::is_nil( anObj ) )
217 aType = (int)anObj->GetShapeType();
221 bool GEOMGUI_Selection::isVisible( const int index ) const
223 #ifdef USE_VISUAL_PROP_MAP
224 QVariant v = visibleProperty( entry( index ), GEOM::propertyName( GEOM::Visibility ) );
225 if ( v.canConvert( QVariant::Bool ) )
231 GEOM::GEOM_Object_var obj = getObject( index );
232 SALOME_View* view = GEOM_Displayer::GetActiveView();
233 if ( !CORBA::is_nil( obj ) && view ) {
234 Handle(SALOME_InteractiveObject) io = new SALOME_InteractiveObject( entry( index ).toLatin1().constData(), "GEOM", "TEMP_IO" );
235 res = view->isVisible( io );
241 bool GEOMGUI_Selection::isAutoColor( const int index ) const
243 GEOM::GEOM_Object_var obj = getObject( index );
244 if ( !CORBA::is_nil( obj ) )
245 return obj->GetAutoColor();
249 bool GEOMGUI_Selection::isImported( const int index ) const
251 GEOM::GEOM_Object_var obj = getObject( index );
252 if ( !CORBA::is_nil( obj ) )
253 return obj->GetType() == GEOM_IMPORT;
257 bool GEOMGUI_Selection::hasImported() const
260 for ( int i = 0; i < count() && !res; i++ )
261 res = isImported( i );
265 bool GEOMGUI_Selection::allImported() const
268 for ( int i = 0; i < count() && res; i++ )
269 res = isImported( i );
273 QVariant GEOMGUI_Selection::visibleProperty( const QString& entry, const QString& propName ) const
276 LightApp_Study* aStudy = study();
278 LightApp_Application* anApp = ::qobject_cast<LightApp_Application*>( aStudy->application() );
279 if ( anApp && anApp->activeViewManager() ) {
280 int id = anApp->activeViewManager()->getGlobalId();
281 v = aStudy->getObjectProperty( id, entry, propName, QVariant() );
287 QString GEOMGUI_Selection::displayMode( const int index ) const
290 QString viewType = activeViewType();
292 #ifdef USE_VISUAL_PROP_MAP
293 QVariant v = visibleProperty( entry( index ), GEOM::propertyName( GEOM::DisplayMode ) );
294 if ( v.canConvert( QVariant::Int ) ) {
296 if ( viewType == OCCViewer_Viewer::Type() ) {
297 OCC_DISPLAY_MODE_TO_STRING( res, dm );
298 } else if ( viewType == SVTK_Viewer::Type() ) {
299 VTK_DISPLAY_MODE_TO_STRING( res, dm );
305 SALOME_View* view = GEOM_Displayer::GetActiveView();
306 if ( view /*fix for 9320==>*/&& ( viewType == OCCViewer_Viewer::Type() || viewType == SVTK_Viewer::Type() ) ) {
307 SALOME_Prs* prs = view->CreatePrs( entry( index ).toLatin1().constData() );
309 if ( viewType == OCCViewer_Viewer::Type() ) { // assuming OCC
310 SOCC_Prs* occPrs = (SOCC_Prs*) prs;
311 AIS_ListOfInteractive lst;
312 occPrs->GetObjects( lst );
313 if ( lst.Extent() ) {
314 Handle(AIS_InteractiveObject) io = lst.First();
315 if ( !io.IsNull() ) {
317 Handle(GEOM_AISShape) aSh = Handle(GEOM_AISShape)::DownCast(io);
319 dm = aSh->isTopLevel() ? aSh->prevDisplayMode() : aSh->DisplayMode();
321 dm = io->DisplayMode();
323 OCC_DISPLAY_MODE_TO_STRING( res, dm );
324 if ( res.isEmpty() ) { // return default display mode of AIS_InteractiveContext
325 OCCViewer_Viewer* occViewer = (OCCViewer_Viewer*)SUIT_Session::session()->activeApplication()->
326 desktop()->activeWindow()->getViewManager()->getViewModel();
327 Handle(AIS_InteractiveContext) ic = occViewer->getAISContext();
328 dm = ic->DisplayMode();
329 OCC_DISPLAY_MODE_TO_STRING( res, dm );
334 else if ( viewType == SVTK_Viewer::Type() ) { // assuming VTK
335 SVTK_Prs* vtkPrs = dynamic_cast<SVTK_Prs*>( prs );
336 vtkActorCollection* lst = vtkPrs ? vtkPrs->GetObjects() : 0;
338 lst->InitTraversal();
339 vtkActor* actor = lst->GetNextActor();
341 SALOME_Actor* salActor = dynamic_cast<SALOME_Actor*>( actor );
343 int dm = salActor->getDisplayMode();
344 VTK_DISPLAY_MODE_TO_STRING( res, dm );
345 } // if ( salome actor )
347 } // if ( lst == vtkPrs->GetObjects() )
355 bool GEOMGUI_Selection::isVectorsMode( const int index ) const
357 #ifdef USE_VISUAL_PROP_MAP
358 QVariant v = visibleProperty( entry( index ), GEOM::propertyName( GEOM::EdgesDirection ) );
359 if ( v.canConvert( QVariant::Bool ) )
365 SALOME_View* view = GEOM_Displayer::GetActiveView();
366 QString viewType = activeViewType();
367 if ( view && ( viewType == OCCViewer_Viewer::Type() || viewType == SVTK_Viewer::Type() ) ) {
368 SALOME_Prs* prs = view->CreatePrs( entry( index ).toLatin1().constData() );
370 if ( viewType == OCCViewer_Viewer::Type() ) { // assuming OCC
371 SOCC_Prs* occPrs = (SOCC_Prs*) prs;
372 AIS_ListOfInteractive lst;
373 occPrs->GetObjects( lst );
374 if ( lst.Extent() ) {
375 Handle(AIS_InteractiveObject) io = lst.First();
376 if ( !io.IsNull() ) {
377 Handle(GEOM_AISShape) aSh = Handle(GEOM_AISShape)::DownCast(io);
379 res = aSh->isShowVectors();
383 else if ( viewType == SVTK_Viewer::Type() ) { // assuming VTK
384 SVTK_Prs* vtkPrs = dynamic_cast<SVTK_Prs*>( prs );
385 vtkActorCollection* lst = vtkPrs ? vtkPrs->GetObjects() : 0;
387 lst->InitTraversal();
388 vtkActor* actor = lst->GetNextActor();
390 GEOM_Actor* aGeomActor = GEOM_Actor::SafeDownCast(actor);
392 res = aGeomActor->GetVectorMode();
402 bool GEOMGUI_Selection::hasChildren( const _PTR(SObject)& obj )
404 // as soon as Use Case browser data tree was added
405 return obj->GetStudy()->GetUseCaseBuilder()->HasChildren( obj );
408 bool GEOMGUI_Selection::expandable( const _PTR(SObject)& obj )
411 _PTR(GenericAttribute) anAttr;
412 if ( obj && obj->FindAttribute( anAttr, "AttributeExpandable" ) ) {
413 _PTR(AttributeExpandable) aAttrExp = anAttr;
414 exp = aAttrExp->IsExpandable();
419 bool GEOMGUI_Selection::isCompoundOfVertices( GEOM::GEOM_Object_ptr obj )
423 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>
424 (SUIT_Session::session()->activeApplication()->activeStudy());*/
425 if ( /*appStudy && */!CORBA::is_nil( obj ) )
426 ret = obj->GetShapeType() == GEOM::COMPOUND && obj->GetMaxShapeType() == GEOM::VERTEX;
430 bool GEOMGUI_Selection::isFolder( const _PTR(SObject)& obj )
433 _PTR(GenericAttribute) anAttr;
434 if ( obj->FindAttribute(anAttr, "AttributeLocalID") ) {
435 _PTR(AttributeLocalID) aLocalID( anAttr );
436 ret = aLocalID->Value() == 999;
441 bool GEOMGUI_Selection::hasChildren( const int index ) const
444 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( study() );
447 QString anEntry = entry( index );
448 _PTR(Study) study = appStudy->studyDS();
449 if ( study && !anEntry.isEmpty() ) {
450 _PTR(SObject) aSO( study->FindObjectID( anEntry.toStdString() ) );
451 ok = hasChildren( aSO );
457 int GEOMGUI_Selection::nbChildren( const int index ) const
460 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( study() );
463 QString anEntry = entry( index );
464 _PTR(Study) study = appStudy->studyDS();
465 if ( study && !anEntry.isEmpty() ) {
466 _PTR(SObject) aSO( study->FindObjectID( anEntry.toStdString() ) );
467 if ( aSO->GetStudy()->GetUseCaseBuilder()->IsUseCaseNode(aSO) ) {
468 _PTR(UseCaseIterator) it = aSO->GetStudy()->GetUseCaseBuilder()->GetUseCaseIterator( aSO );
469 for (it->Init(false); it->More(); it->Next()) nb++;
476 bool GEOMGUI_Selection::hasConcealedChildren( const int index ) const
479 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( study() );
482 QString anEntry = entry( index );
483 _PTR(Study) study = appStudy->studyDS();
484 if ( study && !anEntry.isEmpty() ) {
485 _PTR(SObject) aSO( study->FindObjectID( anEntry.toStdString() ) );
486 OK = !expandable( aSO ) && hasChildren( aSO );
492 bool GEOMGUI_Selection::hasDisclosedChildren( const int index ) const
495 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( study() );
498 QString anEntry = entry( index );
499 _PTR(Study) study = appStudy->studyDS();
500 if ( study && !anEntry.isEmpty() ) {
501 _PTR(SObject) aSO( study->FindObjectID( anEntry.toStdString() ) );
502 OK = expandable( aSO ) && hasChildren( aSO );
508 bool GEOMGUI_Selection::compoundOfVertices( const int index ) const
510 GEOM::GEOM_Object_var obj = getObject( index );
511 return isCompoundOfVertices( obj );
514 bool GEOMGUI_Selection::isComponent( const int index ) const
516 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( study() );
519 QString anEntry = entry( index );
520 _PTR(Study) study = appStudy->studyDS();
521 if ( study && !anEntry.isNull() ) {
522 _PTR(SObject) aSO( study->FindObjectID( anEntry.toStdString() ) );
523 if ( aSO && aSO->GetFatherComponent() )
524 return aSO->GetFatherComponent()->GetIOR() == aSO->GetIOR();
530 GEOM::GEOM_Object_ptr GEOMGUI_Selection::getObject( const int index ) const
532 GEOM::GEOM_Object_var o;
533 if ( 0 <= index && index < myObjects.size() )
534 o = GEOM::GEOM_Object::_duplicate( myObjects[index] );
538 QString GEOMGUI_Selection::selectionMode() const
540 SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( study()->application() );
542 GeometryGUI* aGeomGUI = dynamic_cast<GeometryGUI*>( app->module( "Geometry" ) );
544 switch ( aGeomGUI->getLocalSelectionMode() )
546 case GEOM_POINT : return "VERTEX";
547 case GEOM_EDGE : return "EDGE";
548 case GEOM_WIRE : return "WIRE";
549 case GEOM_FACE : return "FACE";
550 case GEOM_SHELL : return "SHELL";
551 case GEOM_SOLID : return "SOLID";
552 case GEOM_COMPOUND : return "COMPOUND";
553 case GEOM_ALLOBJECTS : return "ALL";
561 bool GEOMGUI_Selection::topLevel( const int index ) const
563 #ifdef USE_VISUAL_PROP_MAP
564 QVariant v = visibleProperty( entry( index ), GEOM::propertyName( GEOM::TopLevel ) );
565 if ( v.canConvert<bool>() )
571 SALOME_View* view = GEOM_Displayer::GetActiveView();
572 QString viewType = activeViewType();
573 if ( view && viewType == OCCViewer_Viewer::Type() ) {
574 SALOME_Prs* prs = view->CreatePrs( entry( index ).toLatin1().constData() );
576 if ( viewType == OCCViewer_Viewer::Type() ) { // assuming OCC
577 SOCC_Prs* occPrs = (SOCC_Prs*) prs;
578 AIS_ListOfInteractive lst;
579 occPrs->GetObjects( lst );
580 if ( lst.Extent() ) {
581 Handle(AIS_InteractiveObject) io = lst.First();
582 if ( !io.IsNull() ) {
583 Handle(GEOM_AISShape) aSh = Handle(GEOM_AISShape)::DownCast(io);
585 res = (bool)aSh->isTopLevel();
594 bool GEOMGUI_Selection::isPhysicalMaterial( const int idx ) const
596 #ifdef USE_VISUAL_PROP_MAP
597 QVariant v = visibleProperty( entry( idx ), GEOM::propertyName( GEOM::Material ) );
598 if ( v.canConvert<QString>() ) {
599 Material_Model material;
600 material.fromProperties( v.toString() );
601 return material.isPhysical();
607 SALOME_View* view = GEOM_Displayer::GetActiveView();
608 QString viewType = activeViewType();
610 SALOME_Prs* prs = view->CreatePrs( entry( idx ).toLatin1().constData() );
612 if ( viewType == OCCViewer_Viewer::Type() ) { // assuming OCC
613 SOCC_Prs* occPrs = (SOCC_Prs*) prs;
614 AIS_ListOfInteractive lst;
615 occPrs->GetObjects( lst );
616 if ( lst.Extent() ) {
617 Handle(AIS_InteractiveObject) io = lst.First();
618 if ( !io.IsNull() ) {
619 Handle(GEOM_AISShape) aSh = Handle(GEOM_AISShape)::DownCast(io);
621 res = (bool) aSh->Attributes()->ShadingAspect()->
622 Material(Aspect_TOFM_BOTH_SIDE).MaterialType( Graphic3d_MATERIAL_PHYSIC );
626 else if ( viewType == SVTK_Viewer::Type() ) { // assuming VTK
627 SVTK_Prs* vtkPrs = dynamic_cast<SVTK_Prs*>( prs );
628 vtkActorCollection* lst = vtkPrs ? vtkPrs->GetObjects() : 0;
630 lst->InitTraversal();
631 vtkActor* actor = lst->GetNextActor();
633 GEOM_Actor* aGeomGActor = GEOM_Actor::SafeDownCast( actor );
635 GEOM_VTKPropertyMaterial* mat = GEOM_VTKPropertyMaterial::SafeDownCast(aGeomGActor->GetProperty());
636 res = mat->GetPhysical();
637 } // if ( salome actor )
639 } // if ( lst == vtkPrs->GetObjects() )
647 bool GEOMGUI_Selection::isFolder( const int index ) const
650 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( study() );
653 QString anEntry = entry( index );
654 _PTR(Study) study = appStudy->studyDS();
655 if ( study && !anEntry.isNull() ) {
656 _PTR(SObject) aSO( study->FindObjectID( anEntry.toStdString() ) );
657 if ( aSO ) res = isFolder( aSO );