itkMultiVariableFunctorVectorImageFilter.h

Go to the documentation of this file.

#ifndef itkMultiVariableFunctorVectorImageFilter_h
#define itkMultiVariableFunctorVectorImageFilter_h

#include "itkFunctorBaseVectorImageFilter.h"

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

  itkNewMacro(Self);

  itkTypeMacro(MultiVariableFunctorVectorImageFilter, 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:
  MultiVariableFunctorVectorImageFilter();
  virtual ~MultiVariableFunctorVectorImageFilter() {}

  void BeforeThreadedGenerateData () ITK_OVERRIDE;

  virtual void PropagateRequestedRegion(DataObject *output) ITK_OVERRIDE
    {
    }

  virtual void GenerateOutputInformation() ITK_OVERRIDE;

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

protected: 
  

private:
  MultiVariableFunctorVectorImageFilter(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
MultiVariableVectorOPImage(const std::vector<itk::SmartPointer<ImageType> >& images, itk::SmartPointer<ImageOutType>& outImage, const OpFunctor& func, const itk::SmartPointer<MaskImageType>& mask=nullptr, int numberOfThreads=-1, int vectorAxis=3)
{
  typedef itk::MultiVariableFunctorVectorImageFilter<ImageType, ImageOutType, OpFunctor, MaskImageType> FunctorImageFilterType;
  typename FunctorImageFilterType::Pointer filter = FunctorImageFilterType::New();

  if (!itk::IsImageEmpty(mask))
    filter->SetMaskImage(mask);
  for ( int i = 0; i < images.size(); ++i ) 
    {
    filter->SetInput(i, images[i]);
    }
  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(__itkMultiVariableFunctorVectorImageFilter_hxx)
#include "itkMultiVariableFunctorVectorImageFilter.hxx"
#endif

#endif