common/vvFromITK.h

   1 /*=========================================================================
   2   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
   3
   4   Authors belong to:
   5   - University of LYON              http://www.universite-lyon.fr/
   6   - Léon Bérard cancer center       http://oncora1.lyon.fnclcc.fr
   7   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
   8
   9   This software is distributed WITHOUT ANY WARRANTY; without even
  10   the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  11   PURPOSE.  See the copyright notices for more information.
  12
  13   It is distributed under dual licence
  14
  15   - BSD        See included LICENSE.txt file
  16   - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  17 ======================================================================-====*/
  18 #ifndef VVIMAGEFROMITK_H
  19 #define VVIMAGEFROMITK_H
  20
  21 // vv
  22 #include "vvImage.h"
  23
  24 // itk
  25 #include <itkExtractImageFilter.h>
  26 #include <itkImageToVTKImageFilter.h>
  27
  28 //------------------------------------------------------------------------------
  29 /**Converts the itk image to vv, handling the 4D problem
  30  * The time_sequence boolean specifies that the image is to be interpreted as a time sequence,
  31  * even if its dim is < 4. */
  32 template<unsigned int Dim, class PixelType> vvImage::Pointer vvImageFromITK(typename itk::Image<PixelType,Dim>::Pointer input, bool time_sequence=false)
  33 {
  34     assert(Dim < 5 && Dim > 0); // We don't handle anything higher than 4-dimensional (for the moment :-p)
  35     vvImage::Pointer vv_image=vvImage::New();
  36     vv_image->Init(); //Delete any existing images
  37     typedef itk::Image< PixelType, Dim > InputImageType;
  38
  39     if (Dim == 4 || time_sequence) //The time sequence case: create a series of VTK images
  40     {
  41         typedef itk::Image< PixelType,  Dim - 1 >    ConnectorImageType;
  42         typedef itk::ImageToVTKImageFilter <ConnectorImageType> ConnectorType;
  43         typedef itk::ExtractImageFilter<InputImageType,ConnectorImageType> FilterType;
  44
  45         typename FilterType::Pointer filter = FilterType::New();
  46         typename ConnectorType::Pointer connector = ConnectorType::New();
  47
  48         //extract the 3D slices and put them in a std::vector<vtkImageData*>
  49         typename InputImageType::RegionType inputRegion = input->GetLargestPossibleRegion();
  50         typename InputImageType::SizeType inputSize = inputRegion.GetSize();
  51
  52         typename InputImageType::SizeType extractedRegionSize = inputSize;
  53         typename InputImageType::RegionType extractedRegion;
  54         extractedRegionSize[Dim - 1] = 0;
  55         extractedRegion.SetSize(extractedRegionSize);
  56
  57         filter->SetInput(input);
  58         connector->SetInput(filter->GetOutput());
  59
  60         typename InputImageType::IndexType start = inputRegion.GetIndex();
  61
  62         for (unsigned int i = 0; i < inputSize[Dim - 1]; i++) {
  63             start[Dim - 1] = i;
  64             extractedRegion.SetIndex(start);
  65             filter->SetExtractionRegion(extractedRegion);
  66             try {
  67                 filter->Update();
  68             }
  69             catch ( itk::ExceptionObject & err ) {
  70                 std::cerr << "Error while setting vvImage from ITK (Dim==4) [Extract phase]"
  71                           << " " << err << std::endl;
  72                 return vv_image;
  73             }
  74             try {
  75                 connector->Update();
  76             }
  77             catch ( itk::ExceptionObject & err ) {
  78                 std::cerr << "Error while setting vvImage from ITK (Dim==4) [Connect phase]"
  79                           << " " << err << std::endl;
  80                 return vv_image;
  81             }
  82             vtkImageData *image = vtkImageData::New();
  83             image->DeepCopy(connector->GetOutput());
  84             vv_image->AddImage(image);
  85         }
  86         vv_image->SetTimeSpacing(input->GetSpacing()[Dim-1]);
  87         vv_image->SetTimeOrigin(input->GetOrigin()[Dim-1]);
  88     }
  89     else //Dim == 1,2,3 and not time_sequence
  90     {
  91         typedef itk::Image< PixelType,  Dim >    ConnectorImageType;
  92         typedef itk::ImageToVTKImageFilter <ConnectorImageType> ConnectorType;
  93         typename ConnectorType::Pointer connector = ConnectorType::New();
  94         connector->SetInput(input);
  95
  96         try {
  97             connector->Update();
  98         }
  99         catch ( itk::ExceptionObject & err ) {
 100             std::cerr << "Error while setting vvImage from ITK (Dim==3)"
 101                       << " " << err << std::endl;
 102             return vv_image;
 103         }
 104         vtkImageData *image = vtkImageData::New();
 105         image->DeepCopy(connector->GetOutput());
 106         vv_image->AddImage(image);
 107     }
 108     return vv_image;
 109 }
 110 //------------------------------------------------------------------------------
 111
 112 #endif //vvImageFromITK