X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=packages%2Fvtk%2Fsrc%2FbbvtkImageCastToUShort.cxx;h=b27dbb6c5f2a789d13a433962bc033b7c742e17e;hb=ce796ef8e6c470b95b8f00cad76c25a9e4a88df1;hp=6543e0620e721bbb2e5a3ada7b9177460b4d4b04;hpb=18f3f2238cb97854f590dd8af0e9ab2a07de77a6;p=bbtk.git

diff --git a/packages/vtk/src/bbvtkImageCastToUShort.cxx b/packages/vtk/src/bbvtkImageCastToUShort.cxx
index 6543e06..b27dbb6 100644
--- 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();