itkBSplineOperator.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

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

#ifndef itkBSplineOperator_h
#define itkBSplineOperator_h

#include <itkNeighborhoodOperator.h>
#include <math.h>

namespace itk {

template<class TPixel,unsigned int VDimension=2,
  class TAllocator = NeighborhoodAllocator<TPixel> >
class ITK_EXPORT BSplineOperator
  : public NeighborhoodOperator<TPixel, VDimension, TAllocator>
{
public:
  
  typedef BSplineOperator                                         Self;
  typedef NeighborhoodOperator<TPixel, VDimension, TAllocator>    Superclass;

  BSplineOperator() { this->m_Spacing = 1.0; };

  BSplineOperator(const Self &other)
    : NeighborhoodOperator<TPixel, VDimension, TAllocator>(other)
    {
      m_Spacing = other.m_Spacing;
    }

  Self &operator=(const Self &other)
  {
    Superclass::operator=(other);
    m_Spacing = other.m_Spacing;
    return *this;
  }

  void SetSpacing(double spacing) { this->m_Spacing = spacing; }
  double GetSpacing() { return this->m_Spacing; }
  
protected:

  typedef typename Superclass::CoefficientVector CoefficientVector;

  CoefficientVector GenerateCoefficients();

  void Fill(const CoefficientVector& coeff)
    {
      this->FillCenteredDirectional(coeff);  
    }

private:

  double m_Spacing;

  const char *GetNameOfClass() { return "itkBSplineOperator"; }

};

template <class TPixel, unsigned int VDimension, class TContainer>
std::ostream & operator<<(std::ostream &os, const BSplineOperator<TPixel,VDimension,TContainer> &bspline)
{
  os << "BSplineOperator:" << std::endl;
  os << "    Radius:" << bspline.GetRadius() << std::endl;
  os << "    Size:" << bspline.GetSize() << std::endl;
  os << "    DataBuffer:" << bspline.GetBufferReference() << std::endl;

  for (unsigned int i = 0; i < bspline.Size(); i++)
    {
      os << "       " << i << ":" << bspline[i] << std::endl;  
    }
  return os;
}

} // namespace itk


#ifndef ITK_MANUAL_INSTANTIATION
#include "itkBSplineOperator.txx"
#endif

#endif