VTK7

[creaVtk.git] / bbtk_creaVtk_PKG / src / bbcreaVtkScreenShot.cxx

diff --git a/bbtk_creaVtk_PKG/src/bbcreaVtkScreenShot.cxx b/bbtk_creaVtk_PKG/src/bbcreaVtkScreenShot.cxx
index 4ee0b66d14a1f495628ec992f58eeed6e97ac13c..08f4089c3dae5547eb2701706f3a714ab7474091 100644 (file)
--- 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();        
 
 /*