#3233 creaVtk Feature New Normal - GetCameraInfo SetCameraInfo ScreenShot

[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 35a5998a0f05ec31e1987ada7cd368b4336bc4f8..4ee0b66d14a1f495628ec992f58eeed6e97ac13c 100644 (file)
--- a/bbtk_creaVtk_PKG/src/bbcreaVtkScreenShot.cxx
+++ b/bbtk_creaVtk_PKG/src/bbcreaVtkScreenShot.cxx
@@ -37,17 +37,21 @@ void ScreenShot::Process()
 //    bbSetOutputOut( bbGetInputIn() );
 //    std::cout << "Output value = " <<bbGetOutputOut() << std::endl;
   
-      vtkWindowToImageFilter *wif = vtkWindowToImageFilter::New();
+       vtkWindowToImageFilter *wif = vtkWindowToImageFilter::New();
+               if (bbGetInputSize().size()==2) 
+               {
+                       bbGetInputvtkRenderer()->GetVTKWindow()->SetSize( bbGetInputSize()[0], bbGetInputSize()[1] );
+               }
        wif->SetInput( bbGetInputvtkRenderer()->GetVTKWindow() );
        wif->SetMagnification( bbGetInputScale()[0] );
 //             wif->SetScale( bbGetInputScale()[0], bbGetInputScale()[1] );
        wif->SetInputBufferTypeToRGB();
        wif->ReadFrontBufferOff();
        wif->Update();
-      vtkPNGWriter *writer = vtkPNGWriter::New();
-       writer->SetFileName( bbGetInputFileName().c_str() );
-       writer->SetInput(wif->GetOutput());
-       writer->Write();        
+       vtkPNGWriter *writer = vtkPNGWriter::New();
+               writer->SetFileName( bbGetInputFileName().c_str() );
+               writer->SetInput(wif->GetOutput());
+               writer->Write();        
 
 /*
        vtkFFMPEGWriter *writer2 = vtkFFMPEGWriter::New();