itkFastMarchingExtensionImageFilter.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkFastMarchingExtensionImageFilter.h,v $
  Language:  C++
  Date:      $Date: 2008-12-21 19:13:11 $
  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 __itkFastMarchingExtensionImageFilter_h
#define __itkFastMarchingExtensionImageFilter_h

#include "itkFastMarchingImageFilter.h"
#include "itkImage.h"
#include "itkLevelSet.h"

namespace itk
{

template <
  class TLevelSet, 
  class TAuxValue,
  unsigned int VAuxDimension = 1,
  class TSpeedImage = Image<float,::itk::GetImageDimension<TLevelSet>::ImageDimension>
>
class ITK_EXPORT FastMarchingExtensionImageFilter :
    public FastMarchingImageFilter<TLevelSet,TSpeedImage>
{
public:
  typedef FastMarchingExtensionImageFilter               Self;
  typedef FastMarchingImageFilter<TLevelSet,TSpeedImage> Superclass;
  typedef SmartPointer<Self>                             Pointer;
  typedef SmartPointer<const Self>                       ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(FastMarchingExtensionImageFilter, FastMarchingImageFilter);

  typedef typename Superclass::LevelSetType       LevelSetType;
  typedef typename Superclass::SpeedImageType     SpeedImageType;
  typedef typename Superclass::LevelSetImageType  LevelSetImageType;

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

  itkStaticConstMacro(AuxDimension, unsigned int,VAuxDimension);

  typedef AuxVarTypeDefault<TAuxValue,
                            itkGetStaticConstMacro(AuxDimension),
                            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 Index<itkGetStaticConstMacro(SetDimension)> IndexType;

  AuxImageType * GetAuxiliaryImage( unsigned int idx );

  void SetAuxiliaryAliveValues( AuxValueContainer * values )
    {
    m_AuxAliveValues = values;
    }

  AuxValueContainer * GetAuxiliaryAliveValues(void)
    {
    return m_AuxAliveValues.GetPointer();
    }

  void SetAuxiliaryTrialValues( AuxValueContainer * values )
    {
    m_AuxTrialValues = values;
    }

  typename AuxValueContainer::Pointer GetAuxiliaryTrialValues()
    {
    return m_AuxTrialValues;
    }

#ifdef ITK_USE_CONCEPT_CHECKING

  itkConceptMacro(AuxValueHasNumericTraitsCheck,
                  (Concept::HasNumericTraits<TAuxValue>));

#endif

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

  virtual void Initialize( LevelSetImageType * );
  virtual double UpdateValue( const IndexType & index,
                              const SpeedImageType * speed, LevelSetImageType * output);

  virtual void GenerateOutputInformation();
  virtual void EnlargeOutputRequestedRegion( DataObject *output );

private:
  FastMarchingExtensionImageFilter(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented
  
  typename AuxValueContainer::Pointer    m_AuxAliveValues;
  typename AuxValueContainer::Pointer    m_AuxTrialValues;
  
};

} // namespace itk


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

#endif