itkSpatiallyDenseSparseVectorImageNeighborhoodAccessorFunctor.h

Go to the documentation of this file.

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

 Program:   Spatially Dense Sparse Vector Image Neighborhood Accessor Functor

 Copyright (c) Pew-Thian Yap. All rights reserved.
 See http://www.unc.edu/~ptyap/ 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 __itkSpatiallyDenseSparseVectorImageNeighborhoodAccessorFunctor_h
#define __itkSpatiallyDenseSparseVectorImageNeighborhoodAccessorFunctor_h

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


namespace itk
{

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

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

  typedef ImageBoundaryCondition< ImageType > const
                          *ImageBoundaryConditionConstPointerType;

  SpatiallyDenseSparseVectorImageNeighborhoodAccessorFunctor(
    PixelType fillBufferValue , VectorLengthType length)
    : m_FillBufferValue( fillBufferValue ), m_VectorLength( length ) {};
  SpatiallyDenseSparseVectorImageNeighborhoodAccessorFunctor()
    : m_FillBufferValue( NumericTraits<PixelType>::Zero, m_VectorLength( 0 ) ) {};

  inline void SetBegin( const InternalPixelType * begin )  // NOTE: begin is always 0
    { this->m_Begin = const_cast< InternalPixelType * >( begin ); }

  inline PixelType Get( const InternalPixelType *pixelPointer ) const
    {
    PixelType pixel;
    pixel.SetSize(m_VectorLength);

    const typename InternalPixelType::InternalDataType *map =
        pixelPointer->GetDataPointer();

    for ( VectorLengthType i = 0; i < m_VectorLength; i++ )
      {
      typename InternalPixelType::InternalDataType::const_iterator it = map->find( i );

      if ( it == map->end() )
        {
        pixel[i] = m_FillBufferValue[i];
        }
      else
        {
        pixel[i] = it->second;
        }
      }

    return pixel;
    }

  inline void Set( InternalPixelType* &pixelPointer, const PixelType &p ) const
    {
    typename InternalPixelType::InternalDataType *map =
        pixelPointer->GetDataPointer();

    map->clear();

    for ( VectorLengthType i = 0; i < m_VectorLength; i++ )
      {
      if ( p[i] != 0 )
        {
        (*map)[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;
    }

  VectorLengthType GetVectorLength()
    {
    return m_VectorLength;
    }

private:
  PixelType m_FillBufferValue;
  InternalPixelType *m_Begin;  // Begin of the buffer, always 0

  VectorLengthType m_VectorLength;

};

} // end namespace itk
#endif