vtk/vtkgdcmViewer.cxx

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