lineSegmentSet.hpp

#ifndef LINE_SEGMENT_SET
#define LINE_SEGMENT_SET

#include <vector>
#include <iostream> 
#include "lines/lineSegment.hpp"


namespace jafar{
  namespace lines{
    
    class LineSegment;
  

    class LineSegmentSet{
    public:
      int eraseInvalid();
      
      void clearPredFlags();
      
      void clearParameters();

      void addLine(LineSegment* lineSegment);
      
      void addLine(LineSegment& lineSegment);

      void clear();
      
      //bool load(String filename, int numberOfParallels=0);
      //bool save(String filename);
      
      void print();
      
      void extractGreyscales(jafar::image::Image* image, int parallel_flag=0);
      
      void extractGradientscales(jafar::image::Image* image, int gradientimage, int parallel_flag=0);
      
      void extractLaplacescales(jafar::image::Image* image, int laplaceimage, int parallel_flag=0);
      
      void changeNumberOfNeighbours(int number);
      
      void calcHistogramDescriptor(jafar::image::Image* image, int nParallels=1, int d=3, jafar::image::Image* colorImage=0);
      
      void calcHistogramDescriptor();
      
      void normaliseHistogramDescriptors();

      void calcGreyspaceDescriptor(jafar::image::Image* image);

      void assignContourPts(jafar::image::Image* image);

      
      void orientLines(jafar::image::Image* image);
    
      void growLinesParallel( jafar::image::Image* image, int dist=3, int thresh=5, int distThresh=20);
      
      void growLines( jafar::image::Image* image, int gapThresh=1, int thresh=40, int maxOffset=2);
      
      void fitLines( jafar::image::Image* image, int gradientimage=0, int nPoints=20, int pDist=25, jafar::image::Image* colorImage=0);
      
      void fitLinesOrientation(LineSegmentSet lsSet, jafar::image::Image* gradX, jafar::image::Image* gradY, int nPoints=10, int pDist=25, double aDist=M_PI*0.5, double minLength=0, double extend=0, jafar::image::Image* colorImage=0);
    
      static void clone(LineSegmentSet& old, LineSegmentSet& clone);
      
      void drawLineSegments(jafar::image::Image* image, CvScalar color, int width);
      void drawLineSegments(jafar::image::Image* image, int width);
      
      std::vector<LineSegment> lineSegments;
    
      unsigned int size(){return lineSegments.size();}
      
      LineSegment& ls(unsigned int index){return lineSegments[index];}
    };
    
  } // namespace lines
} // namespace jafar
#endif