itkReinitializeLevelSetImageFilter.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkReinitializeLevelSetImageFilter.h,v $
  Language:  C++
  Date:      $Date: 2009-04-23 03:53:37 $
  Version:   $Revision: 1.19 $

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

#include "itkImageToImageFilter.h"
#include "itkLevelSet.h"
#include "itkLevelSetNeighborhoodExtractor.h"
#include "itkFastMarchingImageFilter.h"

namespace itk
{
template <class TLevelSet>
class ITK_EXPORT ReinitializeLevelSetImageFilter :
    public ImageToImageFilter<TLevelSet,TLevelSet>
{
public:
  typedef ReinitializeLevelSetImageFilter         Self;
  typedef ImageToImageFilter<TLevelSet,TLevelSet> Superclass;
  typedef SmartPointer<Self>                      Pointer;
  typedef SmartPointer<const Self>                ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(ReinitializeLevelSetImageFilter, ImageToImageFilter);

  typedef LevelSetTypeDefault<TLevelSet>               LevelSetType;
  typedef typename LevelSetType::LevelSetImageType     LevelSetImageType;
  typedef typename LevelSetType::LevelSetPointer       LevelSetPointer;
  typedef typename LevelSetType::LevelSetConstPointer  LevelSetConstPointer;
  typedef typename LevelSetType::PixelType             PixelType;
  typedef typename LevelSetType::NodeType              NodeType;
  typedef typename LevelSetType::NodeContainer         NodeContainer;
  typedef typename LevelSetType::NodeContainerPointer  NodeContainerPointer;

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

  itkSetMacro( LevelSetValue, double );
  itkGetConstMacro( LevelSetValue, double );

  itkSetMacro( NarrowBanding, bool );
  itkGetConstMacro( NarrowBanding, bool );
  itkBooleanMacro( NarrowBanding );

  itkSetClampMacro( InputNarrowBandwidth, double, 0.0, 
                    NumericTraits<double>::max());
  itkGetConstMacro( InputNarrowBandwidth, double );

  itkSetClampMacro( OutputNarrowBandwidth, double, 0.0, 
                    NumericTraits<double>::max());
  itkGetConstMacro( OutputNarrowBandwidth, double );

  void SetNarrowBandwidth( double value )
    {
    this->SetInputNarrowBandwidth(value);
    this->SetOutputNarrowBandwidth(value);
    }

  void SetInputNarrowBand( NodeContainer * ptr );
  NodeContainerPointer GetInputNarrowBand() const
    { return m_InputNarrowBand; }

  NodeContainerPointer GetOutputNarrowBand() const
  { return m_OutputNarrowBand; }

#ifdef ITK_USE_CONCEPT_CHECKING

  itkConceptMacro(LevelSetDoubleAdditiveOperatorsCheck,
                  (Concept::AdditiveOperators<PixelType, double>));
  itkConceptMacro(LevelSetOStreamWritableCheck,
                  (Concept::OStreamWritable<PixelType>));

#endif

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

  typedef Image<float, itkGetStaticConstMacro(SetDimension) > SpeedImageType;
  typedef LevelSetNeighborhoodExtractor<TLevelSet>            LocatorType;
  typedef FastMarchingImageFilter<TLevelSet, SpeedImageType>  FastMarchingImageFilterType;

  void GenerateData();
  virtual void GenerateDataFull();
  virtual void GenerateDataNarrowBand();
  virtual void AllocateOutput();

  virtual void GenerateInputRequestedRegion();
  virtual void EnlargeOutputRequestedRegion( DataObject * );

  void SetOutputNarrowBand( NodeContainer *ptr )
    { m_OutputNarrowBand = ptr; }

private:
  ReinitializeLevelSetImageFilter(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented
  
  double                                m_LevelSetValue;

  typename LocatorType::Pointer                    m_Locator;
  typename FastMarchingImageFilterType::Pointer    m_Marcher;

  bool                                  m_NarrowBanding;
  double                                m_InputNarrowBandwidth;
  double                                m_OutputNarrowBandwidth;
  NodeContainerPointer                  m_InputNarrowBand;
  NodeContainerPointer                  m_OutputNarrowBand;

};

} // namespace itk

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

#endif