itkLevelSetVelocityNeighborhoodExtractor.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkLevelSetVelocityNeighborhoodExtractor.h,v $
  Language:  C++
  Date:      $Date: 2009-01-24 20:02:58 $
  Version:   $Revision: 1.23 $

  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 __itkLevelSetVelocityNeighborhoodExtractor_h
#define __itkLevelSetVelocityNeighborhoodExtractor_h

#include "itkLevelSetNeighborhoodExtractor.h"
#include "itkLevelSet.h"
#include "itkIndex.h"

namespace itk
{

template <
  class TLevelSet,
  class TAuxValue,
  unsigned int VAuxDimension = 1
  >
class ITK_EXPORT LevelSetVelocityNeighborhoodExtractor :
    public LevelSetNeighborhoodExtractor<TLevelSet>
{
public:
  typedef LevelSetVelocityNeighborhoodExtractor    Self;
  typedef LevelSetNeighborhoodExtractor<TLevelSet> Superclass;
  typedef SmartPointer<Self>                       Pointer;
  typedef SmartPointer<const Self>                 ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(LevelSetVelocityNeighborhoodExtractor, 
               LevelSetNeighborhoodExtractor);

  typedef LevelSetTypeDefault<TLevelSet>  LevelSetType;

  itkStaticConstMacro(SetDimension, unsigned int,
                      LevelSetType::SetDimension);

  typedef ::itk::Index<itkGetStaticConstMacro(SetDimension)> Index;

  typedef AuxVarTypeDefault<TAuxValue,VAuxDimension,itkGetStaticConstMacro(SetDimension)>
                                                    AuxVarType;
  typedef typename AuxVarType::AuxValueType         AuxValueType;
  typedef typename AuxVarType::AuxValueVectorType   AuxValueVectorType;
  typedef typename AuxVarType::AuxValueContainer    AuxValueContainer;
  typedef typename AuxVarType::AuxImageType         AuxImageType;
  typedef typename AuxVarType::AuxImagePointer      AuxImagePointer;
  typedef typename AuxVarType::AuxImageConstPointer AuxImageConstPointer;

  void SetAuxImage( const AuxImageType * ptr, unsigned int idx = 0 )
    { 
    if ( idx < VAuxDimension && m_AuxImage[idx] != ptr )
      {
      m_AuxImage[idx] = ptr;
      }
    this->Modified();
    }

  AuxImageConstPointer GetAuxImage( unsigned int idx = 0 )
    {
    if ( idx >= VAuxDimension ) 
      { 
      return NULL; 
      }
    else 
      { 
      return m_AuxImage[idx]; 
      }
    }

  itkGetObjectMacro( AuxInsideValues, AuxValueContainer );

  itkGetObjectMacro( AuxOutsideValues, AuxValueContainer );

protected:
  LevelSetVelocityNeighborhoodExtractor();
  ~LevelSetVelocityNeighborhoodExtractor(){};
  void PrintSelf( std::ostream& os, Indent indent ) const;

  virtual void Initialize();
  virtual double CalculateDistance( Index & index );

private:
  LevelSetVelocityNeighborhoodExtractor(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

  typename AuxValueContainer::Pointer    m_AuxInsideValues;
  typename AuxValueContainer::Pointer    m_AuxOutsideValues;
  AuxImageConstPointer                   m_AuxImage[VAuxDimension];

};

} // namespace itk

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

#endif