ITK  4.9.0
Insight Segmentation and Registration Toolkit
WikiExamples/ImageProcessing/MaskImageFilter.cxx
#include "itkConfigure.h"
#if ( ITK_VERSION_MAJOR < 4 ) //These are all defaults in ITKv4
// Not supported in ITKv3.
int main(int argc, char *argv[])
{
return 0;
}
#else
#include "itkImage.h"
#include "QuickView.h"
typedef itk::Image<unsigned char, 2> ImageType;
void CreateHalfMask(ImageType::Pointer image, ImageType::Pointer &mask);
int main(int argc, char *argv[])
{
if(argc < 2)
{
std::cerr << "Usage: " << argv[0] << " filename" << std::endl;
return EXIT_FAILURE;
}
typedef itk::ImageFileReader<ImageType> ReaderType;
ReaderType::Pointer reader = ReaderType::New();
reader->SetFileName(argv[1]);
reader->Update();
ImageType::Pointer mask = ImageType::New();
CreateHalfMask(reader->GetOutput(), mask);
MaskFilterType::Pointer maskFilter = MaskFilterType::New();
maskFilter->SetInput(reader->GetOutput());
maskFilter->SetMaskImage(mask);
mask->Print(std::cout);
QuickView viewer;
viewer.AddImage(
reader->GetOutput(),true,
itksys::SystemTools::GetFilenameName(argv[1]));
std::stringstream desc;
desc << "Mask";
viewer.AddImage(
mask.GetPointer(),
true,
desc.str());
std::stringstream desc2;
desc2 << "MaskFilter";
viewer.AddImage(
maskFilter->GetOutput(),
true,
desc2.str());
viewer.Visualize();
return EXIT_SUCCESS;
}
void CreateHalfMask(ImageType::Pointer image, ImageType::Pointer &mask)
{
ImageType::RegionType region = image->GetLargestPossibleRegion();
mask->SetRegions(region);
mask->Allocate();
ImageType::SizeType regionSize = region.GetSize();
itk::ImageRegionIterator<ImageType> imageIterator(mask,region);
// Make the left half of the mask white and the right half black
while(!imageIterator.IsAtEnd())
{
if(imageIterator.GetIndex()[0] > static_cast<ImageType::IndexValueType>(regionSize[0]) / 2)
{
imageIterator.Set(0);
}
else
{
imageIterator.Set(255);
}
++imageIterator;
}
}
#endif