SBCluster.h

Go to the documentation of this file.

//by vic

#ifndef __SBCLUSTER_H__
#define __SBCLUSTER_H__

#include "ImageClusterBase.h"

namespace larcv {
 
  class SBCluster : public larcv::ImageClusterBase{
    
  public:
    
    SBCluster(const std::string name = "SBCluster") :
      ImageClusterBase(name),
      _dilation_size ( 5 ),
      _dilation_iter ( 2 ),
      _blur_size     ( 5 ),
      _thresh        ( 1 ),
      _thresh_maxval (255)
    {}
    
    virtual ~SBCluster(){}


  protected:

    void _Configure_(const ::fcllite::PSet &pset);
    
    larcv::ContourArray_t _Process_(const larcv::ContourArray_t& clusters,
                        const ::cv::Mat& img,
                        larcv::ImageMeta& meta);
    
  private:

    int _dilation_size;
    int _dilation_iter;
    int _blur_size;
    float _thresh;
    float _thresh_maxval;
    
  };
  class SBClusterFactory : public ImageClusterFactoryBase {
  public:
    SBClusterFactory() { ImageClusterFactory::get().add_factory("SBCluster",this); }
    ~SBClusterFactory() {}
    ImageClusterBase* create(const std::string instance_name) { return new SBCluster(instance_name); }
  };
  static SBClusterFactory __global_SBClusterFactory__;
  
}
#endif
 // end of doxygen group