ITK/Examples/IO/ReadUnknownImageType

From KitwarePublic

Jump to: navigation, search

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 )
Personal tools