itkLevelSetFunctionWithRefitTerm.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkLevelSetFunctionWithRefitTerm.h,v $
  Language:  C++
  Date:      $Date: 2008-10-16 17:40:09 $
  Version:   $Revision: 1.8 $

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

#include "itkLevelSetFunction.h"
#include "itkSparseImage.h"
#include "itkNumericTraits.h"

namespace itk {

template <class TImageType, class TSparseImageType>
class ITK_EXPORT LevelSetFunctionWithRefitTerm
  : public LevelSetFunction<TImageType>
{
public:

  typedef LevelSetFunctionWithRefitTerm Self;
  typedef LevelSetFunction<TImageType>  Superclass;
  typedef SmartPointer<Self>            Pointer;
  typedef SmartPointer<const Self>      ConstPointer;

  itkTypeMacro( LevelSetFunctionWithRefitTerm, LevelSetFunction );

  itkNewMacro (Self);

  typedef typename Superclass::ImageType              ImageType;
  typedef typename Superclass::FloatOffsetType        FloatOffsetType;
  typedef typename Superclass::ScalarValueType        ScalarValueType;
  typedef typename Superclass::GlobalDataStruct       GlobalDataStruct;
  typedef typename Superclass::NeighborhoodType       NeighborhoodType;
  typedef typename Superclass::NeighborhoodScalesType NeighborhoodScalesType;
  typedef typename Superclass::TimeStepType           TimeStepType;

  typedef typename ImageType::IndexType IndexType;

  typedef TSparseImageType SparseImageType;

  typedef typename SparseImageType::NodeType NodeType;

  typedef typename NodeType::NodeDataType NormalVectorType;

  void SetRefitWeight( const ScalarValueType w )
    {
    m_RefitWeight = w;
    }

  void SetOtherPropagationWeight( const ScalarValueType w )
    {
    m_OtherPropagationWeight = w;
    }

  void SetSparseTargetImage( SparseImageType *im )
    { m_SparseTargetImage = im; }

  SparseImageType* GetSparseTargetImage() const
    { return m_SparseTargetImage; }

  virtual TimeStepType ComputeGlobalTimeStep( void *GlobalData ) const;

protected:
  ScalarValueType m_RefitWeight;

  ScalarValueType m_OtherPropagationWeight;

  LevelSetFunctionWithRefitTerm ();
  void PrintSelf(std::ostream& os, Indent indent) const;

  ScalarValueType ComputeCurvature( const NeighborhoodType & ) const;

  virtual ScalarValueType PropagationSpeed(const NeighborhoodType& ,
                                           const FloatOffsetType &,
                                           GlobalDataStruct * = 0 ) const;

  virtual ScalarValueType OtherPropagationSpeed(const NeighborhoodType& ,
                                                const FloatOffsetType &,
                                                GlobalDataStruct * = 0) const
    {
    return NumericTraits<ScalarValueType>::Zero;
    }

private:
  LevelSetFunctionWithRefitTerm(const Self&); //purposely not implemented
  void operator=(const Self&);   //purposely not implemented
   
  typename SparseImageType::Pointer m_SparseTargetImage;

  ScalarValueType m_MinVectorNorm;

  static const unsigned long   m_NumVertex;
  static const ScalarValueType m_DimConst;
};

} // end namespace itk

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

#endif