itkSimpleFuzzyConnectednessImageFilterBase.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkSimpleFuzzyConnectednessImageFilterBase.h,v $
  Language:  C++
  Date:      $Date: 2009-02-01 13:08:41 $
  Version:   $Revision: 1.4 $

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

#include "itkImage.h"
#include "itkImageToImageFilter.h"
#include <vnl/vnl_matrix_fixed.h>

#include <queue>

namespace itk {

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

  itkNewMacro(Self);

  itkTypeMacro(SimpleFuzzyConnectednessImageFilterBase,ImageToImageFilter);

  itkStaticConstMacro(ImageDimension, unsigned int,
                      TInputImage::ImageDimension);

  typedef TInputImage  InputImageType;
  typedef TOutputImage OutputImageType;

  typedef Image <unsigned short, itkGetStaticConstMacro(ImageDimension)> UShortImage;

  typedef typename TInputImage::IndexType         IndexType;
  typedef typename TInputImage::SizeType          SizeType;
  typedef typename TInputImage::PixelType         PixelType;
  typedef UShortImage                             FuzzySceneType;
  typedef typename FuzzySceneType::Pointer        FuzzyScenePointer;
  typedef typename InputImageType::ConstPointer   InputImageConstPointer;
  typedef std::queue<IndexType>                   QueueType;
  typedef typename OutputImageType::RegionType    RegionType;
  typedef typename OutputImageType::PixelType     OutputPixelType;
  typedef typename OutputImageType::Pointer       OutputImagePointer;
  
  itkSetMacro(Weight, double);
  itkGetMacro(Weight, double);

  itkSetMacro(Threshold, double);
  itkGetMacro(Threshold, double);

  itkSetMacro(InsideValue, OutputPixelType);
  itkGetMacro(InsideValue, OutputPixelType);

  itkSetMacro(OutsideValue, OutputPixelType);
  itkGetMacro(OutsideValue, OutputPixelType);

  itkSetMacro(      ObjectSeed, IndexType );
  itkGetConstReferenceMacro( ObjectSeed, IndexType );

  void MakeSegmentObject();

  const FuzzySceneType * GetFuzzyScene(void) const
    { return m_FuzzyScene; };

  void UpdateThreshold(const double x);

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

  void GenerateData();

  double            m_Weight;
  double            m_Threshold;
  IndexType         m_ObjectSeed;
  SizeType          m_Size;

  InputImageConstPointer    m_InputImage;
  FuzzyScenePointer         m_FuzzyScene;
  OutputImagePointer        m_SegmentObject; 
  
  QueueType m_Queue;

  void PushNeighbors(const IndexType &center);

  virtual double FuzzyAffinity(const PixelType, const PixelType)
    { return 0; }

  double FindStrongPath(const IndexType &center);

private:
  SimpleFuzzyConnectednessImageFilterBase(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

  OutputPixelType  m_InsideValue;
  OutputPixelType  m_OutsideValue;

};


} // end namespace itk.

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

#endif