X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=vtk%2FvtkgdcmViewer2.cxx;h=f991efdedd4bb8036c7654994c309f5e84e2f885;hb=35ee8ee8af32946514713d271177df8bb2298eb2;hp=d9edf475380444c397d481744044e310d96ca716;hpb=cc2d5f1fdd4082c2a93a227c3e488fdc58e3eea7;p=gdcm.git

diff --git a/vtk/vtkgdcmViewer2.cxx b/vtk/vtkgdcmViewer2.cxx
index d9edf475..f991efde 100644
--- 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/12 13:43:10 $
-  Version:   $Revision: 1.13 $
+  Date:      $Date: 2011/03/29 07:36:02 $
+  Version:   $Revision: 1.19 $
                                                                                 
   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
   l'Image). All rights reserved. See Doc/License.txt or
@@ -32,13 +32,17 @@
 // 
 //----------------------------------------------------------------------------
 #include <vtkRenderWindowInteractor.h>
-#if ( (VTK_MAJOR_VERSION >= 5) || ( VTK_MAJOR_VERSION == 4 && VTK_MINOR_VERSION >= 5 ) )
+#if VTK_MAJOR_VERSION >= 5
 #include "vtkImageColorViewer.h"
 #else
 #include <vtkImageViewer2.h>
 #endif
 #include <vtkStructuredPoints.h>
 #include <vtkStructuredPointsWriter.h>
+#include <vtkPNGWriter.h>
+#include <vtkMetaImageWriter.h>
+#include <vtkWindowToImageFilter.h>
+#include <vtkRenderWindow.h>
 #include <vtkCommand.h>
 #include <vtkRenderer.h>
 #include <vtkImageMapToColors.h>
@@ -67,13 +71,16 @@ 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)
+#if VTK_MAJOR_VERSION >= 5
             int max = ImageViewer->GetSliceMax();
             int slice = (ImageViewer->GetSlice() + 1 ) % ++max;
             ImageViewer->SetSlice( slice );
@@ -88,20 +95,33 @@ 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) || ( VTK_MAJOR_VERSION == 4 && VTK_MINOR_VERSION >= 5 ) )
+#if VTK_MAJOR_VERSION >= 5
    vtkImageColorViewer *ImageViewer;
 #else
    vtkImageViewer2 *ImageViewer;
 #endif
 };
 
+// --------------------------------------------------------------------------------
 
 int main(int argc, char *argv[])
 {
    if( argc < 2 )
       return 0;
+
+   bool metaWrite = false;
   
    vtkGdcmReader *reader = vtkGdcmReader::New();
    reader->AllowLookupTableOff();
@@ -113,6 +133,20 @@ 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 (strcmp (argv[argc-1], "m") == 0)
+   {
+      argc--; // Is it allowed?!?
+      metaWrite = true;
+   }
+
    if( argc == 2 )
       reader->SetFileName( argv[1] );
    else
@@ -126,8 +160,23 @@ int main(int argc, char *argv[])
    //print debug info:
    reader->GetOutput()->Print( cout );
 
+
+   //if you wish you can export dicom to a .mhd file
+   //if (metaWrite) {
+   std::cout << "try to write .mhd" << std::endl;
+      vtkMetaImageWriter* w = vtkMetaImageWriter::New();
+      w->SetInput( reader->GetOutput());
+      w->SetFileName( "/home/jpr/Desktop/toto.mhd" );
+      w->SetFileDimensionality(3);
+      w->SetCompression(false);
+      w->Update();
+      w->Write();
+      //w->Delete();
+      std::cout << "end write .mhd" << std::endl;
+    //}
+
    vtkRenderWindowInteractor *iren = vtkRenderWindowInteractor::New();
-#if ( (VTK_MAJOR_VERSION >= 5) || ( VTK_MAJOR_VERSION == 4 && VTK_MINOR_VERSION >= 5 ) )
+#if VTK_MAJOR_VERSION >= 5
    vtkImageColorViewer *viewer = vtkImageColorViewer::New();
 #else
    vtkImageViewer2 *viewer = vtkImageViewer2::New();
@@ -167,7 +216,7 @@ int main(int argc, char *argv[])
    iren->AddObserver(vtkCommand::CharEvent,obs);
    obs->Delete();
 
-#if ( (VTK_MAJOR_VERSION >= 5) || ( VTK_MAJOR_VERSION == 4 && VTK_MINOR_VERSION >= 5 ) )
+#if VTK_MAJOR_VERSION >= 5
    viewer->Render(); // Don't ask why...
 #endif
    iren->Initialize();
@@ -180,6 +229,17 @@ int main(int argc, char *argv[])
    writer->SetFileTypeToBinary();
    //writer->Write();
 
+   //if you wish you can export dicom to a .mhd file
+   /*
+   if (metaWrite) {
+      vtkMetaImageWriter* w = vtkMetaImageWriter::New();
+      w->SetInput( reader->GetOutput());
+      w->SetFileName( "foo.mhd" );
+      w->SetCompression(false);
+      w->Write();
+      w->Delete();
+    }
+*/
    reader->Delete();
    iren->Delete();
    viewer->Delete();