itkLevelSetFunctionWithRefitTerm.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkLevelSetFunctionWithRefitTerm.h,v $
  Language:  C++
  Date:      $Date: 2003/09/10 14:28:51 $
  Version:   $Revision: 1.5 $

  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::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:
  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 

---