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://www.centreleonberard.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
  27 //------------------------------------------------------------------------------
  28 /**Converts the itk image to vv, handling the 4D problem
  29  * The time_sequence boolean specifies that the image is to be interpreted as a time sequence,
  30  * even if its dim is < 4. */
  31
  32 template<unsigned int Dim, class PixelType>
  33 struct vvImageFromITK_Impl
  34 {
  35   static vvImage::Pointer Do (typename itk::Image<PixelType,Dim>::Pointer input, bool time_sequence=false)
  36   {
  37     vvImage::Pointer vv_image=vvImage::New();
  38     typedef itk::Image< PixelType, Dim > InputImageType;
  39     if (time_sequence) //The time sequence case: create a series of VTK images
  40     {
  41       typedef itk::Image< PixelType,  Dim - 1 >    ItkImageType;
  42       typedef itk::ExtractImageFilter<InputImageType, ItkImageType> FilterType;
  43
  44       //extract the 3D slices and put them in a std::vector<vtkImageData*>
  45       input->UpdateOutputInformation();
  46       typename InputImageType::RegionType inputRegion = input->GetLargestPossibleRegion();
  47       typename InputImageType::SizeType inputSize = inputRegion.GetSize();
  48       typename InputImageType::IndexType start = inputRegion.GetIndex();
  49       typename InputImageType::SizeType extractedRegionSize = inputSize;
  50       typename InputImageType::RegionType extractedRegion;
  51       extractedRegionSize[Dim - 1] = 0;
  52       extractedRegion.SetSize(extractedRegionSize);
  53
  54       for (unsigned int i = 0; i < inputSize[Dim - 1]; i++) {
  55         start[Dim - 1] = i;
  56         extractedRegion.SetIndex(start);
  57
  58         typename FilterType::Pointer filter = FilterType::New();
  59 #if ITK_VERSION_MAJOR == 4
  60         filter->SetDirectionCollapseToSubmatrix();
  61 #endif
  62         filter->SetExtractionRegion(extractedRegion);
  63         filter->SetInput(input);
  64         filter->ReleaseDataFlagOn();
  65         vv_image->AddItkImage<ItkImageType>(filter->GetOutput());
  66       }
  67       vv_image->SetTimeSpacing(input->GetSpacing()[Dim-1]);
  68       vv_image->SetTimeOrigin(input->GetOrigin()[Dim-1]);
  69     }
  70     else //Dim == 1,2,3 and not time_sequence
  71     {
  72         vv_image->AddItkImage<InputImageType>(input);
  73     }
  74     return vv_image;
  75   }
  76 };
  77
  78 template<class PixelType>
  79 struct vvImageFromITK_Impl<4u, PixelType>
  80 {
  81   static vvImage::Pointer Do (typename itk::Image<PixelType,4u>::Pointer input, bool time_sequence=false)
  82   {
  83     vvImage::Pointer vv_image=vvImage::New();
  84     typedef itk::Image< PixelType, 4 > InputImageType;
  85
  86     typedef itk::Image< PixelType, 3 >    ItkImageType;
  87     typedef itk::ExtractImageFilter<InputImageType, ItkImageType> FilterType;
  88
  89     //extract the 3D slices and put them in a std::vector<vtkImageData*>
  90     input->UpdateOutputInformation();
  91     typename InputImageType::RegionType inputRegion = input->GetLargestPossibleRegion();
  92     typename InputImageType::SizeType inputSize = inputRegion.GetSize();
  93     typename InputImageType::IndexType start = inputRegion.GetIndex();
  94     typename InputImageType::SizeType extractedRegionSize = inputSize;
  95     typename InputImageType::RegionType extractedRegion;
  96     extractedRegionSize[3] = 0;
  97     extractedRegion.SetSize(extractedRegionSize);
  98
  99     for (unsigned int i = 0; i < inputSize[3]; i++) {
 100       start[3] = i;
 101       extractedRegion.SetIndex(start);
 102
 103       typename FilterType::Pointer filter = FilterType::New();
 104 #if ITK_VERSION_MAJOR == 4
 105       filter->SetDirectionCollapseToSubmatrix();
 106 #endif
 107       filter->SetExtractionRegion(extractedRegion);
 108       filter->SetInput(input);
 109       filter->ReleaseDataFlagOn();
 110       vv_image->AddItkImage<ItkImageType>(filter->GetOutput());
 111     }
 112     vv_image->SetTimeSpacing(input->GetSpacing()[3]);
 113     vv_image->SetTimeOrigin(input->GetOrigin()[3]);
 114     return vv_image;
 115   }
 116 };
 117
 118 template<unsigned int Dim, class PixelType> vvImage::Pointer vvImageFromITK(typename itk::Image<PixelType,Dim>::Pointer input, bool time_sequence=false)
 119 {
 120     assert(Dim < 5 && Dim > 0); // We don't handle anything higher than 4-dimensional (for the moment :-p)
 121     return vvImageFromITK_Impl<Dim, PixelType>::Do(input, time_sequence);
 122 }
 123 //------------------------------------------------------------------------------
 124
 125 #endif //vvImageFromITK