ITK/Examples/IO/ReadUnknownImageType
From KitwarePublic
ReadUnknownImageType.cxx
#include "itkImage.h" #include "itkImageFileReader.h" template<typename TImageType> static void ReadFile(std::string filename, typename TImageType::Pointer image); int main(int argc, char *argv[]) { if(argc < 2) { std::cerr << "Required: filename" << std::endl; return EXIT_FAILURE; } std::string inputFilename = argv[1]; typedef itk::ImageIOBase::IOComponentType ScalarPixelType; itk::ImageIOBase::Pointer imageIO = itk::ImageIOFactory::CreateImageIO( inputFilename.c_str(), itk::ImageIOFactory::ReadMode); imageIO->SetFileName(inputFilename); imageIO->ReadImageInformation(); const ScalarPixelType pixelType = imageIO->GetComponentType(); std::cout << "Pixel Type is " << imageIO->GetComponentTypeAsString(pixelType) // 'double' << std::endl; const size_t numDimensions = imageIO->GetNumberOfDimensions(); std::cout << "numDimensions: " << numDimensions << std::endl; // '2' std::cout << "component size: " << imageIO->GetComponentSize() << std::endl; // '8' std::cout << "pixel type (string): " << imageIO->GetPixelTypeAsString(imageIO->GetPixelType()) << std::endl; // 'vector' std::cout << "pixel type: " << imageIO->GetPixelType() << std::endl; // '5' /* switch (pixelType) { case itk::ImageIOBase::COVARIANTVECTOR: typedef itk::Image<unsigned char, 2> ImageType; ImageType::Pointer image = ImageType::New(); ReadFile<ImageType>(inputFilename, image); break; typedef itk::Image<unsigned char, 2> ImageType; ImageType::Pointer image = ImageType::New(); ReadFile<ImageType>(inputFilename, image); break; default: std::cerr << "Pixel Type (" << imageIO->GetComponentTypeAsString(pixelType) << ") not supported. Exiting." << std::endl; return -1; } */ return EXIT_SUCCESS; } template<typename TImageType> void ReadFile(std::string filename, typename TImageType::Pointer image) { typedef itk::ImageFileReader<TImageType> ReaderType; typename ReaderType::Pointer reader = ReaderType::New(); reader->SetFileName(filename); reader->Update(); image->Graft(reader->GetOutput()); }
CreateImages.cxx
#include "itkImage.h" #include "itkImageFileWriter.h" #include <itkCovariantVector.h> #include <string> template<typename TImageType> void WriteFile(typename TImageType::Pointer image, std::string filename); int main(int, char *[]) { typedef itk::Image< itk::CovariantVector<double, 4> , 2> ImageType4; ImageType4::Pointer image4 = ImageType4::New(); WriteFile<ImageType4>(image4, std::string("image4.mhd")); typedef itk::Image< itk::CovariantVector<double, 5> , 2> ImageType5; ImageType5::Pointer image5 = ImageType5::New(); WriteFile<ImageType5>(image5, std::string("image5.mhd")); return EXIT_SUCCESS; } template<typename TImageType> void WriteFile(typename TImageType::Pointer image, std::string filename) { // Create image typename TImageType::IndexType start; start[0] = 0; start[1] = 0; typename TImageType::SizeType size; size[0] = 20; size[1] = 30; typename TImageType::RegionType region; region.SetSize(size); region.SetIndex(start); image->SetRegions(region); image->Allocate(); typedef typename itk::ImageFileWriter<TImageType> WriterType; typename WriterType::Pointer writer = WriterType::New(); writer->SetInput(image); writer->SetFileName(filename); writer->Update(); } template void WriteFile<itk::Image< itk::CovariantVector<double, 4> , 2> >(itk::Image< itk::CovariantVector<double, 4> , 2>::Pointer, std::string); template void WriteFile<itk::Image< itk::CovariantVector<double, 5> , 2> >(itk::Image< itk::CovariantVector<double, 5> , 2>::Pointer, std::string);
CMakeLists.txt
cmake_minimum_required(VERSION 2.8) project(ReadUnknownImageType) find_package(ITK REQUIRED) include(${ITK_USE_FILE}) add_executable(ReadUnknownImageType ReadUnknownImageType.cxx) if( "${ITK_VERSION_MAJOR}" LESS 4 ) target_link_libraries(ReadUnknownImageType ITKReview ${ITK_LIBRARIES}) else( "${ITK_VERSION_MAJOR}" LESS 4 ) target_link_libraries(ReadUnknownImageType ${ITK_LIBRARIES}) endif( "${ITK_VERSION_MAJOR}" LESS 4 )