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
#include "fhiclcpp/ParameterSet.h"
#include "messagefacility/MessageLogger/MessageLogger.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"
#include "art/Framework/Services/Registry/ServiceMacros.h"
#include "art/Framework/Services/Optional/TFileService.h"
#include "art/Framework/Services/Optional/TFileDirectory.h"
#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/Handle.h"
#include "canvas/Persistency/Common/FindManyP.h"
#include "canvas/Persistency/Common/Ptr.h"
#include "canvas/Persistency/Common/PtrVector.h"

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

// STL
#include <set>
#include <vector>
#include <sstream>

// ROOT
#include <TString.h>
#include <TTree.h>

namespace cluster
{

  class ClusterMergeHelper {
    
  public:

    ClusterMergeHelper(){};

    virtual ~ClusterMergeHelper(){}

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

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

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

    void Process();

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

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

    void AppendResult(art::EDProducer &ed,
                      art::Event      &ev,
                      std::vector<recob::Cluster>           &out_clusters,
                      art::Assns<recob::Cluster,recob::Hit> &assns) const;
      
  protected:

    void SetClusters(const std::vector<std::vector<util::PxHit> > &clusters)
    { 
      fMgr.Reset();
      fMgr.SetClusters(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