WikiExamples/Smoothing/VectorMedianImageFilter.cxx

#include "itkMedianImageFilter.h"

#include "itkImage.h"
#include "itkImageFileReader.h"
#include "itkSubtractImageFilter.h"
#include "itkRGBPixel.h"
#include "itkVector.h"
#include "itkVectorCastImageFilter.h"

#include "itksys/SystemTools.hxx"

#include <sstream>

#include "QuickView.h"

namespace itk
{
template< typename TComponent, unsigned int NVectorDimension = 3 >
class myVectorPixel : public Vector<TComponent,NVectorDimension>
{
public:
  using Self = myVectorPixel;
  using Superclass = Vector<TComponent,NVectorDimension>;

  bool operator<=(const Self & r) const
  {
    if (this->GetNorm() <= r.GetNorm())
      {
      return true;
      }
    else
      {
      return false;
      }
  }
  bool operator<(const Self & r) const
  {
    if (this->GetNorm() < r.GetNorm())
      {
      return true;
      }
    else
      {
      return false;
      }
  }
};
}

int main(int argc, char * argv[])
{
  // Verify command line arguments
  if( argc < 2 )
    {
    std::cerr << "Usage: " << std::endl;
    std::cerr << argv[0] << " InputImageFile [radius]" << std::endl;
    return EXIT_FAILURE;
    }
  std::string inputFilename = argv[1];

  // Setup types
  using PixelType = itk::myVectorPixel<unsigned char,3>;
  using ImageType = itk::Image<PixelType, 2 >;
  using RGBPixelType = itk::RGBPixel<unsigned char>;
  using RGBImageType = itk::Image<RGBPixelType, 2 >;

  using ReaderType = itk::ImageFileReader<ImageType>;
  using FilterType = itk::MedianImageFilter<ImageType, ImageType >;
  using CastType = itk::VectorCastImageFilter<ImageType, RGBImageType>;

  // Create and setup a reader
  ReaderType::Pointer reader = ReaderType::New();
  reader->SetFileName( inputFilename );

  // Create and setup a median filter
  FilterType::Pointer medianFilter = FilterType::New();
  FilterType::InputSizeType radius;
  radius.Fill(2);
  if (argc > 2)
    {
    radius.Fill(atoi(argv[2]));
    }

  medianFilter->SetRadius(radius);
  medianFilter->SetInput(reader->GetOutput());

  // Cast the custom myVector's to RGBPixel's
  CastType::Pointer castReader = CastType::New();
  castReader->SetInput(reader->GetOutput());

  CastType::Pointer castMedian = CastType::New();
  castMedian->SetInput(medianFilter->GetOutput());

  QuickView viewer;
  viewer.AddRGBImage(
    castReader->GetOutput(),
    true,
    itksys::SystemTools::GetFilenameName(inputFilename));  

  std::stringstream desc;
  desc << "MedianImageFilter, radius = " << radius;
  viewer.AddRGBImage(
    castMedian->GetOutput(),
    true,
    desc.str());  

  viewer.Visualize();

  return EXIT_SUCCESS;
}