filters/clitkSplitImageGenericFilter.cxx

   1 /*=========================================================================
   2
   3   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
   4   l'Image). All rights reserved. See Doc/License.txt or
   5   http://www.creatis.insa-lyon.fr/Public/Gdcm/License.html for details.
   6
   7      This software is distributed WITHOUT ANY WARRANTY; without even
   8      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
   9      PURPOSE.  See the above copyright notices for more information.
  10
  11 =========================================================================*/
  12
  13 /**
  14  -------------------------------------------------------------------
  15  * @file   clitkSplitImageGenericFilter.cxx
  16  * @author Joël Schaerer
  17  * @date   20 April 2009
  18
  19  * @brief
  20  -------------------------------------------------------------------*/
  21
  22 #include "clitkSplitImageGenericFilter.h"
  23
  24 #include "clitkSplitImageGenericFilter.txx"
  25 //--------------------------------------------------------------------
  26 clitk::SplitImageGenericFilter::SplitImageGenericFilter():
  27   clitk::ImageToImageGenericFilter<Self>("SplitImage") {
  28   mSplitDimension = 0;
  29   InitializeImageType<3>();
  30   InitializeImageType<4>();
  31 }
  32 //--------------------------------------------------------------------
  33
  34
  35 //--------------------------------------------------------------------
  36 template<unsigned int Dim>
  37 void clitk::SplitImageGenericFilter::InitializeImageType() {
  38   ADD_IMAGE_TYPE(Dim, char);
  39   ADD_IMAGE_TYPE(Dim, unsigned char);
  40   ADD_IMAGE_TYPE(Dim, short);
  41   ADD_IMAGE_TYPE(Dim, int);
  42   ADD_IMAGE_TYPE(Dim, float);
  43   typedef itk::Vector<float, Dim> DVF;
  44   ADD_IMAGE_TYPE(Dim, DVF);
  45 }
  46 //--------------------------------------------------------------------
  47
  48 //--------------------------------------------------------------------
  49 template<class ImageType>
  50 void clitk::SplitImageGenericFilter::UpdateWithInputImageType() {
  51
  52   // Read input
  53   typedef typename ImageType::PixelType PixelType;
  54   typedef itk::Image<PixelType,ImageType::ImageDimension-1> OutputImageType;
  55   typename ImageType::Pointer input = this->GetInput<ImageType>(0);
  56   typedef itk::ExtractImageFilter<ImageType,OutputImageType> FilterType;
  57   typename FilterType::Pointer filter= FilterType::New();
  58
  59   filter->SetInput(input);
  60   typename ImageType::SizeType size=input->GetLargestPossibleRegion().GetSize();
  61   size[mSplitDimension]=0;
  62   typename ImageType::RegionType extracted_region;
  63   extracted_region.SetSize(size);
  64   filter->SetExtractionRegion(extracted_region);
  65   filter->Update();
  66
  67   typename ImageType::IndexType index=input->GetLargestPossibleRegion().GetIndex();
  68   std::string base_filename=GetOutputFilename();
  69   unsigned int number_of_output_images=input->GetLargestPossibleRegion().GetSize()[mSplitDimension];
  70   for (unsigned int i=0;i<number_of_output_images;i++)
  71   {
  72       std::ostringstream ss;
  73       ss << i;
  74       index[mSplitDimension]=i;
  75       extracted_region.SetIndex(index);
  76       filter->SetExtractionRegion(extracted_region);
  77       filter->Update();
  78       SetOutputFilename(base_filename+"_"+ss.str()+".mhd");
  79       typename OutputImageType::Pointer output=filter->GetOutput();
  80       SetNextOutput<OutputImageType>(output);
  81   }
  82 }
  83 //--------------------------------------------------------------------