vtk/vtkgdcmViewer.cxx

   1 #include <iostream>
   2
   3 #include <vtkRenderWindowInteractor.h>
   4 #include <vtkImageViewer2.h>
   5 #include <vtkStructuredPoints.h>
   6 #include <vtkStructuredPointsWriter.h>
   7 #include <vtkCommand.h>
   8 #include <vtkRenderer.h>
   9 #include <vtkImageMapToColors.h>
  10 #include <vtkLookupTable.h>
  11
  12 #include "vtkGdcmReader.h"
  13
  14 //----------------------------------------------------------------------------
  15 // Callback for the interaction
  16 class vtkgdcmObserver : public vtkCommand
  17 {
  18   public:
  19   virtual char const *GetClassName() const { return "vtkgdcmObserver";}
  20   static vtkgdcmObserver *New()
  21     { return new vtkgdcmObserver; }
  22   vtkgdcmObserver()
  23     {
  24       this->ImageViewer = NULL;
  25     }
  26   virtual void Execute(vtkObject *wdg, unsigned long event, void* calldata)
  27     {
  28       if ( this->ImageViewer )
  29         {
  30           if ( event == vtkCommand::CharEvent )
  31           {
  32             int max = ImageViewer->GetWholeZMax();
  33             int slice = (ImageViewer->GetZSlice() + 1 ) % ++max;
  34             ImageViewer->SetZSlice( slice );
  35             ImageViewer->GetRenderer()->ResetCameraClippingRange();
  36             ImageViewer->Render();
  37           }
  38         }
  39     }
  40     vtkImageViewer2 *ImageViewer;
  41 };
  42
  43
  44 int main(int argc, char *argv[])
  45 {
  46   if( argc < 2 )
  47     return 0;
  48
  49   vtkGdcmReader *reader = vtkGdcmReader::New();
  50
  51   if( argc == 2 )
  52     reader->SetFileName( argv[1] );
  53   else
  54     for(int i=1; i< argc; i++)
  55       reader->AddFileName( argv[i] );
  56
  57   reader->DebugOn();
  58   reader->Update();
  59
  60   //print debug info:
  61   reader->GetOutput()->Print( std::cout );
  62
  63   vtkRenderWindowInteractor *iren = vtkRenderWindowInteractor::New();
  64
  65   vtkImageViewer2 *viewer = vtkImageViewer2::New();
  66   if( reader->GetLookupTable() )
  67   {
  68     //convert to color:
  69     vtkImageMapToColors *map = vtkImageMapToColors::New ();
  70     map->SetInput (reader->GetOutput());
  71     map->SetLookupTable (reader->GetLookupTable());
  72     map->SetOutputFormatToRGB();
  73     viewer->SetInput ( map->GetOutput() );
  74     map->Delete();
  75   }
  76   else
  77   {
  78     viewer->SetInput ( reader->GetOutput() );
  79   }
  80   viewer->SetupInteractor (iren);
  81
  82 //  float *range = reader->GetOutput()->GetScalarRange();
  83 //  viewer->SetColorWindow (range[1] - range[0]);
  84 //  viewer->SetColorLevel (0.5 * (range[1] + range[0]));
  85
  86   // Here is where we setup the observer,
  87   vtkgdcmObserver *obs = vtkgdcmObserver::New();
  88   obs->ImageViewer = viewer;
  89   iren->AddObserver(vtkCommand::CharEvent,obs);
  90   obs->Delete();
  91
  92   iren->Initialize();
  93   iren->Start();
  94
  95   //if you wish you can export dicom to a vtk file
  96   vtkStructuredPointsWriter *writer = vtkStructuredPointsWriter::New();
  97   writer->SetInput( reader->GetOutput());
  98   writer->SetFileName( "foo.vtk" );
  99   writer->SetFileTypeToBinary();
 100   writer->Write();
 101
 102   reader->Delete();
 103   iren->Delete();
 104   viewer->Delete();
 105   writer->Delete();
 106
 107   return 0;
 108 }