Salome HOME
Merge remote-tracking branch 'origin/BR_1321_ECW' into BR_DEMO
[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 }
50
51 void HYDROGUI_ShapeLandCoverMap::update( bool theIsUpdateViewer, bool isDeactivateSelection )
52 {
53   setIsToUpdate( false );
54
55   Handle(HYDROData_LandCoverMap) aLandCoverMapObj =
56     Handle(HYDROData_LandCoverMap)::DownCast( getObject() );
57
58   TopoDS_Shape aLandCoverMapShape = aLandCoverMapObj->GetShape();
59   if ( !aLandCoverMapShape.IsNull() ) {
60     setShape( aLandCoverMapShape, false, false );
61   }
62
63   Handle(HYDROGUI_LandCoverMapPrs) aLandCoverMapPrs = 
64     Handle(HYDROGUI_LandCoverMapPrs)::DownCast( getAISObjects()[0] );
65   if ( !aLandCoverMapPrs.IsNull() )
66     aLandCoverMapPrs->UpdateColors();
67
68   HYDROGUI_Shape::update( theIsUpdateViewer, isDeactivateSelection );
69 }
70
71 bool HYDROGUI_ShapeLandCoverMap::isScalarMapModeEnabled() const
72 {
73   return myIsScalarMapMode;
74 }
75
76 void HYDROGUI_ShapeLandCoverMap::setScalarMapModeEnabled( const bool theIsToEnable )
77 {
78   myIsScalarMapMode = theIsToEnable;
79   update( false, false );
80 }
81
82 void HYDROGUI_ShapeLandCoverMap::setVisible( const bool theState,
83                                              const bool theIsUpdateViewer )
84 {
85   HYDROGUI_Shape::setVisible( theState, theIsUpdateViewer );
86   myDisplayer->SetToUpdateColorScale();
87 }
88
89 void HYDROGUI_ShapeLandCoverMap::displayShape( const bool theIsUpdateViewer )
90 {
91   HYDROGUI_Shape::displayShape( theIsUpdateViewer );
92   myDisplayer->SetToUpdateColorScale();
93 }
94
95 QList<Handle(AIS_InteractiveObject)> HYDROGUI_ShapeLandCoverMap::createShape() const
96 {
97   QList<Handle(AIS_InteractiveObject)> shapes;
98
99   Handle(HYDROData_LandCoverMap) aLandCoverMap = Handle(HYDROData_LandCoverMap)::DownCast( getObject() );
100   if ( !aLandCoverMap.IsNull() )
101   {
102     Handle(HYDROGUI_LandCoverMapPrs) aLandCoverMapPrs = new HYDROGUI_LandCoverMapPrs( aLandCoverMap );    
103     if ( myIsScalarMapMode )
104     {
105       LightApp_Application* anApp = myDisplayer->module()->getApp();
106       OCCViewer_ViewManager* aViewManager = ::qobject_cast<OCCViewer_ViewManager*>( 
107         anApp->getViewManager( OCCViewer_Viewer::Type(), true ) );
108       if ( aViewManager )
109       {
110         if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
111         {
112           int aViewerId = (size_t)aViewer;
113
114           // Set Strickler table
115           Handle(HYDROData_StricklerTable) aTable = 
116             myDisplayer->module()->getLandCoverColoringTable( aViewerId );
117           if( !aTable.IsNull() )
118             aLandCoverMapPrs->SetTable( aTable );
119
120           // Set color scale
121           aLandCoverMapPrs->SetColorScale( myDisplayer->GetColorScale( aViewerId) );
122         }
123       }
124     }
125     shapes.append( aLandCoverMapPrs );
126   }
127   return shapes;
128 }
129
130 void HYDROGUI_ShapeLandCoverMap::display( const bool theIsUpdateViewer )
131 {
132   HYDROGUI_Shape::display( theIsUpdateViewer );
133   myDisplayer->SetToUpdateColorScale();
134 }
135
136 void HYDROGUI_ShapeLandCoverMap::erase( const bool theIsUpdateViewer )
137 {
138   HYDROGUI_Shape::erase( theIsUpdateViewer );
139   myDisplayer->SetToUpdateColorScale();
140 }