itkHoughTransform2DCirclesImageFilter.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkHoughTransform2DCirclesImageFilter.h,v $
  Language:  C++
  Date:      $Date: 2009-04-25 12:27:26 $
  Version:   $Revision: 1.13 $

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

#ifdef _MSC_VER
#pragma warning ( disable : 4786 )
#endif

#include "itkImageToImageFilter.h"
#include "itkImage.h"
#include "itkEllipseSpatialObject.h"

namespace itk
{

template<typename TInputPixelType, typename TOutputPixelType>
class ITK_EXPORT HoughTransform2DCirclesImageFilter :
    public ImageToImageFilter< Image<TInputPixelType,2>, Image<TOutputPixelType,2> >
{
public:
   
  typedef HoughTransform2DCirclesImageFilter Self;

  typedef Image<TInputPixelType,2>              InputImageType;
  typedef typename InputImageType::Pointer      InputImagePointer;
  typedef typename InputImageType::ConstPointer InputImageConstPointer;

  typedef Image<TOutputPixelType,2>         OutputImageType;
  typedef typename OutputImageType::Pointer OutputImagePointer;


  typedef ImageToImageFilter<
    Image<TInputPixelType,2>
    , Image<TOutputPixelType,2> >  Superclass;

  typedef SmartPointer<Self>        Pointer;
  typedef SmartPointer<const Self>  ConstPointer;

  typedef typename InputImageType::IndexType IndexType;

  typedef typename InputImageType::PixelType PixelType;

  typedef typename InputImageType::RegionType OutputImageRegionType;

  typedef EllipseSpatialObject<2>      CircleType;
  typedef typename CircleType::Pointer CirclePointer;
  typedef std::list<CirclePointer>     CirclesListType;

  itkTypeMacro( HoughTransform2DCirclesImageFilter, ImageToImageFilter );

  itkNewMacro(Self);  

  void GenerateData();

  void SetRadius(double radius);

  itkSetMacro(MinimumRadius,double);

  itkSetMacro(MaximumRadius,double);

  itkSetMacro(Threshold,double);

  itkGetConstMacro(Threshold,double);

  itkGetObjectMacro(RadiusImage,OutputImageType);

  itkSetMacro(SigmaGradient,double);

  itkGetConstMacro(SigmaGradient,double);

  CirclesListType & GetCircles(unsigned int n=0);

  itkSetMacro(NumberOfCircles,unsigned int);
  itkGetConstMacro(NumberOfCircles,unsigned int);

  itkSetMacro(DiscRadiusRatio,float);
  itkGetConstMacro(DiscRadiusRatio,float);

  itkSetMacro(Variance,float);
  itkGetConstMacro(Variance,float);

  itkSetMacro(SweepAngle,float);
  itkGetConstMacro(SweepAngle,float);

#ifdef ITK_USE_CONCEPT_CHECKING

  itkConceptMacro(IntConvertibleToOutputCheck,
    (Concept::Convertible<int, TOutputPixelType>));
  itkConceptMacro(InputGreaterThanDoubleCheck,
    (Concept::GreaterThanComparable<PixelType, double>));
  itkConceptMacro(OutputPlusIntCheck,
    (Concept::AdditiveOperators<TOutputPixelType, int>));
  itkConceptMacro(OutputDividedByIntCheck,
    (Concept::DivisionOperators<TOutputPixelType, int>));

#endif

protected:

  HoughTransform2DCirclesImageFilter();
  virtual ~HoughTransform2DCirclesImageFilter() {};

  HoughTransform2DCirclesImageFilter(const Self&) {}
  void operator=(const Self&) {}

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

  void GenerateInputRequestedRegion();

  void EnlargeOutputRequestedRegion(DataObject *itkNotUsed(output));

private:

  float  m_SweepAngle;
  double m_MinimumRadius;
  double m_MaximumRadius;
  double m_Threshold;
  double m_SigmaGradient;

  OutputImagePointer m_RadiusImage;
  CirclesListType    m_CirclesList;
  unsigned int       m_NumberOfCircles;
  float              m_DiscRadiusRatio;
  float              m_Variance;
  unsigned long      m_OldModifiedTime;
  unsigned long      m_OldNumberOfCircles;

};

} // end namespace itk

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

#endif