otbLineSegmentDetector.h

Go to the documentation of this file.

/*
 * Copyright (C) 2005-2022 Centre National d'Etudes Spatiales (CNES)
 *
 * This file is part of Orfeo Toolbox
 *
 *     https://www.orfeo-toolbox.org/
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#ifndef otbLineSegmentDetector_h
#define otbLineSegmentDetector_h
 
#include "otbImage.h"
#include "otbVectorDataSource.h"
#include "otbVectorData.h"
 
#include "itkUnaryFunctorImageFilter.h"
#include "itkGradientRecursiveGaussianImageFilter.h"
#include "itkGradientImageFilter.h"
 
namespace otb
{
 
namespace Functor
{
template <class TInputPixel, class TOutputPixel>
class MagnitudeFunctor
{
public:
  inline TOutputPixel operator()(const TInputPixel& input)
  {
    return static_cast<TOutputPixel>(2 * std::sqrt(input[0] * input[0] + input[1] * input[1]));
  }
};
 
template <class TInputPixel, class TOutputPixel>
class OrientationFunctor
{
public:
  inline TOutputPixel operator()(const TInputPixel& input)
  {
    return static_cast<TOutputPixel>(std::atan2(input[0], -input[1]));
  }
};
} // end namespace Functor
 
template <class TInputImage, class TPrecision = double>
class ITK_EXPORT LineSegmentDetector : public VectorDataSource<otb::VectorData<TPrecision>>
{
public:
  typedef LineSegmentDetector                      Self;
  typedef VectorDataSource<VectorData<TPrecision>> Superclass;
  typedef itk::SmartPointer<Self>                  Pointer;
  typedef itk::SmartPointer<const Self>            ConstPointer;
 
  itkNewMacro(Self);
 
  itkTypeMacro(LineSegmentDetector, VectorDataSource);
 
  typedef TInputImage                          InputImageType;
  typedef typename InputImageType::PixelType   InputPixelType;
  typedef typename InputImageType::IndexType   InputIndexType;
  typedef typename InputImageType::SizeType    SizeType;
  typedef typename InputImageType::RegionType  RegionType;
  typedef typename InputImageType::SpacingType SpacingType;
  typedef typename InputImageType::PointType   OriginType;
 
  typedef VectorData<TPrecision>                VectorDataType;
  typedef typename VectorDataType::DataNodeType DataNodeType;
  typedef typename VectorDataType::LineType     LineType;
  typedef typename VectorDataType::PointType    PointType;
  typedef typename LineType::VertexType         VertexType;
 
  typedef Image<TPrecision, 2> OutputImageType;
  typedef typename OutputImageType::PixelType OutputPixelType;
  typedef typename OutputImageType::IndexType OutputIndexType;
  typedef typename OutputImageType::SizeType  OutputSizeType;
 
  typedef std::vector<OutputIndexType>               IndexVectorType;
  typedef typename IndexVectorType::iterator         IndexVectorIteratorType;
  typedef std::vector<IndexVectorType>               CoordinateHistogramType;
  typedef typename CoordinateHistogramType::iterator CoordinateHistogramIteratorType;
 
  typedef std::vector<IndexVectorType>           VectorOfIndexVectorType;
  typedef std::vector<float>                     DirectionVectorType;
  typedef typename DirectionVectorType::iterator DirectionVectorIteratorType;
 
  typedef itk::GradientRecursiveGaussianImageFilter<OutputImageType> GradientFilterType;
  typedef typename GradientFilterType::Pointer                       GradientFilterPointerType;
  typedef typename GradientFilterType::OutputImageType               GradientOutputImageType;
 
  typedef itk::UnaryFunctorImageFilter<GradientOutputImageType, OutputImageType,
                                       Functor::MagnitudeFunctor<typename GradientOutputImageType::PixelType, TPrecision>>
                                                                   MagnitudeFilterType;
  typedef typename MagnitudeFilterType::Pointer                    MagnitudeFilterPointerType;
  typedef typename MagnitudeFilterType::OutputImageType::PixelType MagnitudePixelType;
  typedef typename MagnitudeFilterType::OutputImageType            MagnitudeImageType;
  typedef typename MagnitudeImageType::Pointer                     MagnitudeImagePointerType;
 
  typedef itk::UnaryFunctorImageFilter<GradientOutputImageType, OutputImageType,
                                       Functor::OrientationFunctor<typename GradientOutputImageType::PixelType, TPrecision>>
                                                          OrientationFilterType;
  typedef typename OrientationFilterType::Pointer         OrientationFilterPointerType;
  typedef typename OrientationFilterType::OutputImageType OutputImageDirType;
  typedef typename OutputImageDirType::RegionType         OutputImageDirRegionType;
 
  typedef otb::Image<unsigned char, 2> LabelImageType;
  typedef typename LabelImageType::Pointer LabelImagePointerType;
 
  typedef std::vector<double>                  RectangleType;
  typedef typename RectangleType::iterator     RectangleIteratorType;
  typedef std::vector<RectangleType>           RectangleListType;
  typedef typename RectangleListType::iterator RectangleListTypeIterator;
 
  using Superclass::SetInput;
  virtual void SetInput(const InputImageType* input);
  virtual const InputImageType* GetInput(void);
 
  LabelImagePointerType GetMap()
  {
    return m_UsedPointImage;
  }
  MagnitudeImagePointerType GetGradMod()
  {
    return m_MagnitudeFilter->GetOutput();
  }
  typename OutputImageDirType::Pointer GetGradOri()
  {
    return m_OrientationFilter->GetOutput();
  }
 
protected:
  LineSegmentDetector();
  ~LineSegmentDetector() override
  {
  }
 
  void GenerateInputRequestedRegion() override;
 
  void GenerateData() override;
 
  virtual CoordinateHistogramType SortImageByModulusValue(MagnitudeImagePointerType modulusImage);
 
  virtual void LineSegmentDetection(CoordinateHistogramType& CoordinateHistogram);
 
  virtual bool IsNotUsed(InputIndexType& index) const;
 
  virtual bool IsUsed(InputIndexType& index) const;
 
  virtual bool IsNotIni(InputIndexType& index) const;
 
  virtual void SetPixelToUsed(InputIndexType index);
 
  virtual void SetPixelToNotIni(InputIndexType index);
 
  virtual void SetRegionToNotIni(IndexVectorType region);
 
  virtual bool GrowRegion(InputIndexType index, IndexVectorType& region, double& regionAngle);
 
  virtual bool IsAligned(double Angle, double regionAngle, double prec) const;
 
  virtual int ComputeRectangles();
 
  virtual RectangleType Region2Rectangle(IndexVectorType region, double regionAngle);
 
  virtual double ComputeRegionOrientation(IndexVectorType region, double x, double y, double angleRegion) const;
 
  virtual double angle_diff(double a, double b) const;
 
  virtual double ComputeRectNFA(const RectangleType& rec) const;
 
  virtual double ImproveRectangle(RectangleType& rectangle) const;
 
  virtual double NFA(int n, int k, double p, double logNT) const;
 
  virtual void CopyRectangle(RectangleType& rDst, RectangleType& rSrc) const;
 
  void PrintSelf(std::ostream& os, itk::Indent indent) const override;
 
private:
  LineSegmentDetector(const Self&) = delete;
  void operator=(const Self&) = delete;
 
  VectorOfIndexVectorType m_RegionList;
  DirectionVectorType     m_DirectionVector;
  LabelImagePointerType   m_UsedPointImage;
  RectangleListType       m_RectangleList;
 
  double       m_Threshold;
  double       m_Prec;
  double       m_DirectionsAllowed;
  unsigned int m_MinimumRegionSize;
 
  GradientFilterPointerType m_GradientFilter;
 
  MagnitudeFilterPointerType m_MagnitudeFilter;
 
  OrientationFilterPointerType m_OrientationFilter;
};
} // end namespace otb
 
#ifndef OTB_MANUAL_INSTANTIATION
#include "otbLineSegmentDetector.hxx"
#endif
 
#endif