#3107 BBTK Bug New Normal - branch vtk7itk4 compilation with vtk7

[bbtk.git] / packages / vtk / src / bbvtkImageCastToUShort.cxx

diff --git a/packages/vtk/src/bbvtkImageCastToUShort.cxx b/packages/vtk/src/bbvtkImageCastToUShort.cxx
index 6543e0620e721bbb2e5a3ada7b9177460b4d4b04..b27dbb6c5f2a789d13a433962bc033b7c742e17e 100644 (file)
--- a/packages/vtk/src/bbvtkImageCastToUShort.cxx
+++ b/packages/vtk/src/bbvtkImageCastToUShort.cxx
@@ -48,7 +48,15 @@ void ImageCastToUShort::Process()
 //      (the one provided in the attribute 'type' of the tag 'input') 
 
        vtkImageData* img = bbGetInputIn();     
+
+//EED 2017-01-01 Migration VTK7
+#if (VTK_MAJOR_VERSION <= 5) 
        cast->SetInput(img);
+#endif
+#if (VTK_MAJOR_VERSION >= 6) 
+       cast->SetInputData(img);
+#endif
+
        cast->SetOutputScalarTypeToUnsignedShort();
        cast->Modified();
        cast->Update();