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_WHITE;
  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   Set( myLCMap->GetShape() );
  63   SetMaterial( Graphic3d_NOM_PLASTIC );
  64   HYDROData_LandCoverMap::Explorer anIt( myLCMap );
  65   for( ; anIt.More(); anIt.Next() )
  66   {
  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 );
  72   }
  73 }
  74
  75 Handle(Aspect_ColorScale) HYDROGUI_LandCoverMapPrs::GetColorScale() const
  76 {
  77   return myColorScale;
  78 }
  79
  80 void HYDROGUI_LandCoverMapPrs::SetColorScale( const Handle(Aspect_ColorScale)& theColorScale )
  81 {
  82   myColorScale = theColorScale;
  83   double aMin = 0, aMax = 0;
  84   if( myTable.IsNull() )
  85   {
  86     //TODO: go through all Strickler tables in the document to get the global range
  87   }
  88   else
  89     myTable->GetCoefficientRange( aMin, aMax );
  90
  91   myColorScale->SetRange( aMin, aMax );
  92   UpdateColors();
  93 }
  94
  95 Handle(HYDROData_StricklerTable) HYDROGUI_LandCoverMapPrs::GetTable() const
  96 {
  97   return myTable;
  98 }
  99
 100 void HYDROGUI_LandCoverMapPrs::SetTable( const Handle(HYDROData_StricklerTable)& theTable )
 101 {
 102   myTable = theTable;
 103 }
 104
 105 Quantity_Color HYDROGUI_LandCoverMapPrs::GetColor( const QString& theStricklerType ) const
 106 {
 107   Quantity_Color aResult;
 108
 109   if( !myColorScale.IsNull() && !myTable.IsNull() )
 110   {
 111     double aCoeff = myTable->Get( theStricklerType, 0.0 );
 112     Standard_Boolean isOK = myColorScale->FindColor( aCoeff, aResult );
 113     if( isOK )
 114       return aResult;
 115     else
 116       return Quantity_Color();
 117   }
 118
 119   QColor aColor = HYDROData_Document::Document( myLCMap->Label())->GetAssociatedColor(theStricklerType, myTable);
 120
 121   if (aColor.isValid())
 122     return HYDROData_Tool::toOccColor(aColor);
 123   else
 124     return Quantity_Color();
 125 }
 126
 127 void HYDROGUI_LandCoverMapPrs::Compute( const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
 128                                         const Handle(Prs3d_Presentation)& thePresentation,
 129                                         const Standard_Integer theMode )
 130 {
 131   thePresentation->Clear();
 132
 133   myDrawer->UIsoAspect()->SetNumber( 0 );
 134   myDrawer->VIsoAspect()->SetNumber( 0 );
 135   myDrawer->LineAspect()->SetColor( EDGES_COLOR );
 136   myDrawer->FaceBoundaryAspect()->SetColor( EDGES_COLOR );
 137   myDrawer->FreeBoundaryAspect()->SetColor( EDGES_COLOR );
 138
 139   switch( theMode )
 140   {
 141   case AIS_WireFrame:
 142   case AIS_Shaded:
 143     AIS_ColoredShape::Compute( thePresentationManager, thePresentation, theMode );
 144     break;
 145   }
 146
 147   if( theMode==AIS_Shaded )
 148     StdPrs_WFDeflectionShape::Add( thePresentation, Shape(), myDrawer );
 149 }
 150
 151 void HYDROGUI_LandCoverMapPrs::HilightSelected( const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
 152                                                 const SelectMgr_SequenceOfOwner& theOwners )
 153 {
 154   Handle(Prs3d_Presentation) aSelectPrs = GetSelectPresentation( thePresentationManager );
 155
 156   SetHilightAttributes( EDGES_COLOR );
 157
 158   for( int i=1, n=theOwners.Length(); i<=n; i++ )
 159   {
 160     Handle(StdSelect_BRepOwner) anOwner = Handle(StdSelect_BRepOwner)::DownCast( theOwners.Value( i ) );
 161     if( !anOwner.IsNull() )
 162       StdPrs_WFDeflectionShape::Add( aSelectPrs, anOwner->Shape(), HilightAttributes() );
 163   }
 164
 165   HilightAttributes()->UIsoAspect()->SetNumber( 0 );
 166   HilightAttributes()->VIsoAspect()->SetNumber( 0 );
 167
 168   aSelectPrs->SetDisplayPriority( 9 );
 169   aSelectPrs->Display();
 170 }
 171
 172 void HYDROGUI_LandCoverMapPrs::SetHilightAttributes( const Quantity_Color& theEdgesColor )
 173 {
 174   HilightAttributes()->UIsoAspect()->SetNumber( HILIGHT_ISO_NB );
 175   HilightAttributes()->UIsoAspect()->SetColor( theEdgesColor );
 176   HilightAttributes()->VIsoAspect()->SetNumber( HILIGHT_ISO_NB );
 177   HilightAttributes()->VIsoAspect()->SetColor( theEdgesColor );
 178   HilightAttributes()->LineAspect()->SetColor( theEdgesColor );
 179   HilightAttributes()->FaceBoundaryAspect()->SetColor( theEdgesColor );
 180   HilightAttributes()->FreeBoundaryAspect()->SetColor( theEdgesColor );
 181 }