WikiExamples/VectorImages/VectorMagnitudeImageFilter.cxx

#include "itkImage.h"
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include "itkRescaleIntensityImageFilter.h"
#include "itkVectorImage.h"
#include "itkVectorMagnitudeImageFilter.h"

int main(int argc, char * argv[])
{
  // Verify command line arguments
  if( argc < 3 )
    {
    std::cerr << "Usage: " << std::endl;
    std::cerr << argv[0] << " inputImageFile outputImageFile" << std::endl;
    return EXIT_FAILURE;
    }

  // Parse command line arguments
  std::string inputFilename = argv[1];
  std::string outputFilename = argv[2];

  // Setup types
  typedef itk::VectorImage< float,  2 >    VectorImageType;
  typedef itk::Image< unsigned char, 2 >   UnsignedCharImageType;

  // Create and setup a reader
  typedef itk::ImageFileReader< VectorImageType >  ReaderType;
  ReaderType::Pointer reader = ReaderType::New();
  reader->SetFileName( inputFilename );

  typedef itk::ImageFileWriter< UnsignedCharImageType >  WriterType;

  typedef itk::VectorMagnitudeImageFilter<
    VectorImageType, UnsignedCharImageType >  VectorMagnitudeFilterType;
  VectorMagnitudeFilterType::Pointer magnitudeFilter =
    VectorMagnitudeFilterType::New();
  magnitudeFilter->SetInput( reader->GetOutput() );

  // To write the magnitude image file, we should rescale the gradient values
  // to a reasonable range
  typedef itk::RescaleIntensityImageFilter<
    UnsignedCharImageType, UnsignedCharImageType > rescaleFilterType;

  rescaleFilterType::Pointer rescaler =
    rescaleFilterType::New();
  rescaler->SetOutputMinimum(0);
  rescaler->SetOutputMaximum(255);
  rescaler->SetInput( magnitudeFilter->GetOutput() );
  
  WriterType::Pointer writer =
    WriterType::New();

  writer->SetFileName( outputFilename );
  writer->SetInput( rescaler->GetOutput() );
  writer->Update();

  return EXIT_SUCCESS;
}