Add vtkGdcmReader::SetFlipY(bool) (default : true, to keep old behaviour)

[gdcm.git] / vtk / vtkgdcmViewer2.cxx

diff --git a/vtk/vtkgdcmViewer2.cxx b/vtk/vtkgdcmViewer2.cxx
index deefba6fec822f4498304025f52c19f937cd8fa8..b840ea1cd471f390c95c2a7b3bac67deda703ce1 100644 (file)
--- a/vtk/vtkgdcmViewer2.cxx
+++ b/vtk/vtkgdcmViewer2.cxx
@@ -3,8 +3,8 @@
   Program:   gdcm
   Module:    $RCSfile: vtkgdcmViewer2.cxx,v $
   Language:  C++
-  Date:      $Date: 2007/09/13 11:32:53 $
-  Version:   $Revision: 1.14 $
+  Date:      $Date: 2009/04/18 14:42:51 $
+  Version:   $Revision: 1.16 $
                                                                                 
   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
   l'Image). All rights reserved. See Doc/License.txt or
@@ -39,6 +39,9 @@
 #endif
 #include <vtkStructuredPoints.h>
 #include <vtkStructuredPointsWriter.h>
+#include <vtkPNGWriter.h>
+#include <vtkWindowToImageFilter.h>
+#include <vtkRenderWindow.h>
 #include <vtkCommand.h>
 #include <vtkRenderer.h>
 #include <vtkImageMapToColors.h>
@@ -67,11 +70,14 @@ public:
    {
       this->ImageViewer = NULL;
    }
-   virtual void Execute(vtkObject *, unsigned long event, void* )
+   virtual void Execute(vtkObject *caller, unsigned long event, void* /*callData*/ )
    {
       if ( this->ImageViewer )
       {
-         if ( event == vtkCommand::CharEvent )
+         vtkRenderWindowInteractor * rwi = vtkRenderWindowInteractor::SafeDownCast( caller );
+         char keycode = 0;
+         if( rwi ) keycode = rwi->GetKeyCode();
+         if ( event == vtkCommand::CharEvent && keycode != 's' )
          {
 #if VTK_MAJOR_VERSION >= 5
             int max = ImageViewer->GetSliceMax();
@@ -88,6 +94,16 @@ public:
 #endif
             ImageViewer->Render();
          }
+         else if ( keycode == 's' )
+         {
+           vtkPNGWriter * writer = vtkPNGWriter::New();
+           vtkWindowToImageFilter * w2i = vtkWindowToImageFilter::New();
+           w2i->SetInput( rwi->GetRenderWindow() );
+           writer->SetInput( w2i->GetOutput() );
+           writer->SetFileName( "snapshot.png" );
+           writer->Write();
+           //std::cerr << "Screenshort saved to snapshot.png" << std::endl;
+         }
       }
    }
 #if VTK_MAJOR_VERSION >= 5
@@ -113,6 +129,14 @@ int main(int argc, char *argv[])
       reader->SetKeepOverlays(true);
    }
 
+// not a very clever way to pass several params
+// but it's just for checking
+   if (strcmp (argv[argc-1], "n") == 0)
+   {
+      argc--; // Is it allowed?!?
+      reader->SetFlipY(false);
+   }
+
    if( argc == 2 )
       reader->SetFileName( argv[1] );
    else