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.
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.
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
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #include <HYDROGUI_LandCoverMapPrs.h>
20 #include <HYDROData_Iterator.h>
21 #include <HYDROData_Tool.h>
22 #include <AIS_DisplayMode.hxx>
23 #include <Prs3d_IsoAspect.hxx>
24 #include <SelectMgr_SequenceOfOwner.hxx>
25 #include <StdPrs_WFDeflectionShape.hxx>
26 #include <StdSelect_BRepOwner.hxx>
27 #include <TopoDS_Face.hxx>
31 const Quantity_Color EDGES_COLOR = Quantity_NOC_WHITE;
32 const int HILIGHT_ISO_NB = 10;
34 IMPLEMENT_STANDARD_HANDLE( HYDROGUI_LandCoverMapPrs, AIS_ColoredShape )
35 IMPLEMENT_STANDARD_RTTIEXT( HYDROGUI_LandCoverMapPrs, AIS_ColoredShape )
37 HYDROGUI_LandCoverMapPrs::HYDROGUI_LandCoverMapPrs( const Handle(HYDROData_LandCoverMap)& theMap )
38 : AIS_ColoredShape( theMap->GetShape() )
40 SetLandCoverMap( theMap );
41 SetAutoHilight( Standard_False );
42 SetHilightAttributes( EDGES_COLOR );
45 HYDROGUI_LandCoverMapPrs::~HYDROGUI_LandCoverMapPrs()
49 Handle(HYDROData_LandCoverMap) HYDROGUI_LandCoverMapPrs::GetLandCoverMap() const
54 void HYDROGUI_LandCoverMapPrs::SetLandCoverMap( const Handle(HYDROData_LandCoverMap)& theMap )
60 void HYDROGUI_LandCoverMapPrs::UpdateColors()
62 Set( myLCMap->GetShape() );
63 SetMaterial( Graphic3d_NOM_PLASTIC );
64 HYDROData_LandCoverMap::Iterator anIt( myLCMap );
65 for( ; anIt.More(); anIt.Next() )
67 TopoDS_Face aFace = anIt.Face();
68 QString aStricklerType = anIt.StricklerType();
69 Quantity_Color aColor = GetColor( aStricklerType );
70 SetCustomColor( aFace, aColor );
71 SetCustomWidth( aFace, 1.0 );
75 Handle(Aspect_ColorScale) HYDROGUI_LandCoverMapPrs::GetColorScale() const
80 void HYDROGUI_LandCoverMapPrs::SetColorScale( const Handle(Aspect_ColorScale)& theColorScale )
82 myColorScale = theColorScale;
83 double aMin = 0, aMax = 0;
84 if( myTable.IsNull() )
86 //TODO: go through all Strickler tables in the document to get the global range
89 myTable->GetCoefficientRange( aMin, aMax );
91 myColorScale->SetRange( aMin, aMax );
95 Handle(HYDROData_StricklerTable) HYDROGUI_LandCoverMapPrs::GetTable() const
100 void HYDROGUI_LandCoverMapPrs::SetTable( const Handle(HYDROData_StricklerTable)& theTable )
105 Quantity_Color HYDROGUI_LandCoverMapPrs::GetColor( const QString& theStricklerType ) const
107 Quantity_Color aResult;
109 if( !myColorScale.IsNull() && !myTable.IsNull() )
111 double aCoeff = myTable->Get( theStricklerType, 0.0 );
112 Standard_Boolean isOK = myColorScale->FindColor( aCoeff, aResult );
116 return Quantity_Color();
119 if( !myTable.IsNull() && myTable->HasType( theStricklerType ) )
121 QColor aColor = myTable->GetColor( theStricklerType );
122 return HYDROData_Tool::toOccColor( aColor );
125 HYDROData_Iterator anIt( HYDROData_Document::Document( myLCMap->Label() ), KIND_STRICKLER_TABLE );
126 for( ; anIt.More(); anIt.Next() )
128 Handle(HYDROData_StricklerTable) aTable = Handle(HYDROData_StricklerTable)::DownCast( anIt.Current() );
129 if( aTable->HasType( theStricklerType ) )
131 QColor aColor = aTable->GetColor( theStricklerType );
132 return HYDROData_Tool::toOccColor( aColor );
136 return Quantity_Color();
139 void HYDROGUI_LandCoverMapPrs::Compute( const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
140 const Handle(Prs3d_Presentation)& thePresentation,
141 const Standard_Integer theMode )
143 thePresentation->Clear();
145 myDrawer->UIsoAspect()->SetNumber( 0 );
146 myDrawer->VIsoAspect()->SetNumber( 0 );
147 myDrawer->LineAspect()->SetColor( EDGES_COLOR );
148 myDrawer->FaceBoundaryAspect()->SetColor( EDGES_COLOR );
149 myDrawer->FreeBoundaryAspect()->SetColor( EDGES_COLOR );
155 AIS_ColoredShape::Compute( thePresentationManager, thePresentation, theMode );
159 if( theMode==AIS_Shaded )
160 StdPrs_WFDeflectionShape::Add( thePresentation, Shape(), myDrawer );
163 void HYDROGUI_LandCoverMapPrs::HilightSelected( const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
164 const SelectMgr_SequenceOfOwner& theOwners )
166 Handle(Prs3d_Presentation) aSelectPrs = GetSelectPresentation( thePresentationManager );
168 SetHilightAttributes( EDGES_COLOR );
170 for( int i=1, n=theOwners.Length(); i<=n; i++ )
172 Handle(StdSelect_BRepOwner) anOwner = Handle(StdSelect_BRepOwner)::DownCast( theOwners.Value( i ) );
173 if( !anOwner.IsNull() )
174 StdPrs_WFDeflectionShape::Add( aSelectPrs, anOwner->Shape(), HilightAttributes() );
177 HilightAttributes()->UIsoAspect()->SetNumber( 0 );
178 HilightAttributes()->VIsoAspect()->SetNumber( 0 );
180 aSelectPrs->SetDisplayPriority( 9 );
181 aSelectPrs->Display();
184 void HYDROGUI_LandCoverMapPrs::SetHilightAttributes( const Quantity_Color& theEdgesColor )
186 HilightAttributes()->UIsoAspect()->SetNumber( HILIGHT_ISO_NB );
187 HilightAttributes()->UIsoAspect()->SetColor( theEdgesColor );
188 HilightAttributes()->VIsoAspect()->SetNumber( HILIGHT_ISO_NB );
189 HilightAttributes()->VIsoAspect()->SetColor( theEdgesColor );
190 HilightAttributes()->LineAspect()->SetColor( theEdgesColor );
191 HilightAttributes()->FaceBoundaryAspect()->SetColor( theEdgesColor );
192 HilightAttributes()->FreeBoundaryAspect()->SetColor( theEdgesColor );