Salome HOME
bug #232: fit all on image show
[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
38 #include <OCCViewer_ViewModel.h>
39 #include <SVTK_ViewModel.h>
40
41 #include <SUIT_ViewManager.h>
42
43 HYDROGUI_ShowHideOp::HYDROGUI_ShowHideOp( HYDROGUI_Module* theModule, int theId )
44 : HYDROGUI_Operation( theModule ),
45   myId( theId )
46 {
47   QString aName;
48   switch( myId )
49   {
50     case ShowId:     aName = tr( "SHOW" ); break;
51     case ShowOnlyId: aName = tr( "SHOW_ONLY" ); break;
52     case ShowAllId:  aName = tr( "SHOW_ALL" ); break;
53     case HideId:     aName = tr( "HIDE" ); break;
54     case HideAllId:  aName = tr( "HIDE_ALL" ); break;
55     default: break;
56   }
57   setName( aName );
58 }
59
60 HYDROGUI_ShowHideOp::~HYDROGUI_ShowHideOp()
61 {
62 }
63
64 void HYDROGUI_ShowHideOp::startOperation()
65 {
66   HYDROGUI_Operation::startOperation();
67
68   size_t aViewId = HYDROGUI_Tool::GetActiveViewId( module() );
69
70   // for all objects
71   if( myId == ShowOnlyId || myId == ShowAllId || myId == HideAllId )
72   {
73     bool aVisibility = myId == ShowAllId;
74     HYDROData_Iterator anIterator( doc() );
75     for( ; anIterator.More(); anIterator.Next() )
76     {
77       Handle(HYDROData_Entity) anObject = anIterator.Current();
78       if( !anObject.IsNull() )
79         module()->setObjectVisible( aViewId, anObject, aVisibility );
80     }
81   }
82
83   int anUpdateFlags = 0;
84   SUIT_ViewManager* aVTKMgr = 0;
85   SUIT_ViewManager* aViewMgr = module()->getApp()->activeViewManager();
86   if ( aViewMgr )
87   {
88     if ( aViewMgr->getType() == GraphicsView_Viewer::Type() )
89     {
90       anUpdateFlags |= UF_Viewer;
91     }
92     else if ( aViewMgr->getType() == OCCViewer_Viewer::Type() )
93     {
94       anUpdateFlags |= UF_OCCViewer;
95     }
96     else if ( aViewMgr->getType() == SVTK_Viewer::Type() )
97     {
98       anUpdateFlags |= UF_VTKViewer;
99     }
100   }
101
102   bool isFoundImage = false;
103   // for selected objects
104   if( myId == ShowId || myId == ShowOnlyId || myId == HideId )
105   {
106     HYDROData_SequenceOfObjects aSeq = HYDROGUI_Tool::GetSelectedObjects( module() );
107   
108     bool aVisibility = myId == ShowId || myId == ShowOnlyId || myId == ShowAllId;
109     Handle( HYDROData_Entity ) anObject;
110     for( Standard_Integer anIndex = 1, aLength = aSeq.Length(); anIndex <= aLength; anIndex++ )
111     {
112       anObject = aSeq.Value( anIndex );
113       if( !anObject.IsNull() )
114       {
115         if ( anObject->GetKind() == KIND_IMAGE )
116           isFoundImage = true;
117
118         module()->setObjectVisible( aViewId, anObject, aVisibility );
119         if ( anObject->GetKind() == KIND_REGION )
120         {
121           Handle( HYDROData_Region ) aRegion = Handle( HYDROData_Region )::DownCast( anObject );
122           if ( !aRegion.IsNull() )
123           {
124             HYDROData_SequenceOfObjects aZonesSeq = aRegion->GetZones();
125             for( Standard_Integer aZoneIdx = 1, aNbZones = aZonesSeq.Length(); aZoneIdx <= aNbZones; aZoneIdx++ )
126             {
127               anObject = aZonesSeq.Value( aZoneIdx );
128               if( !anObject.IsNull() )
129               {
130                 module()->setObjectVisible( aViewId, anObject, aVisibility );
131               }
132             }
133           }
134         }
135         else if ( anObject->GetKind() == KIND_BATHYMETRY && aVisibility )
136         {
137           if ( !(anUpdateFlags & UF_VTKViewer) )
138           {
139             // Activate VTK viewer if show a bathymetry
140             aVTKMgr = module()->getApp()->viewManager( SVTK_Viewer::Type() );
141             if ( !aVTKMgr )
142             {
143               aVTKMgr = module()->getApp()->createViewManager( SVTK_Viewer::Type() );
144             }
145             if ( aVTKMgr )
146             {
147               module()->setObjectVisible( (size_t)aVTKMgr->getViewModel(), anObject, aVisibility );
148             }
149           }
150         }
151       }
152     }
153   }
154
155   if ( myId == ShowOnlyId || myId == ShowId || myId == ShowAllId )
156   {
157     if( isFoundImage || myId == ShowAllId )
158       anUpdateFlags |= UF_FitAll;
159   }
160
161   // Set VTK viewer active if show a bathymetry
162   if ( aVTKMgr )
163   {
164     anUpdateFlags |= UF_VTKViewer;
165     aVTKMgr->setShown( true );
166   }
167
168   module()->update( anUpdateFlags );
169   commit();
170 }