ITK  5.0.0
Insight Segmentation and Registration Toolkit
WikiExamples/ImageProcessing/BinaryFunctorImageFilterCustom.cxx
#include "itkVectorImage.h"
#include "itkVector.h"
using ImageType = itk::Image<float, 2>;
static void CreateImage(ImageType::Pointer image);
namespace Functor
{
template< class TPixel>
class MySquaredDifference
{
public:
MySquaredDifference() {}
~MySquaredDifference() {}
bool operator!=(const MySquaredDifference &) const
{
return false;
}
bool operator==(const MySquaredDifference & other) const
{
return !( *this != other );
}
inline TPixel operator()(const TPixel & A,
const TPixel & B) const
{
const auto dA = static_cast< double >( A );
const auto dB = static_cast< double >( B );
const double diff = dA - dB;
return static_cast< TPixel >( diff * diff );
}
};
}
int main(int, char *[])
{
ImageType::Pointer image1 = ImageType::New();
CreateImage(image1);
image1->FillBuffer(2);
ImageType::Pointer image2 = ImageType::New();
CreateImage(image2);
image2->FillBuffer(5);
using FilterType = itk::BinaryFunctorImageFilter< ImageType, ImageType, ImageType,
Functor::MySquaredDifference<ImageType::PixelType> >;
FilterType::Pointer filter = FilterType::New();
filter->SetInput1(image1);
filter->SetInput2(image2);
filter->Update();
itk::Index<2> pixelIndex;
pixelIndex.Fill(0);
ImageType::PixelType input1PixelValue = image1->GetPixel(pixelIndex);
ImageType::PixelType input2PixelValue = image2->GetPixel(pixelIndex);
ImageType::PixelType outputPixelValue = filter->GetOutput()->GetPixel(pixelIndex);
std::cout << "pixel1 was = " << input1PixelValue << std::endl;
std::cout << "pixel2 was = " << input2PixelValue << std::endl;
std::cout << "output is = " << outputPixelValue << std::endl;
return EXIT_SUCCESS;
}
void CreateImage(ImageType::Pointer image)
{
start.Fill(0);
size.Fill(10);
ImageType::RegionType region(start,size);
image->SetRegions(region);
image->Allocate();
}