ClusterMergeHelper.h

Go to the documentation of this file.

// \file ClusterMergeHelper.h
//
// \brief ClusterMergeHelper header file
//
// \author kazuhiro@nevis.columbia.edu
//

#ifndef CLUSTERMERGEHELPER_H
#define CLUSTERMERGEHELPER_H

// ART includes

namespace art {
  class Event;
}

#include "canvas/Persistency/Common/Assns.h"
#include "canvas/Persistency/Common/Ptr.h"

namespace fhicl {
  class ParameterSet;
}

// LArSoft
#include "lardata/Utilities/GeometryUtilities.h"
#include "lardataobj/RecoBase/Cluster.h"
#include "lardataobj/RecoBase/Hit.h"
#include "larreco/RecoAlg/CMTool/CMToolBase/CMergeManager.h"
#include "larreco/RecoAlg/ClusterRecoUtil/ClusterParamsAlg.h"

// STL
#include <vector>

namespace cluster {

  class ClusterMergeHelper {
  public:
    ::cmtool::CMergeManager& GetManager() { return fMgr; }

    void SetClusters(util::GeometryUtilities const& gser,
                     const std::vector<std::vector<art::Ptr<recob::Hit>>>& clusters);

    void SetClusters(util::GeometryUtilities const& gser,
                     const art::Event& evt,
                     const std::string& cluster_module_label);

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

    const std::vector<std::vector<art::Ptr<recob::Hit>>>& GetMergedClusterHits() const;

    const std::vector<cluster::ClusterParamsAlg>& GetMergedCPAN() const;

    void AppendResult(util::GeometryUtilities const& gser,
                      art::Event& ev,
                      std::vector<recob::Cluster>& out_clusters,
                      art::Assns<recob::Cluster, recob::Hit>& assns) const;

  protected:
    void SetClusters(util::GeometryUtilities const& gser,
                     const std::vector<std::vector<util::PxHit>>& clusters)
    {
      fMgr.Reset();
      fMgr.SetClusters(gser, clusters);
    }

  protected:
    ::cmtool::CMergeManager fMgr;

    ::util::GeometryUtilities fGeoU;

    std::vector<std::vector<art::Ptr<recob::Hit>>> fInputClusters;

    std::vector<std::vector<art::Ptr<recob::Hit>>> fOutputClusters;

  }; // class ClusterMergeHelper

} //namespace cluster
#endif