CMManagerBase.h

Go to the documentation of this file.

#ifndef RECOTOOL_CMMANAGERBASE_H
#define RECOTOOL_CMMANAGERBASE_H

#include <map>
#include <set>
#include <stddef.h>
#include <vector>

#include "RtypesCore.h"
class TFile;

#include "lardata/Utilities/PxUtils.h"
#include "larreco/RecoAlg/ClusterRecoUtil/ClusterParamsAlg.h"

namespace util {
  class GeometryUtilities;
}

namespace cmtool {

  class CPriorityAlgoBase;

  class CMManagerBase {
  public:
    enum CMMSGLevel_t {
      kPerMerging,
      kPerIteration,
      kPerEvent,
      kNone
    };

    CMManagerBase();

    virtual ~CMManagerBase() = default;

    void DebugMode(CMMSGLevel_t level) { _debug_mode = level; }

    void ReportTimings(bool time_report = true) { _time_report = time_report; }

    void Reset();

    void AddPriorityAlgo(CPriorityAlgoBase* algo) { _priority_algo = algo; }

    void MergeTillConverge(bool doit = true) { _merge_till_converge = doit; }

    void SetClusters(util::GeometryUtilities const& gser,
                     const std::vector<std::vector<util::PxHit>>& clusters);

    void SetClusters(const std::vector<cluster::ClusterParamsAlg>& clusters);

    const std::vector<cluster::ClusterParamsAlg>& GetInputClusters() const { return _in_clusters; }

    void SetMinNHits(unsigned int n) { _min_nhits = n; }

    void Process(util::GeometryUtilities const& gser);

    void SetAnaFile(TFile* fout) { _fout = fout; }

  protected:
    void ComputePriority(const std::vector<cluster::ClusterParamsAlg>& clusters);

    virtual void EventBegin() {}

    virtual void IterationBegin() {}

    virtual bool IterationProcess(util::GeometryUtilities const& gser) = 0;

    virtual void IterationEnd() {}

    virtual void EventEnd() {}

  protected:
    bool _time_report;

    unsigned int _min_nhits;

    CMMSGLevel_t _debug_mode;

    std::vector<cluster::ClusterParamsAlg> _in_clusters;

    ::cmtool::CPriorityAlgoBase* _priority_algo;

    TFile* _fout;

    std::multimap<float, size_t> _priority;

    bool _merge_till_converge;

    std::set<UChar_t> _planes;
  };
}

#endif
 // end of doxygen group