Salome HOME
Minor changes.
[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_Displayer.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       LightApp_Displayer().EraseAll( true );
113   }
114
115   bool isFoundImage = false;
116   // for selected objects
117   if( myId == ShowId || myId == ShowOnlyId || myId == HideId )
118   {
119     HYDROData_SequenceOfObjects aSeq = HYDROGUI_Tool::GetSelectedObjects( aModule );
120   
121     bool aVisibility = myId == ShowId || myId == ShowOnlyId || myId == ShowAllId;
122     Handle( HYDROData_Entity ) anObject;
123     for( Standard_Integer anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
124     {
125       anObject = aSeq.Value( anIndex );
126       if( !anObject.IsNull() )
127       {
128         if ( anObject->GetKind() == KIND_IMAGE )
129           isFoundImage = true;
130
131         aModule->setObjectVisible( aViewId, anObject, aVisibility );
132         if ( anObject->GetKind() == KIND_REGION )
133         {
134           Handle( HYDROData_Region ) aRegion = Handle( HYDROData_Region )::DownCast( anObject );
135           if ( !aRegion.IsNull() )
136           {
137             HYDROData_SequenceOfObjects aZonesSeq = aRegion->GetZones();
138             for( Standard_Integer aZoneIdx = 1, aNbZones = aZonesSeq.Length(); aZoneIdx <= aNbZones; aZoneIdx++ )
139             {
140               anObject = aZonesSeq.Value( aZoneIdx );
141               if( !anObject.IsNull() )
142               {
143                 aModule->setObjectVisible( aViewId, anObject, aVisibility );
144               }
145             }
146           }
147         }
148         else if ( anObject->GetKind() == KIND_BATHYMETRY && aVisibility )
149         {
150           if ( !(anUpdateFlags & UF_VTKViewer) )
151           {
152             // Activate VTK viewer if show a bathymetry
153             aVTKMgr = aModule->getApp()->viewManager( SVTK_Viewer::Type() );
154             if ( !aVTKMgr )
155             {
156               aVTKMgr = aModule->getApp()->createViewManager( SVTK_Viewer::Type() );
157             }
158             if ( aVTKMgr )
159             {
160               aModule->setObjectVisible( (size_t)aVTKMgr->getViewModel(), anObject, aVisibility );
161             }
162           }
163         }
164       }
165     }
166   }
167
168   if ( myId == ShowOnlyId || myId == ShowId || myId == ShowAllId )
169   {
170     if( isFoundImage || myId == ShowAllId )
171       anUpdateFlags |= UF_FitAll;
172   }
173
174   // Set VTK viewer active if show a bathymetry
175   if ( aVTKMgr )
176   {
177     anUpdateFlags |= UF_VTKViewer;
178     aVTKMgr->setShown( true );
179   }
180
181   aModule->update( anUpdateFlags );
182   commit();
183 }