DirectSegmentsBase.hpp

/* $Id$ */

#ifndef _SEGMENTS_PLUS_BASE_H_
#define _SEGMENTS_PLUS_BASE_H_

#include <vector>
#include <map>

#include <kernel/jafarMacro.hpp>
#include <image/Image.hpp>
#include <jmath/jblas.hpp>

namespace jafar {
  namespace dseg {
    class SegmentsSet;
    class SegmentProbabilityEstimator;
    class SegmentHypothesis;
    class GrowStopCondition;
    struct PreprocessedImage {
      jafar::image::Image image;
      jafar::image::Image proba;
      jafar::image::Image dx;
      jafar::image::Image dy;
    jafar::image::Image inasegment;
   };
    struct SegmentDebug {
      jafar::image::Image inasegment;
      jafar::image::Image proba;
      jafar::image::Image dx;
      jafar::image::Image dy;
    };
    struct GrowSegmentContext {
      SegmentHypothesis* segHyp;
      jafar::image::Image originalImage; //< only usefull when ENABLE_SEGMENT_HYPOTHESIS_DEBUG
      jafar::image::Image proba;
      jafar::image::Image dx;
      jafar::image::Image dy;
      jafar::image::Image* inasegment;
      std::map<int, int> segmentToCount;
      double x_1;
      double x_2;
      double y_1;
      double y_2;
      int k_p;
      int k_n;
      GrowStopCondition* growStopCondition;
    };
    class DirectSegmentsBase {
      public:
        DirectSegmentsBase();
        virtual ~DirectSegmentsBase();
      protected:
        bool searchPoint( GrowSegmentContext& _context, double _searchStep, const jblas::vec2& _pi, double& _xr, double& _yr, double& _angler, int& _bestI, int _searchNbSteps);
        bool smartSearchPoint( GrowSegmentContext& _context, double _searchStep, const jblas::vec2& _pi, double& _xr, double& _yr, double& _angler, int& _bestI, int _searchNbSteps);
        void coloriage( double x_1, double y_1, double x_2, double y_2, image::Image& inasegment, int _value);
        void computeGradientValuesOnLine( double& a, double& b, double angle, double cosA, double sinA, float* _mag1, float* _mag2, float* _mag3, int j ) const;
        bool hasMaximumGradient( double _angle, double _cosA, double _sinA, float* _mag1 , float* _mag2, float* _mag3, int j ) const;
        bool growSegment( GrowSegmentContext& _context, int _step = 1 );
        SegmentHypothesis* attemptToMerge( SegmentHypothesis* segHyp, const image::Image& inasegment, SegmentsSet&  segmentHypothesises , const std::map<int, int>& segmentToCount);
      private:
        JFR_DEFINE_PARAM_RW( int, searchNbSteps, setSearchNbSteps)
        JFR_DEFINE_PARAM_RW( double, maximalSearchStep, setMaximalSearchStep)
        JFR_DEFINE_PARAM_RW( double, thresholdGradColinearity, setThresholdGradColinearity )
        JFR_DEFINE_PARAM_RW( double, thresholdMaxGradient, setThresholdMaxGradient )
        JFR_DEFINE_PARAM_RO( double, minimumSegmentLength)
        JFR_DEFINE_PARAM_RW( SegmentProbabilityEstimator*, segmentProbabilityEstimator, setSegmentProbabilityEstimator )
        JFR_DEFINE_PARAM_RW( double, maximalSearchLength, setMaximalSearchLength)
      public:
        void setMinimumSegmentLength(double _minimumSegmentLength);
      protected:
        double minimumSegmentLengthPow2() const;
      protected:
        int segCount;
        void adjustStep( double& _step, int &_searchNbSteps ) const;
      private:
        double m_minimumSegmentLengthPow2;
    };
  }
}


#endif