src/HYDROGUI/HYDROGUI_LandCoverMapPrs.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_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>
  28 #include <QColor>
  29 #include <QString>
  30
  31 const Quantity_Color EDGES_COLOR = Quantity_NOC_SLATEGRAY;
  32 const int HILIGHT_ISO_NB = 10;
  33
  34 IMPLEMENT_STANDARD_HANDLE( HYDROGUI_LandCoverMapPrs, AIS_ColoredShape )
  35 IMPLEMENT_STANDARD_RTTIEXT( HYDROGUI_LandCoverMapPrs, AIS_ColoredShape )
  36
  37 HYDROGUI_LandCoverMapPrs::HYDROGUI_LandCoverMapPrs( const Handle(HYDROData_LandCoverMap)& theMap )
  38   : AIS_ColoredShape( theMap->GetShape() )
  39 {
  40   SetLandCoverMap( theMap );
  41   SetAutoHilight( Standard_False );
  42   SetHilightAttributes( EDGES_COLOR );
  43 }
  44
  45 HYDROGUI_LandCoverMapPrs::~HYDROGUI_LandCoverMapPrs()
  46 {
  47 }
  48
  49 Handle(HYDROData_LandCoverMap) HYDROGUI_LandCoverMapPrs::GetLandCoverMap() const
  50 {
  51   return myLCMap;
  52 }
  53
  54 void HYDROGUI_LandCoverMapPrs::SetLandCoverMap( const Handle(HYDROData_LandCoverMap)& theMap )
  55 {
  56   myLCMap = theMap;
  57   UpdateColors();
  58 }
  59
  60 void HYDROGUI_LandCoverMapPrs::UpdateColors()
  61 {
  62   SetMaterial( Graphic3d_NOM_PLASTIC );
  63   if( myLCMap.IsNull() )
  64   {
  65     Set( TopoDS_Shape() );
  66     return;
  67   }
  68
  69   Set( myLCMap->GetShape() );
  70   HYDROData_LandCoverMap::Explorer anIt( myLCMap );
  71   for( ; anIt.More(); anIt.Next() )
  72   {
  73     TopoDS_Face aFace = anIt.Face();
  74     QString aStricklerType = anIt.StricklerType();
  75     Quantity_Color aColor = GetColor( aStricklerType );
  76     SetCustomColor( aFace, aColor );
  77     SetCustomWidth( aFace, 1.0 );
  78   }
  79   SetTransparency( myLCMap->GetTransparency() );
  80 }
  81
  82 Handle(Aspect_ColorScale) HYDROGUI_LandCoverMapPrs::GetColorScale() const
  83 {
  84   return myColorScale;
  85 }
  86
  87 void HYDROGUI_LandCoverMapPrs::SetColorScale( const Handle(Aspect_ColorScale)& theColorScale )
  88 {
  89   myColorScale = theColorScale;
  90   double aMin = 0, aMax = 0;
  91   if( myTable.IsNull() )
  92   {
  93     //TODO: go through all Strickler tables in the document to get the global range
  94   }
  95   else
  96     myTable->GetCoefficientRange( aMin, aMax );
  97
  98   myColorScale->SetRange( aMin, aMax );
  99   UpdateColors();
 100 }
 101
 102 Handle(HYDROData_StricklerTable) HYDROGUI_LandCoverMapPrs::GetTable() const
 103 {
 104   return myTable;
 105 }
 106
 107 void HYDROGUI_LandCoverMapPrs::SetTable( const Handle(HYDROData_StricklerTable)& theTable )
 108 {
 109   myTable = theTable;
 110 }
 111
 112 Quantity_Color HYDROGUI_LandCoverMapPrs::GetColor( const QString& theStricklerType ) const
 113 {
 114   Quantity_Color aResult;
 115
 116   if( !myColorScale.IsNull() && !myTable.IsNull() )
 117   {
 118     double aCoeff = myTable->Get( theStricklerType, 0.0 );
 119     Standard_Boolean isOK = myColorScale->FindColor( aCoeff, aResult );
 120     if( isOK )
 121       return aResult;
 122     else
 123       return Quantity_Color();
 124   }
 125
 126   QColor aColor = HYDROData_Document::Document( myLCMap->Label())->GetAssociatedColor(theStricklerType, myTable);
 127
 128   if (aColor.isValid())
 129     return HYDROData_Tool::toOccColor(aColor);
 130   else
 131     return Quantity_Color();
 132 }
 133
 134 void HYDROGUI_LandCoverMapPrs::Compute( const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
 135                                         const Handle(Prs3d_Presentation)& thePresentation,
 136                                         const Standard_Integer theMode )
 137 {
 138   thePresentation->Clear();
 139
 140   myDrawer->UIsoAspect()->SetNumber( 0 );
 141   myDrawer->VIsoAspect()->SetNumber( 0 );
 142   myDrawer->LineAspect()->SetColor( EDGES_COLOR );
 143   myDrawer->FaceBoundaryAspect()->SetColor( EDGES_COLOR );
 144   myDrawer->FreeBoundaryAspect()->SetColor( EDGES_COLOR );
 145
 146   switch( theMode )
 147   {
 148   case AIS_WireFrame:
 149   case AIS_Shaded:
 150     AIS_ColoredShape::Compute( thePresentationManager, thePresentation, theMode );
 151     break;
 152   }
 153
 154   if( theMode==AIS_Shaded )
 155     StdPrs_WFDeflectionShape::Add( thePresentation, Shape(), myDrawer );
 156 }
 157
 158 void HYDROGUI_LandCoverMapPrs::HilightSelected( const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
 159                                                 const SelectMgr_SequenceOfOwner& theOwners )
 160 {
 161   Handle(Prs3d_Presentation) aSelectPrs = GetSelectPresentation( thePresentationManager );
 162
 163   SetHilightAttributes( EDGES_COLOR );
 164
 165   for( int i=1, n=theOwners.Length(); i<=n; i++ )
 166   {
 167     Handle(StdSelect_BRepOwner) anOwner = Handle(StdSelect_BRepOwner)::DownCast( theOwners.Value( i ) );
 168     if( !anOwner.IsNull() )
 169       StdPrs_WFDeflectionShape::Add( aSelectPrs, anOwner->Shape(), HilightAttributes() );
 170   }
 171
 172   HilightAttributes()->UIsoAspect()->SetNumber( 0 );
 173   HilightAttributes()->VIsoAspect()->SetNumber( 0 );
 174
 175   aSelectPrs->SetDisplayPriority( 9 );
 176   aSelectPrs->Display();
 177 }
 178
 179 void HYDROGUI_LandCoverMapPrs::SetHilightAttributes( const Quantity_Color& theEdgesColor )
 180 {
 181   HilightAttributes()->UIsoAspect()->SetNumber( HILIGHT_ISO_NB );
 182   HilightAttributes()->UIsoAspect()->SetColor( theEdgesColor );
 183   HilightAttributes()->VIsoAspect()->SetNumber( HILIGHT_ISO_NB );
 184   HilightAttributes()->VIsoAspect()->SetColor( theEdgesColor );
 185   HilightAttributes()->LineAspect()->SetColor( theEdgesColor );
 186   HilightAttributes()->FaceBoundaryAspect()->SetColor( theEdgesColor );
 187   HilightAttributes()->FreeBoundaryAspect()->SetColor( theEdgesColor );
 188 }