X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=packages%2Fvtk%2Fsrc%2FbbvtkImageCastToUShort.cxx;fp=packages%2Fvtk%2Fsrc%2FbbvtkImageCastToUShort.cxx;h=b27dbb6c5f2a789d13a433962bc033b7c742e17e;hb=088ed2b7bdbe14eb85a7245afaa693ddb7a53a97;hp=6543e0620e721bbb2e5a3ada7b9177460b4d4b04;hpb=79d06fa7c6441e99291ad3a1949d241f336573d5;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();