ImageClusterManager.h

Go to the documentation of this file.

#ifndef __IMAGECLUSTERMANAGER_H__
#define __IMAGECLUSTERMANAGER_H__

#include <iostream>
#include "ImageClusterBase.h"
namespace larcv {
  class ImageClusterManager : public laropencv_base{
    
  public:
    
    ImageClusterManager(const std::string name="ImageClusterManager");
    
    ~ImageClusterManager(){}

    const std::string& Name() const { return _name; }
    void Reset();
    //AlgorithmID_t AddAlg(ImageClusterBase* alg); // deprecated: use configuration method + factory
    ImageClusterBase* GetAlg(const AlgorithmID_t id) const;
    ImageClusterBase* GetAlg(const std::string name) const;
    AlgorithmID_t GetAlgID(const std::string name) const;
    void Configure(const ::fcllite::PSet& main_cfg);
    void Process(const ::cv::Mat& img, const larcv::ImageMeta& meta);
    const ImageMeta& MetaData(const AlgorithmID_t alg_id) const;
    const Contour_t& Cluster(const ClusterID_t cluster_id, const AlgorithmID_t alg_id=kINVALID_ALGO_ID) const;
    const ContourArray_t& Clusters(const AlgorithmID_t alg_id=kINVALID_ALGO_ID) const;
    ClusterID_t ClusterID(const double x, const double y, AlgorithmID_t alg_id=kINVALID_ALGO_ID) const;
    void Report() const;
    
  private:
    std::string _name;
    bool _configured;
    std::vector<larcv::ImageClusterBase*> _alg_v;
    std::map<std::string,larcv::AlgorithmID_t> _alg_m;
    std::vector<larcv::ContourArray_t> _clusters_v;
    std::vector<larcv::ImageMeta> _meta_v;
    larcv::ImageMeta _orig_meta;
    bool _profile;
    Watch _watch;
    size_t _process_count;
    double _process_time;
  };
}
#endif
 // end of doxygen group