1 /*=========================================================================
2 Program: vv http://www.creatis.insa-lyon.fr/rio/vv
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
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.
13 It is distributed under dual licence
15 - BSD See included LICENSE.txt file
16 - CeCILL-B http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
17 ===========================================================================**/
18 #ifndef clitkNVectorImageTo4DImageGenericFilter_txx
19 #define clitkNVectorImageTo4DImageGenericFilter_txx
21 /* =================================================
22 * @file clitkNVectorImageTo4DImageGenericFilter.txx
28 ===================================================*/
30 #include "itkVectorImageToImageAdaptor.h"
36 //-------------------------------------------------------------------
37 // Update with the number of dimensions
38 //-------------------------------------------------------------------
39 template<unsigned int Dimension>
41 NVectorImageTo4DImageGenericFilter::UpdateWithDim(std::string PixelType, unsigned int Components)
43 if (m_Verbose) std::cout << "Image was detected to be "<<Dimension<<"D and "<< PixelType<<"..."<<std::endl;
45 if (PixelType == "short") {
46 UpdateWithDimAndPixelType<Dimension, short>();
47 } else if (PixelType == "unsigned short") {
48 UpdateWithDimAndPixelType<Dimension, unsigned short>();
49 } else if (PixelType == "unsigned_short") {
50 UpdateWithDimAndPixelType<Dimension, unsigned short>();
51 } else if (PixelType == "char") {
52 UpdateWithDimAndPixelType<Dimension, char>();
53 } else if (PixelType == "unsigned_char") {
54 UpdateWithDimAndPixelType<Dimension, unsigned char>();
55 } else if (PixelType == "int") {
56 UpdateWithDimAndPixelType<Dimension, int>();
57 } else if (PixelType == "unsigned_int") {
58 UpdateWithDimAndPixelType<Dimension, unsigned int>();
59 } else if (PixelType == "double") {
60 UpdateWithDimAndPixelType<Dimension, double>();
61 } else if (PixelType == "float") {
62 UpdateWithDimAndPixelType<Dimension, float>();
64 std::cerr << "Error, pixel type : \"" << PixelType << "\" unknown !" << std::endl;
69 //-------------------------------------------------------------------
70 // Update with the number of dimensions and the pixeltype
71 //-------------------------------------------------------------------
72 template <unsigned int Dimension, class PixelType>
74 NVectorImageTo4DImageGenericFilter::UpdateWithDimAndPixelType()
77 typedef itk::VectorImage<PixelType, Dimension> InputImageType;
78 typedef itk::Image<PixelType, Dimension+1> OutputImageType;
81 typedef itk::ImageFileReader<InputImageType> InputReaderType;
82 typename InputReaderType::Pointer reader = InputReaderType::New();
83 reader->SetFileName( m_InputFileName);
85 typename InputImageType::Pointer input= reader->GetOutput();
88 typedef itk::VectorImageToImageAdaptor<PixelType, Dimension> ImageAdaptorType;
89 typedef itk::ImageFileWriter<OutputImageType> WriterType;
90 typename ImageAdaptorType::Pointer adaptor = ImageAdaptorType::New();
91 typename OutputImageType::Pointer output = OutputImageType::New();
92 typename WriterType::Pointer writer = WriterType::New();
94 adaptor->SetExtractComponentIndex(0);
95 adaptor->SetImage(input);
96 std::string fileName=m_ArgsInfo.output_arg;
99 typename OutputImageType::IndexType index;
101 typename OutputImageType::SizeType size;
102 size.Fill(input->GetNumberOfComponentsPerPixel());
103 typename OutputImageType::SpacingType spacing;
105 typename OutputImageType::PointType origin;
107 for (unsigned int pixelDim=0; pixelDim<Dimension; ++pixelDim)
109 size[pixelDim]=adaptor->GetLargestPossibleRegion().GetSize(pixelDim);
110 spacing[pixelDim]=input->GetSpacing()[pixelDim];
111 origin[pixelDim]=input->GetOrigin()[pixelDim];
113 typename OutputImageType::RegionType region;
114 region.SetSize(size);
115 region.SetIndex(index);
116 output->SetRegions(region);
117 output->SetOrigin(origin);
118 output->SetSpacing(spacing);
120 writer->SetInput(output);
123 for (unsigned int pixelDim=0; pixelDim<input->GetNumberOfComponentsPerPixel(); ++pixelDim)
125 adaptor->SetExtractComponentIndex(pixelDim);
127 itk::ImageRegionIterator<InputImageType> imageIterator(input,input->GetLargestPossibleRegion());
129 while(!imageIterator.IsAtEnd())
131 typename OutputImageType::IndexType indexVector;
133 for (unsigned int indexDim=0; indexDim<Dimension; ++indexDim)
135 indexVector[indexDim]=imageIterator.GetIndex().GetElement(indexDim);
137 indexVector[Dimension]=pixelDim;
139 output->SetPixel(indexVector, adaptor->GetPixel(imageIterator.GetIndex()));
144 writer->SetFileName(fileName);
149 #endif //#define clitkNVectorImageTo4DImageGenericFilter_txx