common/vvFromITK.h

   1 #ifndef VVIMAGEFROMITK_H
   2 #define VVIMAGEFROMITK_H
   3 #include "vvImage.h"
   4 //#include <itkImage.h>
   5 #include <itkExtractImageFilter.h>
   6 #include <itkImageToVTKImageFilter.h>
   7
   8 /**Converts the itk image to vv, handling the 4D problem
   9  * The time_sequence boolean specifies that the image is to be interpreted as a time sequence,
  10  * even if its dim is < 4. */
  11 template<unsigned int Dim, class PixelType> vvImage::Pointer vvImageFromITK(typename itk::Image<PixelType,Dim>::Pointer input, bool time_sequence=false)
  12 {
  13     assert(Dim < 5 && Dim > 0); // We don't handle anything higher than 4-dimensional (for the moment :-p)
  14     vvImage::Pointer vv_image=vvImage::New();
  15     vv_image->Init(); //Delete any existing images
  16     typedef itk::Image< PixelType, Dim > InputImageType;
  17
  18     if (Dim == 4 || time_sequence) //The time sequence case: create a series of VTK images
  19     {
  20         typedef itk::Image< PixelType,  Dim - 1 >    ConnectorImageType;
  21         typedef itk::ImageToVTKImageFilter <ConnectorImageType> ConnectorType;
  22         typedef itk::ExtractImageFilter<InputImageType,ConnectorImageType> FilterType;
  23
  24         typename FilterType::Pointer filter = FilterType::New();
  25         typename ConnectorType::Pointer connector = ConnectorType::New();
  26
  27         //extract the 3D slices and put them in a std::vector<vtkImageData*>
  28         typename InputImageType::RegionType inputRegion = input->GetLargestPossibleRegion();
  29         typename InputImageType::SizeType inputSize = inputRegion.GetSize();
  30
  31         typename InputImageType::SizeType extractedRegionSize = inputSize;
  32         typename InputImageType::RegionType extractedRegion;
  33         extractedRegionSize[Dim - 1] = 0;
  34         extractedRegion.SetSize(extractedRegionSize);
  35
  36         filter->SetInput(input);
  37         connector->SetInput(filter->GetOutput());
  38
  39         typename InputImageType::IndexType start = inputRegion.GetIndex();
  40
  41         for (unsigned int i = 0; i < inputSize[Dim - 1]; i++) {
  42             start[Dim - 1] = i;
  43             extractedRegion.SetIndex(start);
  44             filter->SetExtractionRegion(extractedRegion);
  45             try {
  46                 filter->Update();
  47             }
  48             catch ( itk::ExceptionObject & err ) {
  49                 std::cerr << "Error while setting vvImage from ITK (Dim==4) [Extract phase]"
  50                           << " " << err << std::endl;
  51                 return vv_image;
  52             }
  53             try {
  54                 connector->Update();
  55             }
  56             catch ( itk::ExceptionObject & err ) {
  57                 std::cerr << "Error while setting vvImage from ITK (Dim==4) [Connect phase]"
  58                           << " " << err << std::endl;
  59                 return vv_image;
  60             }
  61             vtkImageData *image = vtkImageData::New();
  62             image->DeepCopy(connector->GetOutput());
  63             vv_image->AddImage(image);
  64         }
  65     }
  66     else //Dim == 1,2,3 and not time_sequence
  67     {
  68         typedef itk::Image< PixelType,  Dim >    ConnectorImageType;
  69         typedef itk::ImageToVTKImageFilter <ConnectorImageType> ConnectorType;
  70         typename ConnectorType::Pointer connector = ConnectorType::New();
  71         connector->SetInput(input);
  72
  73         try {
  74             connector->Update();
  75         }
  76         catch ( itk::ExceptionObject & err ) {
  77             std::cerr << "Error while setting vvImage from ITK (Dim==3)"
  78                       << " " << err << std::endl;
  79             return vv_image;
  80         }
  81         vtkImageData *image = vtkImageData::New();
  82         image->DeepCopy(connector->GetOutput());
  83         vv_image->AddImage(image);
  84     }
  85     return vv_image;
  86 }
  87
  88 #endif //vvImageFromITK