]> SALOME platform Git repositories - modules/geom.git/blob - src/DisplayGUI/DisplayGUI.cxx
Salome HOME
Additional fix for bug NPAL19028 (see remarks from Olivier Giorgis).
[modules/geom.git] / src / DisplayGUI / DisplayGUI.cxx
1 //  GEOM GEOMGUI : GUI for Geometry component
2 //
3 //  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS 
5 // 
6 //  This library is free software; you can redistribute it and/or 
7 //  modify it under the terms of the GNU Lesser General Public 
8 //  License as published by the Free Software Foundation; either 
9 //  version 2.1 of the License. 
10 // 
11 //  This library is distributed in the hope that it will be useful, 
12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of 
13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
14 //  Lesser General Public License for more details. 
15 // 
16 //  You should have received a copy of the GNU Lesser General Public 
17 //  License along with this library; if not, write to the Free Software 
18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA 
19 // 
20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 //
23 //
24 //  File   : DisplayGUI.cxx
25 //  Author : Vadim SANDLER
26 //  Module : GEOM
27 //  $Header$
28
29 #include "DisplayGUI.h"
30 #include "GeometryGUI.h"
31 #include "GEOM_Displayer.h"
32
33 #include <SUIT_Desktop.h>
34 #include <SUIT_Session.h>
35 #include <SUIT_ViewWindow.h>
36 #include <SUIT_OverrideCursor.h>
37
38 #include <OCCViewer_ViewManager.h>
39 #include <OCCViewer_ViewModel.h>
40 #include <OCCViewer_ViewWindow.h>
41
42 #include <SALOME_ListIteratorOfListIO.hxx>
43
44 #include <SVTK_ViewWindow.h>
45 #include <SVTK_View.h>
46 #include <SVTK_ViewModel.h>
47 #include <SOCC_ViewModel.h>
48 #include <SVTK_Prs.h>
49 #include <SOCC_Prs.h>
50
51 #include <QtxActionMenuMgr.h>
52
53 #include <SalomeApp_Application.h>
54 #include <LightApp_SelectionMgr.h>
55 #include <SalomeApp_Study.h>
56
57 #include <AIS_ListIteratorOfListOfInteractive.hxx>
58
59 #include <qmenubar.h>
60
61
62 //=======================================================================
63 // function : DisplayGUI::DisplayGUI()
64 // purpose  : Constructor
65 //=======================================================================
66 DisplayGUI::DisplayGUI( GeometryGUI* parent ) : GEOMGUI( parent )
67 {
68 }
69
70 //=======================================================================
71 // function : DisplayGUI::~DisplayGUI()
72 // purpose  : Destructor
73 //=======================================================================
74 DisplayGUI::~DisplayGUI()
75 {
76 }
77
78
79 //=======================================================================
80 // function : DisplayGUI::OnGUIEvent()
81 // purpose  : Dispatch menu command
82 //=======================================================================
83 bool DisplayGUI::OnGUIEvent(int theCommandID, SUIT_Desktop* parent)
84 {
85   SalomeApp_Application* app = getGeometryGUI()->getApp();
86   if (!app) return false;
87
88   LightApp_SelectionMgr *Sel = app->selectionMgr();
89   SALOME_ListIO selected;
90   Sel->selectedObjects( selected );
91
92   switch (theCommandID) {
93   case 211: // MENU VIEW - WIREFRAME/SHADING
94     {
95       InvertDisplayMode();
96       int newMode = GetDisplayMode();
97       getGeometryGUI()->action( 211 )->setMenuText
98         ( newMode == 1 ? tr( "GEOM_MEN_WIREFRAME" ) : tr("GEOM_MEN_SHADING") );
99       getGeometryGUI()->menuMgr()->update();
100       break;
101     }
102   case 212: // MENU VIEW - DISPLAY ALL
103     {
104       getGeometryGUI()->EmitSignalDeactivateDialog();
105       DisplayAll();
106       break;
107     }
108   case 213: // MENU VIEW - DISPLAY ONLY
109     {
110       getGeometryGUI()->EmitSignalDeactivateDialog();
111       DisplayOnly();
112       break;
113     }
114   case 214: // MENU VIEW - ERASE ALL
115     {
116       EraseAll();
117       break;
118     }
119   case 215: // MENU VIEW - ERASE
120     {
121       Erase();
122       break;
123     }
124   case 216: // MENU VIEW - DISPLAY
125     {
126       getGeometryGUI()->EmitSignalDeactivateDialog();
127       Display();
128       break;
129     }
130   case 80311: // POPUP VIEWER - WIREFRAME
131     {
132       ChangeDisplayMode( 0 );
133       break;
134     }
135   case 80312: // POPUP VIEWER - SHADING
136     {
137       ChangeDisplayMode( 1 );
138       break;
139     }
140   default:
141     {
142       app->putInfo(tr("GEOM_PRP_COMMAND").arg(theCommandID));
143       break;
144     }
145   }
146   Sel->setSelectedObjects( selected );
147   return true;
148 }
149
150 //=====================================================================================
151 // function : DisplayGUI::DisplayAll()
152 // purpose  : Display all GEOM objects
153 //=====================================================================================
154 void DisplayGUI::DisplayAll()
155 {
156   SalomeApp_Application* app = getGeometryGUI()->getApp();
157   if ( !app ) return;
158
159   SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
160   if ( !appStudy ) return;
161   _PTR(Study) aStudy = appStudy->studyDS();
162   if ( !aStudy ) return;
163   _PTR(SComponent) SC ( aStudy->FindComponent( "GEOM" ) );
164   if ( !SC )
165     return;
166
167   SALOME_ListIO listIO;
168   _PTR(ChildIterator) anIter ( aStudy->NewChildIterator( SC ) );
169   anIter->InitEx( true );
170
171   SUIT_OverrideCursor();
172
173   while( anIter->More() ) {
174     _PTR(SObject) valSO ( anIter->Value() );
175     _PTR(SObject) refSO;
176     if ( !valSO->ReferencedObject( refSO ) ) {
177       listIO.Append( new SALOME_InteractiveObject(valSO->GetID().c_str(),
178                                                   SC->ComponentDataType().c_str(),
179                                                   valSO->GetName().c_str()) );
180     } 
181     anIter->Next();
182   }
183   GEOM_Displayer( appStudy ).Display( listIO, true );
184 }
185
186 //=====================================================================================
187 // function : DisplayGUI::EraseAll()
188 // purpose  : Erase all GEOM objects
189 //=====================================================================================
190 void DisplayGUI::EraseAll()
191 {
192   SUIT_OverrideCursor();
193
194   SUIT_Application* app = getGeometryGUI()->getApp();
195   if ( app ) {
196     SUIT_ViewWindow* vw = app->desktop()->activeWindow();
197     if ( vw ) {
198       SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
199       SUIT_ViewManager* vman = vw->getViewManager();
200       if ( vman->getType() == OCCViewer_Viewer::Type() || 
201            vman->getType() == SVTK_Viewer::Type() ) {
202         GEOM_Displayer( appStudy ).EraseAll();
203       }
204     }
205   }
206 }
207
208 //=====================================================================================
209 // function : DisplayGUI::DisplayOnly()
210 // purpose  : Display selected GEOM objects and erase other
211 //=====================================================================================
212 void DisplayGUI::DisplayOnly()
213 {
214   EraseAll();
215   Display();
216 }
217
218 //=====================================================================================
219 // function : DisplayGUI::Display()
220 // purpose  : Display selected GEOM objects
221 //=====================================================================================
222 void DisplayGUI::Display()
223 {
224   SALOME_ListIO listIO;
225
226   SalomeApp_Application* app = getGeometryGUI()->getApp();
227   if ( !app ) return;
228
229   SalomeApp_Study* anActiveStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
230   if ( !anActiveStudy ) return;
231
232   //get SalomeApp selection manager
233   LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
234   if ( !aSelMgr ) return;
235
236   SALOME_ListIO aList;
237   aSelMgr->selectedObjects( aList );
238   SALOME_ListIteratorOfListIO It( aList );
239
240   SUIT_OverrideCursor();
241
242   for( ;It.More();It.Next() ) {
243     Handle(SALOME_InteractiveObject) anIObject = It.Value();
244     if ( anIObject->hasEntry() ) {
245       _PTR(SObject) SO ( anActiveStudy->studyDS()->FindObjectID( anIObject->getEntry() ) );
246       if ( SO && QString(SO->GetID().c_str()) == QString(SO->GetFatherComponent()->GetID().c_str()) ) {
247         _PTR(SComponent) SC ( SO->GetFatherComponent() );
248         // if component is selected
249         listIO.Clear();
250         _PTR(ChildIterator) anIter ( anActiveStudy->studyDS()->NewChildIterator( SO ) );
251         anIter->InitEx( true );
252         while( anIter->More() ) {
253           _PTR(SObject) valSO ( anIter->Value() );
254           _PTR(SObject) refSO;
255           if ( !valSO->ReferencedObject( refSO ) ) {
256             listIO.Append( new SALOME_InteractiveObject(valSO->GetID().c_str(),
257                                                         SC->ComponentDataType().c_str(),
258                                                         valSO->GetName().c_str()) );
259           }
260           anIter->Next();
261         }
262         break;
263       }
264       else {
265         listIO.Append( anIObject );
266       }
267     }
268     else {
269       listIO.Append( anIObject );
270     }
271   }
272   GEOM_Displayer( anActiveStudy ).Display( listIO, true );
273 }
274
275
276 //=====================================================================================
277 // function : DisplayGUI::Erase()
278 // purpose  : Erase selected GEOM objects
279 //=====================================================================================
280 void DisplayGUI::Erase()
281 {
282   SALOME_ListIO listIO;
283
284   SalomeApp_Application* app = getGeometryGUI()->getApp();
285   if ( !app ) return;
286
287   SalomeApp_Study* anActiveStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
288   if ( !anActiveStudy ) return;
289
290   //get SalomeApp selection manager
291   LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
292   if ( !aSelMgr ) return;
293
294   SALOME_ListIO aList;
295   aSelMgr->selectedObjects( aList );
296   SALOME_ListIteratorOfListIO It( aList );
297
298   SUIT_OverrideCursor();
299
300   for( ; It.More(); It.Next() ) {
301     Handle(SALOME_InteractiveObject) anIObject = It.Value();
302     if ( anIObject->hasEntry() ) {
303       _PTR(SObject) SO ( anActiveStudy->studyDS()->FindObjectID( anIObject->getEntry() ) );
304       if ( SO && QString(SO->GetID().c_str()) == QString(SO->GetFatherComponent()->GetID().c_str()) ) {
305         _PTR(SComponent) SC ( SO->GetFatherComponent() );
306         // if component is selected
307         listIO.Clear();
308         _PTR(ChildIterator) anIter ( anActiveStudy->studyDS()->NewChildIterator( SO ) );
309         anIter->InitEx( true );
310         while( anIter->More() ) {
311           _PTR(SObject) valSO ( anIter->Value() );
312           _PTR(SObject) refSO;
313           if ( !valSO->ReferencedObject( refSO ) ) {
314             listIO.Append( new SALOME_InteractiveObject(valSO->GetID().c_str(),
315                                                         SC->ComponentDataType().c_str(),
316                                                         valSO->GetName().c_str()) );
317           }
318           anIter->Next();
319         }
320         break;
321       }
322       else {
323         listIO.Append( anIObject );
324       }
325     }
326     else {
327       listIO.Append( anIObject );
328     }
329   }
330
331   SUIT_ViewWindow* viewWindow = app->desktop()->activeWindow();
332   bool aIsForced = true;
333   if(viewWindow->getViewManager()->getType() == SVTK_Viewer::Type())
334     aIsForced = false;
335
336   GEOM_Displayer(anActiveStudy).Erase( listIO, aIsForced);
337   getGeometryGUI()->getApp()->selectionMgr()->clearSelected();
338 }
339
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 )
345 {
346   SUIT_OverrideCursor();
347
348   if ( !viewWindow ) 
349     viewWindow = getGeometryGUI()->getApp()->desktop()->activeWindow();
350   if ( viewWindow->getViewManager()->getType() == SVTK_Viewer::Type() ) {
351     SVTK_View* aView = ((SVTK_ViewWindow*)viewWindow)->getView();
352     aView->SetDisplayMode( mode );
353   } 
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 );
363
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 );
369       }
370       ite.Next();
371     }
372
373     ic->SetDisplayMode( newmode, Standard_False );
374   }
375 }
376
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 )
382 {
383   int dispMode = 0;
384   if ( !viewWindow ) 
385     viewWindow = getGeometryGUI()->getApp()->desktop()->activeWindow();
386   if ( viewWindow->getViewManager()->getType() == SVTK_Viewer::Type() ) {
387     SVTK_View* aView = ((SVTK_ViewWindow*)viewWindow)->getView();
388     dispMode = aView->GetDisplayMode();
389   } 
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 );
395   }
396   return dispMode;
397 }
398
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 )
405 {
406   SetDisplayMode( 1 - GetDisplayMode( viewWindow ) );
407 }
408
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 )
415 {
416   SalomeApp_Application* app = getGeometryGUI()->getApp();
417   if ( !app ) return;
418
419   if ( !viewWindow ) 
420     viewWindow = app->desktop()->activeWindow();
421
422   LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
423   if ( !aSelMgr ) return;
424
425   SUIT_OverrideCursor();
426
427   SALOME_ListIO aList;
428
429   if ( viewWindow->getViewManager()->getType() == SVTK_Viewer::Type() ) {
430     SVTK_ViewWindow* vw = dynamic_cast<SVTK_ViewWindow*>( viewWindow );
431     SVTK_View* aView = vw->getView();
432
433     aSelMgr->selectedObjects( aList );
434     SALOME_ListIteratorOfListIO It( aList );
435
436     for( ;It.More(); It.Next() ) {
437       SVTK_Viewer* stvkViewer = dynamic_cast<SVTK_Viewer*>(vw->getViewManager()->getViewModel());
438       SVTK_Prs* vtkPrs =
439         stvkViewer ? dynamic_cast<SVTK_Prs*>( stvkViewer->CreatePrs( It.Value()->getEntry() ) ) : 0;
440       if ( vtkPrs && !vtkPrs->IsNull() ) {
441         if ( mode == 0 )
442           aView->ChangeRepresentationToWireframe( vtkPrs->GetObjects() );
443         else if ( mode == 1 )
444           aView->ChangeRepresentationToSurface( vtkPrs->GetObjects() );
445       }
446     }
447     aView->Repaint();
448   }
449   else if ( viewWindow->getViewManager()->getType() == OCCViewer_Viewer::Type() ) {
450     OCCViewer_Viewer* v3d = ((OCCViewer_ViewManager*)(viewWindow->getViewManager()))->getOCCViewer();
451     Handle(AIS_InteractiveContext) ic = v3d->getAISContext();
452
453     aSelMgr->selectedObjects( aList );
454     SALOME_ListIteratorOfListIO It( aList );
455
456     for( ;It.More(); It.Next() ) {
457       SOCC_Viewer* soccViewer = (SOCC_Viewer*)(viewWindow->getViewManager()->getViewModel());
458       SOCC_Prs* occPrs = dynamic_cast<SOCC_Prs*>( soccViewer->CreatePrs( It.Value()->getEntry() ) );
459       if ( occPrs && !occPrs->IsNull() ) {
460         AIS_ListOfInteractive shapes; occPrs->GetObjects( shapes );
461         AIS_ListIteratorOfListOfInteractive interIter( shapes );
462         for ( ; interIter.More(); interIter.Next() ) {
463           if ( mode == 0 )
464             ic->SetDisplayMode( interIter.Value(), AIS_WireFrame, false );
465           else if ( mode == 1 )
466             ic->SetDisplayMode( interIter.Value(), AIS_Shaded, false );
467         }
468       }
469     }
470     ic->UpdateCurrentViewer();
471   }
472 }
473
474 //=====================================================================================
475 // EXPORTED METHODS
476 //=====================================================================================
477 extern "C"
478 {
479  GEOM_DISPLAYGUI_EXPORT
480   GEOMGUI* GetLibGUI( GeometryGUI* parent )
481   {
482     return new DisplayGUI( parent );
483   }
484 }