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   ADD_VEC_IMAGE_TYPE(Dim, 3,float);
  44 }
  45 //--------------------------------------------------------------------
  46
  47 //--------------------------------------------------------------------
  48 template<class ImageType>
  49 void clitk::SplitImageGenericFilter::UpdateWithInputImageType() {
  50
  51   // Read input
  52   typedef typename ImageType::PixelType PixelType;
  53   typedef itk::Image<PixelType,ImageType::ImageDimension-1> OutputImageType;
  54   typename ImageType::Pointer input = this->GetInput<ImageType>(0);
  55   typedef itk::ExtractImageFilter<ImageType,OutputImageType> FilterType;
  56   typename FilterType::Pointer filter= FilterType::New();
  57
  58   filter->SetInput(input);
  59   typename ImageType::SizeType size=input->GetLargestPossibleRegion().GetSize();
  60   size[mSplitDimension]=0;
  61   typename ImageType::RegionType extracted_region;
  62   extracted_region.SetSize(size);
  63   filter->SetExtractionRegion(extracted_region);
  64   filter->Update();
  65
  66   typename ImageType::IndexType index=input->GetLargestPossibleRegion().GetIndex();
  67   std::string base_filename=GetOutputFilename();
  68   unsigned int number_of_output_images=input->GetLargestPossibleRegion().GetSize()[mSplitDimension];
  69   for (unsigned int i=0;i<number_of_output_images;i++)
  70   {
  71       std::ostringstream ss;
  72       ss << i;
  73       index[mSplitDimension]=i;
  74       extracted_region.SetIndex(index);
  75       filter->SetExtractionRegion(extracted_region);
  76       filter->Update();
  77       SetOutputFilename(base_filename+"_"+ss.str()+".mhd");
  78       typename OutputImageType::Pointer output=filter->GetOutput();
  79       SetNextOutput<OutputImageType>(output);
  80   }
  81 }
  82 //--------------------------------------------------------------------