CMManagerBase.h

Go to the documentation of this file.

#ifndef RECOTOOL_CMMANAGERBASE_H
#define RECOTOOL_CMMANAGERBASE_H

#include <iostream>

#include "CPriorityAlgoBase.h"
#include "TStopwatch.h"

namespace cmtool {

  class CMManagerBase{

  public:

    enum CMMSGLevel_t {
      kPerMerging,
      kPerIteration,
      kPerEvent,
      kNone
    };
    
  public:
    
    CMManagerBase();
    
    virtual ~CMManagerBase(){}

    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(const std::vector<std::vector<util::PxHit> > &clusters);
    //void SetClusters(const std::vector<std::vector<larutil::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();

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

  protected:

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

    virtual void EventBegin(){}

    virtual void IterationBegin(){}

    virtual bool IterationProcess()=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