lib/cpExtensions/Algorithms/ImageFunctorFilter.h

   1 // -------------------------------------------------------------------------
   2 // @author Leonardo Florez-Valencia (florez-l@javeriana.edu.co)
   3 // -------------------------------------------------------------------------
   4
   5 #ifndef __CPEXTENSIONS__ALGORITHMS__IMAGEFUNCTORFILTER__H__
   6 #define __CPEXTENSIONS__ALGORITHMS__IMAGEFUNCTORFILTER__H__
   7
   8 #include <itkImageToImageFilter.h>
   9 #include <itkImageScanlineConstIterator.h>
  10 #include <itkImageScanlineIterator.h>
  11 #include <itkProgressReporter.h>
  12
  13 namespace cpExtensions
  14 {
  15   namespace Algorithms
  16   {
  17     /**
  18      */
  19     template< class I, class O, class F >
  20     class ImageFunctorFilter
  21       : public itk::ImageToImageFilter< I, O >
  22     {
  23     public:
  24       typedef ImageFunctorFilter             Self;
  25       typedef itk::ImageToImageFilter< I, O > Superclass;
  26       typedef itk::SmartPointer< Self >       Pointer;
  27       typedef itk::SmartPointer< const Self > ConstPointer;
  28
  29       typedef I TInputImage;
  30       typedef O TOutputImage;
  31       typedef F TFunctor;
  32
  33       typedef typename O::RegionType TRegion;
  34
  35     public:
  36       itkNewMacro( Self );
  37       itkTypeMacro( ImageFunctorFilter, itkImageToImageFilter );
  38
  39       itkGetMacro( Functor, F );
  40       itkGetConstMacro( Functor, F );
  41
  42     protected:
  43       ImageFunctorFilter( )
  44         : Superclass( )
  45         {
  46         }
  47       virtual ~ImageFunctorFilter( )
  48         {
  49         }
  50
  51       virtual void ThreadedGenerateData(
  52         const TRegion& region,
  53         itk::ThreadIdType threadId
  54         )
  55         {
  56           const typename TRegion::SizeType& regionSize = region.GetSize( );
  57           if( regionSize[ 0 ] == 0 )
  58             return;
  59           const I* in = this->GetInput( );
  60           O* out = this->GetOutput( 0 );
  61
  62           const size_t nLines = region.GetNumberOfPixels( ) / regionSize[ 0 ];
  63           itk::ProgressReporter progress( this, threadId, nLines );
  64
  65           // Define the iterators
  66           itk::ImageScanlineConstIterator< I > inIt( in, region );
  67           itk::ImageScanlineIterator< O > outIt( out, region );
  68
  69           inIt.GoToBegin( );
  70           outIt.GoToBegin( );
  71           while( !inIt.IsAtEnd( ) )
  72           {
  73             while( !inIt.IsAtEndOfLine( ) )
  74             {
  75               outIt.Set( this->m_Functor( inIt.Get( ) ) );
  76               ++inIt;
  77               ++outIt;
  78
  79             } // elihw
  80             inIt.NextLine( );
  81             outIt.NextLine( );
  82             progress.CompletedPixel( );
  83
  84           } // elihw
  85         }
  86
  87     private:
  88       // Purposely not implemented.
  89       ImageFunctorFilter( const Self& );
  90       void operator=( const Self& );
  91
  92     protected:
  93       F m_Functor;
  94     };
  95
  96   } // ecapseman
  97
  98 } // ecapseman
  99
 100 #endif // __CPEXTENSIONS__ALGORITHMS__IMAGEFUNCTORFILTER__H__
 101
 102 // eof - $RCSfile$