itkVectorImageNeighborhoodAccessorFunctor.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkVectorImageNeighborhoodAccessorFunctor.h,v $
  Language:  C++
  Date:      $Date: 2009-03-03 15:11:40 $
  Version:   $Revision: 1.4 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef __itkVectorImageNeighborhoodAccessorFunctor_h
#define __itkVectorImageNeighborhoodAccessorFunctor_h

#include "itkVariableLengthVector.h"
#include "itkImageBoundaryCondition.h"
#include "itkNeighborhood.h"
#include "itkImageBase.h"

namespace itk
{
 
template< class TImage >
class VectorImageNeighborhoodAccessorFunctor
{
public:
  typedef TImage                                ImageType;
  typedef typename ImageType::PixelType         PixelType;
  typedef typename ImageType::InternalPixelType InternalPixelType;
  typedef unsigned int                          VectorLengthType;
  typedef typename ImageType::OffsetType        OffsetType;

  typedef Neighborhood< InternalPixelType *,
          ::itk::GetImageDimension< TImage >::ImageDimension> NeighborhoodType;

  typedef ImageBoundaryCondition< ImageType > const 
                          *ImageBoundaryConditionConstPointerType;
  
  VectorImageNeighborhoodAccessorFunctor( VectorLengthType length ) 
      : m_VectorLength( length ), m_OffsetMultiplier( length -1 ) { };
  VectorImageNeighborhoodAccessorFunctor() 
      : m_VectorLength( 0 ), m_OffsetMultiplier( 0 ) {};
  
  inline void SetBegin( const InternalPixelType * begin ) 
    { this->m_Begin = const_cast< InternalPixelType * >( begin ); }

  inline PixelType Get( const InternalPixelType *pixelPointer ) const 
    { 
    return PixelType(pixelPointer+(pixelPointer-m_Begin)*m_OffsetMultiplier, m_VectorLength);
    }

  inline void Set( InternalPixelType* &pixelPointer, const PixelType &p ) const
    {
    InternalPixelType *truePixelPointer = 
      pixelPointer + (pixelPointer-m_Begin)*m_OffsetMultiplier;
    for( VectorLengthType i=0; i< m_VectorLength; i++ ) 
      { 
      truePixelPointer[i] = p[i];
      }
    }

  inline PixelType BoundaryCondition( 
      const OffsetType& point_index,
      const OffsetType &boundary_offset, 
      const NeighborhoodType *data,
      const ImageBoundaryConditionConstPointerType boundaryCondition) const
    {
    return boundaryCondition->operator()(point_index, boundary_offset, data, *this);
    }


  void SetVectorLength( VectorLengthType length )
    { 
    m_VectorLength = length;
    m_OffsetMultiplier = length - 1;
    }

  VectorLengthType GetVectorLength()
    {
    return m_VectorLength;
    }

private:
  VectorLengthType m_VectorLength; 
  VectorLengthType m_OffsetMultiplier; // m_OffsetMultiplier = m_VectorLength-1
                                        // (precomputed for speedup).
  InternalPixelType *m_Begin; // Begin of the buffer. 
};


} // end namespace itk
#endif