#include "HYDROGUI_Module.h"
#include "HYDROGUI_VTKPrs.h"
#include "HYDROGUI_VTKPrsBathymetryDriver.h"
+#include "HYDROGUI_VTKPrsShapeDriver.h"
#include "HYDROGUI_Tool.h"
#include "HYDROData_Tool.h"
#include <vtkLookupTable.h>
#include <vtkRenderer.h>
+#include <vtkTextProperty.h>
#include <vtkWindow.h>
#include <vtkActor2DCollection.h>
//#define SATURATION_START 1.0
//#define SATURATION_END 0.4
-#define HUE_START 0.0
-#define HUE_END 0.7
+#define HUE_START 0.7
+#define HUE_END 0.0
#define SATURATION_START 1.0
#define SATURATION_END 1.0
HYDROGUI_VTKPrsDisplayer::HYDROGUI_VTKPrsDisplayer( HYDROGUI_Module* theModule )
-: HYDROGUI_AbstractDisplayer( theModule ), myDriver( NULL )
+: HYDROGUI_AbstractDisplayer( theModule ), myDriver( NULL ), myShapeDriver( NULL )
{
}
{
aViewer->EraseAll( true );
module()->removeViewVTKPrs( (size_t)aViewer );
+ EraseScalarBar( theViewerId );
}
}
-void HYDROGUI_VTKPrsDisplayer::DeleteScalarBar( const int theViewerId )
+void HYDROGUI_VTKPrsDisplayer::EraseScalarBar( const int theViewerId, const bool theIsDelete )
{
SVTK_Viewer* aViewer = module()->getVTKViewer( theViewerId );
if( aViewer )
{
- SVTK_ViewWindow* aView = dynamic_cast<SVTK_ViewWindow*>(
- aViewer->getViewManager()->getActiveView() );
- if ( aView )
+ if ( myScalarBars.contains( (size_t)aViewer ) )
{
- if ( myScalarBars.contains( (size_t)aViewer ) )
+ SUIT_ViewManager* aViewMgr = dynamic_cast<SUIT_ViewManager*>( aViewer->getViewManager() );
+ if ( aViewMgr && aViewMgr->getViewsCount() > 0 )
{
- vtkScalarBarActor* aScalarBar = myScalarBars[ (size_t)aViewer ];
- if ( aView->getRenderer()->HasViewProp( aScalarBar ) )
+ SVTK_ViewWindow* aView = dynamic_cast<SVTK_ViewWindow*>( aViewMgr->getActiveView() );
+ if ( aView )
{
- aView->getRenderer()->RemoveActor2D( aScalarBar );
+ vtkScalarBarActor* aScalarBar = myScalarBars[ (size_t)aViewer ];
+ if ( aView->getRenderer()->HasViewProp( aScalarBar ) )
+ {
+ aView->getRenderer()->RemoveActor2D( aScalarBar );
+ }
}
}
+ if ( theIsDelete )
+ {
+ myScalarBars.remove( (size_t)aViewer );
+ }
}
}
}
int anInvalidZ = HYDROGUI_VTKPrs::InvalidZValue();
bool isChanged = false;
+ bool isScalarBarNeeded = false;
HYDROGUI_VTKPrs* aPrs;
for ( int i = 1, n = theObjs.Length(); i <= n; i++ )
{
{
if ( anIsVisible )
{
-
- // Extend the global Z range if necessary
- double* aGlobalRange = GetZRange( (size_t)aViewer );
- double* aRange = aPrs->getInternalZRange();
- bool anIsUpdate = false;
- if ( aRange[0] < aGlobalRange[0] || ValuesEquals( aGlobalRange[0], anInvalidZ ) )
+ if ( aPrs->needScalarBar() )
{
- aGlobalRange[0] = aRange[0];
- anIsUpdate = true;
- }
- if ( aRange[1] > aGlobalRange[1] || ValuesEquals( aGlobalRange[1], anInvalidZ ) )
- {
- aGlobalRange[1] = aRange[1];
- anIsUpdate = true;
- }
-
- if ( anIsUpdate )
- {
- module()->updateVTKZRange( (size_t)aViewer, aGlobalRange );
+ // Extend the global Z range if necessary
+ double* aGlobalRange = GetZRange( (size_t)aViewer );
+ double* aRange = aPrs->getInternalZRange();
+ bool anIsUpdate = false;
+ if ( aRange[0] < aGlobalRange[0] || ValuesEquals( aGlobalRange[0], anInvalidZ ) )
+ {
+ aGlobalRange[0] = aRange[0];
+ anIsUpdate = true;
+ }
+ if ( aRange[1] > aGlobalRange[1] || ValuesEquals( aGlobalRange[1], anInvalidZ ) )
+ {
+ aGlobalRange[1] = aRange[1];
+ anIsUpdate = true;
+ }
+
+ if ( anIsUpdate )
+ {
+ module()->updateVTKZRange( (size_t)aViewer, aGlobalRange );
+ }
}
aViewer->Display( aPrs );
+ isScalarBarNeeded = isScalarBarNeeded || aPrs->needScalarBar();
}
else
if ( aView )
{
- if ( isChanged && aScalarBar )
+ if ( isChanged && isScalarBarNeeded && aScalarBar )
{
// Show colors legend bar
aView->getRenderer()->AddActor2D( aScalarBar );
}
aDriver = myDriver;
}
+ else
+ {
+ if ( !myShapeDriver )
+ {
+ myShapeDriver = new HYDROGUI_VTKPrsShapeDriver( myScalarBars[ theViewId ] );
+ }
+ aDriver = myShapeDriver;
+ }
return aDriver;
}
aTable->Build();
vtkSmartPointer<vtkScalarBarActor> aScalarBar = vtkScalarBarActor::New();
aScalarBar->SetLookupTable( aTable );
+ aScalarBar->SetNumberOfLabels( NB_COLORS * 0.75 );
+ aScalarBar->SetWidth( aScalarBar->GetWidth() / 1.5 );
+ aScalarBar->SetTextureGridWidth( aScalarBar->GetTextureGridWidth() * 4. );
aTable->Delete();
myScalarBars.insert( theViewId, aScalarBar );
}