ImageMultiplication.cxx

Go to the documentation of this file.

#include "utlITK.h"

#include "ImageMultiplicationCLP.h"
#include "itkMultiplyByConstantVectorImageFilter.h"


int 
main (int argc, char const* argv[])
{
  // GenerateCLP
  PARSE_ARGS;

  typedef float FloatType;

  // NOTE: use 4D to work for VectorImage<FloatType,3> and Image<FloatType,4>
  typedef itk::VectorImage<FloatType,4> InputImageType;
  typedef itk::VectorImage<FloatType,4> OutputImageType;
  typedef InputImageType::PixelType  ContainerType;

  InputImageType::Pointer input = InputImageType::New();
  itk::ReadImage(_InputFile, input);
  int pixelDimension = input->GetNumberOfComponentsPerPixel();

  typedef itk::MultiplyByConstantVectorImageFilter<InputImageType, ContainerType, OutputImageType> MultiplyFilterType;
  MultiplyFilterType::Pointer filter = MultiplyFilterType::New();


  ContainerType vec;
  vec.SetSize(pixelDimension);
  for ( int i = 0; i < pixelDimension; i += 1 ) 
    vec[i] = _Scale;

  filter->SetInput(input);
  filter->SetConstantVector(vec);

  filter->Update();
  
  OutputImageType::Pointer output = filter->GetOutput();

  itk::SaveImage(output, _OutputFile);
  
  return 0;
}