ConnectedComponentAnalyzer.h

Go to the documentation of this file.

/*******************************************************************************
 *  ConnectedComponentAnalyzer.cpp
 *
 *  (C) 2007 AG Aktives Sehen <agas@uni-koblenz.de>
 *           Universitaet Koblenz-Landau
 *
 *  Additional information:  
 *  $Id: ConnectedComponentAnalyzer.h 44313 2011-04-06 22:46:28Z agas $ 
 *******************************************************************************/

#ifndef CONNECTEDCOMPONENTANALYZER_H
#define CONNECTEDCOMPONENTANALYZER_H

#include <vector>
#include <opencv2/opencv.hpp>
#include "ConnectedComponent.h"


class ConnectedComponentAnalyzer {

public:

  ConnectedComponentAnalyzer();

  ~ConnectedComponentAnalyzer();

  std::vector< ConnectedComponent > segmentByLimit( cv::Mat* pi_img, double pi_limit );


  std::vector< ConnectedComponent > segment(unsigned char* binImage, unsigned width, unsigned height, unsigned minSegmentSize=1, float minOccupiedArea=0.0, unsigned expandSegmentSize=0);
  
  std::vector< ConnectedComponent > segment(const float* floatImage, unsigned width, unsigned height, float maxDistance, unsigned minSegmentSize=1, float minOccupiedArea=0.0, unsigned expandSegmentSize=0);
  std::vector< ConnectedComponent > segment(const float* floatImage, const unsigned char* mask, unsigned width, unsigned height, float maxDistance, unsigned minSegmentSize=1, float minOccupiedArea=0.0, unsigned expandSegmentSize=0, int maxr=1);

  static void isolateLargestSegment(unsigned char* binImage, unsigned width, unsigned height);
  
  unsigned* getLastLabelImage() { return m_LabelImage; }
  unsigned  getImageWidth() { return m_Width; }
  unsigned  getImageHeight() { return m_Height; }
                                                       
private:
  
  std::vector<ConnectedComponent> extractComponents(unsigned max_label, unsigned minSegmentSize,
                                                    float minOccupiedArea, unsigned expandSegmentSize);
  
  template < typename T >
  struct IntensitySimilarity
  {
    private:
      T m_MaxDistance;
    public:
      IntensitySimilarity( T maxDistance ) : m_MaxDistance( maxDistance ) {}
      ~IntensitySimilarity() {}
      bool operator()( const T& l, const T& r ) const
      {
        T diff;
        if ( l < r ) {
          diff = r - l ;
        } else {
          diff = l - r;
        }
        return diff <= m_MaxDistance;
      }
  };
  
  template < typename MaskT, typename PixelT >
  struct MaskVadility
  {
    private:
      const MaskT* mask;
      const int colstep;
      const MaskT MASKED;
      
    public:
      MaskVadility( MaskT masked, const MaskT* mask, int colstep)
      : mask(mask), colstep(colstep), MASKED(masked)
      {}
      
      bool operator()(int idx, PixelT intensity ) const
      { return mask[ idx ] != MASKED; }
      
      
  };
  
  unsigned* m_LabelImage;
  unsigned  m_Width, m_Height;
};


#endif