Salome HOME
refs #1501 p.2
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_ShapeLandCoverMap.cxx
1 // Copyright (C) 2014-2015  EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
6 //
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
10 // Lesser General Public License for more details.
11 //
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
15 //
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
17 //
18
19 #include "HYDROGUI_ShapeLandCoverMap.h"
20 #include "HYDROGUI_OCCDisplayer.h"
21 #include "HYDROGUI_LandCoverMapPrs.h"
22 #include "HYDROGUI_Tool.h"
23 #include "HYDROGUI_Module.h"
24
25 #include <LightApp_Application.h>
26 #include <OCCViewer_ViewManager.h>
27 #include <OCCViewer_ViewModel.h>
28 #include <OCCViewer_ViewWindow.h>
29 #include <OCCViewer_ViewPort3d.h>
30
31 #include <HYDROData_LandCoverMap.h>
32
33 #include <AIS_InteractiveContext.hxx>
34
35 HYDROGUI_ShapeLandCoverMap::HYDROGUI_ShapeLandCoverMap( HYDROGUI_OCCDisplayer* theDisplayer,
36                                                         const Handle(AIS_InteractiveContext)& theContext,
37                                                         const Handle(HYDROData_LandCoverMap)& theLandCoverMap,
38                                                         const int                             theZLayer,
39                                                         const bool                            theIsScalarMode )
40 : HYDROGUI_Shape( theContext, theLandCoverMap, theZLayer ),
41   myDisplayer( theDisplayer ),
42   myIsScalarMapMode( theIsScalarMode )
43 {
44   setFillingColor( QColor(), false, false );
45 }
46
47 HYDROGUI_ShapeLandCoverMap::~HYDROGUI_ShapeLandCoverMap()
48 {
49 #ifndef LIGHT_MODE
50   if( myDisplayer )
51     myDisplayer->SetToUpdateColorScale();
52 #endif
53 }
54
55 void HYDROGUI_ShapeLandCoverMap::update( bool theIsUpdateViewer, bool isDeactivateSelection )
56 {
57   setIsToUpdate( false );
58
59   Handle(HYDROData_LandCoverMap) aLandCoverMapObj =
60     Handle(HYDROData_LandCoverMap)::DownCast( getObject() );
61
62   TopoDS_Shape aLandCoverMapShape = aLandCoverMapObj->GetShape();
63   if ( !aLandCoverMapShape.IsNull() ) {
64     setShape( aLandCoverMapShape, false, false );
65   }
66
67   Handle(HYDROGUI_LandCoverMapPrs) aLandCoverMapPrs = 
68     Handle(HYDROGUI_LandCoverMapPrs)::DownCast( getAISObjects()[0] );
69   if ( !aLandCoverMapPrs.IsNull() )
70     aLandCoverMapPrs->UpdateColors();
71
72   HYDROGUI_Shape::update( theIsUpdateViewer, isDeactivateSelection );
73 }
74
75 bool HYDROGUI_ShapeLandCoverMap::isScalarMapModeEnabled() const
76 {
77   return myIsScalarMapMode;
78 }
79
80 void HYDROGUI_ShapeLandCoverMap::setScalarMapModeEnabled( const bool theIsToEnable )
81 {
82   myIsScalarMapMode = theIsToEnable;
83   update( false, false );
84 }
85
86 void HYDROGUI_ShapeLandCoverMap::setVisible( const bool theState,
87                                              const bool theIsUpdateViewer )
88 {
89   HYDROGUI_Shape::setVisible( theState, theIsUpdateViewer );
90   myDisplayer->SetToUpdateColorScale();
91 }
92
93 void HYDROGUI_ShapeLandCoverMap::displayShape( const bool theIsUpdateViewer )
94 {
95   HYDROGUI_Shape::displayShape( theIsUpdateViewer );
96   myDisplayer->SetToUpdateColorScale();
97 }
98
99 QList<Handle(AIS_InteractiveObject)> HYDROGUI_ShapeLandCoverMap::createShape() const
100 {
101   QList<Handle(AIS_InteractiveObject)> shapes;
102
103   Handle(HYDROData_LandCoverMap) aLandCoverMap = Handle(HYDROData_LandCoverMap)::DownCast( getObject() );
104   if ( !aLandCoverMap.IsNull() )
105   {
106     Handle(HYDROGUI_LandCoverMapPrs) aLandCoverMapPrs = new HYDROGUI_LandCoverMapPrs( aLandCoverMap );    
107     if ( myIsScalarMapMode )
108     {
109       LightApp_Application* anApp = myDisplayer->module()->getApp();
110       OCCViewer_ViewManager* aViewManager = ::qobject_cast<OCCViewer_ViewManager*>( 
111         anApp->getViewManager( OCCViewer_Viewer::Type(), true ) );
112       if ( aViewManager )
113       {
114         if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
115         {
116           int aViewerId = (size_t)aViewer;
117
118           // Set Strickler table
119           Handle(HYDROData_StricklerTable) aTable = 
120             myDisplayer->module()->getLandCoverColoringTable( aViewerId );
121           if( !aTable.IsNull() )
122             aLandCoverMapPrs->SetTable( aTable );
123
124           // Set color scale
125           aLandCoverMapPrs->SetColorScale( myDisplayer->GetColorScale( aViewerId) );
126         }
127       }
128     }
129     shapes.append( aLandCoverMapPrs );
130   }
131   return shapes;
132 }
133
134 void HYDROGUI_ShapeLandCoverMap::display( const bool theIsUpdateViewer )
135 {
136   HYDROGUI_Shape::display( theIsUpdateViewer );
137   myDisplayer->SetToUpdateColorScale();
138 }
139
140 void HYDROGUI_ShapeLandCoverMap::erase( const bool theIsUpdateViewer )
141 {
142   HYDROGUI_Shape::erase( theIsUpdateViewer );
143   myDisplayer->SetToUpdateColorScale();
144 }