itkNarrowBandLevelSetImageFilter.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkNarrowBandLevelSetImageFilter.h,v $
  Language:  C++
  Date:      $Date: 2003/09/10 14:28:36 $
  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 __itkNarrowBandLevelSetImageFilter_h_
#define __itkNarrowBandLevelSetImageFilter_h_

#include "itkNarrowBandImageFilterBase.h"
#include "itkSegmentationLevelSetFunction.h"
#include "itkFastChamferDistanceImageFilter.h"
#include "itkIsoContourDistanceImageFilter.h"

namespace itk {

template <class TInputImage,
          class TFeatureImage,
          class TOutputPixelType = float,
          class TOutputImage = Image<TOutputPixelType,
                                     ::itk::GetImageDimension<TInputImage>::ImageDimension> >
class ITK_EXPORT NarrowBandLevelSetImageFilter
  : public NarrowBandImageFilterBase<TInputImage, TOutputImage>
{
public:
  typedef NarrowBandLevelSetImageFilter Self;
  typedef NarrowBandImageFilterBase<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::InputImageType  InputImageType;

  typedef TOutputImage   OutputImageType;
  typedef TFeatureImage FeatureImageType;

  typedef SegmentationLevelSetFunction <OutputImageType, FeatureImageType>
  SegmentationFunctionType;

  typedef typename SegmentationFunctionType::VectorImageType VectorImageType;
    
  itkTypeMacro(NarrowBandLevelSetImageFilter, NarrowBandImageFilterBase);

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

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

  virtual void SetFeatureImage(const FeatureImageType *f)
  {
    this->ProcessObject::SetNthInput( 1, const_cast< FeatureImageType * >(f) );
    m_SegmentationFunction->SetFeatureImage(f);
  }
  virtual FeatureImageType * GetFeatureImage()
  {
    return ( static_cast< FeatureImageType *>(this->ProcessObject::GetInput(1)) );
  }

  virtual void SetInitialImage(InputImageType *f)
  {
    this->SetInput(f);
  }
  
  virtual const typename SegmentationFunctionType::ImageType *GetSpeedImage() const
  { return m_SegmentationFunction->GetSpeedImage(); }

  virtual const typename SegmentationFunctionType::VectorImageType *GetAdvectionImage() const
  { return m_SegmentationFunction->GetAdvectionImage(); }

  void SetUseNegativeFeaturesOn()
  {
    itkWarningMacro( << "SetUseNegativeFeaturesOn has been deprecated.  Please use ReverseExpansionDirectionOn() instead" );
    this->ReverseExpansionDirectionOn();
  }
  void SetUseNegativeFeaturesOff()
  {
    itkWarningMacro( << "SetUseNegativeFeaturesOff has been deprecated.  Please use ReverseExpansionDirectionOff() instead" );
    this->ReverseExpansionDirectionOff();
  }

  void SetUseNegativeFeatures( bool u )
  {
    itkWarningMacro( << "SetUseNegativeFeatures has been deprecated.  Please use SetReverseExpansionDirection instead" );
    if (u == true)
      {
      this->SetReverseExpansionDirection(false);
      }
    else
      {
      this->SetReverseExpansionDirection(true);
      }
  }
  bool GetUseNegativeFeatures() const
  {
    itkWarningMacro( << "GetUseNegativeFeatures has been deprecated.  Please use GetReverseExpansionDirection() instead" );
    if ( this->GetUseNegativeFeatures() == false)
      {
      return true;
      }
    else
      {
      return false;
      }
  }

  itkSetMacro(ReverseExpansionDirection, bool);
  itkGetMacro(ReverseExpansionDirection, bool);
  itkBooleanMacro(ReverseExpansionDirection);
  
  void SetFeatureScaling(ValueType v)
  {
    if (v != m_SegmentationFunction->GetPropagationWeight())
      {        
      this->SetPropagationScaling(v);
      }
    if (v != m_SegmentationFunction->GetAdvectionWeight())
      {
      this->SetAdvectionScaling(v);
      }
  }

  void SetPropagationScaling(ValueType v)
  {
    if (v != m_SegmentationFunction->GetPropagationWeight())
      {        
      m_SegmentationFunction->SetPropagationWeight(v);
      }
  }
  ValueType GetPropagationScaling() const
  {
    return m_SegmentationFunction->GetPropagationWeight();
  }

  void SetAdvectionScaling(ValueType v)
  {
    if (v != m_SegmentationFunction->GetAdvectionWeight())
      {        
      m_SegmentationFunction->SetAdvectionWeight(v);
      }
  }
  ValueType GetAdvectionScaling() const
  {
    return m_SegmentationFunction->GetAdvectionWeight();
  }

  void SetCurvatureScaling(ValueType v)
  {
    if (v != m_SegmentationFunction->GetCurvatureWeight())
      {        
      m_SegmentationFunction->SetCurvatureWeight(v);
      }
  }
  ValueType GetCurvatureScaling() const
  {
    return m_SegmentationFunction->GetCurvatureWeight();
  }

  virtual void SetSegmentationFunction(SegmentationFunctionType *s);
  virtual SegmentationFunctionType *GetSegmentationFunction()
  { return m_SegmentationFunction; }
  
protected:
  virtual ~NarrowBandLevelSetImageFilter() {}
  NarrowBandLevelSetImageFilter();
  NarrowBandLevelSetImageFilter(const Self &); //purposely not implemented

  virtual void PrintSelf(std::ostream& os, Indent indent) const;
  void operator=(const Self&); //purposely not implemented
  
  virtual void InitializeIteration()
  {
    Superclass::InitializeIteration();
    // Estimate the progress of the filter
    this->SetProgress( (float) ((float)this->GetElapsedIterations()
                                / (float)this->GetMaximumIterations()) );
  }
  
  virtual void CreateNarrowBand ();

  void GenerateData();

  bool Halt();

  bool m_ReverseExpansionDirection;
  
  typedef IsoContourDistanceImageFilter<OutputImageType,OutputImageType>
          IsoFilterType;
  typedef FastChamferDistanceImageFilter<OutputImageType,OutputImageType>
          ChamferFilterType; 

  typename IsoFilterType::Pointer     m_IsoFilter;
  typename ChamferFilterType::Pointer m_ChamferFilter;

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

} // end namespace itk

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

#endif

---