1 // GEOM GEOMGUI : GUI for Geometry component
3 // Copyright (C) 2003 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
24 // File : DisplayGUI.cxx
25 // Author : Vadim SANDLER
29 #include "DisplayGUI.h"
30 #include "GeometryGUI.h"
31 #include "GEOM_Displayer.h"
33 #include <SUIT_Desktop.h>
34 #include <SUIT_Session.h>
35 #include <SUIT_ViewWindow.h>
36 #include <SUIT_OverrideCursor.h>
38 #include <OCCViewer_ViewManager.h>
39 #include <OCCViewer_ViewModel.h>
40 #include <OCCViewer_ViewWindow.h>
42 #include <SALOME_ListIteratorOfListIO.hxx>
44 #include <SVTK_ViewWindow.h>
45 #include <SVTK_View.h>
46 #include <SVTK_ViewModel.h>
47 #include <SOCC_ViewModel.h>
51 #include <QtxActionMenuMgr.h>
53 #include <SalomeApp_Application.h>
54 #include <LightApp_SelectionMgr.h>
55 #include <SalomeApp_Study.h>
57 #include <AIS_ListIteratorOfListOfInteractive.hxx>
62 DisplayGUI* DisplayGUI::myGUIObject = 0;
64 //=======================================================================
65 // function : DisplayGUI::GetDisplayGUI()
66 // purpose : Get the only DisplayGUI object [ static ]
67 //=======================================================================
68 DisplayGUI* DisplayGUI::GetDisplayGUI( GeometryGUI* parent )
70 if ( myGUIObject == 0 ) {
71 // init DisplayGUI only once
72 myGUIObject = new DisplayGUI( parent );
77 //=======================================================================
78 // function : DisplayGUI::DisplayGUI()
79 // purpose : Constructor
80 //=======================================================================
81 DisplayGUI::DisplayGUI( GeometryGUI* parent ) : GEOMGUI( parent )
86 //=======================================================================
87 // function : DisplayGUI::~DisplayGUI()
88 // purpose : Destructor
89 //=======================================================================
90 DisplayGUI::~DisplayGUI()
95 //=======================================================================
96 // function : DisplayGUI::OnGUIEvent()
97 // purpose : Dispatch menu command
98 //=======================================================================
99 bool DisplayGUI::OnGUIEvent(int theCommandID, SUIT_Desktop* parent)
101 DisplayGUI* myDisplayGUI = GetDisplayGUI( getGeometryGUI() );
102 LightApp_SelectionMgr *Sel = getGeometryGUI()->getApp()->selectionMgr();
103 SALOME_ListIO selected;
104 Sel->selectedObjects( selected );
106 switch (theCommandID) {
107 case 211: // MENU VIEW - WIREFRAME/SHADING
109 myDisplayGUI->InvertDisplayMode();
110 int newMode = myDisplayGUI->GetDisplayMode();
111 getGeometryGUI()->action( 211 )->setMenuText( newMode == 1 ? tr( "GEOM_MEN_WIREFRAME" ) : tr("GEOM_MEN_SHADING") );
112 getGeometryGUI()->menuMgr()->update();
113 // SUIT_Session::session()->activeApplication()->desktop()->menuBar()->
114 // changeItem( 211, newMode == 1 ? tr( "GEOM_MEN_WIREFRAME" ) : tr("GEOM_MEN_SHADING") );
117 case 212: // MENU VIEW - DISPLAY ALL
119 getGeometryGUI()->EmitSignalDeactivateDialog();
120 myDisplayGUI->DisplayAll();
123 case 213: // MENU VIEW - DISPLAY ONLY
125 getGeometryGUI()->EmitSignalDeactivateDialog();
126 myDisplayGUI->DisplayOnly();
129 case 214: // MENU VIEW - ERASE ALL
131 myDisplayGUI->EraseAll();
134 case 215: // MENU VIEW - ERASE
136 myDisplayGUI->Erase();
139 case 216: // MENU VIEW - DISPLAY
141 getGeometryGUI()->EmitSignalDeactivateDialog();
142 myDisplayGUI->Display();
145 case 80311: // POPUP VIEWER - WIREFRAME
147 myDisplayGUI->ChangeDisplayMode( 0 );
150 case 80312: // POPUP VIEWER - SHADING
152 myDisplayGUI->ChangeDisplayMode( 1 );
157 SUIT_Session::session()->activeApplication()->putInfo(tr("GEOM_PRP_COMMAND").arg(theCommandID));
161 Sel->setSelectedObjects( selected );
165 //=====================================================================================
166 // function : DisplayGUI::DisplayAll()
167 // purpose : Display all GEOM objects
168 //=====================================================================================
169 void DisplayGUI::DisplayAll()
171 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( SUIT_Session::session()->activeApplication()->activeStudy() );
172 if ( !appStudy ) return;
173 _PTR(Study) aStudy = appStudy->studyDS();
174 if ( !aStudy ) return;
175 _PTR(SComponent) SC ( aStudy->FindComponent( "GEOM" ) );
179 SALOME_ListIO listIO;
180 _PTR(ChildIterator) anIter ( aStudy->NewChildIterator( SC ) );
181 anIter->InitEx( true );
183 SUIT_OverrideCursor();
185 while( anIter->More() ) {
186 _PTR(SObject) valSO ( anIter->Value() );
188 if ( !valSO->ReferencedObject( refSO ) ) {
189 listIO.Append( new SALOME_InteractiveObject( valSO->GetID().c_str(), SC->ComponentDataType().c_str() ,valSO->GetName().c_str() ) );
193 GEOM_Displayer( appStudy ).Display( listIO, true );
196 //=====================================================================================
197 // function : DisplayGUI::EraseAll()
198 // purpose : Erase all GEOM objects
199 //=====================================================================================
200 void DisplayGUI::EraseAll()
202 SUIT_OverrideCursor();
204 SUIT_Application* app = SUIT_Session::session()->activeApplication();
206 SUIT_ViewWindow* vw = app->desktop()->activeWindow();
208 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
209 SUIT_ViewManager* vman = vw->getViewManager();
210 if ( vman->getType() == OCCViewer_Viewer::Type() ||
211 vman->getType() == SVTK_Viewer::Type() ) {
212 GEOM_Displayer( appStudy ).EraseAll();
218 //=====================================================================================
219 // function : DisplayGUI::DisplayOnly()
220 // purpose : Display selected GEOM objects and erase other
221 //=====================================================================================
222 void DisplayGUI::DisplayOnly()
228 //=====================================================================================
229 // function : DisplayGUI::Display()
230 // purpose : Display selected GEOM objects
231 //=====================================================================================
232 void DisplayGUI::Display()
234 SALOME_ListIO listIO;
236 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
239 SalomeApp_Study* anActiveStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
240 if ( !anActiveStudy ) return;
242 //get SalomeApp selection manager
243 LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
244 if ( !aSelMgr ) return;
247 aSelMgr->selectedObjects( aList );
248 SALOME_ListIteratorOfListIO It( aList );
250 SUIT_OverrideCursor();
252 for( ;It.More();It.Next() ) {
253 Handle(SALOME_InteractiveObject) anIObject = It.Value();
254 if ( anIObject->hasEntry() ) {
255 _PTR(SObject) SO ( anActiveStudy->studyDS()->FindObjectID( anIObject->getEntry() ) );
256 if ( SO && QString( SO->GetID().c_str() ) == QString( SO->GetFatherComponent()->GetID().c_str() ) ) {
257 _PTR(SComponent) SC ( SO->GetFatherComponent() );
258 // if component is selected
260 _PTR(ChildIterator) anIter ( anActiveStudy->studyDS()->NewChildIterator( SO ) );
261 anIter->InitEx( true );
262 while( anIter->More() ) {
263 _PTR(SObject) valSO ( anIter->Value() );
265 if ( !valSO->ReferencedObject( refSO ) ) {
266 listIO.Append( new SALOME_InteractiveObject( valSO->GetID().c_str(), SC->ComponentDataType().c_str() ,valSO->GetName().c_str() ) );
273 listIO.Append( anIObject );
277 listIO.Append( anIObject );
280 GEOM_Displayer( anActiveStudy ).Display( listIO, true );
284 //=====================================================================================
285 // function : DisplayGUI::Erase()
286 // purpose : Erase selected GEOM objects
287 //=====================================================================================
288 void DisplayGUI::Erase()
290 SALOME_ListIO listIO;
292 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
295 SalomeApp_Study* anActiveStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
296 if ( !anActiveStudy ) return;
298 //get SalomeApp selection manager
299 LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
300 if ( !aSelMgr ) return;
303 aSelMgr->selectedObjects( aList );
304 SALOME_ListIteratorOfListIO It( aList );
306 SUIT_OverrideCursor();
308 for( ;It.More();It.Next() ) {
309 Handle(SALOME_InteractiveObject) anIObject = It.Value();
310 if ( anIObject->hasEntry() ) {
311 _PTR(SObject) SO ( anActiveStudy->studyDS()->FindObjectID( anIObject->getEntry() ) );
312 if ( SO && QString( SO->GetID().c_str() ) == QString( SO->GetFatherComponent()->GetID().c_str() ) ) {
313 _PTR(SComponent) SC ( SO->GetFatherComponent() );
314 // if component is selected
316 _PTR(ChildIterator) anIter ( anActiveStudy->studyDS()->NewChildIterator( SO ) );
317 anIter->InitEx( true );
318 while( anIter->More() ) {
319 _PTR(SObject) valSO ( anIter->Value() );
321 if ( !valSO->ReferencedObject( refSO ) ) {
322 listIO.Append( new SALOME_InteractiveObject( valSO->GetID().c_str(), SC->ComponentDataType().c_str() ,valSO->GetName().c_str() ) );
329 listIO.Append( anIObject );
333 listIO.Append( anIObject );
336 GEOM_Displayer(anActiveStudy).Erase( listIO, true );
337 ((SalomeApp_Application*)(SUIT_Session::session()->activeApplication()))->selectionMgr()->clearSelected();
340 //=====================================================================================
341 // function : DisplayGUI::SetDisplayMode()
342 // purpose : Set display mode for the viewer (current viewer if <viewWindow> - 0 )
343 //=====================================================================================
344 void DisplayGUI::SetDisplayMode( const int mode, SUIT_ViewWindow* viewWindow )
346 SUIT_OverrideCursor();
349 viewWindow = SUIT_Session::session()->activeApplication()->desktop()->activeWindow();
350 if ( viewWindow->getViewManager()->getType() == SVTK_Viewer::Type() ) {
351 SVTK_View* aView = ((SVTK_ViewWindow*)viewWindow)->getView();
352 aView->SetDisplayMode( mode );
354 else if ( viewWindow->getViewManager()->getType() == OCCViewer_Viewer::Type() ) {
355 OCCViewer_Viewer* v3d = ((OCCViewer_ViewManager*)(viewWindow->getViewManager()))->getOCCViewer();
356 Handle(AIS_InteractiveContext) ic = v3d->getAISContext();
357 AIS_DisplayMode newmode = (mode == 1 ? AIS_Shaded : AIS_WireFrame);
358 AIS_ListOfInteractive List;
359 ic->DisplayedObjects( List );
360 AIS_ListOfInteractive List1;
361 ic->ObjectsInCollector( List1 );
362 List.Append( List1 );
364 AIS_ListIteratorOfListOfInteractive ite( List );
365 while( ite.More() ) {
366 if( ite.Value()->IsInstance( STANDARD_TYPE(GEOM_AISShape) ) ) {
367 Handle(GEOM_AISShape) aSh = Handle(GEOM_AISShape)::DownCast( ite.Value() );
368 ic->SetDisplayMode( aSh, Standard_Integer( newmode ),true );
373 ic->SetDisplayMode( newmode, Standard_False );
377 //=====================================================================================
378 // function : DisplayGUI::GetDisplayMode()
379 // purpose : Get display mode of the viewer (current viewer if <viewWindow> - 0 )
380 //=====================================================================================
381 int DisplayGUI::GetDisplayMode( SUIT_ViewWindow* viewWindow )
385 viewWindow = SUIT_Session::session()->activeApplication()->desktop()->activeWindow();
386 if ( viewWindow->getViewManager()->getType() == SVTK_Viewer::Type() ) {
387 SVTK_View* aView = ((SVTK_ViewWindow*)viewWindow)->getView();
388 dispMode = aView->GetDisplayMode();
390 else if ( viewWindow->getViewManager()->getType() == OCCViewer_Viewer::Type() ) {
391 OCCViewer_Viewer* v3d = ((OCCViewer_ViewManager*)(viewWindow->getViewManager()))->getOCCViewer();
392 Handle(AIS_InteractiveContext) ic = v3d->getAISContext();
393 AIS_DisplayMode mode = (AIS_DisplayMode)ic->DisplayMode();
394 dispMode = (mode == AIS_WireFrame ? 0 : 1 );
399 //=====================================================================================
400 // function : DisplayGUI::InvertDisplayMode()
401 // purpose : Invert display mode ( shadin <-> wireframe ) for the viewer
402 // (current viewer if <viewWindow> = 0 )
403 //=====================================================================================
404 void DisplayGUI::InvertDisplayMode( SUIT_ViewWindow* viewWindow )
406 SetDisplayMode( 1 - GetDisplayMode( viewWindow ) );
409 //=====================================================================================
410 // function : DisplayGUI::ChangeDisplayMode()
411 // purpose : Set display mode for selected objects in the viewer given
412 // (current viewer if <viewWindow> = 0 )
413 //=====================================================================================
414 void DisplayGUI::ChangeDisplayMode( const int mode, SUIT_ViewWindow* viewWindow )
417 viewWindow = SUIT_Session::session()->activeApplication()->desktop()->activeWindow();
419 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
422 LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
423 if ( !aSelMgr ) return;
425 SUIT_OverrideCursor();
429 if ( viewWindow->getViewManager()->getType() == SVTK_Viewer::Type() ) {
430 SVTK_ViewWindow* vw = dynamic_cast<SVTK_ViewWindow*>( viewWindow );
431 SVTK_View* aView = vw->getView();
433 aSelMgr->selectedObjects( aList );
434 SALOME_ListIteratorOfListIO It( aList );
436 for( ;It.More(); It.Next() ) {
437 SVTK_Viewer* stvkViewer = dynamic_cast<SVTK_Viewer*>(vw->getViewManager()->getViewModel());
438 SVTK_Prs* vtkPrs = stvkViewer ? dynamic_cast<SVTK_Prs*>( stvkViewer->CreatePrs( It.Value()->getEntry() ) ) : 0;
439 if ( vtkPrs && !vtkPrs->IsNull() ) {
441 aView->ChangeRepresentationToWireframe( vtkPrs->GetObjects() );
442 else if ( mode == 1 )
443 aView->ChangeRepresentationToSurface( vtkPrs->GetObjects() );
448 else if ( viewWindow->getViewManager()->getType() == OCCViewer_Viewer::Type() ) {
449 OCCViewer_Viewer* v3d = ((OCCViewer_ViewManager*)(viewWindow->getViewManager()))->getOCCViewer();
450 Handle(AIS_InteractiveContext) ic = v3d->getAISContext();
452 aSelMgr->selectedObjects( aList );
453 SALOME_ListIteratorOfListIO It( aList );
455 for( ;It.More(); It.Next() ) {
456 SOCC_Viewer* soccViewer = (SOCC_Viewer*)(viewWindow->getViewManager()->getViewModel());
457 SOCC_Prs* occPrs = dynamic_cast<SOCC_Prs*>( soccViewer->CreatePrs( It.Value()->getEntry() ) );
458 if ( occPrs && !occPrs->IsNull() ) {
459 AIS_ListOfInteractive shapes; occPrs->GetObjects( shapes );
460 AIS_ListIteratorOfListOfInteractive interIter( shapes );
461 for ( ; interIter.More(); interIter.Next() ) {
463 ic->SetDisplayMode( interIter.Value(), AIS_WireFrame, false );
464 else if ( mode == 1 )
465 ic->SetDisplayMode( interIter.Value(), AIS_Shaded, false );
469 ic->UpdateCurrentViewer();
473 //=====================================================================================
475 //=====================================================================================
479 __declspec( dllexport )
481 GEOMGUI* GetLibGUI( GeometryGUI* parent )
483 return DisplayGUI::GetDisplayGUI( parent );