ImageClusterBase.h

Go to the documentation of this file.

#ifndef __IMAGECLUSTERBASE_H__
#define __IMAGECLUSTERBASE_H__

#include <iostream>
#include "FhiclLite/PSet.h"
#include "Core/ImageMeta.h"
#include "Core/laropencv_base.h"
#include "Utils/Watch.h"
#include "ImageClusterTypes.h"
#include "ImageClusterFactory.h"
namespace larcv {
  class ImageClusterBase : public laropencv_base{
    
  public:
    
    ImageClusterBase(const std::string name = "noname");
    
    virtual ~ImageClusterBase(){}

    const std::string& Name() const { return _name; };

    void Configure(const ::fcllite::PSet &pset);

    void Profile(bool doit=true) { _profile = doit; }

    larcv::ContourArray_t Process(const larcv::ContourArray_t& clusters,
                  const ::cv::Mat& img,
                  larcv::ImageMeta& meta);

    size_t ProcessCount() const { return _proc_count; }
    double ProcessTime() const { return _proc_time; }

  protected:

    virtual void _Configure_(const ::fcllite::PSet &pset) = 0;

    virtual larcv::ContourArray_t _Process_(const larcv::ContourArray_t& clusters,
                        const ::cv::Mat& img,
                        larcv::ImageMeta& meta) = 0;

  private:

    std::string _name;   

    larcv::Watch _watch; 

    double _proc_time;   

    size_t _proc_count;  

    bool _profile;       
  };

}
#endif
 // end of doxygen group