CMergeManager.h

Go to the documentation of this file.

#ifndef RECOTOOL_CMERGEMANAGER_H
#define RECOTOOL_CMERGEMANAGER_H

#include "larreco/RecoAlg/CMTool/CMToolBase/CMManagerBase.h"
#include "larreco/RecoAlg/CMTool/CMToolBase/CMergeBookKeeper.h"

#include "larreco/RecoAlg/ClusterRecoUtil/ClusterParamsAlg.h"
#include <vector>

namespace cmtool {

  class CBoolAlgoBase;

  class CMergeManager : public CMManagerBase {
  public:
    CMergeManager();

    virtual ~CMergeManager() {}

    virtual void Reset();

    void AddMergeAlgo(CBoolAlgoBase* algo) { _merge_algo = algo; }

    void AddSeparateAlgo(CBoolAlgoBase* algo) { _separate_algo = algo; }

    const std::vector<cluster::ClusterParamsAlg>& GetClusters() const { return _out_clusters; }

    const CMergeBookKeeper& GetBookKeeper() const { return _book_keeper; }

  protected:
    //
    // FMWK functions override
    //

    virtual void EventBegin();

    virtual void IterationBegin();

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

    virtual void IterationEnd();

    virtual void EventEnd();

  protected:
    void RunMerge(const std::vector<cluster::ClusterParamsAlg>& in_clusters,
                  CMergeBookKeeper& book_keeper) const;

    void RunMerge(const std::vector<cluster::ClusterParamsAlg>& in_clusters,
                  const std::vector<bool>& merge_flag,
                  CMergeBookKeeper& book_keeper) const;

    void RunSeparate(const std::vector<cluster::ClusterParamsAlg>& in_clusters,
                     CMergeBookKeeper& book_keeper) const;

  protected:
    std::vector<cluster::ClusterParamsAlg> _out_clusters;

    CMergeBookKeeper _book_keeper;

    ::cmtool::CBoolAlgoBase* _merge_algo;

    ::cmtool::CBoolAlgoBase* _separate_algo;

    size_t _iter_ctr;

    std::vector<CMergeBookKeeper> _book_keeper_v;

    std::vector<std::vector<unsigned short>> _tmp_merged_indexes;

    std::vector<cluster::ClusterParamsAlg> _tmp_merged_clusters;
  };
}

#endif
 // end of doxygen group