itkHoughTransform2DLinesImageFilter.h

Go to the documentation of this file.

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

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

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

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

#include "itkImageToImageFilter.h"
#include "itkImage.h"
#include "itkLineSpatialObject.h"

namespace itk
{

template<typename TInputPixelType, typename TOutputPixelType>
class ITK_EXPORT HoughTransform2DLinesImageFilter :
    public ImageToImageFilter< Image<TInputPixelType,2>, Image<TOutputPixelType,2> >
{
public:
   
  typedef HoughTransform2DLinesImageFilter 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 SmartPointer<Self>        Pointer;
  typedef SmartPointer<const Self>  ConstPointer;

  typedef LineSpatialObject<2>         LineType;
  typedef typename LineType::Pointer   LinePointer;
  typedef std::list<LinePointer>       LinesListType;
  typedef LineType::LinePointType      LinePointType;

  typedef ImageToImageFilter<InputImageType, OutputImageType>  Superclass;

  typedef typename InputImageType::IndexType IndexType;

  typedef typename InputImageType::PixelType PixelType;

  typedef typename InputImageType::RegionType OutputImageRegionType;

  itkTypeMacro( HoughTransform2DLinesImageFilter, ImageToImageFilter );

  itkNewMacro(Self);  

  void GenerateData();

  itkSetMacro(Threshold,float);

  itkGetConstMacro(Threshold,float);

  itkSetMacro(AngleResolution,float);

  itkGetConstMacro(AngleResolution,float);

  void Simplify(void);

  itkGetObjectMacro(SimplifyAccumulator,OutputImageType);

  LinesListType & GetLines(unsigned int n=0);

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

  itkSetMacro(DiscRadius,float);
  itkGetConstMacro(DiscRadius,float);

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

#ifdef ITK_USE_CONCEPT_CHECKING

  itkConceptMacro(IntConvertibleToOutputCheck,
    (Concept::Convertible<int, TOutputPixelType>));
  itkConceptMacro(InputGreaterThanFloatCheck,
    (Concept::GreaterThanComparable<PixelType, float>));
  itkConceptMacro(OutputPlusIntCheck,
    (Concept::AdditiveOperators<TOutputPixelType, int>));

#endif

protected:

  HoughTransform2DLinesImageFilter();
  virtual ~HoughTransform2DLinesImageFilter() {};

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

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

  void GenerateInputRequestedRegion();

  void GenerateOutputInformation();

  void EnlargeOutputRequestedRegion(DataObject *output);

private:

  float              m_AngleResolution;
  float              m_Threshold;
  OutputImagePointer m_SimplifyAccumulator;
  LinesListType      m_LinesList;
  unsigned int       m_NumberOfLines;
  float              m_DiscRadius;
  float              m_Variance;
  unsigned long      m_OldModifiedTime;
  unsigned long      m_OldNumberOfLines;
};

} // end namespace itk

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

#endif