itkUnaryFunctorVectorImageFilter.h

Go to the documentation of this file.

#ifndef itkUnaryFunctorVectorImageFilter_h
#define itkUnaryFunctorVectorImageFilter_h

#include "itkFunctorBaseVectorImageFilter.h"

namespace itk
{
template< typename TInputImage, typename TOutputImage, typename TFunction, class TMaskImage=Image<double,3> >
class ITK_EXPORT UnaryFunctorVectorImageFilter
  :public FunctorBaseVectorImageFilter< TInputImage, TOutputImage, TFunction, TMaskImage >
{
public:
  typedef UnaryFunctorVectorImageFilter                         Self;
  typedef FunctorBaseVectorImageFilter< TInputImage, TOutputImage, TFunction, TMaskImage > Superclass;
  typedef SmartPointer< Self >                            Pointer;
  typedef SmartPointer< const Self >                      ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(UnaryFunctorVectorImageFilter, FunctorBaseVectorImageFilter);

  typedef TFunction FunctorType;

  typedef TInputImage                             InputImageType;
  typedef typename InputImageType::Pointer        InputImagePointer;
  typedef typename InputImageType::ConstPointer   InputImageConstPointer;
  typedef typename InputImageType::IndexType      InputImageIndexType;
  typedef typename InputImageType::SizeType       InputImageSizeType;
  typedef typename InputImageType::SpacingType    InputImageSpacingType;
  typedef typename InputImageType::PixelType      InputImagePixelType;
  typedef typename InputImageType::RegionType     InputImageRegionType;
  
  typedef TOutputImage                            OutputImageType;
  typedef typename OutputImageType::Pointer       OutputImagePointer;
  typedef typename OutputImageType::IndexType     OutputImageIndexType;
  typedef typename OutputImageType::SizeType      OutputImageSizeType;
  typedef typename OutputImageType::SpacingType   OutputImageSpacingType;
  typedef typename OutputImageType::PixelType     OutputImagePixelType;
  typedef typename OutputImageType::RegionType    OutputImageRegionType;
  
  typedef typename Superclass::MaskImageType      MaskImageType;

protected:
  UnaryFunctorVectorImageFilter();
  virtual ~UnaryFunctorVectorImageFilter() {}

  void BeforeThreadedGenerateData () ITK_OVERRIDE;

  virtual void GenerateOutputInformation() ITK_OVERRIDE;

  void ThreadedGenerateData(const OutputImageRegionType & outputRegionForThread,
                            ThreadIdType threadId) ITK_OVERRIDE;

protected: 
  

private:
  UnaryFunctorVectorImageFilter(const Self &) ITK_DELETE_FUNCTION;
  void operator=(const Self &) ITK_DELETE_FUNCTION;

};


template <class ImageType, class ImageOutType, class OpFunctor, class MaskImageType=Image<double,4> >
void
UnaryVectorOPImage(const itk::SmartPointer<ImageType>& image, itk::SmartPointer<ImageOutType>& outImage, const OpFunctor& func, const itk::SmartPointer<MaskImageType>& mask=nullptr, int numberOfThreads=-1, int vectorAxis=3)
{
  typedef itk::UnaryFunctorVectorImageFilter<ImageType, ImageOutType, OpFunctor, MaskImageType> UnaryFunctorFilterType;
  typename UnaryFunctorFilterType::Pointer filter = UnaryFunctorFilterType::New();

  if (!itk::IsImageEmpty(mask))
    filter->SetMaskImage(mask);
  filter->SetInput(image);
  if (vectorAxis>=0)
    filter->SetVectorAxis(vectorAxis);
  filter->SetFunctor(func);
  filter->SetDebug(utl::IsLogDebug());
  filter->SetLogLevel(utl::LogLevel);
  if (numberOfThreads>0)
    filter->SetNumberOfThreads(numberOfThreads);

  filter->Update();

  outImage = filter->GetOutput();       
}

} // end namespace itk

#if !defined(ITK_MANUAL_INSTANTIATION) && !defined(__itkUnaryFunctorVectorImageFilter_hxx)
#include "itkUnaryFunctorVectorImageFilter.hxx"
#endif

#endif