itkSegmentationLevelSetImageFilter.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkSegmentationLevelSetImageFilter.h,v $
  Language:  C++
  Date:      $Date: 2002/09/30 16:39:32 $
  Version:   $Revision: 1.5 $

  Copyright (c) 2002 Insight 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 __itkSegmentationLevelSetImageFilter_h_
#define __itkSegmentationLevelSetImageFilter_h_

#include "itkSparseFieldLevelSetImageFilter.h"
#include "itkSegmentationLevelSetFunction.h"

namespace itk {

template <class TInputImage, class TOutputImage>
class ITK_EXPORT SegmentationLevelSetImageFilter
  : public SparseFieldLevelSetImageFilter<TInputImage, TOutputImage>
{
public:
  typedef SegmentationLevelSetImageFilter Self;
  typedef SparseFieldLevelSetImageFilter<TInputImage, TOutputImage> Superclass;
  typedef SmartPointer<Self>  Pointer;
  typedef SmartPointer<const Self>  ConstPointer;

  typedef typename Superclass::ValueType ValueType;
  typedef typename Superclass::IndexType IndexType;
  typedef typename Superclass::TimeStepType TimeStepType;
  typedef typename Superclass::OutputImageType OutputImageType;
  typedef typename Superclass::InputImageType  InputImageType;

  typedef SegmentationLevelSetFunction<TOutputImage> SegmentationFunctionType;

  typedef typename SegmentationFunctionType::FeatureImageType FeatureImageType;
  
  itkTypeMacro(SegmentationLevelSetImageFilter, SparseFieldLevelSetImageFilter);

  itkSetMacro(MaximumRMSError, ValueType);
  itkGetMacro(MaximumRMSError, ValueType);

  itkSetMacro(MaximumIterations, unsigned int);
  itkGetMacro(MaximumIterations, unsigned int); 

  virtual void SetFeatureImage(FeatureImageType *f)
  {
    m_SegmentationFunction->SetFeatureImage(f);
    this->Modified();
  }
  virtual FeatureImageType * GetFeatureImage() const
  { return m_SegmentationFunction->GetFeatureImage(); }

  virtual typename SegmentationFunctionType::ImageType *GetSpeedImage() const
  { return m_SegmentationFunction->GetSpeedImage(); }

  
  void SetUseNegativeFeaturesOn()
  {
    this->SetUseNegativeFeatures(true);
  }
  void SetUseNegativeFeaturesOff()
  {
    this->SetUseNegativeFeatures(false);
  }

  itkSetMacro(UseNegativeFeatures, bool);
  itkGetMacro(UseNegativeFeatures, bool);

  itkSetMacro(FeatureScaling, ValueType);
  itkGetMacro(FeatureScaling, ValueType);

  itkSetMacro(CurvatureScaling, ValueType);
  itkGetMacro(CurvatureScaling, ValueType);

  virtual void SetSegmentationFunction(SegmentationFunctionType *s);
  virtual SegmentationFunctionType *GetSegmentationFunction()
  { return m_SegmentationFunction; }
  
protected:
  virtual ~SegmentationLevelSetImageFilter() {}
  SegmentationLevelSetImageFilter();

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

  virtual void InitializeIteration()
  {
    Superclass::InitializeIteration();
    // Estimate the progress of the filter
    this->SetProgress( (float) ((float)this->GetElapsedIterations()
                                / (float)this->GetMaximumIterations()) );
  }
  
  
  void GenerateData();

  bool Halt();

  bool m_UseNegativeFeatures;

  ValueType m_FeatureScaling;

  ValueType m_CurvatureScaling;
  
private:
  unsigned int m_MaximumIterations;
  SegmentationFunctionType *m_SegmentationFunction;
  ValueType m_MaximumRMSError;

};

} // end namespace itk

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

#endif

---