Salome HOME
Hide objects from other modules if hide all requested (Bug #355).
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_ShowHideOp.cxx
1 // Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  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 #include "HYDROGUI_ShowHideOp.h"
24
25 #include <GraphicsView_Viewer.h>
26
27 #include "HYDROGUI_Module.h"
28 #include "HYDROGUI_Operations.h"
29 #include "HYDROGUI_Tool.h"
30 #include "HYDROGUI_UpdateFlags.h"
31
32 #include <HYDROData_Iterator.h>
33 #include <HYDROData_Entity.h>
34 #include <HYDROData_Region.h>
35
36 #include <LightApp_Application.h>
37 #include <LightApp_Study.h>
38
39 #include <OCCViewer_ViewModel.h>
40 #include <OCCViewer_ViewManager.h>
41
42 #include <SVTK_ViewModel.h>
43
44 #include <SUIT_ViewManager.h>
45
46 HYDROGUI_ShowHideOp::HYDROGUI_ShowHideOp( HYDROGUI_Module* theModule, int theId )
47 : HYDROGUI_Operation( theModule ),
48   myId( theId )
49 {
50   QString aName;
51   switch( myId )
52   {
53     case ShowId:     aName = tr( "SHOW" ); break;
54     case ShowOnlyId: aName = tr( "SHOW_ONLY" ); break;
55     case ShowAllId:  aName = tr( "SHOW_ALL" ); break;
56     case HideId:     aName = tr( "HIDE" ); break;
57     case HideAllId:  aName = tr( "HIDE_ALL" ); break;
58     default: break;
59   }
60   setName( aName );
61 }
62
63 HYDROGUI_ShowHideOp::~HYDROGUI_ShowHideOp()
64 {
65 }
66
67 void HYDROGUI_ShowHideOp::startOperation()
68 {
69   HYDROGUI_Operation::startOperation();
70
71   HYDROGUI_Module* aModule = module();
72
73   size_t aViewId = HYDROGUI_Tool::GetActiveViewId( aModule );
74
75   int anUpdateFlags = 0;
76
77   SUIT_ViewManager* aVTKMgr           = NULL;
78   OCCViewer_ViewManager* anOCCManager = NULL;
79
80   SUIT_ViewManager* aViewMgr = aModule->getApp()->activeViewManager();
81   if ( aViewMgr )
82   {
83     if ( aViewMgr->getType() == GraphicsView_Viewer::Type() )
84     {
85       anUpdateFlags |= UF_Viewer;
86     }
87     else if ( aViewMgr->getType() == OCCViewer_Viewer::Type() )
88     {
89       anUpdateFlags |= UF_OCCViewer;
90       anOCCManager = ::qobject_cast<OCCViewer_ViewManager*>( aViewMgr );
91     }
92     else if ( aViewMgr->getType() == SVTK_Viewer::Type() )
93     {
94       anUpdateFlags |= UF_VTKViewer;
95     }
96   }
97
98   // for all objects
99   if( myId == ShowOnlyId || myId == ShowAllId || myId == HideAllId )
100   {
101     bool aVisibility = myId == ShowAllId;
102     HYDROData_Iterator anIterator( doc() );
103     for( ; anIterator.More(); anIterator.Next() )
104     {
105       Handle(HYDROData_Entity) anObject = anIterator.Current();
106       if( !anObject.IsNull() )
107         aModule->setObjectVisible( aViewId, anObject, aVisibility );
108     }
109
110     // For occ viewer we do the additional step to hide objects from other modules
111     if ( anOCCManager != NULL && !aVisibility )
112     {
113       if ( SUIT_ViewModel* vmod = anOCCManager->getViewModel() )
114       {
115         SALOME_View* vf = dynamic_cast<SALOME_View*>( vmod );
116         if ( vf )
117           vf->EraseAll( true );
118       }
119       
120       LightApp_Study* aStudy = dynamic_cast<LightApp_Study*>( aModule->getApp()->activeStudy() );  
121       if ( aStudy )
122         aStudy->setVisibilityStateForAll( Qtx::HiddenState );
123     }
124   }
125
126   bool isFoundImage = false;
127   // for selected objects
128   if( myId == ShowId || myId == ShowOnlyId || myId == HideId )
129   {
130     HYDROData_SequenceOfObjects aSeq = HYDROGUI_Tool::GetSelectedObjects( aModule );
131   
132     bool aVisibility = myId == ShowId || myId == ShowOnlyId || myId == ShowAllId;
133     Handle( HYDROData_Entity ) anObject;
134     for( Standard_Integer anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
135     {
136       anObject = aSeq.Value( anIndex );
137       if( !anObject.IsNull() )
138       {
139         if ( anObject->GetKind() == KIND_IMAGE )
140           isFoundImage = true;
141
142         aModule->setObjectVisible( aViewId, anObject, aVisibility );
143         if ( anObject->GetKind() == KIND_REGION )
144         {
145           Handle( HYDROData_Region ) aRegion = Handle( HYDROData_Region )::DownCast( anObject );
146           if ( !aRegion.IsNull() )
147           {
148             HYDROData_SequenceOfObjects aZonesSeq = aRegion->GetZones();
149             for( Standard_Integer aZoneIdx = 1, aNbZones = aZonesSeq.Length(); aZoneIdx <= aNbZones; aZoneIdx++ )
150             {
151               anObject = aZonesSeq.Value( aZoneIdx );
152               if( !anObject.IsNull() )
153               {
154                 aModule->setObjectVisible( aViewId, anObject, aVisibility );
155               }
156             }
157           }
158         }
159         else if ( anObject->GetKind() == KIND_BATHYMETRY && aVisibility )
160         {
161           if ( !(anUpdateFlags & UF_VTKViewer) )
162           {
163             // Activate VTK viewer if show a bathymetry
164             aVTKMgr = aModule->getApp()->viewManager( SVTK_Viewer::Type() );
165             if ( !aVTKMgr )
166             {
167               aVTKMgr = aModule->getApp()->createViewManager( SVTK_Viewer::Type() );
168             }
169             if ( aVTKMgr )
170             {
171               aModule->setObjectVisible( (size_t)aVTKMgr->getViewModel(), anObject, aVisibility );
172             }
173           }
174         }
175       }
176     }
177   }
178
179   if ( myId == ShowOnlyId || myId == ShowId || myId == ShowAllId )
180   {
181     if( isFoundImage || myId == ShowAllId )
182       anUpdateFlags |= UF_FitAll;
183   }
184
185   // Set VTK viewer active if show a bathymetry
186   if ( aVTKMgr )
187   {
188     anUpdateFlags |= UF_VTKViewer;
189     aVTKMgr->setShown( true );
190   }
191
192   aModule->update( anUpdateFlags );
193   commit();
194 }