lsExtractor.hpp

#ifndef LS_EXTRACTOR
#define LS_EXTRACTOR

#include <image/Image.hpp>
#include "lineSegment.hpp"
#include "lineSegmentSet.hpp"
#include "histogram.hpp"
#include "lsMisc.hpp"
#include "ctdef.hpp"

namespace jafar{
  namespace lines{
    

    class LsExtractor{
      public:
        LsExtractor();
      

        bool procCanny(jafar::image::Image* image, jafar::image::Image* cannyIm);
        

        static bool procCannySt(jafar::image::Image* image, jafar::image::Image* cannyIm, int lowerThresh, int higherThresh, int ap);
        

        void extractLineSegments(jafar::image::Image* image, jafar::image::Image* cannyIm, LineSegmentSet& segmentStorage,double thresh=15);
        

        static void extractLineSegmentsSt(jafar::image::Image* image, jafar::image::Image* cannyIm, LineSegmentSet& segmentStorage, double thresh=15, int lowerThresh=75, int higherThresh=150, int ap=3);
        

        bool procHough(jafar::image::Image* image, LineSegmentSet& newSegments);
        
        bool procCvFindContour(jafar::image::Image* image, LineSegmentSet& newSegments, double thresh=15 );
        
        static void findLinesDP(jafar::image::Image* image, LineSegmentSet& newSegments, double thresh );
    
        static void findLinesCalife(jafar::image::Image* image, LineSegmentSet& newSegments, double thresh=15 );
        
        static int dp(short *x, short *y, int startIdx, int endIdx, std::vector<CvPoint> &endpoints, double thresh=1);
    
        void setCannyLowerThresh(int lowerThresh){ cannyLowerThresh = lowerThresh;}
        void setCannyHigherThresh(int higherThresh){ cannyHigherThresh = higherThresh;}
        void setCannyAp(int ap){ if(ap>0) cannyAp = ap; } 
        void setCannyPara(int lowerThresh, int higherThresh, int ap){
          cannyLowerThresh = lowerThresh;
          cannyHigherThresh = higherThresh;
          cannyAp = ap;
        }
        
        void setHoughMethod(int method){ houghMethod = method;} 
        void setHoughRho(double rho){ if(rho>0) houghRho = rho;}
        void setHoughTheta(double theta){ if(theta>0) houghTheta = theta;}
        void setHoughThreshold(int thresh){ houghThreshold = thresh;}
        void setHoughParam1(double param){ houghParam1 = param;}
        void setHoughParam2(double param){ houghParam2 = param;}
        void setHoughPara(int method, double rho, double theta, int thresh, int param1, int param2){
          houghMethod = method;
          if(rho>0)houghRho = rho;
          if(theta>0) houghTheta = theta;
          houghThreshold = thresh;
          houghParam1 = param1;
          houghParam2 = param2;
        }
        
      protected:
    
        // parameters for Canny
        int cannyLowerThresh;   
        int cannyHigherThresh;  
        int cannyAp;            
        
        // parameters for Hough
        int houghMethod;      
        double houghRho;      
        double houghTheta;    
        int houghThreshold;   
        double houghParam1;   
        double houghParam2;   
    };
    
  } // namespace lines
} // namespace jafar

#endif