PandoraInterfaceHelper.h

#ifndef PANDORAINTERFACEHELPER_H
#define PANDORAINTERFACEHELPER_H

#include "HelperBase.h"

#include "canvas/Persistency/Common/FindManyP.h"
#include "canvas/Persistency/Common/FindOneP.h"
#include "lardataobj/RecoBase/PFParticle.h"
#include "lardataobj/RecoBase/SpacePoint.h"
#include "lardataobj/RecoBase/Hit.h"
#include "larpandora/LArPandoraInterface/LArPandoraHelper.h"

#include "nusimdata/SimulationBase/MCParticle.h"
#include "nusimdata/SimulationBase/MCTruth.h"
#include "lardataobj/AnalysisBase/BackTrackerMatchingData.h"

namespace lar_pandora
{
typedef std::map<art::Ptr<recob::PFParticle>, art::Ptr<simb::MCParticle>> PFParticlesToMCParticles;
}

namespace lee
{

typedef std::map<art::Ptr<recob::PFParticle>, unsigned int>
    RecoParticleToNMatchedHits;
typedef std::map<art::Ptr<simb::MCParticle>, RecoParticleToNMatchedHits>
    ParticleMatchingMap;
typedef std::set<art::Ptr<recob::PFParticle>> PFParticleSet;
typedef std::set<art::Ptr<simb::MCParticle>> MCParticleSet;

class PandoraInterfaceHelper : public HelperBase
{
  public:
    PandoraInterfaceHelper();
    ~PandoraInterfaceHelper() {}

    void reconfigure(fhicl::ParameterSet const &pset);

    void traversePFParticleTree(
        const art::ValidHandle<std::vector<recob::PFParticle>> pfparticles,
        size_t top_index, std::vector<size_t> &unordered_daugthers,
        std::string _pfp_producer);

    std::vector<double> calculateChargeCenter(
        size_t ipf,
        const art::ValidHandle<std::vector<recob::PFParticle>> pfparticles,
        const art::Event &evt,
        std::string _pfp_producer);

    void get_daughter_tracks(std::vector<size_t> pf_ids, const art::Event &evt,
                              std::vector<art::Ptr<recob::Track>> &tracks,
                              std::string _pfp_producer);

    void get_daughter_showers(std::vector<size_t> pf_ids, const art::Event &evt,
                              std::vector<art::Ptr<recob::Shower>> &showers,
                              std::string _pfp_producer);


    void GetRecoToTrueMatches(lar_pandora::PFParticlesToMCParticles &matchedParticles);

    void Configure(art::Event const &e,
                    std::string _pfp_producer,
                    std::string _spacepoint_producer,
                    std::string _hitfinder_producer,
                    std::string _geant_producer,
                    std::string _hit_mcp_producer);

    art::Ptr<simb::MCTruth> TrackIDToMCTruth(art::Event const &e, std::string _geant_producer, int geant_track_id);

    void CollectMCParticles(const art::Event &evt,
                            const std::string &label,
                            lar_pandora::MCParticleVector &particleVector);
    void CollectMCParticles(const art::Event &evt,
                            const std::string &label,
                            lar_pandora::MCTruthToMCParticles &truthToParticles,
                            lar_pandora::MCParticlesToMCTruth &particlesToTruth);

  protected:
    lar_pandora::HitsToMCParticles _hit_to_mcps_map; 
    lar_pandora::PFParticlesToHits _pfp_to_hits_map; 
  private:
    bool _configured = false;
    bool m_isOverlaidSample = false;
    bool _debug = false;
    bool _verbose = false;
};
}

#endif