X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROGUI%2FHYDROGUI_VTKPrsBathymetry.cxx;h=6833acec714a898feb6e226e969ab106b4f9303c;hb=58bb6b7459bebeeb089c9ed486c4683a8bae7288;hp=16983bf278d1978c61f3590d5ac8b5e6c1561aaf;hpb=d5288d46b0d85156f7666edf2cef050dc6655eae;p=modules%2Fhydro.git

diff --git a/src/HYDROGUI/HYDROGUI_VTKPrsBathymetry.cxx b/src/HYDROGUI/HYDROGUI_VTKPrsBathymetry.cxx
index 16983bf2..6833acec 100644
--- a/src/HYDROGUI/HYDROGUI_VTKPrsBathymetry.cxx
+++ b/src/HYDROGUI/HYDROGUI_VTKPrsBathymetry.cxx
@@ -1,12 +1,8 @@
-// Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
-//
-// Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
-// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
-//
+// Copyright (C) 2014-2015  EDF-R&D
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 // License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
 //
 // This library is distributed in the hope that it will be useful,
 // but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -23,16 +19,27 @@
 #include "HYDROGUI_VTKPrsBathymetry.h"
 
 #include <HYDROData_Entity.h>
+#include <HYDROData_Tool.h>
 
 #include <SALOME_Actor.h>
-#include <gp_XYZ.hxx>
+#include <gp_Pnt.hxx>
+#include <vtkDoubleArray.h>
 #include <vtkPoints.h>
 #include <vtkPolyData.h>
+#include <vtkPointData.h>
 #include <vtkPolyDataMapper.h>
 #include <vtkVertex.h>
+#include <vtkScalarBarActor.h>
 
 #include <QString>
 
+/*! \def Z_MAX
+    \brief Maximum Z value used in bathymetry presentation.
+
+    This value is used instead of invalid values.
+*/
+#define Z_MAX 1
+
 //=======================================================================
 // name    : HYDROGUI_VTKPrsBathymetry
 // Purpose : Constructor
@@ -56,13 +63,15 @@ HYDROGUI_VTKPrsBathymetry::~HYDROGUI_VTKPrsBathymetry()
 //================================================================
 void HYDROGUI_VTKPrsBathymetry::compute()
 {
+  HYDROGUI_VTKPrs::compute();
+
   if ( !getObject().IsNull() )
   {
     Handle(HYDROData_Bathymetry) aBathymetry = Handle(HYDROData_Bathymetry)::DownCast( getObject() );
     if ( !aBathymetry.IsNull() )
     {
       HYDROData_Bathymetry::AltitudePoints anAltPoints = aBathymetry->GetAltitudePoints();
-      int aNbPoints = anAltPoints.length();
+      int aNbPoints = anAltPoints.size();
 
       HYDROData_Bathymetry::AltitudePoint anAltPnt;
       vtkPoints* aPoints = vtkPoints::New();
@@ -71,28 +80,72 @@ void HYDROGUI_VTKPrsBathymetry::compute()
       vtkPolyData* aVertexGrid = vtkPolyData::New();
       aVertexGrid->Allocate( aNbPoints );
 
+      vtkDoubleArray* aZValues = vtkDoubleArray::New();
+      aZValues->Allocate( aNbPoints );
+
       vtkVertex* aVertex = vtkVertex::New();
 
+      double aZ;
+      int anInvalidZ = InvalidZValue();
       for (int i = 0; i < aNbPoints; i++ )
       {
-        anAltPnt = anAltPoints.at( i );
-        aPoints->InsertPoint( i, anAltPnt.X(), anAltPnt.Y(), anAltPnt.Z() );
-        aVertex->GetPointIds()->SetId( 0, i );
+        anAltPnt = anAltPoints[i];
+        aZ = anAltPnt.Z;
+        if ( ValuesLessEquals( aZ, anInvalidZ ) )
+        {
+          aZ = Z_MAX; // If Z value is invalid then use Z_MAX
+        }
+        aPoints->InsertPoint( i - 1, anAltPnt.X, anAltPnt.Y, aZ );
+        aVertex->GetPointIds()->SetId( 0, i - 1 );
         aVertexGrid->InsertNextCell( aVertex->GetCellType(), aVertex->GetPointIds());
+        aZValues->InsertNextValue( aZ );
       }
 
       aVertex->Delete();
 
       aVertexGrid->SetPoints( aPoints );
-      vtkPolyDataMapper* aMapper = vtkPolyDataMapper::New();
-      aMapper->SetInputData( aVertexGrid );
-      SALOME_Actor* anActor = SALOME_Actor::New();
-      anActor->SetMapper( aMapper );
+      aVertexGrid->GetPointData()->SetScalars( aZValues );
+      
+      // Update the lookup table range if this bathymetry is out of it
+      if ( myLookupTable )
+      {
+        aZValues->GetRange( myInternalZRange );
+        double* aGlobalRange = myLookupTable->GetRange();
+        // If the global range is not yet initialized or the current one is out of scope then update the global
+        bool anIsUpdated = false;
+        if (  ValuesEquals( aGlobalRange[0], anInvalidZ ) || ( aGlobalRange[0] > myInternalZRange[0] ) )
+        {
+          aGlobalRange[0] = myInternalZRange[0];
+          anIsUpdated = true;
+        }
+
+        if (  ValuesEquals( aGlobalRange[1], anInvalidZ ) || ( aGlobalRange[1] < myInternalZRange[1] ) )
+        {
+          aGlobalRange[1] = myInternalZRange[1];
+          anIsUpdated = true;
+        }
+
+        if ( anIsUpdated )
+        {
+          myLookupTable->SetRange( aGlobalRange );
+          myLookupTable->Build();
+        }
+
+        myMapper->SetScalarRange( aGlobalRange );
+        myMapper->ScalarVisibilityOn();
+        myMapper->SetScalarModeToUsePointData();
+        myMapper->SetLookupTable( myLookupTable );
+      }
+
+      myMapper->SetInputData( aVertexGrid );
+      
+      SALOME_Actor* anActor = getActor<SALOME_Actor>(this);
+      anActor->SetMapper( myMapper.GetPointer() );
       anActor->setIO( getIO() );
-      //anActor->setName( aBathymetry->GetName().toLatin1() );
       AddObject( anActor );
-      //anActor.AddPosition(0, 0, 6);
-      //anActor.GetProperty().SetDiffuseColor(1, 1, 1);
+
+      aVertexGrid->Delete();
+      aZValues->Delete();
     }
   }
 }