itkFunctorBaseVectorImageFilter.h

Go to the documentation of this file.

#ifndef itkFunctorBaseVectorImageFilter_h
#define itkFunctorBaseVectorImageFilter_h

#include "itkMaskedImageToImageFilter.h"
#include "utlCoreMacro.h"
#include "utlTypeinfo.h"

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

  itkNewMacro(Self);

  itkTypeMacro(FunctorBaseVectorImageFilter, MaskedImageToImageFilter);

  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;

  itkSetGetMacro(VectorAxis, int);

  FunctorType &       GetFunctor() { return m_Functor; }
  const FunctorType & GetFunctor() const { return m_Functor; }

  void SetFunctor(const FunctorType & functor)
  {
    if ( m_Functor != functor )
      {
      m_Functor = functor;
      this->Modified();
      }
  }

protected:
  FunctorBaseVectorImageFilter() 
    {
    this->SetNumberOfRequiredInputs(1);
    }
  virtual ~FunctorBaseVectorImageFilter() {}


protected: 
  
  void PrintSelf(std::ostream & os, Indent indent) const ITK_OVERRIDE
    {
    Superclass::PrintSelf(os, indent);
    utlLogOSVar(os << indent, m_VectorAxis);
    m_Functor.Print(os <<indent << "m_Functor : "<<  utl::TypeName(m_Functor)<< " ("<< &m_Functor <<") = \n");
    }
  
  typename LightObject::Pointer InternalClone() const ITK_OVERRIDE
    {
    typename LightObject::Pointer loPtr = Superclass::InternalClone();

    typename Self::Pointer rval = dynamic_cast<Self *>(loPtr.GetPointer());
    if(rval.IsNull())
      {
      itkExceptionMacro(<< "downcast to type " << this->GetNameOfClass()<< " failed.");
      }
    rval->m_Functor = m_Functor;
    rval->m_VectorAxis = m_VectorAxis;
    return loPtr;
    }

  FunctorType m_Functor;

  int m_VectorAxis=3;

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

};
} // end namespace itk

#endif