HierarchicalDirectSegmentsDetector.hpp

/* $Id$ */

#ifndef _DSEG_HIERARCHICAL_DIRECT_SEGMENTS_DETECTOR_HPP_
#define _DSEG_HIERARCHICAL_DIRECT_SEGMENTS_DETECTOR_HPP_

#include <dseg/DirectSegmentsDetectorBase.hpp>

namespace jafar {
  namespace dseg {
    class PyramidSP;
    class HierarchicalDirectSegmentsDetector : public DirectSegmentsDetectorBase {
      public:
        HierarchicalDirectSegmentsDetector(kernel::IdFactory<>* _idMaker = 0);
      public:
        void detectSegment( const jafar::image::Image& _image, const image::ConvexRoi *roiPtr, SegmentsSet& segmentStorage, SegmentDebug* _sdebug = 0);
      void detectSegment( PyramidSP* pyramid, const jafar::image::Image& _image, const image::ConvexRoi *roiPtr, SegmentsSet& segmentStorage, SegmentDebug* _sdebug = 0);

      PyramidSP* computePyramid(const jafar::image::Image& _image );

    public:
        JFR_DEFINE_PARAM_RW( int, levels, setLevels )
      protected:
        virtual void detectedSegment( SegmentHypothesis* , DirectSegmentsDetectorBase::LoopDetectionContext* _context, GrowSegmentContext&);
        virtual bool isInASegment( DirectSegmentsDetectorBase::LoopDetectionContext* _context, int _x, int _y );
      private:
        void detectSegmentAtScale( SegmentHypothesis* segHyp, DirectSegmentsDetectorBase::LoopDetectionContext* _context, int level);
        SegmentHypothesis* findSegHyp( const jblas::vec2& _center, double angle, std::vector< GrowSegmentContext >& hyps, int _level, DirectSegmentsDetectorBase::LoopDetectionContext* _context, double* grads, double scaledMinLength2  );
        struct Interval;
        void updateIntervals( SegmentHypothesis* _segHyp, const Interval& _currentInterval, std::list<Interval>& _intervals, int _level, const jblas::vec2& _center, const jblas::vec2& _dir, double _scaledMinLength );
      private:
        PyramidSP* m_pyramidSP; // Should be in the LoopDetectionContext
        int m_topToBottomScale;
    };
  }
}

#endif