X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=bbtk_creaVtk_PKG%2Fsrc%2FbbcreaVtkScreenShot.cxx;h=08f4089c3dae5547eb2701706f3a714ab7474091;hb=2e8c7cadbf9be911a9621142f95d81807d6c0402;hp=4ee0b66d14a1f495628ec992f58eeed6e97ac13c;hpb=d1ff777024c5613e1c10ed20c784bb323c8407a1;p=creaVtk.git

diff --git a/bbtk_creaVtk_PKG/src/bbcreaVtkScreenShot.cxx b/bbtk_creaVtk_PKG/src/bbcreaVtkScreenShot.cxx
index 4ee0b66..08f4089 100644
--- a/bbtk_creaVtk_PKG/src/bbcreaVtkScreenShot.cxx
+++ b/bbtk_creaVtk_PKG/src/bbcreaVtkScreenShot.cxx
@@ -50,7 +50,15 @@ void ScreenShot::Process()
       	wif->Update();
       	vtkPNGWriter *writer = vtkPNGWriter::New();
 		writer->SetFileName( bbGetInputFileName().c_str() );
-		writer->SetInput(wif->GetOutput());
+
+//EED 2017-01-01 Migration VTK7
+#if VTK_MAJOR_VERSION <= 5
+		writer->SetInput( wif->GetOutput() );
+#else
+		writer->SetInputData( wif->GetOutput() );
+#endif
+
+
 		writer->Write();	
 
 /*