Implementation of the "0022102: EDF 1496 SMESH : Displaying of discrete elements...

[modules/gui.git] / src / VTKViewer / VTKViewer_DataSetMapper.cxx

diff --git a/src/VTKViewer/VTKViewer_DataSetMapper.cxx b/src/VTKViewer/VTKViewer_DataSetMapper.cxx
index d06e0a74226c22d6b624e83e5373c773340d3b96..5d9888fa872bfc363ccfe7f57e4a09202c95de86 100644 (file)
--- a/src/VTKViewer/VTKViewer_DataSetMapper.cxx
+++ b/src/VTKViewer/VTKViewer_DataSetMapper.cxx
@@ -29,6 +29,7 @@ vtkStandardNewMacro(VTKViewer_DataSetMapper);
 VTKViewer_DataSetMapper::VTKViewer_DataSetMapper()
 {
   this->MarkerEnabled = false;
+  this->BallEnabled = false;
   this->MarkerType    = VTK::MT_NONE;
   this->MarkerScale   = VTK::MS_NONE;
   this->MarkerId      = 0;
@@ -54,7 +55,8 @@ void VTKViewer_DataSetMapper::Render(vtkRenderer *ren, vtkActor *act)
       pm->SetMarkerStd( this->MarkerType, this->MarkerScale );
     else
       pm->SetMarkerTexture( this->MarkerId, this->MarkerTexture );
-
+    pm->SetBallEnabled( this->BallEnabled );
+    
     this->GeometryExtractor = gf;
     this->PolyDataMapper = pm;
   }
@@ -70,6 +72,17 @@ void VTKViewer_DataSetMapper::SetMarkerEnabled( bool theMarkerEnabled )
       aMapper->SetMarkerEnabled( theMarkerEnabled );
 }
 
+
+//-----------------------------------------------------------------------------
+void VTKViewer_DataSetMapper::SetBallEnabled( bool theBallEnabled )
+{
+  this->BallEnabled = theBallEnabled;
+  if( this->PolyDataMapper )
+    if( VTKViewer_PolyDataMapper* aMapper = dynamic_cast<VTKViewer_PolyDataMapper*>( this->PolyDataMapper ) )
+      aMapper->SetBallEnabled( theBallEnabled );
+}
+
+
 //----------------------------------------------------------------------------
 void VTKViewer_DataSetMapper::SetMarkerStd( VTK::MarkerType theMarkerType, VTK::MarkerScale theMarkerScale )
 {